diff --git a/src/agents/tools/cron-tool.test-helpers.ts b/src/agents/tools/cron-tool.test-helpers.ts index 9125c8a86dc..9045636864b 100644 --- a/src/agents/tools/cron-tool.test-helpers.ts +++ b/src/agents/tools/cron-tool.test-helpers.ts @@ -1,6 +1,11 @@ import { vi } from "vitest"; -export const callGatewayMock = vi.fn(); +type GatewayMockFn = ((opts: unknown) => unknown) & { + mockReset: () => void; + mockResolvedValue: (value: unknown) => void; +}; + +export const callGatewayMock = vi.fn() as GatewayMockFn; vi.mock("../../gateway/call.js", () => ({ callGateway: (opts: unknown) => callGatewayMock(opts),