diff --git a/src/agents/tools/cron-tool.e2e.test.ts b/src/agents/tools/cron-tool.e2e.test.ts index 7b6d1310e4a..b850f74ad36 100644 --- a/src/agents/tools/cron-tool.e2e.test.ts +++ b/src/agents/tools/cron-tool.e2e.test.ts @@ -1,15 +1,6 @@ -import { beforeEach, describe, expect, it, vi } from "vitest"; - -const callGatewayMock = vi.fn(); -vi.mock("../../gateway/call.js", () => ({ - callGateway: (opts: unknown) => callGatewayMock(opts), -})); - -vi.mock("../agent-scope.js", () => ({ - resolveSessionAgentId: () => "agent-123", -})); - +import { beforeEach, describe, expect, it } from "vitest"; import { createCronTool } from "./cron-tool.js"; +import { callGatewayMock, resetCronToolGatewayMock } from "./cron-tool.test-helpers.js"; describe("cron tool", () => { async function executeAddAndReadDelivery(params: { @@ -35,8 +26,7 @@ describe("cron tool", () => { } beforeEach(() => { - callGatewayMock.mockReset(); - callGatewayMock.mockResolvedValue({ ok: true }); + resetCronToolGatewayMock(); }); it.each([ diff --git a/src/agents/tools/cron-tool.flat-params.test.ts b/src/agents/tools/cron-tool.flat-params.test.ts index 2a96b451073..cac2ac75647 100644 --- a/src/agents/tools/cron-tool.flat-params.test.ts +++ b/src/agents/tools/cron-tool.flat-params.test.ts @@ -1,20 +1,10 @@ -import { beforeEach, describe, expect, it, vi } from "vitest"; - -const callGatewayMock = vi.fn(); -vi.mock("../../gateway/call.js", () => ({ - callGateway: (opts: unknown) => callGatewayMock(opts), -})); - -vi.mock("../agent-scope.js", () => ({ - resolveSessionAgentId: () => "agent-123", -})); - +import { beforeEach, describe, expect, it } from "vitest"; import { createCronTool } from "./cron-tool.js"; +import { callGatewayMock, resetCronToolGatewayMock } from "./cron-tool.test-helpers.js"; describe("cron tool flat-params", () => { beforeEach(() => { - callGatewayMock.mockReset(); - callGatewayMock.mockResolvedValue({ ok: true }); + resetCronToolGatewayMock(); }); it("preserves explicit top-level sessionKey during flat-params recovery", async () => { diff --git a/src/agents/tools/cron-tool.test-helpers.ts b/src/agents/tools/cron-tool.test-helpers.ts new file mode 100644 index 00000000000..9125c8a86dc --- /dev/null +++ b/src/agents/tools/cron-tool.test-helpers.ts @@ -0,0 +1,16 @@ +import { vi } from "vitest"; + +export const callGatewayMock = vi.fn(); + +vi.mock("../../gateway/call.js", () => ({ + callGateway: (opts: unknown) => callGatewayMock(opts), +})); + +vi.mock("../agent-scope.js", () => ({ + resolveSessionAgentId: () => "agent-123", +})); + +export function resetCronToolGatewayMock() { + callGatewayMock.mockReset(); + callGatewayMock.mockResolvedValue({ ok: true }); +}