74 lines
3.0 KiB
Go
74 lines
3.0 KiB
Go
package database
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestMySQLRealTaskReportMigrationRecovery(t *testing.T) {
|
|
f := startRealMySQL(t)
|
|
baseline := f.database(t)
|
|
if err := Migrate(t.Context(), baseline.DB, "mysql"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
expectedSchema := mysqlSchemaSnapshot(t, baseline.DB)
|
|
content, err := json.Marshal(map[string]string{"message": strings.Repeat("部署报告", 20000)})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, boundary := range []string{"before_create", "after_create", "with_report"} {
|
|
t.Run(boundary, func(t *testing.T) {
|
|
db := f.database(t)
|
|
mysqlPrepareVersion(t, db.DB, 19)
|
|
if boundary != "before_create" {
|
|
mysqlDeploymentExec(t, db.DB, mysqlDeploymentStatements(t, 20)[0])
|
|
}
|
|
if boundary == "with_report" {
|
|
if _, err := db.DB.ExecContext(t.Context(), `INSERT INTO task_deployment_reports
|
|
(task_id, attempt_id, attempt_number, report_json) VALUES ('report-task', 'report-attempt', 1, ?)`, string(content)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
mysqlAssertUpgrade(t, db.DB, expectedSchema)
|
|
if boundary == "with_report" {
|
|
var stored string
|
|
if err := db.DB.QueryRowContext(t.Context(), `SELECT report_json FROM task_deployment_reports
|
|
WHERE task_id = 'report-task' AND attempt_id = 'report-attempt'`).Scan(&stored); err != nil || stored != string(content) {
|
|
t.Fatalf("恢复迁移必须保留超过普通文本字段容量的中文报告：%v", err)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
for _, invalid := range []struct {
|
|
name string
|
|
statement string
|
|
}{
|
|
{"short_report", "ALTER TABLE task_deployment_reports MODIFY COLUMN report_json TEXT NOT NULL"},
|
|
{"wrong_primary_order", "ALTER TABLE task_deployment_reports DROP PRIMARY KEY, ADD PRIMARY KEY (attempt_id, task_id)"},
|
|
{"nullable_attempt", "ALTER TABLE task_deployment_reports MODIFY COLUMN attempt_number INT NULL"},
|
|
{"wrong_engine", "ALTER TABLE task_deployment_reports ENGINE=MyISAM"},
|
|
{"extra_column", "ALTER TABLE task_deployment_reports ADD COLUMN unexpected INT NOT NULL DEFAULT 0"},
|
|
} {
|
|
t.Run(invalid.name, func(t *testing.T) {
|
|
db := f.database(t)
|
|
mysqlPrepareVersion(t, db.DB, 19)
|
|
mysqlDeploymentExec(t, db.DB, mysqlDeploymentStatements(t, 20)[0])
|
|
mysqlDeploymentExec(t, db.DB, `INSERT INTO task_deployment_reports
|
|
(task_id, attempt_id, attempt_number, report_json) VALUES ('kept-task', 'kept-attempt', 1, '{"status":"failed"}')`)
|
|
mysqlDeploymentExec(t, db.DB, invalid.statement)
|
|
before := fmt.Sprint(mysqlSchemaSnapshot(t, db.DB))
|
|
assertMySQLDeploymentMigrationRejected(t, db.DB, 19)
|
|
if after := fmt.Sprint(mysqlSchemaSnapshot(t, db.DB)); after != before {
|
|
t.Fatal("报告表冲突不得被自动覆盖或修复")
|
|
}
|
|
var stored string
|
|
if err := db.DB.QueryRowContext(t.Context(), `SELECT report_json FROM task_deployment_reports
|
|
WHERE task_id = 'kept-task' AND attempt_id = 'kept-attempt'`).Scan(&stored); err != nil || stored != `{"status":"failed"}` {
|
|
t.Fatalf("拒绝迁移时不得修改已有报告：%v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|