68 lines
2.1 KiB
Go
68 lines
2.1 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/testsupport/postgrestest"
|
|
)
|
|
|
|
func TestPostgresRealRecoveryDispositionMigration(t *testing.T) {
|
|
server := postgrestest.New(t)
|
|
for version := 22; version <= 25; version++ {
|
|
t.Run(fmt.Sprint(version), func(t *testing.T) {
|
|
cfg := server.Database(t)
|
|
db, err := Open(t.Context(), cfg)
|
|
if err != nil {
|
|
t.Fatal("隔离迁移连接失败")
|
|
}
|
|
t.Cleanup(func() { _ = db.Close() })
|
|
if _, err := db.DB.ExecContext(t.Context(), migrationTableSQL(cfg.Driver)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for previous := 1; previous < version; previous++ {
|
|
if err := applyMigration(t.Context(), db.DB, cfg.Driver, previous); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
content, err := migrationFiles.ReadFile(fmt.Sprintf("migrations/postgres/%04d_%s.sql", version, migrationName(version)))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// 已执行 DDL 但版本尚未落库时中止事务，不能留下半迁移结构。
|
|
tx, err := db.DB.BeginTx(t.Context(), nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, query := range migrationStatements(content) {
|
|
if _, err := tx.ExecContext(t.Context(), query); err != nil {
|
|
_ = tx.Rollback()
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if err := tx.Rollback(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
current, err := currentMigrationVersion(t.Context(), db.DB)
|
|
if err != nil || current != version-1 {
|
|
t.Fatal("回滚后版本记录改变", err)
|
|
}
|
|
for range 2 {
|
|
if err := Migrate(t.Context(), db.DB, cfg.Driver); err != nil {
|
|
t.Fatal("中断后的迁移重入失败", err)
|
|
}
|
|
}
|
|
current, err = currentMigrationVersion(t.Context(), db.DB)
|
|
if err != nil || current != latestMigrationVersion {
|
|
t.Fatal("迁移未达到当前版本", err)
|
|
}
|
|
for _, table := range []string{"recovery_dispositions", "recovery_claims", "recovery_events"} {
|
|
var count int
|
|
if err := db.DB.QueryRowContext(t.Context(), "SELECT COUNT(*) FROM "+table).Scan(&count); err != nil || count != 0 {
|
|
t.Fatal("处置表未正确初始化", err)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|