diff --git a/src/gateway/gateway.e2e.test.ts b/src/gateway/gateway.test.ts similarity index 100% rename from src/gateway/gateway.e2e.test.ts rename to src/gateway/gateway.test.ts diff --git a/src/gateway/openai-http.e2e.test.ts b/src/gateway/openai-http.test.ts similarity index 100% rename from src/gateway/openai-http.e2e.test.ts rename to src/gateway/openai-http.test.ts diff --git a/src/gateway/openresponses-http.e2e.test.ts b/src/gateway/openresponses-http.test.ts similarity index 100% rename from src/gateway/openresponses-http.e2e.test.ts rename to src/gateway/openresponses-http.test.ts diff --git a/src/gateway/server.agent.gateway-server-agent-b.e2e.test.ts b/src/gateway/server.agent.gateway-server-agent-b.test.ts similarity index 99% rename from src/gateway/server.agent.gateway-server-agent-b.e2e.test.ts rename to src/gateway/server.agent.gateway-server-agent-b.test.ts index 9468a7e8c..0515c14c1 100644 --- a/src/gateway/server.agent.gateway-server-agent-b.e2e.test.ts +++ b/src/gateway/server.agent.gateway-server-agent-b.test.ts @@ -18,6 +18,7 @@ import { rpcReq, startServerWithClient, testState, + trackConnectChallengeNonce, withGatewayServer, writeSessionStore, } from "./test-helpers.js"; @@ -341,6 +342,7 @@ describe("gateway server agent", () => { await withGatewayServer(async ({ port }) => { const dial = async () => { const ws = new WebSocket(`ws://127.0.0.1:${port}`); + trackConnectChallengeNonce(ws); await new Promise((resolve) => ws.once("open", resolve)); await connectOk(ws); return ws; diff --git a/src/gateway/server.auth.e2e.test.ts b/src/gateway/server.auth.test.ts similarity index 100% rename from src/gateway/server.auth.e2e.test.ts rename to src/gateway/server.auth.test.ts diff --git a/src/gateway/server.canvas-auth.e2e.test.ts b/src/gateway/server.canvas-auth.test.ts similarity index 100% rename from src/gateway/server.canvas-auth.e2e.test.ts rename to src/gateway/server.canvas-auth.test.ts diff --git a/src/gateway/server.chat.gateway-server-chat.e2e.test.ts b/src/gateway/server.chat.gateway-server-chat.test.ts similarity index 99% rename from src/gateway/server.chat.gateway-server-chat.e2e.test.ts rename to src/gateway/server.chat.gateway-server-chat.test.ts index 91961c6fb..d72e54f6b 100644 --- a/src/gateway/server.chat.gateway-server-chat.e2e.test.ts +++ b/src/gateway/server.chat.gateway-server-chat.test.ts @@ -12,6 +12,7 @@ import { onceMessage, rpcReq, testState, + trackConnectChallengeNonce, writeSessionStore, } from "./test-helpers.js"; import { agentCommand } from "./test-helpers.mocks.js"; @@ -78,6 +79,7 @@ describe("gateway server chat", () => { webchatWs = new WebSocket(`ws://127.0.0.1:${port}`, { headers: { origin: `http://127.0.0.1:${port}` }, }); + trackConnectChallengeNonce(webchatWs); await new Promise((resolve) => webchatWs?.once("open", resolve)); await connectOk(webchatWs, { client: { @@ -361,6 +363,7 @@ describe("gateway server chat", () => { const webchatWs = new WebSocket(`ws://127.0.0.1:${port}`, { headers: { origin: `http://127.0.0.1:${port}` }, }); + trackConnectChallengeNonce(webchatWs); await new Promise((resolve) => webchatWs.once("open", resolve)); await connectOk(webchatWs, { client: { diff --git a/src/gateway/server.config-apply.e2e.test.ts b/src/gateway/server.config-apply.test.ts similarity index 96% rename from src/gateway/server.config-apply.e2e.test.ts rename to src/gateway/server.config-apply.test.ts index da13d93e2..85a55caae 100644 --- a/src/gateway/server.config-apply.e2e.test.ts +++ b/src/gateway/server.config-apply.test.ts @@ -6,6 +6,7 @@ import { installGatewayTestHooks, onceMessage, startGatewayServer, + trackConnectChallengeNonce, } from "./test-helpers.js"; installGatewayTestHooks({ scope: "suite" }); @@ -24,6 +25,7 @@ afterAll(async () => { const openClient = async () => { const ws = new WebSocket(`ws://127.0.0.1:${port}`); + trackConnectChallengeNonce(ws); await new Promise((resolve) => ws.once("open", resolve)); await connectOk(ws); return ws; diff --git a/src/gateway/server.e2e-ws-harness.ts b/src/gateway/server.e2e-ws-harness.ts index ab585d56f..f304ad34f 100644 --- a/src/gateway/server.e2e-ws-harness.ts +++ b/src/gateway/server.e2e-ws-harness.ts @@ -1,6 +1,11 @@ import { WebSocket } from "ws"; import { captureEnv } from "../test-utils/env.js"; -import { connectOk, getFreePort, startGatewayServer } from "./test-helpers.js"; +import { + connectOk, + getFreePort, + startGatewayServer, + trackConnectChallengeNonce, +} from "./test-helpers.js"; export type GatewayWsClient = { ws: WebSocket; @@ -22,6 +27,7 @@ export async function startGatewayServerHarness(): Promise const openClient = async (opts?: Parameters[1]): Promise => { const ws = new WebSocket(`ws://127.0.0.1:${port}`); + trackConnectChallengeNonce(ws); await new Promise((resolve) => ws.once("open", resolve)); const hello = await connectOk(ws, opts); return { ws, hello }; diff --git a/src/gateway/server.health.e2e.test.ts b/src/gateway/server.health.test.ts similarity index 100% rename from src/gateway/server.health.e2e.test.ts rename to src/gateway/server.health.test.ts diff --git a/src/gateway/server.models-voicewake-misc.e2e.test.ts b/src/gateway/server.models-voicewake-misc.test.ts similarity index 98% rename from src/gateway/server.models-voicewake-misc.e2e.test.ts rename to src/gateway/server.models-voicewake-misc.test.ts index 1d7c954a3..f5a33b42f 100644 --- a/src/gateway/server.models-voicewake-misc.e2e.test.ts +++ b/src/gateway/server.models-voicewake-misc.test.ts @@ -26,6 +26,7 @@ import { startServerWithClient, testState, testTailnetIPv4, + trackConnectChallengeNonce, } from "./test-helpers.js"; installGatewayTestHooks({ scope: "suite" }); @@ -134,6 +135,7 @@ describe("gateway server models + voicewake", () => { test("pushes voicewake.changed to nodes on connect and on updates", async () => { await withTempHome(async () => { const nodeWs = new WebSocket(`ws://127.0.0.1:${port}`); + trackConnectChallengeNonce(nodeWs); await new Promise((resolve) => nodeWs.once("open", resolve)); const firstEventP = onceMessage( nodeWs, @@ -389,7 +391,12 @@ describe("gateway server misc", () => { type: "req", id, method: "send", - params: { to: "+15550000000", message: "hi", idempotencyKey: idem }, + params: { + to: "+15550000000", + channel: "whatsapp", + message: "hi", + idempotencyKey: idem, + }, }), ); sendReq("a1"); diff --git a/src/gateway/server.sessions.gateway-server-sessions-a.e2e.test.ts b/src/gateway/server.sessions.gateway-server-sessions-a.test.ts similarity index 99% rename from src/gateway/server.sessions.gateway-server-sessions-a.e2e.test.ts rename to src/gateway/server.sessions.gateway-server-sessions-a.test.ts index acac30ef3..a6864d8b7 100644 --- a/src/gateway/server.sessions.gateway-server-sessions-a.e2e.test.ts +++ b/src/gateway/server.sessions.gateway-server-sessions-a.test.ts @@ -14,6 +14,7 @@ import { piSdkMock, rpcReq, testState, + trackConnectChallengeNonce, writeSessionStore, } from "./test-helpers.js"; @@ -1063,6 +1064,7 @@ describe("gateway server sessions", () => { const ws = new WebSocket(`ws://127.0.0.1:${harness.port}`, { headers: { origin: `http://127.0.0.1:${harness.port}` }, }); + trackConnectChallengeNonce(ws); await new Promise((resolve) => ws.once("open", resolve)); await connectOk(ws, { client: { diff --git a/src/gateway/test-helpers.server.ts b/src/gateway/test-helpers.server.ts index c6ba81a16..d697f23ae 100644 --- a/src/gateway/test-helpers.server.ts +++ b/src/gateway/test-helpers.server.ts @@ -585,6 +585,7 @@ export async function connectWebchatClient(params: { const ws = new WebSocket(`ws://127.0.0.1:${params.port}`, { headers: { origin }, }); + trackConnectChallengeNonce(ws); await new Promise((resolve, reject) => { const timer = setTimeout(() => reject(new Error("timeout waiting for ws open")), 10_000); const onOpen = () => {