empayre/fleet
14
0
Fork 0
mirror of https://github.com/empayre/fleet.git synced 2024-11-06 17:05:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update mysql test setup (#343)

Browse Source 
reuse a single conneciton pool
close connection gracefully at the end of tests
setup and teardown tables in between each test
This commit is contained in:
Victor Vrantchan 2016-10-25 11:01:14 -04:00 committed by GitHub
parent ee3d96eb53
commit efffe50a55
1 changed files with 28 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
server/datastore/gorm_test.go
View File

@ -6,44 +6,47 @@ import (
"testing"
"github.com/kolide/kolide-ose/server/kolide"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func setupGorm(t *testing.T) kolide.Datastore {
user := "kolide"
password := "kolide"
dbName := "kolide"
func setupGorm(t *testing.T) (ds kolide.Datastore, teardown func()) {
var (
user = "kolide"
password = "kolide"
dbName = "kolide"
host = "127.0.0.1"
)
// try container first
host := os.Getenv("MYSQL_PORT_3306_TCP_ADDR")
if host == "" {
host = "127.0.0.1"
// use linked container if available.
if h, ok := os.LookupEnv("MYSQL_PORT_3306_TCP_ADDR"); ok {
host = h
}
host = fmt.Sprintf("%s:3306", host)
connString := fmt.Sprintf("%s:%s@(%s)/%s?charset=utf8&parseTime=True&loc=Local", user, password, host, dbName)
connString := fmt.Sprintf("%s:%s@(%s:3306)/%s?charset=utf8&parseTime=True&loc=Local", user, password, host, dbName)
ds, err := New("gorm-mysql", connString)
err = ds.Migrate()
assert.Nil(t, err)
return ds
}
func teardownGorm(t *testing.T, ds kolide.Datastore) {
err := ds.Drop()
assert.Nil(t, err)
require.Nil(t, err)
teardown = func() {
db, ok := ds.(gormDB)
if !ok {
panic("expected gormDB datastore")
}
db.DB.Close()
}
return ds, teardown
}
func TestGorm(t *testing.T) {
address := os.Getenv("MYSQL_ADDR")
if address == "" {
if _, ok := os.LookupEnv("MYSQL_TEST"); !ok {
t.SkipNow()
}
ds, teardown := setupGorm(t)
defer teardown()
for _, f := range testFunctions {
t.Run(functionName(f), func(t *testing.T) {
ds := setupGorm(t)
defer teardownGorm(t, ds)
err := ds.Migrate()
require.Nil(t, err)
f(t, ds)
err = ds.Drop()
require.Nil(t, err)
})
}
}