From a1e2d2bf42f2f4c6fcc632dec4ed2598cd92dc11 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Tue, 31 Mar 2026 15:47:21 +0100 Subject: [PATCH] test: repair stale task and image mocks --- packages/tasks-host-sdk/src/flow-runtime.test.ts | 10 +++++----- .../tasks-host-sdk/src/task-executor.test.ts | 4 ++-- .../tasks-host-sdk/src/task-registry.test.ts | 8 ++++---- src/media-understanding/image.test.ts | 16 ++++++++-------- 4 files changed, 19 insertions(+), 19 deletions(-) diff --git a/packages/tasks-host-sdk/src/flow-runtime.test.ts b/packages/tasks-host-sdk/src/flow-runtime.test.ts index 3f9aabeb1bd..7e8437d1424 100644 --- a/packages/tasks-host-sdk/src/flow-runtime.test.ts +++ b/packages/tasks-host-sdk/src/flow-runtime.test.ts @@ -24,25 +24,25 @@ vi.mock("./task-registry-delivery-runtime.js", () => ({ sendMessage: (...args: unknown[]) => mocks.sendMessageMock(...args), })); -vi.mock("../infra/system-events.js", () => ({ +vi.mock("../../../src/infra/system-events.js", () => ({ enqueueSystemEvent: (...args: unknown[]) => mocks.enqueueSystemEventMock(...args), })); -vi.mock("../infra/heartbeat-wake.js", () => ({ +vi.mock("../../../src/infra/heartbeat-wake.js", () => ({ requestHeartbeatNow: (...args: unknown[]) => mocks.requestHeartbeatNowMock(...args), })); -vi.mock("../infra/agent-events.js", () => ({ +vi.mock("../../../src/infra/agent-events.js", () => ({ onAgentEvent: () => () => {}, })); -vi.mock("../acp/control-plane/manager.js", () => ({ +vi.mock("../../../src/acp/control-plane/manager.js", () => ({ getAcpSessionManager: () => ({ cancelSession: vi.fn(), }), })); -vi.mock("../agents/subagent-control.js", () => ({ +vi.mock("../../../src/agents/subagent-control.js", () => ({ killSubagentRunAdmin: vi.fn(), })); diff --git a/packages/tasks-host-sdk/src/task-executor.test.ts b/packages/tasks-host-sdk/src/task-executor.test.ts index 0e081d2d28d..6b449acf9cd 100644 --- a/packages/tasks-host-sdk/src/task-executor.test.ts +++ b/packages/tasks-host-sdk/src/task-executor.test.ts @@ -41,13 +41,13 @@ vi.mock("./task-registry-delivery-runtime.js", () => ({ sendMessage: hoisted.sendMessageMock, })); -vi.mock("../acp/control-plane/manager.js", () => ({ +vi.mock("../../../src/acp/control-plane/manager.js", () => ({ getAcpSessionManager: () => ({ cancelSession: hoisted.cancelSessionMock, }), })); -vi.mock("../agents/subagent-control.js", () => ({ +vi.mock("../../../src/agents/subagent-control.js", () => ({ killSubagentRunAdmin: (params: unknown) => hoisted.killSubagentRunAdminMock(params), })); diff --git a/packages/tasks-host-sdk/src/task-registry.test.ts b/packages/tasks-host-sdk/src/task-registry.test.ts index fbc7b0fcf90..cbbca737028 100644 --- a/packages/tasks-host-sdk/src/task-registry.test.ts +++ b/packages/tasks-host-sdk/src/task-registry.test.ts @@ -52,13 +52,13 @@ vi.mock("./task-registry-delivery-runtime.js", () => ({ sendMessage: hoisted.sendMessageMock, })); -vi.mock("../acp/control-plane/manager.js", () => ({ +vi.mock("../../../src/acp/control-plane/manager.js", () => ({ getAcpSessionManager: () => ({ cancelSession: hoisted.cancelSessionMock, }), })); -vi.mock("../agents/subagent-control.js", () => ({ +vi.mock("../../../src/agents/subagent-control.js", () => ({ killSubagentRunAdmin: (params: unknown) => hoisted.killSubagentRunAdminMock(params), })); @@ -67,12 +67,12 @@ async function loadFreshTaskRegistryModulesForControlTest() { vi.doMock("./task-registry-delivery-runtime.js", () => ({ sendMessage: hoisted.sendMessageMock, })); - vi.doMock("../acp/control-plane/manager.js", () => ({ + vi.doMock("../../../src/acp/control-plane/manager.js", () => ({ getAcpSessionManager: () => ({ cancelSession: hoisted.cancelSessionMock, }), })); - vi.doMock("../agents/subagent-control.js", () => ({ + vi.doMock("../../../src/agents/subagent-control.js", () => ({ killSubagentRunAdmin: (params: unknown) => hoisted.killSubagentRunAdminMock(params), })); return await import("./task-registry.js"); diff --git a/src/media-understanding/image.test.ts b/src/media-understanding/image.test.ts index 11c6c07bfef..d42e3e1bd96 100644 --- a/src/media-understanding/image.test.ts +++ b/src/media-understanding/image.test.ts @@ -226,10 +226,10 @@ describe("describeImageWithModel", () => { it("normalizes deprecated google flash ids before lookup and keeps profile auth selection", async () => { const findMock = vi.fn((provider: string, modelId: string) => { expect(provider).toBe("google"); - expect(modelId).toBe("gemini-3.1-flash-preview"); + expect(modelId).toBe("gemini-3-flash-preview"); return { provider: "google", - id: "gemini-3.1-flash-preview", + id: "gemini-3-flash-preview", input: ["text", "image"], baseUrl: "https://generativelanguage.googleapis.com/v1beta", }; @@ -239,7 +239,7 @@ describe("describeImageWithModel", () => { role: "assistant", api: "google-generative-ai", provider: "google", - model: "gemini-3.1-flash-preview", + model: "gemini-3-flash-preview", stopReason: "stop", timestamp: Date.now(), content: [{ type: "text", text: "flash ok" }], @@ -260,7 +260,7 @@ describe("describeImageWithModel", () => { expect(result).toEqual({ text: "flash ok", - model: "gemini-3.1-flash-preview", + model: "gemini-3-flash-preview", }); expect(findMock).toHaveBeenCalledOnce(); expect(getApiKeyForModelMock).toHaveBeenCalledWith( @@ -274,10 +274,10 @@ describe("describeImageWithModel", () => { it("normalizes gemini 3.1 flash-lite ids before lookup and keeps profile auth selection", async () => { const findMock = vi.fn((provider: string, modelId: string) => { expect(provider).toBe("google"); - expect(modelId).toBe("gemini-3.1-flash-lite"); + expect(modelId).toBe("gemini-3.1-flash-lite-preview"); return { provider: "google", - id: "gemini-3.1-flash-lite", + id: "gemini-3.1-flash-lite-preview", input: ["text", "image"], baseUrl: "https://generativelanguage.googleapis.com/v1beta", }; @@ -287,7 +287,7 @@ describe("describeImageWithModel", () => { role: "assistant", api: "google-generative-ai", provider: "google", - model: "gemini-3.1-flash-lite", + model: "gemini-3.1-flash-lite-preview", stopReason: "stop", timestamp: Date.now(), content: [{ type: "text", text: "flash lite ok" }], @@ -308,7 +308,7 @@ describe("describeImageWithModel", () => { expect(result).toEqual({ text: "flash lite ok", - model: "gemini-3.1-flash-lite", + model: "gemini-3.1-flash-lite-preview", }); expect(findMock).toHaveBeenCalledOnce(); expect(getApiKeyForModelMock).toHaveBeenCalledWith(