36 lines
1.3 KiB
TypeScript
36 lines
1.3 KiB
TypeScript
import { test } from "bun:test";
|
|
import { connect, migrate, type Db } from "@stc/db";
|
|
import { runMigrations } from "../src/server/migrate";
|
|
|
|
const PG = process.env.STC_TEST_PG_URL;
|
|
|
|
/** Fresh migrated in-memory SQLite (and a fresh Postgres schema when STC_TEST_PG_URL is set), with learner migrations applied. */
|
|
export function forEachDb(name: string, fn: (db: Db) => Promise<void>, timeout?: number): void {
|
|
test(`${name} [sqlite]`, async () => {
|
|
const db = await connect("sqlite://:memory:");
|
|
try { await migrate(db); await runMigrations(db); await fn(db); } finally { await db.close(); }
|
|
}, timeout);
|
|
if (!PG) return;
|
|
test(`${name} [postgres]`, async () => {
|
|
const { db, drop } = await pgSchemaDb();
|
|
try { await migrate(db); await runMigrations(db); await fn(db); } finally { await drop(); }
|
|
}, timeout);
|
|
}
|
|
|
|
export async function pgSchemaDb(): Promise<{ db: Db; drop(): Promise<void> }> {
|
|
const schema = `t_${Bun.randomUUIDv7().replaceAll("-", "")}`;
|
|
const admin = await connect(PG!);
|
|
await admin.sql.unsafe(`CREATE SCHEMA ${schema}`);
|
|
const u = new URL(PG!);
|
|
u.searchParams.set("options", `-c search_path=${schema}`);
|
|
const db = await connect(u.toString());
|
|
return {
|
|
db,
|
|
async drop() {
|
|
await db.close();
|
|
await admin.sql.unsafe(`DROP SCHEMA ${schema} CASCADE`);
|
|
await admin.close();
|
|
},
|
|
};
|
|
}
|