wehub-resource-sync
54 lines
1.5 KiB
TypeScript
54 lines
1.5 KiB
TypeScript
// Per-worker access to the shared TestServer started by globalSetup. Each
|
|
// test file imports `getTestServer()` once at module top-level; the returned
|
|
// value is a singleton within that worker process.
|
|
//
|
|
// `webapp.fetch(path)` prepends the shared baseUrl. The PrismaClient is
|
|
// constructed lazily and disconnected on test-suite end via afterAll in the
|
|
// importing file (or left to the worker shutting down).
|
|
|
|
import { PrismaClient } from "@trigger.dev/database";
|
|
import { afterAll, inject } from "vitest";
|
|
|
|
interface SharedWebapp {
|
|
baseUrl: string;
|
|
fetch(path: string, init?: RequestInit): Promise<Response>;
|
|
}
|
|
|
|
interface SharedTestServer {
|
|
webapp: SharedWebapp;
|
|
prisma: PrismaClient;
|
|
}
|
|
|
|
let cached: SharedTestServer | undefined;
|
|
|
|
export function getTestServer(): SharedTestServer {
|
|
if (cached) return cached;
|
|
|
|
const baseUrl = inject("baseUrl");
|
|
const databaseUrl = inject("databaseUrl");
|
|
|
|
if (!baseUrl || !databaseUrl) {
|
|
throw new Error(
|
|
"globalSetup didn't provide baseUrl/databaseUrl — run via vitest.e2e.full.config.ts"
|
|
);
|
|
}
|
|
|
|
const prisma = new PrismaClient({ datasources: { db: { url: databaseUrl } } });
|
|
|
|
cached = {
|
|
webapp: {
|
|
baseUrl,
|
|
fetch: (path, init) => fetch(`${baseUrl}${path}`, init),
|
|
},
|
|
prisma,
|
|
};
|
|
|
|
// Disconnect the PrismaClient when the worker is done. globalSetup's
|
|
// teardown stops the container; this just releases the per-worker pool.
|
|
afterAll(async () => {
|
|
await prisma.$disconnect().catch(() => {});
|
|
});
|
|
|
|
return cached;
|
|
}
|