Arve Knudsen
2a78d2a61c
* pkg/services: Check errors * pkg/services: Don't treat context.Canceled|context.DeadlineExceeded as error
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package migrations
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/go-xorm/xorm"
|
|
. "github.com/grafana/grafana/pkg/services/sqlstore/migrator"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore/sqlutil"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestMigrations(t *testing.T) {
|
|
testDBs := []sqlutil.TestDB{
|
|
sqlutil.TestDB_Sqlite3,
|
|
}
|
|
|
|
for _, testDB := range testDBs {
|
|
sql := `select count(*) as count from migration_log`
|
|
r := struct {
|
|
Count int64
|
|
}{}
|
|
|
|
Convey("Initial "+testDB.DriverName+" migration", t, func() {
|
|
x, err := xorm.NewEngine(testDB.DriverName, testDB.ConnStr)
|
|
So(err, ShouldBeNil)
|
|
|
|
err = NewDialect(x).CleanDB()
|
|
So(err, ShouldBeNil)
|
|
|
|
_, err = x.SQL(sql).Get(&r)
|
|
So(err, ShouldNotBeNil)
|
|
|
|
mg := NewMigrator(x)
|
|
AddMigrations(mg)
|
|
|
|
err = mg.Start()
|
|
So(err, ShouldBeNil)
|
|
|
|
has, err := x.SQL(sql).Get(&r)
|
|
So(err, ShouldBeNil)
|
|
So(has, ShouldBeTrue)
|
|
expectedMigrations := mg.MigrationsCount() //we currently skip to migrations. We should rewrite skipped migrations to write in the log as well. until then we have to keep this
|
|
So(r.Count, ShouldEqual, expectedMigrations)
|
|
|
|
mg = NewMigrator(x)
|
|
AddMigrations(mg)
|
|
|
|
err = mg.Start()
|
|
So(err, ShouldBeNil)
|
|
|
|
has, err = x.SQL(sql).Get(&r)
|
|
So(err, ShouldBeNil)
|
|
So(has, ShouldBeTrue)
|
|
So(r.Count, ShouldEqual, expectedMigrations)
|
|
})
|
|
}
|
|
}
|