whale-code 6.5.6 → 6.5.7

package/dist/cli/services/tools/__tests__/agent-tools.test.js

@@ -0,0 +1,340 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+
+// ============================================================================
+// Mocks
+// ============================================================================
+
+vi.mock("fs", () => ({
+  readFileSync: vi.fn(path => {
+    if (path.includes("MEMORY.md")) return "# Memory\nSome memory content here";
+    if (path.includes("commit.md")) return "---\ntitle: commit\n---\nCommit instructions for $1 with $ARGS";
+    throw new Error("File not found");
+  }),
+  existsSync: vi.fn(path => {
+    if (path.includes("MEMORY.md")) return true;
+    if (path.includes("commit.md")) return true;
+    return false;
+  })
+}));
+vi.mock("../../subagent.js", () => ({
+  runSubagent: vi.fn(async opts => ({
+    success: true,
+    output: `Subagent completed: ${opts.prompt}`
+  })),
+  runSubagentBackground: vi.fn(async () => ({
+    agentId: "agent-abc123",
+    outputFile: "/tmp/agent-abc123.out"
+  }))
+}));
+vi.mock("../../telemetry.js", () => ({
+  createTurnContext: vi.fn(() => ({
+    traceId: "trace-1",
+    spanId: "span-1",
+    conversationId: "conv-1",
+    userId: "user-1",
+    userEmail: "test@test.com"
+  })),
+  getTurnNumber: vi.fn(() => 5)
+}));
+vi.mock("../../team-lead.js", () => ({
+  runAgentTeam: vi.fn(async config => ({
+    success: true,
+    durationMs: 5000,
+    tokensUsed: {
+      input: 1000,
+      output: 500
+    },
+    taskResults: config.tasks.map(t => ({
+      description: t.description,
+      status: "completed",
+      result: `Done: ${t.description}`
+    }))
+  }))
+}));
+vi.mock("../../background-processes.js", () => ({
+  readAgentOutput: vi.fn(id => {
+    if (id === "agent-running") return {
+      status: "running",
+      output: "still working..."
+    };
+    if (id === "agent-done") return {
+      status: "completed",
+      output: "result data"
+    };
+    return null;
+  }),
+  stopBackgroundAgent: vi.fn(id => ({
+    success: true,
+    message: `Stopped ${id}`
+  })),
+  readProcessOutput: vi.fn(_id => ({
+    status: "completed",
+    exitCode: 0,
+    newOutput: "shell output",
+    newErrors: ""
+  })),
+  killProcess: vi.fn(() => ({
+    success: true,
+    message: "Process killed"
+  }))
+}));
+vi.mock("../../agent-events.js", () => {
+  const emitter = {
+    emit: vi.fn()
+  };
+  return {
+    getGlobalEmitter: vi.fn(() => emitter)
+  };
+});
+vi.mock("../../lsp-manager.js", () => ({
+  executeLSP: vi.fn(async op => ({
+    success: true,
+    output: `LSP result for ${op}`
+  }))
+}));
+vi.mock("../../debug-log.js", () => ({
+  debugLog: vi.fn()
+}));
+vi.mock("../../permission-modes.js", () => ({
+  setPermissionMode: vi.fn(mode => ({
+    success: true,
+    message: `Mode set to ${mode}`
+  })),
+  getPermissionMode: vi.fn(() => "default")
+}));
+vi.mock("../../model-manager.js", () => ({
+  getModel: vi.fn(() => "claude-sonnet-4-20250514"),
+  setModel: vi.fn(value => {
+    if (value === "invalid-model") return {
+      success: false,
+      error: "Unknown model"
+    };
+    return {
+      success: true,
+      model: value
+    };
+  })
+}));
+import { configTool, askUser, lspTool, skillTool } from "../agent-tools.js";
+import { getGlobalEmitter } from "../../agent-events.js";
+beforeEach(() => {
+  vi.clearAllMocks();
+});
+
+// ============================================================================
+// configTool
+// ============================================================================
+
+describe("configTool", () => {
+  it("returns current model when no value provided", () => {
+    const result = configTool({
+      setting: "model"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("Current model");
+  });
+  it("sets model with valid value", () => {
+    const result = configTool({
+      setting: "model",
+      value: "claude-opus-4-20250514"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("Model set to");
+  });
+  it("returns error for invalid model", () => {
+    const result = configTool({
+      setting: "model",
+      value: "invalid-model"
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("Unknown model");
+  });
+  it("returns current permission mode", () => {
+    const result = configTool({
+      setting: "mode"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("Current mode: default");
+  });
+  it("sets permission mode", () => {
+    const result = configTool({
+      setting: "mode",
+      value: "yolo"
+    });
+    expect(result.success).toBe(true);
+  });
+  it("also accepts 'permission_mode' alias", () => {
+    const result = configTool({
+      setting: "permission_mode",
+      value: "plan"
+    });
+    expect(result.success).toBe(true);
+  });
+  it("rejects invalid permission mode", () => {
+    const result = configTool({
+      setting: "mode",
+      value: "badmode"
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("Invalid mode");
+  });
+  it("reads memory file", () => {
+    const result = configTool({
+      setting: "memory"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("Memory");
+    expect(result.output).toContain("Some memory content");
+  });
+  it("rejects writing to memory directly", () => {
+    const result = configTool({
+      setting: "memory",
+      value: "new memory"
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("read_file/write_file");
+  });
+  it("returns error for unknown setting", () => {
+    const result = configTool({
+      setting: "unknown_setting"
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("Unknown setting");
+  });
+});
+
+// ============================================================================
+// askUser
+// ============================================================================
+
+describe("askUser", () => {
+  it("formats question with options", () => {
+    const result = askUser({
+      question: "Which framework?",
+      options: [{
+        label: "React",
+        description: "Popular UI library"
+      }, {
+        label: "Vue",
+        description: "Progressive framework"
+      }]
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("Which framework?");
+    expect(result.output).toContain("React");
+    expect(result.output).toContain("Vue");
+    expect(result.output).toContain("Popular UI library");
+  });
+  it("emits ask_user event via global emitter", () => {
+    askUser({
+      question: "Pick one",
+      options: [{
+        label: "A",
+        description: "option A"
+      }, {
+        label: "B",
+        description: "option B"
+      }]
+    });
+    const emitter = getGlobalEmitter();
+    expect(emitter.emit).toHaveBeenCalledWith("ask_user", expect.objectContaining({
+      question: "Pick one"
+    }));
+  });
+  it("requires question parameter", () => {
+    const result = askUser({
+      options: [{
+        label: "A",
+        description: "a"
+      }, {
+        label: "B",
+        description: "b"
+      }]
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("question is required");
+  });
+  it("requires at least 2 options", () => {
+    const result = askUser({
+      question: "test",
+      options: [{
+        label: "Only",
+        description: "one"
+      }]
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("at least 2 options");
+  });
+  it("fails with no options", () => {
+    const result = askUser({
+      question: "test"
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("at least 2 options");
+  });
+});
+
+// ============================================================================
+// lspTool
+// ============================================================================
+
+describe("lspTool", () => {
+  it("requires operation parameter", async () => {
+    const result = await lspTool({});
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("operation is required");
+  });
+  it("delegates to executeLSP", async () => {
+    const result = await lspTool({
+      operation: "hover",
+      file: "test.ts",
+      line: 10
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("LSP result for hover");
+  });
+});
+
+// ============================================================================
+// skillTool
+// ============================================================================
+
+describe("skillTool", () => {
+  it("requires skill name", () => {
+    const result = skillTool({});
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("skill name is required");
+  });
+  it("returns error when skill not found", () => {
+    const result = skillTool({
+      skill: "nonexistent-skill"
+    });
+    expect(result.success).toBe(false);
+    expect(result.output).toContain("Skill not found");
+  });
+  it("loads a skill template and expands arguments", () => {
+    const result = skillTool({
+      skill: "commit",
+      args: "fix-bug"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("[Skill: commit");
+    expect(result.output).toContain("Commit instructions for fix-bug");
+  });
+  it("expands $ARGS with all arguments", () => {
+    const result = skillTool({
+      skill: "commit",
+      args: "arg1 arg2 arg3"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("arg1 arg2 arg3");
+  });
+  it("strips frontmatter from skill template", () => {
+    const result = skillTool({
+      skill: "commit"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).not.toContain("---");
+    expect(result.output).not.toContain("title: commit");
+  });
+});

package/dist/cli/services/tools/__tests__/file-ops-cache.test.js

@@ -0,0 +1,152 @@
+/**
+ * Tests for file-ops — readFile cache behavior and searchContent/searchFiles.
+ */
+
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { vol } from "memfs";
+
+// ============================================================================
+// Mocks — must be declared before imports
+// ============================================================================
+
+vi.mock("fs", async () => {
+  const memfs = await import("memfs");
+  return {
+    ...memfs.fs,
+    default: memfs.fs
+  };
+});
+vi.mock("../../file-history.js", () => ({
+  backupFile: vi.fn()
+}));
+vi.mock("../../lsp-manager.js", () => ({
+  notifyFileChanged: vi.fn()
+}));
+vi.mock("../../debug-log.js", () => ({
+  debugLog: vi.fn()
+}));
+vi.mock("child_process", () => ({
+  execSync: vi.fn(cmd => {
+    if (cmd.includes("find")) {
+      if (cmd.includes("-name '*.ts'")) return "/project/foo.ts\n/project/bar.ts\n";
+      if (cmd.includes("-maxdepth 4")) return "/project\n/project/src\n/project/src/index.ts\n";
+      return "";
+    }
+    if (cmd.includes("grep")) {
+      if (cmd.includes("'TODO'")) return "/project/foo.ts:5:// TODO: fix this\n";
+      return "";
+    }
+    return "";
+  })
+}));
+import { readFile, writeFile, searchFiles, searchContent, clearReadCache } from "../file-ops.js";
+
+// ============================================================================
+// Setup / Teardown
+// ============================================================================
+
+beforeEach(() => {
+  vol.reset();
+  vi.clearAllMocks();
+  clearReadCache();
+});
+afterEach(() => {
+  vol.reset();
+});
+
+// ============================================================================
+// searchFiles
+// ============================================================================
+
+describe("searchFiles", () => {
+  it("searches for files by pattern", () => {
+    const result = searchFiles({
+      pattern: "*.ts",
+      path: "/project"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("foo.ts");
+  });
+  it("escapes single quotes in pattern", () => {
+    const result = searchFiles({
+      pattern: "file's.ts",
+      path: "/project"
+    });
+    expect(result.success).toBeDefined();
+  });
+});
+
+// ============================================================================
+// searchContent
+// ============================================================================
+
+describe("searchContent", () => {
+  it("searches file content with grep", () => {
+    const result = searchContent({
+      query: "TODO",
+      path: "/project"
+    });
+    expect(result.success).toBe(true);
+    expect(result.output).toContain("TODO");
+  });
+  it("returns no matches for empty grep result", () => {
+    const result = searchContent({
+      query: "TODO",
+      path: "/project"
+    });
+    expect(result.success).toBe(true);
+  });
+});
+
+// ============================================================================
+// readFile — cache behavior
+// ============================================================================
+
+describe("readFile — cache behavior", () => {
+  it("returns same content on second read (cache hit)", async () => {
+    vol.fromJSON({
+      "/test/cached.txt": "cached content"
+    });
+    const result1 = await readFile({
+      path: "/test/cached.txt"
+    });
+    const result2 = await readFile({
+      path: "/test/cached.txt"
+    });
+    expect(result1.output).toBe(result2.output);
+    expect(result2.success).toBe(true);
+  });
+  it("writeFile invalidates cache so next read returns updated content", async () => {
+    vol.fromJSON({
+      "/test/update.txt": "original"
+    });
+    await readFile({
+      path: "/test/update.txt"
+    });
+    writeFile({
+      path: "/test/update.txt",
+      content: "updated"
+    });
+    const result = await readFile({
+      path: "/test/update.txt"
+    });
+    expect(result.output).toBe("updated");
+  });
+  it("files > 100KB are not cached but still returned", async () => {
+    const bigContent = "x".repeat(110_000);
+    vol.fromJSON({
+      "/test/big-cache.txt": bigContent
+    });
+    const result1 = await readFile({
+      path: "/test/big-cache.txt"
+    });
+    expect(result1.success).toBe(true);
+    vol.fromJSON({
+      "/test/big-cache.txt": "replaced"
+    });
+    const result2 = await readFile({
+      path: "/test/big-cache.txt"
+    });
+    expect(result2.output).toBe("replaced");
+  });
+});