From dbcdf05b3eb46192d550042753821cb9f444a83f Mon Sep 17 00:00:00 2001 From: xuans Date: Thu, 1 Oct 2026 17:38:13 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3SQLite=E6=AD=A3=E5=BC=8F?= =?UTF-8?q?=E5=91=BD=E4=BB=A4=E8=A1=8C=E9=AA=8C=E6=94=B6=E7=89=88=E6=9C=AC?= =?UTF-8?q?=E5=B9=B6=E8=A6=86=E7=9B=96=E9=83=A8=E7=BD=B2=E5=8E=86=E5=8F=B2?= =?UTF-8?q?=E6=81=A2=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../internal/cli/sqlite_upgrade_cli_test.go | 4 +- .../cli/sqlite_upgrade_history_test.go | 84 +++++++++++++++++++ 2 files changed, 87 insertions(+), 1 deletion(-) create mode 100644 backend/internal/cli/sqlite_upgrade_history_test.go diff --git a/backend/internal/cli/sqlite_upgrade_cli_test.go b/backend/internal/cli/sqlite_upgrade_cli_test.go index c3e228b..422c643 100644 --- a/backend/internal/cli/sqlite_upgrade_cli_test.go +++ b/backend/internal/cli/sqlite_upgrade_cli_test.go @@ -85,6 +85,7 @@ func TestSQLiteBusinessUpgradeStandaloneCLI(t *testing.T) { runSQLiteProductCLI(t, binary, fixture, "certificate", "list") assertSQLiteCLIUpgradeUsable(t, fixture) createSQLiteProjectForBackup(t, fixture) + seedSQLiteCLIDeploymentHistory(t, fixture) upgraded := snapshotSQLiteCLIState(t, *cfg) currentArchive := filepath.Join(t.TempDir(), "upgraded-backup") runSQLiteProductCLI(t, binary, fixture, "backup", "--output", currentArchive) @@ -110,6 +111,7 @@ func TestSQLiteBusinessUpgradeStandaloneCLI(t *testing.T) { _ = current.Close() t.Fatal(err) } + mutateSQLiteCLIDeploymentHistory(t, current.DB) if err := current.Close(); err != nil { t.Fatal(err) } @@ -289,7 +291,7 @@ func assertSQLiteCLIUpgradeUsable(t *testing.T, fixture sqliteBusinessFixture) { } defer db.Close() var version, bindings int - if err := db.DB.QueryRowContext(t.Context(), "SELECT MAX(version) FROM schema_migrations").Scan(&version); err != nil || version != 14 { + if err := db.DB.QueryRowContext(t.Context(), "SELECT MAX(version) FROM schema_migrations").Scan(&version); err != nil || version != expectedSQLiteCLIMigrationVersion(t, db.DB) { t.Fatalf("正式应用入口未完成历史升级：%d，%v", version, err) } project, err := repository.NewSQLProjectRepository(db.DB).GetProject(t.Context(), domain.DefaultProjectID) diff --git a/backend/internal/cli/sqlite_upgrade_history_test.go b/backend/internal/cli/sqlite_upgrade_history_test.go new file mode 100644 index 0000000..a02af02 --- /dev/null +++ b/backend/internal/cli/sqlite_upgrade_history_test.go @@ -0,0 +1,84 @@ +package cli + +import ( + "database/sql" + "fmt" + "os" + "path/filepath" + "strings" + "testing" + + "git.pcux.cn/xuans/Scert/backend/internal/database" +) + +// 以当批源码目录为预期，同时校验正式程序写入的迁移历史连续且完整。 +func expectedSQLiteCLIMigrationVersion(t *testing.T, db *sql.DB) int { + t.Helper() + entries, err := os.ReadDir(filepath.Join("..", "database", "migrations", "sqlite")) + if err != nil || len(entries) == 0 { + t.Fatalf("无法读取本批 SQLite 迁移目录：%v", err) + } + for index, entry := range entries { + if entry.IsDir() || !strings.HasPrefix(entry.Name(), fmt.Sprintf("%04d_", index+1)) || !strings.HasSuffix(entry.Name(), ".sql") { + t.Fatalf("SQLite 迁移文件序列无效：%s", entry.Name()) + } + } + rows, err := db.QueryContext(t.Context(), "SELECT version FROM schema_migrations ORDER BY version") + if err != nil { + t.Fatal(err) + } + defer rows.Close() + count := 0 + for rows.Next() { + var version int + if err := rows.Scan(&version); err != nil { + t.Fatal(err) + } + count++ + if version != count { + t.Fatalf("正式程序迁移历史不连续：期望 %d，实际 %d", count, version) + } + } + if err := rows.Err(); err != nil { + t.Fatal(err) + } + if count != len(entries) { + t.Fatalf("正式程序迁移不完整：源码 %d 项，数据库 %d 项", len(entries), count) + } + return len(entries) +} + +// 仅在隔离样本中写入终态事务，用于验证存储恢复，不作为远端恢复证据。 +func seedSQLiteCLIDeploymentHistory(t *testing.T, fixture sqliteBusinessFixture) { + t.Helper() + db, err := database.Open(t.Context(), fixture.config.Database) + if err != nil { + t.Fatal(err) + } + defer db.Close() + for _, statement := range []string{ + `INSERT INTO task_idempotency (domain_id, idempotency_key, task_id) VALUES ('domain', 'backup-deploy-key', 'task')`, + `INSERT INTO task_attempts (id, task_id, attempt_number, stage, status, message) VALUES ('backup-attempt', 'task', 1, 'certificate_deploy', 'success', '历史部署尝试')`, + `INSERT INTO task_deployment_results (task_id, attempt_id, attempt_number, result_index, server_id, server_name, status, message, backup_id) VALUES ('task', 'backup-attempt', 1, 0, 'server', '历史服务器', 'success', '历史部署结果', 'backup-transaction')`, + `INSERT INTO deployment_transactions (id, task_id, attempt_id, attempt_number, result_index, server_id, server_name, certificate_path, certificate_chain_path, private_key_path, temporary_paths_json, installed_paths_json, backups_json, lock_paths_json, reload_attempted, phase, status, message, recovery_evidence_json) VALUES ('backup-transaction', 'task', 'backup-attempt', 1, 0, 'server', '历史服务器', '/test/cert.pem', '/test/chain.pem', '/test/key.pem', '[]', '["/test/cert.pem"]', '[]', '[]', 1, 'committed', 'committed', '历史事务', '{"version":1,"lock_token":"fixture-lock","address":"192.0.2.10","port":22}')`, + } { + if _, err := db.DB.ExecContext(t.Context(), statement); err != nil { + t.Fatal(err) + } + } +} + +// 主动改变新增表，避免用空表或未变化的数据宣称恢复成功。 +func mutateSQLiteCLIDeploymentHistory(t *testing.T, db *sql.DB) { + t.Helper() + for _, statement := range []string{ + `UPDATE task_idempotency SET idempotency_key = 'changed-key'`, + `UPDATE task_attempts SET message = '待恢复尝试'`, + `UPDATE task_deployment_results SET message = '待恢复结果'`, + `UPDATE deployment_transactions SET recovery_evidence_json = '{}', message = '待恢复证据'`, + } { + if _, err := db.ExecContext(t.Context(), statement); err != nil { + t.Fatal(err) + } + } +}