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)
				}
			}
		})
	}
}