From f54cf74ef6bb76b703208381d7e978b5ad590f8a Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Sun, 19 Apr 2026 03:58:08 +0100 Subject: [PATCH] test: share BTW sanitized user assertion --- src/agents/btw.test.ts | 32 ++++++++++++++------------------ 1 file changed, 14 insertions(+), 18 deletions(-) diff --git a/src/agents/btw.test.ts b/src/agents/btw.test.ts index 01f25b3d5c4..e37227f0336 100644 --- a/src/agents/btw.test.ts +++ b/src/agents/btw.test.ts @@ -245,6 +245,18 @@ function expectSanitizedAssistantContext(context: unknown, text: string) { }); } +function expectSeedOnlyUserContext(context: unknown) { + expect(context).toMatchObject({ + messages: [ + expect.objectContaining({ + role: "user", + content: [{ type: "text", text: "seed" }], + }), + expect.objectContaining({ role: "user" }), + ], + }); +} + describe("runBtwSideQuestion", () => { beforeEach(() => { streamSimpleMock.mockReset(); @@ -766,15 +778,7 @@ describe("runBtwSideQuestion", () => { ]); const context = await runMathSideQuestionAndCaptureContext(); - expect(context).toMatchObject({ - messages: [ - expect.objectContaining({ - role: "user", - content: [{ type: "text", text: "seed" }], - }), - expect.objectContaining({ role: "user" }), - ], - }); + expectSeedOnlyUserContext(context); }); it("drops assistant thinking blocks from BTW context", async () => { @@ -828,15 +832,7 @@ describe("runBtwSideQuestion", () => { ]); const context = await runMathSideQuestionAndCaptureContext(); - expect(context).toMatchObject({ - messages: [ - expect.objectContaining({ - role: "user", - content: [{ type: "text", text: "seed" }], - }), - expect.objectContaining({ role: "user" }), - ], - }); + expectSeedOnlyUserContext(context); }); it("normalizes malformed assistant content before stripping tool blocks", async () => {