From 6b2b7d12e5d54256ab6fe460ffa4c65f0bc2715e Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 11 May 2026 15:01:06 +0100
Subject: [PATCH] test: tighten media understanding assertions

---
 src/media-understanding/runtime.test.ts | 84 +++++++++++++++----------
 1 file changed, 52 insertions(+), 32 deletions(-)

diff --git a/src/media-understanding/runtime.test.ts b/src/media-understanding/runtime.test.ts
index b9c4989ea34..521fd93263d 100644
--- a/src/media-understanding/runtime.test.ts
+++ b/src/media-understanding/runtime.test.ts
@@ -287,17 +287,26 @@ describe("media-understanding runtime", () => {
     });
 
     expect(mocks.normalizeMediaProviderId).toHaveBeenCalledWith("gemini");
-    expect(describeImage).toHaveBeenCalledWith(
-      expect.objectContaining({
-        buffer: Buffer.from("image-bytes"),
-        fileName: "sample.jpg",
-        mime: "image/jpeg",
-        provider: "gemini",
-        model: "vision-v1",
-        prompt: "Describe the sample.",
-        agentDir: "/tmp/agent",
-      }),
-    );
+    const [[describeImageOptions]] = describeImage.mock.calls as unknown as Array<
+      [
+        {
+          buffer?: Buffer;
+          fileName?: string;
+          mime?: string;
+          provider?: string;
+          model?: string;
+          prompt?: string;
+          agentDir?: string;
+        },
+      ]
+    >;
+    expect(describeImageOptions?.buffer).toEqual(Buffer.from("image-bytes"));
+    expect(describeImageOptions?.fileName).toBe("sample.jpg");
+    expect(describeImageOptions?.mime).toBe("image/jpeg");
+    expect(describeImageOptions?.provider).toBe("gemini");
+    expect(describeImageOptions?.model).toBe("vision-v1");
+    expect(describeImageOptions?.prompt).toBe("Describe the sample.");
+    expect(describeImageOptions?.agentDir).toBe("/tmp/agent");
   });
 
   it("routes structured extraction to a provider by id and model", async () => {
@@ -347,27 +356,38 @@ describe("media-understanding runtime", () => {
       "Vision-Plugin",
       providerRegistry,
     );
-    expect(extractStructured).toHaveBeenCalledWith(
-      expect.objectContaining({
-        input: [
-          { type: "text", text: "Extract the fact." },
-          {
-            type: "image",
-            buffer: Buffer.from("image-bytes"),
-            fileName: "fact.png",
-            mime: "image/png",
-          },
-        ],
-        instructions: "Return JSON.",
-        provider: "Vision-Plugin",
-        model: "vision-json",
-        profile: "work",
-        preferredProfile: "preferred-work",
-        authStore,
-        timeoutMs: 45_000,
-        agentDir: "/tmp/agent",
-      }),
-    );
+    const [[extractOptions]] = extractStructured.mock.calls as unknown as Array<
+      [
+        {
+          input?: unknown;
+          instructions?: string;
+          provider?: string;
+          model?: string;
+          profile?: string;
+          preferredProfile?: string;
+          authStore?: AuthProfileStore;
+          timeoutMs?: number;
+          agentDir?: string;
+        },
+      ]
+    >;
+    expect(extractOptions?.input).toEqual([
+      { type: "text", text: "Extract the fact." },
+      {
+        type: "image",
+        buffer: Buffer.from("image-bytes"),
+        fileName: "fact.png",
+        mime: "image/png",
+      },
+    ]);
+    expect(extractOptions?.instructions).toBe("Return JSON.");
+    expect(extractOptions?.provider).toBe("Vision-Plugin");
+    expect(extractOptions?.model).toBe("vision-json");
+    expect(extractOptions?.profile).toBe("work");
+    expect(extractOptions?.preferredProfile).toBe("preferred-work");
+    expect(extractOptions?.authStore).toBe(authStore);
+    expect(extractOptions?.timeoutMs).toBe(45_000);
+    expect(extractOptions?.agentDir).toBe("/tmp/agent");
   });
 
   it("rejects text-only structured extraction before provider lookup", async () => {