test: cover tool call hydration preview

2025-11-15 17:33:56 +01:00
parent af7d7cedb0
commit e26219c967
4 changed files with 136 additions and 18 deletions
@@ -0,0 +1,74 @@
+import { describe, expect, it } from "vitest";
+
+import { resolveToolCallPreview } from "./tool-call-preview";
+import type { CommandDetails, EditEntry, ReadEntry } from "@/utils/tool-call-parsers";
+
+describe("resolveToolCallPreview", () => {
+  it("prefers parsed hydration payloads when provided", () => {
+    const parsedEdits: EditEntry[] = [
+      {
+        filePath: "README.md",
+        diffLines: [
+          { type: "header", content: "@@ -1 +1 @@" },
+          { type: "remove", content: "-Old" },
+          { type: "add", content: "+New" },
+        ],
+      },
+    ];
+    const parsedReads: ReadEntry[] = [
+      {
+        filePath: "README.md",
+        content: "# Hydrated\nFinal text\n",
+      },
+    ];
+    const parsedCommand: CommandDetails = {
+      command: "ls",
+      cwd: "/tmp/hydrated",
+      output: "README.md\npackages\n",
+      exitCode: 0,
+    };
+
+    const preview = resolveToolCallPreview({
+      parsedEditEntries: parsedEdits,
+      parsedReadEntries: parsedReads,
+      parsedCommandDetails: parsedCommand,
+    });
+
+    expect(preview.editEntries).toBe(parsedEdits);
+    expect(preview.readEntries).toBe(parsedReads);
+    expect(preview.commandDetails).toBe(parsedCommand);
+  });
+
+  it("falls back to derived parser output when hydration metadata is missing", () => {
+    const args = {
+      type: "mcp_tool_use",
+      id: "call_fallback",
+      name: "apply_patch",
+      server: "editor",
+      input: {
+        file_path: "README.md",
+        patch: "*** Begin Patch\n*** Update File: README.md\n@@\n-Old\n+New\n*** End Patch",
+      },
+    };
+    const result = {
+      changes: [
+        {
+          file_path: "README.md",
+          previous_content: "Old\n",
+          content: "New\n",
+        },
+      ],
+    };
+
+    const preview = resolveToolCallPreview({
+      args,
+      result,
+    });
+
+    expect(preview.editEntries[0]?.diffLines.length).toBeGreaterThan(0);
+    expect(
+      preview.editEntries[0]?.diffLines.some((line) => line.content.includes("+New"))
+    ).toBe(true);
+    expect(preview.commandDetails).toBeNull();
+  });
+});