mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-10 14:38:18 -05:00
f553a70d20
Co-authored-by: colinlyguo <colinlyguo@users.noreply.github.com> Co-authored-by: Péter Garamvölgyi <peter@scroll.io>
87 lines
1.6 KiB
Go
87 lines
1.6 KiB
Go
package migrate
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
_ "github.com/lib/pq"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"scroll-tech/common/docker"
|
|
|
|
"scroll-tech/database"
|
|
)
|
|
|
|
var (
|
|
base *docker.App
|
|
pgDB *sqlx.DB
|
|
)
|
|
|
|
func initEnv(t *testing.T) error {
|
|
// Start db container.
|
|
base.RunDBImage(t)
|
|
|
|
// Create db orm handler.
|
|
factory, err := database.NewOrmFactory(base.DBConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pgDB = factory.GetDB()
|
|
return nil
|
|
}
|
|
|
|
func TestMigrate(t *testing.T) {
|
|
base = docker.NewDockerApp()
|
|
if err := initEnv(t); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
t.Run("testCurrent", testCurrent)
|
|
t.Run("testStatus", testStatus)
|
|
t.Run("testResetDB", testResetDB)
|
|
t.Run("testMigrate", testMigrate)
|
|
t.Run("testRollback", testRollback)
|
|
|
|
t.Cleanup(func() {
|
|
base.Free()
|
|
})
|
|
}
|
|
|
|
func testCurrent(t *testing.T) {
|
|
cur, err := Current(pgDB.DB)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0, int(cur))
|
|
}
|
|
|
|
func testStatus(t *testing.T) {
|
|
status := Status(pgDB.DB)
|
|
assert.NoError(t, status)
|
|
}
|
|
|
|
func testResetDB(t *testing.T) {
|
|
assert.NoError(t, ResetDB(pgDB.DB))
|
|
cur, err := Current(pgDB.DB)
|
|
assert.NoError(t, err)
|
|
// total number of tables.
|
|
assert.Equal(t, 11, int(cur))
|
|
}
|
|
|
|
func testMigrate(t *testing.T) {
|
|
assert.NoError(t, Migrate(pgDB.DB))
|
|
cur, err := Current(pgDB.DB)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, cur > 0)
|
|
}
|
|
|
|
func testRollback(t *testing.T) {
|
|
version, err := Current(pgDB.DB)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, version > 0)
|
|
|
|
assert.NoError(t, Rollback(pgDB.DB, nil))
|
|
|
|
cur, err := Current(pgDB.DB)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, cur+1 == version)
|
|
}
|