whale-code 6.5.10 → 6.6.0

package/dist/server/__tests__/tool-router-registry-cache.test.js

@@ -1,383 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
-
-// ---------------------------------------------------------------------------
-// Mocks — must be before imports
-// ---------------------------------------------------------------------------
-
-vi.mock("../../shared/agent-core.js", () => ({
-  sanitizeError: err => err instanceof Error ? err.message : String(err)
-}));
-vi.mock("../validation.js", () => ({
-  validateToolArgs: vi.fn((toolName, args) => ({
-    valid: true,
-    data: args
-  }))
-}));
-vi.mock("../lib/prompt-sanitizer.js", () => ({
-  sanitizeToolDescription: vi.fn(s => s || "")
-}));
-vi.mock("../lib/ssrf-guard.js", () => ({
-  validateUrl: vi.fn(async () => null)
-}));
-vi.mock("../lib/utils.js", () => ({
-  summarizeResult: vi.fn(() => ({
-    rows: 1
-  })),
-  withTimeout: vi.fn(async promise => promise)
-}));
-vi.mock("../handlers/inventory.js", () => ({
-  handleInventory: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleInventoryQuery: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleInventoryAudit: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/supply-chain.js", () => ({
-  handlePurchaseOrders: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleTransfers: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/catalog.js", () => ({
-  handleProducts: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleCollections: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/crm.js", () => ({
-  handleCustomers: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleOrders: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/analytics.js", () => ({
-  handleAnalytics: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/operations.js", () => ({
-  handleLocations: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleSuppliers: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleAlerts: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleAuditTrail: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleStore: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/comms.js", () => ({
-  handleEmail: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleDocuments: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/platform.js", () => ({
-  handleWebSearch: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleTelemetry: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/browser.js", () => ({
-  handleBrowser: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/discovery.js", () => ({
-  handleDiscovery: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/voice.js", () => ({
-  handleVoice: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/workflows.js", () => ({
-  handleWorkflows: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/embeddings.js", () => ({
-  handleEmbeddings: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/llm-providers.js", () => ({
-  handleLLM: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/image-gen.js", () => ({
-  handleImageGen: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/video-gen.js", () => ({
-  handleVideoGen: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/api-keys.js", () => ({
-  handleAPIKeys: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/creations.js", () => ({
-  handleCreations: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/meta-ads.js", () => ({
-  handleMetaAds: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/kali.js", () => ({
-  handleKali: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/local-agent.js", () => ({
-  handleLocalAgent: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/enrichment.js", () => ({
-  handleEnrichment: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-
-// ---------------------------------------------------------------------------
-// Import the module under test
-// ---------------------------------------------------------------------------
-
-import { getCachedToolDefs, getUserToolByPrefixedName, getToolsForAgent, setExtendedToolsCache, getExtendedToolsCache, getExtendedToolsIndex, getFullToolSchemas } from "../tool-router.js";
-
-// ---------------------------------------------------------------------------
-// TESTS
-// ---------------------------------------------------------------------------
-
-beforeEach(() => {
-  vi.clearAllMocks();
-});
-describe("extended tools cache", () => {
-  it("setExtendedToolsCache populates cache and index", () => {
-    const tools = [{
-      name: "kali",
-      description: "Security scanner. Advanced pentesting.",
-      input_schema: {
-        type: "object"
-      }
-    }, {
-      name: "browser",
-      description: "Web browser automation.",
-      input_schema: {
-        type: "object"
-      }
-    }];
-    setExtendedToolsCache(tools);
-    expect(getExtendedToolsCache()).toEqual(tools);
-    const index = getExtendedToolsIndex();
-    expect(index).toHaveLength(2);
-    expect(index[0].description).toBe("Security scanner");
-    expect(index[1].description).toBe("Web browser automation");
-  });
-  it("getFullToolSchemas filters by name", () => {
-    const tools = [{
-      name: "kali",
-      description: "Security",
-      input_schema: {
-        type: "object"
-      }
-    }, {
-      name: "browser",
-      description: "Browser",
-      input_schema: {
-        type: "object"
-      }
-    }, {
-      name: "voice",
-      description: "Voice",
-      input_schema: {
-        type: "object"
-      }
-    }];
-    setExtendedToolsCache(tools);
-    const result = getFullToolSchemas(["kali", "voice"]);
-    expect(result).toHaveLength(2);
-    expect(result.map(t => t.name)).toEqual(["kali", "voice"]);
-  });
-  it("getFullToolSchemas returns empty for empty names", () => {
-    expect(getFullToolSchemas([])).toEqual([]);
-  });
-  it("getFullToolSchemas skips unknown names", () => {
-    setExtendedToolsCache([{
-      name: "kali",
-      description: "x",
-      input_schema: {}
-    }]);
-    expect(getFullToolSchemas(["nonexistent"])).toEqual([]);
-  });
-});
-describe("getUserToolByPrefixedName", () => {
-  const rows = [{
-    id: "1",
-    name: "custom_search",
-    display_name: "Search",
-    description: "desc",
-    input_schema: {},
-    execution_type: "rpc",
-    is_read_only: true,
-    requires_approval: false,
-    http_config: null,
-    rpc_function: "fn",
-    sql_template: null,
-    allowed_tables: null,
-    max_execution_time_ms: 5000
-  }];
-  it("finds tool by prefixed name", () => {
-    const result = getUserToolByPrefixedName(rows, "user_tool__custom_search");
-    expect(result?.name).toBe("custom_search");
-  });
-  it("returns undefined for unknown tool", () => {
-    const result = getUserToolByPrefixedName(rows, "user_tool__nonexistent");
-    expect(result).toBeUndefined();
-  });
-  it("strips only the user_tool__ prefix", () => {
-    const result = getUserToolByPrefixedName(rows, "custom_search");
-    expect(result?.name).toBe("custom_search");
-  });
-});
-describe("getToolsForAgent", () => {
-  const coreTools = [{
-    name: "inventory",
-    description: "Manage inventory",
-    input_schema: {}
-  }, {
-    name: "products",
-    description: "Manage products",
-    input_schema: {}
-  }];
-  const userTools = [{
-    name: "user_tool__search",
-    description: "Custom search",
-    input_schema: {}
-  }, {
-    name: "user_tool__export",
-    description: "Custom export",
-    input_schema: {}
-  }];
-  it("returns core + all user tools when no enabled_tools filter", () => {
-    const agent = {
-      enabled_tools: []
-    };
-    const tools = getToolsForAgent(agent, coreTools, userTools);
-    expect(tools).toHaveLength(4);
-  });
-  it("filters user tools by enabled_tools", () => {
-    const agent = {
-      enabled_tools: ["user_tool__search"]
-    };
-    const tools = getToolsForAgent(agent, coreTools, userTools);
-    expect(tools).toHaveLength(3);
-    expect(tools.map(t => t.name)).toContain("user_tool__search");
-    expect(tools.map(t => t.name)).not.toContain("user_tool__export");
-  });
-  it("core tools are always included regardless of enabled_tools", () => {
-    const agent = {
-      enabled_tools: ["user_tool__search"]
-    };
-    const tools = getToolsForAgent(agent, coreTools, userTools);
-    expect(tools.map(t => t.name)).toContain("inventory");
-    expect(tools.map(t => t.name)).toContain("products");
-  });
-  it("works with no user tools", () => {
-    const agent = {
-      enabled_tools: []
-    };
-    const tools = getToolsForAgent(agent, coreTools);
-    expect(tools).toHaveLength(2);
-  });
-});
-describe("getCachedToolDefs", () => {
-  it("returns tool defs with descriptions", () => {
-    const tools = [{
-      name: "test",
-      description: "A short description",
-      input_schema: {
-        type: "object"
-      }
-    }];
-    const result = getCachedToolDefs(tools);
-    expect(result).toHaveLength(1);
-    expect(result[0].name).toBe("test");
-    expect(result[0].description).toBe("A short description");
-  });
-  it("truncates long descriptions to 150 chars", () => {
-    const longDesc = "x".repeat(200);
-    const tools = [{
-      name: "test",
-      description: longDesc,
-      input_schema: {
-        type: "object"
-      }
-    }];
-    const result = getCachedToolDefs(tools);
-    expect(result[0].description.length).toBeLessThanOrEqual(200);
-  });
-});

package/dist/server/__tests__/tool-router-registry-handlers.test.js

@@ -1,272 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
-
-// ---------------------------------------------------------------------------
-// Mocks — must be before imports
-// ---------------------------------------------------------------------------
-
-vi.mock("../../shared/agent-core.js", () => ({
-  sanitizeError: err => err instanceof Error ? err.message : String(err)
-}));
-vi.mock("../validation.js", () => ({
-  validateToolArgs: vi.fn((toolName, args) => ({
-    valid: true,
-    data: args
-  }))
-}));
-vi.mock("../lib/prompt-sanitizer.js", () => ({
-  sanitizeToolDescription: vi.fn(s => s || "")
-}));
-vi.mock("../lib/ssrf-guard.js", () => ({
-  validateUrl: vi.fn(async () => null)
-}));
-vi.mock("../lib/utils.js", () => ({
-  summarizeResult: vi.fn(() => ({
-    rows: 1
-  })),
-  withTimeout: vi.fn(async promise => promise)
-}));
-vi.mock("../handlers/inventory.js", () => ({
-  handleInventory: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleInventoryQuery: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleInventoryAudit: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/supply-chain.js", () => ({
-  handlePurchaseOrders: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleTransfers: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/catalog.js", () => ({
-  handleProducts: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleCollections: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/crm.js", () => ({
-  handleCustomers: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleOrders: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/analytics.js", () => ({
-  handleAnalytics: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/operations.js", () => ({
-  handleLocations: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleSuppliers: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleAlerts: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleAuditTrail: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleStore: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/comms.js", () => ({
-  handleEmail: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleDocuments: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/platform.js", () => ({
-  handleWebSearch: vi.fn(async () => ({
-    success: true,
-    data: {}
-  })),
-  handleTelemetry: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/browser.js", () => ({
-  handleBrowser: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/discovery.js", () => ({
-  handleDiscovery: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/voice.js", () => ({
-  handleVoice: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/workflows.js", () => ({
-  handleWorkflows: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/embeddings.js", () => ({
-  handleEmbeddings: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/llm-providers.js", () => ({
-  handleLLM: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/image-gen.js", () => ({
-  handleImageGen: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/video-gen.js", () => ({
-  handleVideoGen: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/api-keys.js", () => ({
-  handleAPIKeys: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/creations.js", () => ({
-  handleCreations: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/meta-ads.js", () => ({
-  handleMetaAds: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/kali.js", () => ({
-  handleKali: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/local-agent.js", () => ({
-  handleLocalAgent: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-vi.mock("../handlers/enrichment.js", () => ({
-  handleEnrichment: vi.fn(async () => ({
-    success: true,
-    data: {}
-  }))
-}));
-
-// ---------------------------------------------------------------------------
-// Import the module under test
-// ---------------------------------------------------------------------------
-
-import { flushSpans, getToolMetrics, TOOL_HANDLERS, getRegisteredToolNames } from "../tool-router.js";
-
-// ---------------------------------------------------------------------------
-// TESTS
-// ---------------------------------------------------------------------------
-
-beforeEach(() => {
-  vi.clearAllMocks();
-});
-describe("TOOL_HANDLERS registry", () => {
-  it("is frozen and cannot be modified", () => {
-    expect(Object.isFrozen(TOOL_HANDLERS)).toBe(true);
-  });
-  it("contains all expected tool names", () => {
-    const names = getRegisteredToolNames();
-    expect(names).toContain("inventory");
-    expect(names).toContain("products");
-    expect(names).toContain("customers");
-    expect(names).toContain("orders");
-    expect(names).toContain("analytics");
-    expect(names).toContain("web_search");
-    expect(names).toContain("discover_tools");
-    expect(names).toContain("kali");
-    expect(names).toContain("local_agent");
-    expect(names).toContain("enrichment");
-  });
-  it("getRegisteredToolNames returns only string keys", () => {
-    const names = getRegisteredToolNames();
-    expect(names.length).toBeGreaterThan(10);
-    for (const n of names) {
-      expect(typeof n).toBe("string");
-    }
-  });
-  it("every entry has handler, timeout, and requiresStore", () => {
-    for (const [name, entry] of Object.entries(TOOL_HANDLERS)) {
-      expect(typeof entry.handler).toBe("function");
-      expect(typeof entry.timeout).toBe("number");
-      expect(typeof entry.requiresStore).toBe("boolean");
-    }
-  });
-  it("web_search does not require store", () => {
-    expect(TOOL_HANDLERS["web_search"].requiresStore).toBe(false);
-  });
-  it("inventory requires store", () => {
-    expect(TOOL_HANDLERS["inventory"].requiresStore).toBe(true);
-  });
-  it("kali has 600s timeout", () => {
-    expect(TOOL_HANDLERS["kali"].timeout).toBe(600_000);
-  });
-  it("discover_tools has 5s timeout", () => {
-    expect(TOOL_HANDLERS["discover_tools"].timeout).toBe(5000);
-  });
-});
-describe("getToolMetrics", () => {
-  it("returns an object (may be empty or populated from other tests)", () => {
-    const metrics = getToolMetrics();
-    expect(typeof metrics).toBe("object");
-  });
-});
-describe("flushSpans", () => {
-  it("does nothing when buffer is empty", async () => {
-    await flushSpans();
-  });
-});