From 4ae1780805067e3da32caabfef65d726309d87fd Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Fri, 8 May 2026 20:06:16 +0100
Subject: [PATCH] test: tighten image provider header assertions

---
 .../openai-compatible-image-provider.test.ts      | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/image-generation/openai-compatible-image-provider.test.ts b/src/image-generation/openai-compatible-image-provider.test.ts
index 90d2a80ba3e..71414476daa 100644
--- a/src/image-generation/openai-compatible-image-provider.test.ts
+++ b/src/image-generation/openai-compatible-image-provider.test.ts
@@ -60,6 +60,17 @@ vi.mock("openclaw/plugin-sdk/provider-http", () => ({
   sanitizeConfiguredModelProviderRequest: sanitizeConfiguredModelProviderRequestMock,
 }));
 
+function requireFirstRequestHeaders(mock: ReturnType<typeof vi.fn>): Headers {
+  const request = mock.mock.calls[0]?.[0] as { headers?: Headers } | undefined;
+  const headers = request?.headers;
+  expect(request).toBeDefined();
+  expect(headers).toBeInstanceOf(Headers);
+  if (!headers) {
+    throw new Error("expected request headers");
+  }
+  return headers;
+}
+
 function createProvider(overrides: Partial<OpenAiCompatibleImageProviderOptions> = {}) {
   return createOpenAiCompatibleImageGenerationProvider({
     id: "sample",
@@ -185,7 +196,7 @@ describe("OpenAI-compatible image provider helper", () => {
         },
       }),
     );
-    const headers = postJsonRequestMock.mock.calls[0]?.[0].headers as Headers;
+    const headers = requireFirstRequestHeaders(postJsonRequestMock);
     expect(headers.get("Content-Type")).toBe("application/json");
     expect(result).toMatchObject({
       model: "custom-image",
@@ -212,7 +223,7 @@ describe("OpenAI-compatible image provider helper", () => {
         body: expect.any(FormData),
       }),
     );
-    const headers = postMultipartRequestMock.mock.calls[0]?.[0].headers as Headers;
+    const headers = requireFirstRequestHeaders(postMultipartRequestMock);
     expect(headers.has("Content-Type")).toBe(false);
   });