64 lines
2.2 KiB
Go
64 lines
2.2 KiB
Go
package database
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"path/filepath"
|
||
"testing"
|
||
|
||
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
||
)
|
||
|
||
func TestOpenAndMigrateSQLite(t *testing.T) {
|
||
dir := t.TempDir()
|
||
cfg := config.Default(dir)
|
||
cfg.Database.DSN = filepath.Join(dir, "data", "test.db")
|
||
db, err := Open(context.Background(), cfg.Database)
|
||
if err != nil {
|
||
t.Fatalf("打开 SQLite 失败：%v", err)
|
||
}
|
||
defer db.Close()
|
||
if err := Migrate(context.Background(), db.DB, "sqlite"); err != nil {
|
||
t.Fatalf("SQLite 迁移失败：%v", err)
|
||
}
|
||
var count int
|
||
if err := db.DB.QueryRow("SELECT COUNT(*) FROM schema_migrations").Scan(&count); err != nil {
|
||
t.Fatalf("读取迁移版本失败：%v", err)
|
||
}
|
||
if count != 4 {
|
||
t.Fatalf("迁移版本数量错误：%d", count)
|
||
}
|
||
var autoRenew bool
|
||
if err := db.DB.QueryRow("SELECT auto_renew FROM domains LIMIT 1").Scan(&autoRenew); err != nil && err != sql.ErrNoRows {
|
||
t.Fatalf("检查域名选项字段失败：%v", err)
|
||
}
|
||
}
|
||
|
||
func TestMySQLMigrationStatementsRecognizeRecoverableChanges(t *testing.T) {
|
||
cases := []struct {
|
||
statement string
|
||
table string
|
||
name string
|
||
}{
|
||
{statement: "ALTER TABLE domains ADD COLUMN auto_renew BOOLEAN NOT NULL DEFAULT TRUE", table: "domains", name: "auto_renew"},
|
||
{statement: "ALTER TABLE `tasks` ADD COLUMN `stage` VARCHAR(32) NOT NULL DEFAULT 'queued'", table: "tasks", name: "stage"},
|
||
{statement: "CREATE INDEX idx_tasks_status ON tasks(status)", table: "tasks", name: "idx_tasks_status"},
|
||
{statement: "CREATE INDEX `idx_tasks_idempotency` ON `tasks`(idempotency_key)", table: "tasks", name: "idx_tasks_idempotency"},
|
||
}
|
||
for _, item := range cases {
|
||
if matches := mysqlAddColumnPattern.FindStringSubmatch(item.statement); len(matches) == 3 {
|
||
if matches[1] != item.table || matches[2] != item.name {
|
||
t.Errorf("字段迁移解析结果不正确：statement=%q matches=%v", item.statement, matches)
|
||
}
|
||
continue
|
||
}
|
||
if matches := mysqlCreateIndexPattern.FindStringSubmatch(item.statement); len(matches) == 3 {
|
||
if matches[1] != item.name || matches[2] != item.table {
|
||
t.Errorf("索引迁移解析结果不正确：statement=%q matches=%v", item.statement, matches)
|
||
}
|
||
continue
|
||
}
|
||
t.Errorf("可恢复的 MySQL 迁移语句未被识别：%q", item.statement)
|
||
}
|
||
}
|