mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
49 lines
1.9 KiB
TypeScript
49 lines
1.9 KiB
TypeScript
import crypto from "node:crypto";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import pg from "pg";
|
|
|
|
// Each test file (forks pool, isolated) gets its own Postgres database cloned
|
|
// from the migrated template built in tests/global-setup.ts, plus its own
|
|
// workspace dir. setupFiles run before any app module loads, so
|
|
// apps/api/src/config.ts captures the per-file DATABASE_URL.
|
|
const suffix = `${process.pid}_${crypto.randomUUID().slice(0, 8).replace(/-/g, "")}`;
|
|
const forkDir = path.join(os.tmpdir(), `SnapOtter-test-${suffix}`);
|
|
process.env.WORKSPACE_PATH = path.join(forkDir, "workspace");
|
|
|
|
const baseUrl = process.env.TEST_PG_BASE_URL;
|
|
if (!baseUrl) {
|
|
throw new Error("TEST_PG_BASE_URL missing; tests/global-setup.ts did not run");
|
|
}
|
|
|
|
const redisBaseUrl = process.env.TEST_REDIS_BASE_URL;
|
|
if (!redisBaseUrl) {
|
|
throw new Error("TEST_REDIS_BASE_URL missing; tests/global-setup.ts did not run");
|
|
}
|
|
process.env.REDIS_URL = redisBaseUrl;
|
|
process.env.BULLMQ_PREFIX = `snapotter_test_${suffix}`;
|
|
|
|
// Heavy format conversions can exceed the 8s production default under parallel
|
|
// test forks; 30s keeps tool routes synchronous (200) in tests while production stays at 8s.
|
|
process.env.SYNC_WAIT_MS = "30000";
|
|
const dbName = `snapotter_test_${suffix}`; // pid digits + uuid hex: identifier-safe
|
|
const admin = new pg.Client({ connectionString: baseUrl });
|
|
await admin.connect();
|
|
// Concurrent CREATE DATABASE ... TEMPLATE from parallel forks can transiently
|
|
// conflict; retry briefly.
|
|
let created = false;
|
|
for (let attempt = 0; attempt < 5 && !created; attempt++) {
|
|
try {
|
|
await admin.query(`CREATE DATABASE ${dbName} TEMPLATE snapotter_template`);
|
|
created = true;
|
|
} catch (err) {
|
|
if (attempt === 4) throw err;
|
|
await new Promise((r) => setTimeout(r, 150 + Math.floor(150 * attempt)));
|
|
}
|
|
}
|
|
await admin.end();
|
|
|
|
const forkUrl = new URL(baseUrl);
|
|
forkUrl.pathname = `/${dbName}`;
|
|
process.env.DATABASE_URL = forkUrl.toString();
|