From 2b2ad43a090b5151b3aaa806135b5147d9f1d5ae Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 9 May 2026 16:46:10 +0100
Subject: [PATCH] test: tighten context engine lifecycle assertions

---
 .../harness/context-engine-lifecycle.test.ts  | 33 ++++++++++---------
 1 file changed, 17 insertions(+), 16 deletions(-)

diff --git a/src/agents/harness/context-engine-lifecycle.test.ts b/src/agents/harness/context-engine-lifecycle.test.ts
index 54dc90ff5f1..77859338c50 100644
--- a/src/agents/harness/context-engine-lifecycle.test.ts
+++ b/src/agents/harness/context-engine-lifecycle.test.ts
@@ -64,11 +64,8 @@ describe("harness context engine lifecycle", () => {
       modelId: "gpt-test",
     });
 
-    expect(assemble).toHaveBeenCalledWith(
-      expect.objectContaining({
-        messages: [visibleUser, visibleAssistant],
-      }),
-    );
+    const assembleParams = assemble.mock.calls[0]?.[0];
+    expect(assembleParams?.messages).toEqual([visibleUser, visibleAssistant]);
   });
 
   it("keeps hidden runtime-context custom messages out of afterTurn hooks", async () => {
@@ -103,12 +100,17 @@ describe("harness context engine lifecycle", () => {
       warn: () => {},
     });
 
-    expect(afterTurn).toHaveBeenCalledWith(
-      expect.objectContaining({
-        messages: [beforePromptUser, beforePromptAssistant, turnUser, turnAssistant],
-        prePromptMessageCount: 2,
-      }),
-    );
+    const afterTurnCalls = (afterTurn as unknown as { mock: { calls: unknown[][] } }).mock.calls;
+    const afterTurnParams = afterTurnCalls[0]?.[0] as
+      | { messages?: AgentMessage[]; prePromptMessageCount?: number }
+      | undefined;
+    expect(afterTurnParams?.messages).toEqual([
+      beforePromptUser,
+      beforePromptAssistant,
+      turnUser,
+      turnAssistant,
+    ]);
+    expect(afterTurnParams?.prePromptMessageCount).toBe(2);
   });
 
   it("keeps hidden runtime-context custom messages out of ingestBatch fallbacks", async () => {
@@ -143,10 +145,9 @@ describe("harness context engine lifecycle", () => {
       warn: () => {},
     });
 
-    expect(ingestBatch).toHaveBeenCalledWith(
-      expect.objectContaining({
-        messages: [turnUser, turnAssistant],
-      }),
-    );
+    const ingestBatchCalls = (ingestBatch as unknown as { mock: { calls: unknown[][] } }).mock
+      .calls;
+    const ingestBatchParams = ingestBatchCalls[0]?.[0] as { messages?: AgentMessage[] } | undefined;
+    expect(ingestBatchParams?.messages).toEqual([turnUser, turnAssistant]);
   });
 });