mirror of https://github.com/status-im/migrate.git
44 lines
1019 B
Go
44 lines
1019 B
Go
package stub
|
|
|
|
import (
|
|
"github.com/golang-migrate/migrate/v4"
|
|
"github.com/golang-migrate/migrate/v4/source"
|
|
"github.com/golang-migrate/migrate/v4/source/stub"
|
|
"testing"
|
|
|
|
dt "github.com/golang-migrate/migrate/v4/database/testing"
|
|
)
|
|
|
|
func Test(t *testing.T) {
|
|
s := &Stub{}
|
|
d, err := s.Open("")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dt.Test(t, d, []byte("/* foobar migration */"))
|
|
}
|
|
|
|
func TestMigrate(t *testing.T) {
|
|
s := &Stub{}
|
|
d, err := s.Open("")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
stubMigrations := source.NewMigrations()
|
|
stubMigrations.Append(&source.Migration{Version: 1, Direction: source.Up, Identifier: "CREATE 1"})
|
|
stubMigrations.Append(&source.Migration{Version: 1, Direction: source.Down, Identifier: "DROP 1"})
|
|
src := &stub.Stub{}
|
|
srcDrv, err := src.Open("")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
srcDrv.(*stub.Stub).Migrations = stubMigrations
|
|
m, err := migrate.NewWithInstance("stub", srcDrv, "", d)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
dt.TestMigrate(t, m, []byte("/* foobar migration */"))
|
|
}
|