46 lines
1.8 KiB
Go
46 lines
1.8 KiB
Go
package database
|
|
|
|
import "testing"
|
|
|
|
func TestMySQLDNSExecutionInterruptedMigration(t *testing.T) {
|
|
f := startRealMySQL(t)
|
|
for _, boundary := range []string{"before-create", "after-create", "with-record", "bad-column", "bad-index", "bad-collation"} {
|
|
t.Run(boundary, func(t *testing.T) {
|
|
db := f.database(t)
|
|
mysqlPrepareVersion(t, db.DB, 28)
|
|
statements := mysqlDeploymentStatements(t, 29)
|
|
if boundary != "before-create" {
|
|
mysqlDeploymentExec(t, db.DB, statements[0])
|
|
}
|
|
conflict := true
|
|
switch boundary {
|
|
case "bad-column":
|
|
mysqlDeploymentExec(t, db.DB, "ALTER TABLE dns_execution_leases MODIFY COLUMN account_revision INT NULL")
|
|
case "bad-index":
|
|
mysqlDeploymentExec(t, db.DB, "ALTER TABLE dns_execution_leases DROP INDEX uq_dns_leases_attempt")
|
|
case "bad-collation":
|
|
mysqlDeploymentExec(t, db.DB, "ALTER TABLE dns_execution_leases CONVERT TO CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci")
|
|
default:
|
|
conflict = false
|
|
}
|
|
if boundary == "with-record" {
|
|
mysqlDeploymentExec(t, db.DB, `INSERT INTO dns_execution_leases VALUES ('kept','domain','binding','project','account',9,'cloudflare','secret://dns/test','task','attempt',1,0,'active',NOW(),NOW())`)
|
|
}
|
|
err := Migrate(t.Context(), db.DB, "mysql")
|
|
if (err != nil) != conflict {
|
|
t.Fatal("租约迁移中断恢复或冲突拒绝失败", err)
|
|
}
|
|
version, err := currentMigrationVersion(t.Context(), db.DB)
|
|
if err != nil || (conflict && version != 28) || (!conflict && version != latestMigrationVersion) {
|
|
t.Fatal("错误标记迁移版本", version, err)
|
|
}
|
|
if boundary == "with-record" {
|
|
var status string
|
|
if err := db.DB.QueryRowContext(t.Context(), `SELECT status FROM dns_execution_leases WHERE id = 'kept'`).Scan(&status); err != nil || status != "active" {
|
|
t.Fatal("重入丢失活动秘密引用", err)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|