From 030ff273473f0a5620ba276a370e5119f57179df Mon Sep 17 00:00:00 2001
From: Niall Sheridan <nsheridan@gmail.com>
Date: Mon, 8 Aug 2016 23:39:46 +0100
Subject: SQLite DB support

---
 server/store/store_test.go | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

(limited to 'server/store/store_test.go')

diff --git a/server/store/store_test.go b/server/store/store_test.go
index bf16fa6..629230b 100644
--- a/server/store/store_test.go
+++ b/server/store/store_test.go
@@ -3,7 +3,10 @@ package store
 import (
 	"crypto/rand"
 	"crypto/rsa"
+	"fmt"
+	"io/ioutil"
 	"os"
+	"os/exec"
 	"testing"
 	"time"
 
@@ -94,7 +97,7 @@ func TestMySQLStore(t *testing.T) {
 	if config == "" {
 		t.Skip("No MYSQL_TEST_CONFIG environment variable")
 	}
-	db, err := NewMySQLStore(config)
+	db, err := NewSQLStore(config)
 	if err != nil {
 		t.Error(err)
 	}
@@ -112,3 +115,21 @@ func TestMongoStore(t *testing.T) {
 	}
 	testStore(t, db)
 }
+
+func TestSQLiteStore(t *testing.T) {
+	f, err := ioutil.TempFile("", "sqlite_test_db")
+	if err != nil {
+		t.Error(err)
+	}
+	defer os.Remove(f.Name())
+	// This is so jank.
+	args := []string{"run", "../../cmd/dbinit/dbinit.go", "-db_type", "sqlite", "-db_path", f.Name()}
+	if err := exec.Command("go", args...).Run(); err != nil {
+		t.Error(err)
+	}
+	db, err := NewSQLStore(fmt.Sprintf("sqlite:%s", f.Name()))
+	if err != nil {
+		t.Error(err)
+	}
+	testStore(t, db)
+}
-- 
cgit v1.2.3