chatccc 0.2.4 → 0.2.6

package/src/__tests__/claude-adapter.test.ts

@@ -0,0 +1,529 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { normalizeSdkMessage, createClaudeAdapter } from "../adapters/claude-adapter.ts";
+import type { UnifiedStreamMessage } from "../adapters/adapter-interface.ts";
+
+// ---------------------------------------------------------------------------
+// Mock Claude SDK
+// ---------------------------------------------------------------------------
+
+const mockSend = vi.fn();
+const mockClose = vi.fn();
+const mockStreamNext = vi.fn();
+const mockStreamIterable: AsyncGenerator<any, void> = {
+  next: mockStreamNext,
+  [Symbol.asyncIterator]() { return this; },
+} as any;
+
+function mockSession(overrides: Partial<{ send: any; close: any; streamNext: any }> = {}) {
+  return {
+    send: overrides.send ?? mockSend,
+    close: overrides.close ?? mockClose,
+    stream: () => ({
+      next: overrides.streamNext ?? mockStreamNext,
+      [Symbol.asyncIterator]() { return this; },
+    }),
+  };
+}
+
+vi.mock("@anthropic-ai/claude-agent-sdk", () => {
+  // We only mock the full adapter integration tests; normalizeSdkMessage is pure
+  return {
+    unstable_v2_createSession: vi.fn(),
+    unstable_v2_resumeSession: vi.fn(),
+    getSessionInfo: vi.fn(),
+  };
+});
+
+// ---------------------------------------------------------------------------
+// normalizeSdkMessage — 核心映射逻辑测试（纯函数，不需要 mock SDK）
+// ---------------------------------------------------------------------------
+
+describe("normalizeSdkMessage", () => {
+  // --- assistant messages ---
+
+  it("normalizes assistant message with text block", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "text", text: "Hello world" }] },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.type).toBe("assistant");
+    expect(result!.blocks).toEqual([{ type: "text", text: "Hello world" }]);
+  });
+
+  it("normalizes assistant message with thinking block", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "thinking", thinking: "Let me analyze..." }] },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.blocks).toEqual([{ type: "thinking", thinking: "Let me analyze..." }]);
+  });
+
+  it("normalizes assistant message with tool_use block", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: {
+        content: [{ type: "tool_use", name: "Read", input: { file_path: "/tmp/test.txt" } }],
+      },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.blocks).toEqual([
+      { type: "tool_use", name: "Read", input: { file_path: "/tmp/test.txt" } },
+    ]);
+  });
+
+  it('uses "unknown" for tool_use without name', () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "tool_use", input: { x: 1 } }] },
+    });
+    expect(result!.blocks[0]).toMatchObject({ type: "tool_use", name: "unknown" });
+  });
+
+  it("normalizes user message with tool_result block (success)", () => {
+    const result = normalizeSdkMessage({
+      type: "user",
+      message: {
+        content: [
+          { type: "tool_result", tool_use_id: "abc123", content: "file content here", is_error: false },
+        ],
+      },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.type).toBe("user");
+    const block = result!.blocks[0] as any;
+    expect(block.type).toBe("tool_result");
+    expect(block.tool_use_id).toBe("abc123");
+    expect(block.content).toBe("file content here");
+    expect(block.is_error).toBe(false);
+  });
+
+  it("normalizes user message with tool_result block (error)", () => {
+    const result = normalizeSdkMessage({
+      type: "user",
+      message: {
+        content: [
+          { type: "tool_result", tool_use_id: "err456", content: "permission denied", is_error: true },
+        ],
+      },
+    });
+    expect(result!.blocks[0]).toMatchObject({
+      type: "tool_result",
+      tool_use_id: "err456",
+      is_error: true,
+    });
+  });
+
+  it("normalizes tool_result with array content", () => {
+    const content = [{ type: "text", text: "line 1" }, { type: "text", text: "line 2" }];
+    const result = normalizeSdkMessage({
+      type: "user",
+      message: { content: [{ type: "tool_result", tool_use_id: "arr", content }] },
+    });
+    expect(result!.blocks[0]).toMatchObject({
+      type: "tool_result",
+      tool_use_id: "arr",
+    });
+    expect((result!.blocks[0] as any).content).toEqual(content);
+  });
+
+  it("normalizes redacted_thinking block", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "redacted_thinking" }] },
+    });
+    expect(result!.blocks).toEqual([{ type: "redacted_thinking" }]);
+  });
+
+  it("normalizes search_result block", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: {
+        content: [{ type: "search_result", query: "TypeScript best practices" }],
+      },
+    });
+    expect(result!.blocks).toEqual([
+      { type: "search_result", query: "TypeScript best practices" },
+    ]);
+  });
+
+  it("normalizes search_result without query (empty string)", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "search_result" }] },
+    });
+    expect(result!.blocks).toEqual([{ type: "search_result", query: "" }]);
+  });
+
+  // --- system messages ---
+
+  it("normalizes system compact_boundary message", () => {
+    const result = normalizeSdkMessage({
+      type: "system",
+      subtype: "compact_boundary",
+      compact_metadata: {
+        trigger: "auto",
+        pre_tokens: 15000,
+        post_tokens: 8000,
+      },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.type).toBe("system");
+    expect(result!.blocks).toEqual([
+      { type: "compact_boundary", trigger: "auto", pre_tokens: 15000, post_tokens: 8000 },
+    ]);
+  });
+
+  it("normalizes compact_boundary with manual trigger", () => {
+    const result = normalizeSdkMessage({
+      type: "system",
+      subtype: "compact_boundary",
+      compact_metadata: { trigger: "manual", pre_tokens: 20000 },
+    });
+    expect(result!.blocks).toEqual([
+      { type: "compact_boundary", trigger: "manual", pre_tokens: 20000, post_tokens: undefined },
+    ]);
+  });
+
+  it("normalizes compact_boundary without post_tokens", () => {
+    const result = normalizeSdkMessage({
+      type: "system",
+      subtype: "compact_boundary",
+      compact_metadata: { trigger: "auto", pre_tokens: 10000 },
+    });
+    expect(result!.blocks).toEqual([
+      { type: "compact_boundary", trigger: "auto", pre_tokens: 10000, post_tokens: undefined },
+    ]);
+  });
+
+  // --- mixed blocks ---
+
+  it("normalizes message with multiple block types mixed", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: {
+        content: [
+          { type: "thinking", thinking: "Hmm..." },
+          { type: "tool_use", name: "Grep", input: { pattern: "foo" } },
+          { type: "text", text: "Found it." },
+        ],
+      },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.blocks).toHaveLength(3);
+    expect(result!.blocks[0]).toEqual({ type: "thinking", thinking: "Hmm..." });
+    expect(result!.blocks[1]).toEqual({ type: "tool_use", name: "Grep", input: { pattern: "foo" } });
+    expect(result!.blocks[2]).toEqual({ type: "text", text: "Found it." });
+  });
+
+  // --- edge cases ---
+
+  it("skips unknown block types", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: {
+        content: [
+          { type: "unknown_type", data: "whatever" },
+          { type: "text", text: "still here" },
+        ],
+      },
+    });
+    expect(result!.blocks).toHaveLength(1);
+    expect(result!.blocks[0]).toEqual({ type: "text", text: "still here" });
+  });
+
+  it("returns null for non-assistant/non-user/non-system messages", () => {
+    expect(
+      normalizeSdkMessage({ type: "result", subtype: "success" }),
+    ).toBeNull();
+    expect(
+      normalizeSdkMessage({ type: "stream_event" }),
+    ).toBeNull();
+    expect(
+      normalizeSdkMessage({ type: "status" }),
+    ).toBeNull();
+  });
+
+  it("returns null for system message without compact_boundary subtype", () => {
+    expect(
+      normalizeSdkMessage({ type: "system", subtype: "init" }),
+    ).toBeNull();
+    expect(
+      normalizeSdkMessage({ type: "system", subtype: "notification" }),
+    ).toBeNull();
+  });
+
+  it("returns message with empty blocks for content=[ ]", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [] },
+    });
+    expect(result).not.toBeNull();
+    expect(result!.blocks).toEqual([]);
+  });
+
+  it("returns null for message without content", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: {},
+    });
+    expect(result).toBeNull();
+  });
+
+  it("returns null for message with null content array", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: undefined as any },
+    });
+    expect(result).toBeNull();
+  });
+
+  it("returns null for compact_boundary without metadata", () => {
+    const result = normalizeSdkMessage({
+      type: "system",
+      subtype: "compact_boundary",
+    });
+    expect(result).toBeNull();
+  });
+
+  it("skips thinking block with empty thinking string", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "thinking", thinking: "" }] },
+    });
+    // empty thinking string → falsy check (block.thinking → "") → skipped
+    expect(result!.blocks).toEqual([]);
+  });
+
+  it("skips text block with empty text string", () => {
+    const result = normalizeSdkMessage({
+      type: "assistant",
+      message: { content: [{ type: "text", text: "" }] },
+    });
+    // empty text string → falsy check → skipped
+    expect(result!.blocks).toEqual([]);
+  });
+
+  it("handles message with undefined type gracefully", () => {
+    const result = normalizeSdkMessage({
+      message: { content: [{ type: "text", text: "orphan" }] },
+    });
+    expect(result).toBeNull();
+  });
+});
+
+// ---------------------------------------------------------------------------
+// createClaudeAdapter — 集成测试（mock SDK）
+// ---------------------------------------------------------------------------
+
+describe("createClaudeAdapter", () => {
+  let sdk: any;
+
+  beforeEach(async () => {
+    vi.clearAllMocks();
+    sdk = await import("@anthropic-ai/claude-agent-sdk");
+  });
+
+  it("returns adapter with correct displayName and sessionDescPrefix", () => {
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: (v) => v.trim().toLowerCase() === "default",
+    });
+    expect(adapter.displayName).toBe("Claude Code");
+    expect(adapter.sessionDescPrefix).toBe("Claude Code Session:");
+  });
+
+  it("createSession returns sessionId from init event", async () => {
+    const mockSessionObj = mockSession({
+      streamNext: vi
+        .fn()
+        .mockResolvedValueOnce({
+          done: false,
+          value: { session_id: "test-sid-001", type: "system", subtype: "init" },
+        })
+        .mockResolvedValueOnce({ done: true, value: undefined }),
+    });
+    sdk.unstable_v2_createSession.mockReturnValue(mockSessionObj);
+
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    const result = await adapter.createSession("/tmp");
+    expect(result.sessionId).toBe("test-sid-001");
+    expect(sdk.unstable_v2_createSession).toHaveBeenCalledTimes(1);
+  });
+
+  it("createSession throws when first event has no session_id", async () => {
+    const mockSessionObj = mockSession({
+      streamNext: vi
+        .fn()
+        .mockResolvedValueOnce({
+          done: false,
+          value: { type: "other", no_session_id: true },
+        }),
+    });
+    sdk.unstable_v2_createSession.mockReturnValue(mockSessionObj);
+
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    await expect(adapter.createSession("/tmp")).rejects.toThrow(
+      "No session ID",
+    );
+  });
+
+  it("createSession sends 'ok' to the session", async () => {
+    const sendSpy = vi.fn();
+    const mockSessionObj = mockSession({
+      send: sendSpy,
+      streamNext: vi
+        .fn()
+        .mockResolvedValueOnce({
+          done: false,
+          value: { session_id: "sid", type: "system", subtype: "init" },
+        })
+        .mockResolvedValueOnce({ done: true, value: undefined }),
+    });
+    sdk.unstable_v2_createSession.mockReturnValue(mockSessionObj);
+
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    await adapter.createSession("/tmp");
+    expect(sendSpy).toHaveBeenCalledWith("ok");
+  });
+
+  it("prompt yields normalized messages", async () => {
+    const mockSessionObj = mockSession({
+      streamNext: (() => {
+        let callCount = 0;
+        return vi.fn(async () => {
+          callCount++;
+          if (callCount === 1) {
+            return {
+              done: false,
+              value: {
+                type: "assistant",
+                message: { content: [{ type: "text", text: "Hello!" }] },
+              },
+            };
+          }
+          return { done: true, value: undefined };
+        });
+      })(),
+    });
+    sdk.unstable_v2_resumeSession.mockReturnValue(mockSessionObj);
+
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    const messages: UnifiedStreamMessage[] = [];
+    for await (const msg of adapter.prompt("sid", "hi", "/tmp")) {
+      messages.push(msg);
+    }
+
+    expect(messages).toHaveLength(1);
+    expect(messages[0].type).toBe("assistant");
+    expect(messages[0].blocks).toEqual([{ type: "text", text: "Hello!" }]);
+    expect(sdk.unstable_v2_resumeSession).toHaveBeenCalledTimes(1);
+  });
+
+  it("prompt handles AbortSignal", async () => {
+    const mockSessionObj = mockSession({
+      streamNext: vi
+        .fn()
+        .mockResolvedValueOnce({
+          done: false,
+          value: {
+            type: "assistant",
+            message: { content: [{ type: "text", text: "msg1" }] },
+          },
+        })
+        .mockResolvedValueOnce({
+          done: false,
+          value: {
+            type: "assistant",
+            message: { content: [{ type: "text", text: "msg2" }] },
+          },
+        })
+        .mockResolvedValue({ done: true, value: undefined }),
+    });
+    sdk.unstable_v2_resumeSession.mockReturnValue(mockSessionObj);
+
+    const controller = new AbortController();
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    const messages: UnifiedStreamMessage[] = [];
+    for await (const msg of adapter.prompt("sid", "hi", "/tmp", controller.signal)) {
+      messages.push(msg);
+      if (messages.length === 1) controller.abort();
+    }
+
+    // Should only get the first message before abort stops iteration
+    expect(messages).toHaveLength(1);
+  });
+
+  it("getSessionInfo returns mapped SessionInfo", async () => {
+    sdk.getSessionInfo.mockResolvedValue({
+      sessionId: "sid",
+      cwd: "/home/user",
+      summary: "A test session",
+      lastModified: 1234567890,
+    });
+
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    const info = await adapter.getSessionInfo("sid");
+    expect(info).toEqual({
+      sessionId: "sid",
+      cwd: "/home/user",
+      summary: "A test session",
+      lastModified: 1234567890,
+    });
+  });
+
+  it("getSessionInfo returns undefined when SDK returns undefined", async () => {
+    sdk.getSessionInfo.mockResolvedValue(undefined);
+
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    const info = await adapter.getSessionInfo("nonexistent");
+    expect(info).toBeUndefined();
+  });
+
+  it("closeSession is no-op (does not throw)", async () => {
+    const adapter = createClaudeAdapter({
+      model: "claude-sonnet-4-6",
+      effort: "high",
+      isDefault: () => false,
+    });
+
+    await expect(adapter.closeSession("any-sid")).resolves.toBeUndefined();
+  });
+});