From d1a51a6a2414374ee83ae4716cbffd0ab54e5293 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 9 May 2026 15:20:23 +0100
Subject: [PATCH] test: tighten openai stream wrapper assertions

---
 .../openai-stream-wrappers.test.ts               | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/src/agents/pi-embedded-runner/openai-stream-wrappers.test.ts b/src/agents/pi-embedded-runner/openai-stream-wrappers.test.ts
index 19d41c522b7..5a40812c441 100644
--- a/src/agents/pi-embedded-runner/openai-stream-wrappers.test.ts
+++ b/src/agents/pi-embedded-runner/openai-stream-wrappers.test.ts
@@ -300,10 +300,8 @@ describe("createOpenAIAttributionHeadersWrapper", () => {
     );
 
     expect(codexCalls).toBe(1);
-    expect(capturedHeaders).toMatchObject({
-      originator: "openclaw",
-      "User-Agent": expect.stringMatching(/^openclaw\//),
-    });
+    expect(capturedHeaders?.originator).toBe("openclaw");
+    expect(capturedHeaders?.["User-Agent"]).toMatch(/^openclaw\//);
   });
 
   it("keeps existing wrapped Codex streams so runtime OAuth injection is preserved", () => {
@@ -345,12 +343,8 @@ describe("createOpenAIAttributionHeadersWrapper", () => {
 
     expect(upstreamCalls).toBe(1);
     expect(codexCalls).toBe(0);
-    expect(capturedOptions).toMatchObject({
-      apiKey: "oauth-bearer-token",
-      headers: {
-        originator: "openclaw",
-        "User-Agent": expect.stringMatching(/^openclaw\//),
-      },
-    });
+    expect(capturedOptions?.apiKey).toBe("oauth-bearer-token");
+    expect(capturedOptions?.headers?.originator).toBe("openclaw");
+    expect(capturedOptions?.headers?.["User-Agent"]).toMatch(/^openclaw\//);
   });
 });