xorm/migrate/migrate_test.go
Lunny Xiao 4bfe6853f5 Fix some comments lint and bug (#1888)
Reviewed-on: #1888
Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
Co-committed-by: Lunny Xiao <xiaolunwen@gmail.com>
2021-04-10 10:57:36 +08:00

148 lines
2.8 KiB
Go

package migrate
import (
"fmt"
"log"
"os"
"testing"
_ "github.com/mattn/go-sqlite3"
"github.com/stretchr/testify/assert"
"xorm.io/xorm"
)
type Person struct {
ID int64
Name string
}
type Pet struct {
ID int64
Name string
PersonID int
}
const (
dbName = "testdb.sqlite3"
)
var (
migrations = []*Migration{
{
ID: "201608301400",
Migrate: func(tx *xorm.Engine) error {
return tx.Sync2(&Person{})
},
Rollback: func(tx *xorm.Engine) error {
return tx.DropTables(&Person{})
},
},
{
ID: "201608301430",
Migrate: func(tx *xorm.Engine) error {
return tx.Sync2(&Pet{})
},
Rollback: func(tx *xorm.Engine) error {
return tx.DropTables(&Pet{})
},
},
}
)
func TestMigration(t *testing.T) {
_ = os.Remove(dbName)
db, err := xorm.NewEngine("sqlite3", dbName)
if err != nil {
log.Fatal(err)
}
defer db.Close()
if err = db.DB().Ping(); err != nil {
log.Fatal(err)
}
m := New(db, DefaultOptions, migrations)
err = m.Migrate()
assert.NoError(t, err)
exists, _ := db.IsTableExist(&Person{})
assert.True(t, exists)
exists, _ = db.IsTableExist(&Pet{})
assert.True(t, exists)
assert.Equal(t, 2, tableCount(db, "migrations"))
err = m.RollbackLast()
assert.NoError(t, err)
exists, _ = db.IsTableExist(&Person{})
assert.True(t, exists)
exists, _ = db.IsTableExist(&Pet{})
assert.False(t, exists)
assert.Equal(t, 1, tableCount(db, "migrations"))
err = m.RollbackLast()
assert.NoError(t, err)
exists, _ = db.IsTableExist(&Person{})
assert.False(t, exists)
exists, _ = db.IsTableExist(&Pet{})
assert.False(t, exists)
assert.Equal(t, 0, tableCount(db, "migrations"))
}
func TestInitSchema(t *testing.T) {
os.Remove(dbName)
db, err := xorm.NewEngine("sqlite3", dbName)
if err != nil {
log.Fatal(err)
}
defer db.Close()
if err = db.DB().Ping(); err != nil {
log.Fatal(err)
}
m := New(db, DefaultOptions, migrations)
m.InitSchema(func(tx *xorm.Engine) error {
if err := tx.Sync2(&Person{}); err != nil {
return err
}
return tx.Sync2(&Pet{})
})
err = m.Migrate()
assert.NoError(t, err)
exists, _ := db.IsTableExist(&Person{})
assert.True(t, exists)
exists, _ = db.IsTableExist(&Pet{})
assert.True(t, exists)
assert.Equal(t, 2, tableCount(db, "migrations"))
}
func TestMissingID(t *testing.T) {
os.Remove(dbName)
db, err := xorm.NewEngine("sqlite3", dbName)
assert.NoError(t, err)
if db != nil {
defer db.Close()
}
assert.NoError(t, db.DB().Ping())
migrationsMissingID := []*Migration{
{
Migrate: func(tx *xorm.Engine) error {
return nil
},
},
}
m := New(db, DefaultOptions, migrationsMissingID)
assert.Equal(t, ErrMissingID, m.Migrate())
}
func tableCount(db *xorm.Engine, tableName string) (count int) {
row := db.DB().QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", tableName))
row.Scan(&count)
return
}