From c4aac407dcca7a7ed95f7f5509d551087ac917cb Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Sat, 21 Feb 2026 23:13:51 +0000 Subject: [PATCH] test(gateway): dedupe openai context assertions --- src/gateway/openai-http.e2e.test.ts | 31 +++++++++++++++++++---------- 1 file changed, 21 insertions(+), 10 deletions(-) diff --git a/src/gateway/openai-http.e2e.test.ts b/src/gateway/openai-http.e2e.test.ts index 7e5ebd2b39c..62662b0d029 100644 --- a/src/gateway/openai-http.e2e.test.ts +++ b/src/gateway/openai-http.e2e.test.ts @@ -120,6 +120,19 @@ describe("OpenAI-compatible HTTP API (e2e)", () => { ); await res.text(); }; + const expectMessageContext = ( + message: string, + expected: { history: string[]; current: string[] }, + ) => { + expect(message).toContain(HISTORY_CONTEXT_MARKER); + for (const line of expected.history) { + expect(message).toContain(line); + } + expect(message).toContain(CURRENT_MESSAGE_MARKER); + for (const line of expected.current) { + expect(message).toContain(line); + } + }; try { { @@ -241,11 +254,10 @@ describe("OpenAI-compatible HTTP API (e2e)", () => { const opts = (agentCommand.mock.calls[0] as unknown[] | undefined)?.[0]; const message = (opts as { message?: string } | undefined)?.message ?? ""; - expect(message).toContain(HISTORY_CONTEXT_MARKER); - expect(message).toContain("User: Hello, who are you?"); - expect(message).toContain("Assistant: I am Claude."); - expect(message).toContain(CURRENT_MESSAGE_MARKER); - expect(message).toContain("User: What did I just ask you?"); + expectMessageContext(message, { + history: ["User: Hello, who are you?", "Assistant: I am Claude."], + current: ["User: What did I just ask you?"], + }); await res.text(); } @@ -301,11 +313,10 @@ describe("OpenAI-compatible HTTP API (e2e)", () => { const opts = (agentCommand.mock.calls[0] as unknown[] | undefined)?.[0]; const message = (opts as { message?: string } | undefined)?.message ?? ""; - expect(message).toContain(HISTORY_CONTEXT_MARKER); - expect(message).toContain("User: What's the weather?"); - expect(message).toContain("Assistant: Checking the weather."); - expect(message).toContain(CURRENT_MESSAGE_MARKER); - expect(message).toContain("Tool: Sunny, 70F."); + expectMessageContext(message, { + history: ["User: What's the weather?", "Assistant: Checking the weather."], + current: ["Tool: Sunny, 70F."], + }); await res.text(); }