move from test init()s to TestMain

This commit is contained in:
sentriz
2020-05-03 04:42:29 +01:00
parent 9e4d924fd5
commit 9bf80f4b18
3 changed files with 35 additions and 29 deletions

View File

@@ -3,6 +3,7 @@ package db
import (
"log"
"math/rand"
"os"
"testing"
_ "github.com/jinzhu/gorm/dialects/sqlite"
@@ -10,14 +11,6 @@ import (
var testDB *DB
func init() {
var err error
testDB, err = NewMock()
if err != nil {
log.Fatalf("error opening database: %v\n", err)
}
}
func randKey() string {
letters := []rune("abcdef0123456789")
b := make([]rune, 16)
@@ -44,3 +37,12 @@ func TestGetSetting(t *testing.T) {
t.Errorf("expected %q, got %q", expected, actual)
}
}
func TestMain(m *testing.M) {
var err error
testDB, err = NewMock()
if err != nil {
log.Fatalf("error opening database: %v\n", err)
}
os.Exit(m.Run())
}