test: repair stale task and image mocks

This commit is contained in:
Peter Steinberger 2026-03-31 15:47:21 +01:00
parent c425ef3e74
commit a1e2d2bf42
No known key found for this signature in database
4 changed files with 19 additions and 19 deletions

View File

@ -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(),
}));

View File

@ -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),
}));

View File

@ -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");

View File

@ -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(