test(db): 空库一键迁移验收 + globalSetup 临时库 + 公共夹具收敛(阶段1验收)
- vitest globalSetup:每次运行 CREATE sf_test_<ts> 临时库 → drizzle-orm 程序化迁移器 全量迁移(apps/api/src/db/migrate.ts,drizzle-kit 无法指向运行时建库)→ 设 DATABASE_URL_TEST 让 getDb() 自动回退到临时库;teardown DROP(WITH FORCE)。 有 default 导出时 vitest 不捕获具名 teardown,故用双具名导出。 - db.full-migrate.test.ts:21 表齐 + 四张不可变触发器在库 + sf_test_* 护栏断言 (current_database 不符即红,杜绝静默连开发库)。 - helpers.ts 收敛:rand/rejectionMessage/createSiteFixture/withImmutableCleanup/ withTestDb,Task 6–12 七文件内联版替换为 import(断言、对照组、清场顺序不变)。 - scripts/migrate-test.sh:docker compose up --wait + pnpm test 一键验收。 - 实证:75 测试全绿(72+3);开发库 21 表行数总和 0(零写入);无 sf_test_* 残留。 Co-Authored-By: Claude Code <noreply@anthropic.com>
This commit is contained in:
@@ -1,19 +1,18 @@
|
||||
// 来源域集成测试:快照不可变触发器(UPDATE/DELETE 拒绝)+ 三处唯一约束在真 PG 上生效。
|
||||
// 生命周期纪律:随机后缀插入 + finally 反序清理,可重复运行不残留脏数据。
|
||||
// 注意:source_snapshots 不可变触发器连 DELETE 也拒——清理时先 DISABLE 该触发器再删行后 ENABLE
|
||||
// Task 13 起夹具走公共 helpers(createSiteFixture/withImmutableCleanup/withTestDb/rejectionMessage);
|
||||
// 生命周期纪律不变:随机后缀插入 + finally 反序清理,可重复运行不残留脏数据。
|
||||
// 注意:source_snapshots 不可变触发器连 DELETE 也拒——清场走 withImmutableCleanup
|
||||
// (不可变表的生产语义是永不删;测试清场是唯一例外,走显式 DDL 开关且删除限定本站点行)。
|
||||
import { describe, it, expect } from 'vitest'
|
||||
import { getDb } from '../../apps/api/src/db/client.js'
|
||||
import { createSiteFixture, withImmutableCleanup, withTestDb, rejectionMessage, type Db } from './helpers.js'
|
||||
|
||||
type Db = ReturnType<typeof getDb>
|
||||
type Fixture = { siteId: string; connectionId: string; snapshotId: string; sha256: string; idempotencyKey: string }
|
||||
|
||||
// 建链(FK 依赖顺序):site → source_connection → source_snapshot → import_job
|
||||
async function setupSourceFixture(db: Db): Promise<Fixture> {
|
||||
const rand = `${Date.now()}-${Math.random().toString(36).slice(2, 8)}`
|
||||
const site = (await db.execute(`insert into sites (id, key, name, default_locale, production_origin)
|
||||
values (gen_random_uuid(), 'src-${rand}', 'SrcFixture', 'en', 'https://src.example') returning id`)) as any
|
||||
const siteId = site[0].id
|
||||
const site = await createSiteFixture(db, 'src')
|
||||
const rand = site.rand
|
||||
const siteId = site.siteId
|
||||
const conn = (await db.execute(`insert into source_connections (id, site_id, kind, name, configuration)
|
||||
values (gen_random_uuid(), '${siteId}', 'wxr', 'conn-${rand}', '{}') returning id`)) as any
|
||||
const connectionId = conn[0].id
|
||||
@@ -24,40 +23,24 @@ async function setupSourceFixture(db: Db): Promise<Fixture> {
|
||||
return { siteId, connectionId, snapshotId: snap[0].id, sha256: snap[0].sha256, idempotencyKey: job[0].idempotency_key }
|
||||
}
|
||||
|
||||
// 反序清理:子表常规删除;快照表临时 DISABLE 不可变触发器;最后删 site。
|
||||
// 反序清理:子表常规删除;快照表不可变,清场走 DISABLE→删→ENABLE 封装;最后删 site。
|
||||
async function cleanupSourceFixture(db: Db, f: Fixture) {
|
||||
await db.execute(`delete from source_records where site_id = '${f.siteId}'`)
|
||||
await db.execute(`delete from import_jobs where site_id = '${f.siteId}'`)
|
||||
await db.execute(`alter table source_snapshots disable trigger source_snapshots_immutable`)
|
||||
try {
|
||||
await db.execute(`delete from source_snapshots where site_id = '${f.siteId}'`)
|
||||
} finally {
|
||||
await db.execute(`alter table source_snapshots enable trigger source_snapshots_immutable`)
|
||||
}
|
||||
await withImmutableCleanup(db, 'source_snapshots', 'source_snapshots_immutable', `site_id = '${f.siteId}'`)
|
||||
await db.execute(`delete from source_connections where site_id = '${f.siteId}'`)
|
||||
await db.execute(`delete from sites where id = '${f.siteId}'`)
|
||||
}
|
||||
|
||||
async function withSourceFixture(fn: (db: Db, f: Fixture) => Promise<void>) {
|
||||
const db = getDb()
|
||||
const f = await setupSourceFixture(db)
|
||||
try {
|
||||
await fn(db, f)
|
||||
} finally {
|
||||
await cleanupSourceFixture(db, f)
|
||||
await (db as any).$client.end()
|
||||
}
|
||||
}
|
||||
|
||||
// drizzle 会把 PostgresError 包成 DrizzleQueryError(message 只有 "Failed query: ..."),
|
||||
// 服务端原始错误文本在 cause 里——断言触发器/约束身份时必须看 cause。
|
||||
async function rejectionMessage(db: Db, query: string): Promise<string> {
|
||||
try {
|
||||
await db.execute(query)
|
||||
} catch (e: any) {
|
||||
return String(e?.cause?.message ?? e?.message ?? e)
|
||||
}
|
||||
throw new Error(`expected query to be rejected: ${query}`)
|
||||
await withTestDb(async (db) => {
|
||||
const f = await setupSourceFixture(db)
|
||||
try {
|
||||
await fn(db, f)
|
||||
} finally {
|
||||
await cleanupSourceFixture(db, f)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
describe('来源域约束', () => {
|
||||
|
||||
Reference in New Issue
Block a user