79 lines
3.0 KiB
Go
79 lines
3.0 KiB
Go
package database
|
||
|
||
import (
|
||
"testing"
|
||
)
|
||
|
||
func TestSQLiteProjectMigrationPreservesBusiness(t *testing.T) {
|
||
db, err := Open(t.Context(), sqliteBusinessUpgradeConfig(t))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer db.Close()
|
||
createSQLiteHistoricalBusiness(t, db.DB, 9)
|
||
before := snapshotSQLiteHistoricalBusiness(t, db.DB)
|
||
if err := Migrate(t.Context(), db.DB, "sqlite"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
assertSQLiteHistoricalBusiness(t, db.DB, before)
|
||
var name string
|
||
if err := db.DB.QueryRowContext(t.Context(), "SELECT name FROM projects WHERE id = 'default'").Scan(&name); err != nil || name != "默认项目" {
|
||
t.Fatalf("默认项目未创建：%q，%v", name, err)
|
||
}
|
||
if _, err := db.DB.ExecContext(t.Context(), "UPDATE projects SET name = '调整后的默认项目' WHERE id = 'default'"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
// 模拟建表、种子写入已成功，但版本记录尚未落盘的可重试情况。
|
||
if _, err := db.DB.ExecContext(t.Context(), "DELETE FROM schema_migrations WHERE version = 10"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for round := 0; round < 2; round++ {
|
||
if err := Migrate(t.Context(), db.DB, "sqlite"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
var count int
|
||
if err := db.DB.QueryRowContext(t.Context(), "SELECT COUNT(*) FROM projects").Scan(&count); err != nil || count != 1 {
|
||
t.Fatalf("重复迁移改变项目数量：%d，%v", count, err)
|
||
}
|
||
if err := db.DB.QueryRowContext(t.Context(), "SELECT name FROM projects WHERE id = 'default'").Scan(&name); err != nil || name != "调整后的默认项目" {
|
||
t.Fatalf("重复迁移覆盖用户修改：%q，%v", name, err)
|
||
}
|
||
assertSQLiteHistoricalBusiness(t, db.DB, before)
|
||
}
|
||
|
||
func TestSQLiteProjectMigrationRollsBackAndRetries(t *testing.T) {
|
||
db, err := Open(t.Context(), sqliteBusinessUpgradeConfig(t))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer db.Close()
|
||
createSQLiteHistoricalBusiness(t, db.DB, 9)
|
||
before := snapshotSQLiteHistoricalBusiness(t, db.DB)
|
||
// 在登记版本时制造失败，验证同一事务中的新表及默认项目也会回滚。
|
||
_, err = db.DB.ExecContext(t.Context(), `CREATE TRIGGER fail_project_version BEFORE INSERT ON schema_migrations
|
||
WHEN NEW.version = 10 BEGIN SELECT RAISE(ABORT, 'test-only'); END`)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := Migrate(t.Context(), db.DB, "sqlite"); err == nil {
|
||
t.Fatal("版本登记故障未阻断迁移")
|
||
}
|
||
version, err := currentMigrationVersion(t.Context(), db.DB)
|
||
if err != nil || version != 9 {
|
||
t.Fatalf("失败迁移错误登记版本：%d，%v", version, err)
|
||
}
|
||
var count int
|
||
if err := db.DB.QueryRowContext(t.Context(), "SELECT COUNT(*) FROM sqlite_master WHERE name = 'projects'").Scan(&count); err != nil || count != 0 {
|
||
t.Fatalf("失败迁移未回滚项目表：%d，%v", count, err)
|
||
}
|
||
assertSQLiteHistoricalBusiness(t, db.DB, before)
|
||
if _, err := db.DB.ExecContext(t.Context(), "DROP TRIGGER fail_project_version"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := Migrate(t.Context(), db.DB, "sqlite"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
assertSQLiteHistoricalBusiness(t, db.DB, before)
|
||
}
|