64 lines
2.9 KiB
Go
64 lines
2.9 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", "bad-column-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", "bad-column-collation":
|
|
var before, after string
|
|
query := `SELECT table_collation FROM information_schema.tables WHERE table_schema = DATABASE() AND table_name = 'dns_execution_leases'`
|
|
if err := db.DB.QueryRowContext(t.Context(), query).Scan(&before); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// 必须真正改变排序规则，不能把测试库默认值当作故障注入。
|
|
collation := "utf8mb4_bin"
|
|
if before == collation {
|
|
collation = "utf8mb4_unicode_ci"
|
|
}
|
|
statement := "ALTER TABLE dns_execution_leases CONVERT TO CHARACTER SET utf8mb4 COLLATE " + collation
|
|
if boundary == "bad-column-collation" {
|
|
statement = "ALTER TABLE dns_execution_leases MODIFY COLUMN task_id VARCHAR(64) CHARACTER SET utf8mb4 COLLATE " + collation + " NOT NULL"
|
|
query = `SELECT collation_name FROM information_schema.columns WHERE table_schema = DATABASE() AND table_name = 'dns_execution_leases' AND column_name = 'task_id'`
|
|
}
|
|
mysqlDeploymentExec(t, db.DB, statement)
|
|
if err := db.DB.QueryRowContext(t.Context(), query).Scan(&after); err != nil || before == after {
|
|
t.Fatal("排序规则故障注入未实际生效", err)
|
|
}
|
|
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)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|