From a74251d415f3f25f8013d45718ff8b516a783d02 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Mon, 16 Feb 2026 02:17:37 +0000 Subject: [PATCH] refactor(agents): dedupe fast tool stubs --- src/agents/test-helpers/fast-coding-tools.ts | 23 +------------------- src/agents/test-helpers/fast-core-tools.ts | 22 +------------------ src/agents/test-helpers/fast-tool-stubs.ts | 22 +++++++++++++++++++ 3 files changed, 24 insertions(+), 43 deletions(-) create mode 100644 src/agents/test-helpers/fast-tool-stubs.ts diff --git a/src/agents/test-helpers/fast-coding-tools.ts b/src/agents/test-helpers/fast-coding-tools.ts index 99b4ab351..5cc92f38a 100644 --- a/src/agents/test-helpers/fast-coding-tools.ts +++ b/src/agents/test-helpers/fast-coding-tools.ts @@ -1,22 +1 @@ -import { vi } from "vitest"; - -const stubTool = (name: string) => ({ - name, - description: `${name} stub`, - parameters: { type: "object", properties: {} }, - execute: vi.fn(), -}); - -vi.mock("../tools/image-tool.js", () => ({ - createImageTool: () => stubTool("image"), -})); - -vi.mock("../tools/web-tools.js", () => ({ - createWebSearchTool: () => null, - createWebFetchTool: () => null, -})); - -vi.mock("../../plugins/tools.js", () => ({ - resolvePluginTools: () => [], - getPluginToolMeta: () => undefined, -})); +import "./fast-tool-stubs.js"; diff --git a/src/agents/test-helpers/fast-core-tools.ts b/src/agents/test-helpers/fast-core-tools.ts index d459c8276..5bda64b09 100644 --- a/src/agents/test-helpers/fast-core-tools.ts +++ b/src/agents/test-helpers/fast-core-tools.ts @@ -1,11 +1,5 @@ import { vi } from "vitest"; - -const stubTool = (name: string) => ({ - name, - description: `${name} stub`, - parameters: { type: "object", properties: {} }, - execute: vi.fn(), -}); +import { stubTool } from "./fast-tool-stubs.js"; vi.mock("../tools/browser-tool.js", () => ({ createBrowserTool: () => stubTool("browser"), @@ -14,17 +8,3 @@ vi.mock("../tools/browser-tool.js", () => ({ vi.mock("../tools/canvas-tool.js", () => ({ createCanvasTool: () => stubTool("canvas"), })); - -vi.mock("../tools/image-tool.js", () => ({ - createImageTool: () => stubTool("image"), -})); - -vi.mock("../tools/web-tools.js", () => ({ - createWebSearchTool: () => null, - createWebFetchTool: () => null, -})); - -vi.mock("../../plugins/tools.js", () => ({ - resolvePluginTools: () => [], - getPluginToolMeta: () => undefined, -})); diff --git a/src/agents/test-helpers/fast-tool-stubs.ts b/src/agents/test-helpers/fast-tool-stubs.ts new file mode 100644 index 000000000..8560c9c5b --- /dev/null +++ b/src/agents/test-helpers/fast-tool-stubs.ts @@ -0,0 +1,22 @@ +import { vi } from "vitest"; + +export const stubTool = (name: string) => ({ + name, + description: `${name} stub`, + parameters: { type: "object", properties: {} }, + execute: vi.fn(), +}); + +vi.mock("../tools/image-tool.js", () => ({ + createImageTool: () => stubTool("image"), +})); + +vi.mock("../tools/web-tools.js", () => ({ + createWebSearchTool: () => null, + createWebFetchTool: () => null, +})); + +vi.mock("../../plugins/tools.js", () => ({ + resolvePluginTools: () => [], + getPluginToolMeta: () => undefined, +}));