92 lines
3.1 KiB
Go
92 lines
3.1 KiB
Go
package database
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/testsupport/mysqltest"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/testsupport/postgrestest"
|
|
)
|
|
|
|
func TestDomainChallengeMigration(t *testing.T) {
|
|
for _, driver := range []string{"sqlite", "mysql", "postgres"} {
|
|
t.Run(driver, func(t *testing.T) {
|
|
var cfg config.DatabaseConfig
|
|
switch driver {
|
|
case "sqlite":
|
|
cfg = sqliteBusinessUpgradeConfig(t)
|
|
case "mysql":
|
|
cfg = mysqltest.New(t).Database(t)
|
|
case "postgres":
|
|
cfg = postgrestest.New(t).Database(t)
|
|
}
|
|
db, err := Open(t.Context(), cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
if _, err := db.DB.ExecContext(t.Context(), migrationTableSQL(driver)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for version := 1; version <= 25; version++ {
|
|
if err := applyMigration(t.Context(), db.DB, driver, version); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if _, err := db.DB.ExecContext(t.Context(), `INSERT INTO domains (id,name,certificate_type,sans_json,status) VALUES ('legacy','legacy.example.com','single','["legacy.example.com"]','valid')`); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for round := 0; round < 2; round++ {
|
|
if err := Migrate(t.Context(), db.DB, driver); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var kind, status string
|
|
if err := db.DB.QueryRowContext(t.Context(), `SELECT challenge_type,status FROM domains WHERE id='legacy'`).Scan(&kind, &status); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if kind != "dns-01" || status != "valid" {
|
|
t.Fatal("升级改变历史域名验证方式或状态")
|
|
}
|
|
}
|
|
if _, err := db.DB.ExecContext(t.Context(), `UPDATE domains SET challenge_type='http-01' WHERE id='legacy'`); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := Migrate(t.Context(), db.DB, driver); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var kind string
|
|
if err := db.DB.QueryRowContext(t.Context(), `SELECT challenge_type FROM domains WHERE id='legacy'`).Scan(&kind); err != nil || kind != "http-01" {
|
|
t.Fatal("重复迁移覆盖 HTTP 验证选择：", err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMySQLDomainChallengeInterruptedMigration(t *testing.T) {
|
|
f := startRealMySQL(t)
|
|
for _, conflict := range []bool{false, true} {
|
|
t.Run(map[bool]string{false: "resume", true: "conflict"}[conflict], func(t *testing.T) {
|
|
db := f.database(t)
|
|
mysqlPrepareVersion(t, db.DB, 25)
|
|
statement := "ALTER TABLE domains ADD COLUMN challenge_type VARCHAR(16) NOT NULL DEFAULT 'dns-01'"
|
|
if conflict {
|
|
statement = "ALTER TABLE domains ADD COLUMN challenge_type VARCHAR(8) NULL DEFAULT NULL"
|
|
}
|
|
if _, err := db.DB.ExecContext(t.Context(), statement); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err := Migrate(t.Context(), db.DB, "mysql")
|
|
if (err != nil) != conflict {
|
|
t.Fatal("中断迁移结构检查结果错误：", err)
|
|
}
|
|
version, err := currentMigrationVersion(t.Context(), db.DB)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if conflict && version != 25 || !conflict && version != latestMigrationVersion {
|
|
t.Fatal("迁移版本记录错误")
|
|
}
|
|
})
|
|
}
|
|
}
|