diff --git a/server/src/routes/setup-token-route.test.ts b/server/src/routes/setup-token-route.test.ts index 97357ef7966..b101fdd276a 100644 --- a/server/src/routes/setup-token-route.test.ts +++ b/server/src/routes/setup-token-route.test.ts @@ -361,6 +361,20 @@ function buildTransport(opts: { onSubmit?: "complete" | "throw" | "pending" } = // --- App builder with a capturing request logger ----------------------------- +function loadRouteModules() { + return Promise.all([ + vi.importActual("../routes/agents.js"), + vi.importActual("../middleware/index.js"), + vi.importActual("pino"), + vi.importActual("pino-http"), + vi.importActual( + "../middleware/redact-sensitive.js", + ), + ]); +} + +let routeModules: Awaited>; + interface AppHandle { app: express.Express; logLines: string[]; @@ -372,16 +386,7 @@ async function createApp(opts: { confidentialEdgeTlsTerminated?: boolean; transport?: TransportHandle; } = {}): Promise { - const [{ agentRoutes }, { errorHandler }, pinoModule, pinoHttpModule, redactModule] = - await Promise.all([ - vi.importActual("../routes/agents.js"), - vi.importActual("../middleware/index.js"), - vi.importActual("pino"), - vi.importActual("pino-http"), - vi.importActual( - "../middleware/redact-sensitive.js", - ), - ]); + const [{ agentRoutes }, { errorHandler }, pinoModule, pinoHttpModule, redactModule] = routeModules; const { pino } = pinoModule; const { pinoHttp } = pinoHttpModule; const { redactSensitive } = redactModule; @@ -465,9 +470,10 @@ async function settle(): Promise { } } -beforeAll(() => { +beforeAll(async () => { vi.resetModules(); registerModuleMocks(); + routeModules = await loadRouteModules(); }); beforeEach(() => {