Add Tavily-search

extensions/tavily-search/index.test.ts

@@ -0,0 +1,98 @@
+/**
+ * Smoke test for @jmcombs/pi-tavily-search.
+ *
+ * Verifies the extension's registration surface against a minimal, real-shape
+ * `ExtensionAPI` stub. **No external API is mocked.** End-to-end behavior of
+ * the tool is exercised manually via `pi -e ./packages/tavily-search` against
+ * a real Tavily key (see README).
+ */
+
+import { describe, expect, it } from "vitest";
+import type { ExtensionAPI, ToolDefinition } from "@mariozechner/pi-coding-agent";
+import factory, { type TavilySearchInput } from "./index.js";
+
+interface CapturedTool {
+  name: string;
+  label?: string;
+  description?: string;
+  parameters: unknown;
+}
+
+function createApiStub(): { api: ExtensionAPI; tools: CapturedTool[] } {
+  const tools: CapturedTool[] = [];
+  const notImplemented = (method: string) => () => {
+    throw new Error(`ExtensionAPI.${method} not implemented in test stub`);
+  };
+
+  const api = {
+    on: (() => {
+      /* tavily-search subscribes to no events */
+    }) as unknown as ExtensionAPI["on"],
+    registerTool: ((tool: ToolDefinition) => {
+      tools.push({
+        name: tool.name,
+        label: tool.label,
+        description: tool.description,
+        parameters: tool.parameters,
+      });
+    }) as unknown as ExtensionAPI["registerTool"],
+    registerCommand: notImplemented("registerCommand"),
+    registerShortcut: notImplemented("registerShortcut"),
+    registerFlag: notImplemented("registerFlag"),
+    getFlag: notImplemented("getFlag"),
+    registerMessageRenderer: notImplemented("registerMessageRenderer"),
+    sendMessage: notImplemented("sendMessage"),
+    sendUserMessage: notImplemented("sendUserMessage"),
+    appendEntry: notImplemented("appendEntry"),
+    setSessionName: notImplemented("setSessionName"),
+    getSessionName: notImplemented("getSessionName"),
+    setLabel: notImplemented("setLabel"),
+    exec: notImplemented("exec"),
+    getActiveTools: notImplemented("getActiveTools"),
+    getAllTools: notImplemented("getAllTools"),
+    setActiveTools: notImplemented("setActiveTools"),
+    getCommands: notImplemented("getCommands"),
+    setModel: notImplemented("setModel"),
+  } as unknown as ExtensionAPI;
+
+  return { api, tools };
+}
+
+describe("@jmcombs/pi-tavily-search", () => {
+  it("exports a default factory function", () => {
+    expect(typeof factory).toBe("function");
+  });
+
+  it("registers exactly one tool, named tavily_search", () => {
+    const { api, tools } = createApiStub();
+    factory(api);
+
+    expect(tools).toHaveLength(1);
+    expect(tools[0]?.name).toBe("tavily_search");
+    expect(tools[0]?.label).toBe("Tavily Web Search");
+    expect(tools[0]?.description).toMatch(/tavily/i);
+  });
+
+  it("declares a TypeBox schema requiring a non-empty query string", () => {
+    const { api, tools } = createApiStub();
+    factory(api);
+
+    const params = tools[0]?.parameters as {
+      type: string;
+      properties: { query: { type: string; minLength?: number } };
+      required: string[];
+    };
+
+    expect(params.type).toBe("object");
+    expect(params.properties.query.type).toBe("string");
+    expect(params.properties.query.minLength).toBe(1);
+    expect(params.required).toContain("query");
+  });
+
+  it("publicly exports the TavilySearchInput type for downstream extensions", () => {
+    // Compile-time only: this assignment fails the build if the exported type
+    // ever drifts from its actual schema shape.
+    const sample: TavilySearchInput = { query: "pi coding agent" };
+    expect(sample.query).toBe("pi coding agent");
+  });
+});