mirror of
https://github.com/grafana/grafana.git
synced 2025-09-27 19:23:54 +08:00
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package migrations
|
|
|
|
import (
|
|
"testing"
|
|
|
|
. "github.com/grafana/grafana/pkg/services/sqlstore/migrator"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore/sqlutil"
|
|
"xorm.io/xorm"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestMigrations(t *testing.T) {
|
|
testDBs := []sqlutil.TestDB{
|
|
sqlutil.SQLite3TestDB(),
|
|
}
|
|
|
|
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)
|
|
// we currently skip to migrations. We should rewrite skipped migrations to write in the log as well.
|
|
// until then we have to keep this
|
|
expectedMigrations := mg.MigrationsCount()
|
|
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)
|
|
})
|
|
}
|
|
}
|