SQLStore: Fix migrator locking test for SQLite (#45523)

* SQLStore: Fix migrator locking test for SQLite
pull/45536/head
Sofia Papagiannaki 3 years ago committed by GitHub
parent dac2b7ef73
commit 17dee4700c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 22
      pkg/services/sqlstore/migrations/migrations_test.go

@ -65,7 +65,7 @@ func TestMigrationLock(t *testing.T) {
t.Skip()
}
testDB := getTestDB(dbType)
testDB := getTestDB(t, dbType)
x, err := xorm.NewEngine(testDB.DriverName, testDB.ConnStr)
require.NoError(t, err)
@ -154,7 +154,7 @@ func TestMigrationLock(t *testing.T) {
func TestMigratorLocking(t *testing.T) {
dbType := getDBType()
testDB := getTestDB(dbType)
testDB := getTestDB(t, dbType)
x, err := xorm.NewEngine(testDB.DriverName, testDB.ConnStr)
require.NoError(t, err)
@ -172,7 +172,7 @@ func TestMigratorLocking(t *testing.T) {
i := i // capture i variable
t.Run(fmt.Sprintf("run migration %d", i), func(t *testing.T) {
t.Parallel()
err = mg.Start(true, 0)
err := mg.Start(true, 0)
if err != nil {
if errors.Is(err, ErrMigratorIsLocked) {
atomic.AddInt64(&errorNum, 1)
@ -191,7 +191,7 @@ func TestDatabaseLocking(t *testing.T) {
t.Skip()
}
testDB := getTestDB(dbType)
testDB := getTestDB(t, dbType)
x, err := xorm.NewEngine(testDB.DriverName, testDB.ConnStr)
require.NoError(t, err)
@ -281,14 +281,24 @@ func getDBType() string {
return dbType
}
func getTestDB(dbType string) sqlutil.TestDB {
func getTestDB(t *testing.T, dbType string) sqlutil.TestDB {
switch dbType {
case "mysql":
return sqlutil.MySQLTestDB()
case "postgres":
return sqlutil.PostgresTestDB()
default:
return sqlutil.SQLite3TestDB()
f, err := os.CreateTemp(".", "grafana-test-db-")
require.NoError(t, err)
t.Cleanup(func() {
err := os.Remove(f.Name())
require.NoError(t, err)
})
return sqlutil.TestDB{
DriverName: "sqlite3",
ConnStr: f.Name(),
}
}
}

Loading…
Cancel
Save