68 lines
1.9 KiB
TypeScript
68 lines
1.9 KiB
TypeScript
|
import { execSync } from 'node:child_process';
|
||
|
|
import path from 'node:path';
|
||
|
|
import { fileURLToPath } from 'node:url';
|
||
|
|
import pg from 'pg';
|
||
|
|
import { drizzle } from 'drizzle-orm/node-postgres';
|
||
|
|
import { migrate } from 'drizzle-orm/node-postgres/migrator';
|
||
|
|
import {
|
||
|
|
E2E_CONTAINER,
|
||
|
|
E2E_DATABASE_URL,
|
||
|
|
E2E_DB_NAME,
|
||
|
|
E2E_DB_PASSWORD,
|
||
|
|
E2E_DB_PORT,
|
||
|
|
} from './env';
|
||
|
|
|
||
|
|
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
||
|
|
const MIGRATIONS = path.resolve(__dirname, '../../../packages/db/migrations');
|
||
|
|
|
||
|
|
async function waitForPostgres(timeoutMs = 60_000): Promise<void> {
|
||
|
|
const deadline = Date.now() + timeoutMs;
|
||
|
|
let lastErr: unknown;
|
||
|
|
while (Date.now() < deadline) {
|
||
|
|
const client = new pg.Client({ connectionString: E2E_DATABASE_URL });
|
||
|
|
try {
|
||
|
|
await client.connect();
|
||
|
|
await client.query('select 1');
|
||
|
|
await client.end();
|
||
|
|
return;
|
||
|
|
} catch (err) {
|
||
|
|
lastErr = err;
|
||
|
|
await client.end().catch(() => {});
|
||
|
|
await new Promise((r) => setTimeout(r, 500));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
throw new Error(`e2e postgres did not become ready in ${timeoutMs}ms: ${lastErr}`);
|
||
|
|
}
|
||
|
|
|
||
|
|
export default async function setup() {
|
||
|
|
try {
|
||
|
|
execSync(`docker rm -f ${E2E_CONTAINER}`, { stdio: 'ignore' });
|
||
|
|
} catch {
|
||
|
|
// no stale container — fine
|
||
|
|
}
|
||
|
|
|
||
|
|
execSync(
|
||
|
|
`docker run --rm -d --name ${E2E_CONTAINER} -p ${E2E_DB_PORT}:5432 ` +
|
||
|
|
`-e POSTGRES_PASSWORD=${E2E_DB_PASSWORD} -e POSTGRES_DB=${E2E_DB_NAME} postgres:16-alpine`,
|
||
|
|
{ stdio: 'inherit' },
|
||
|
|
);
|
||
|
|
|
||
|
|
try {
|
||
|
|
await waitForPostgres();
|
||
|
|
|
||
|
|
const pool = new pg.Pool({ connectionString: E2E_DATABASE_URL, max: 2 });
|
||
|
|
try {
|
||
|
|
await migrate(drizzle(pool), { migrationsFolder: MIGRATIONS });
|
||
|
|
} finally {
|
||
|
|
await pool.end();
|
||
|
|
}
|
||
|
|
} catch (err) {
|
||
|
|
execSync(`docker rm -f ${E2E_CONTAINER}`, { stdio: 'ignore' });
|
||
|
|
throw err;
|
||
|
|
}
|
||
|
|
|
||
|
|
return () => {
|
||
|
|
execSync(`docker rm -f ${E2E_CONTAINER}`, { stdio: 'ignore' });
|
||
|
|
};
|
||
|
|
}
|