Add typed mock factory utilities for testing

New testing utilities in common/src/testing/mocks/:
- createMockLogger() - typed logger mock with capture support
- createMockFetch(), installMockFetch() - typed fetch mock with call tracking
- createMockFs() - typed filesystem mock compatible with CodebuffFileSystem
- createMockTimers() - typed timer mock (deprecated, prefer Bun builtins)
- Tree-sitter mocks: createMockTreeSitterParser/Query/Captures

Test fixture improvements:
- createTestAgentRuntimeParams() for agent runtime testing
- mockFileContext, testClientEnv, testCiEnv exports

Bug fixes:
- Fix installMockFetch call-capture bug (calls now captured even after mockImplementation)
- Fix null guard in toContentString (prevents TypeError on null array items)

Updated test files to use typed mock factories:
- packages/code-map tests use tree-sitter mock utilities
- web/agents-transform.test.ts uses explicit type annotations
- Various agent-runtime tests use typed fixtures

Also trimmed verbose JSDoc comments across all testing utility files.

Author: brandonkachen

agents/e2e/context-pruner.e2e.test.ts

@@ -10,6 +10,29 @@ import {
   type ToolMessage,
   type JSONValue,
 } from '@codebuff/sdk'
+
+import type { ToolCallPart } from '@codebuff/common/types/messages/content-part'
+
+/**
+ * Type guard to check if a content part is a tool-call part with toolCallId.
+ */
+function isToolCallPart(part: unknown): part is ToolCallPart {
+  return (
+    typeof part === 'object' &&
+    part !== null &&
+    'type' in part &&
+    part.type === 'tool-call' &&
+    'toolCallId' in part &&
+    typeof (part as ToolCallPart).toolCallId === 'string'
+  )
+}
+
+/**
+ * Type guard to check if a message is a tool message with toolCallId.
+ */
+function isToolMessageWithId(msg: Message): msg is ToolMessage & { toolCallId: string } {
+  return msg.role === 'tool' && 'toolCallId' in msg && typeof msg.toolCallId === 'string'
+}
 /**
  * Integration tests for the context-pruner agent.
  * These tests verify that context-pruner correctly prunes message history
@@ -154,8 +177,8 @@ Do not do anything else. Just spawn context-pruner and then report the result.`,
       for (const msg of finalMessages) {
         if (msg.role === 'assistant' && Array.isArray(msg.content)) {
           for (const part of msg.content) {
-            if (part.type === 'tool-call' && (part as any).toolCallId) {
-              toolCallIds.add((part as any).toolCallId)
+            if (isToolCallPart(part)) {
+              toolCallIds.add(part.toolCallId)
             }
           }
         }
@@ -164,8 +187,8 @@ Do not do anything else. Just spawn context-pruner and then report the result.`,
       // Extract all tool result IDs
       const toolResultIds = new Set<string>()
       for (const msg of finalMessages) {
-        if (msg.role === 'tool' && (msg as any).toolCallId) {
-          toolResultIds.add((msg as any).toolCallId)
+        if (isToolMessageWithId(msg)) {
+          toolResultIds.add(msg.toolCallId)
         }
       }
 
@@ -280,13 +303,13 @@ Do not do anything else. Just spawn context-pruner and then report the result.`,
       for (const msg of finalMessages) {
         if (msg.role === 'assistant' && Array.isArray(msg.content)) {
           for (const part of msg.content) {
-            if (part.type === 'tool-call' && (part as any).toolCallId) {
-              toolCallIds.add((part as any).toolCallId)
+            if (isToolCallPart(part)) {
+              toolCallIds.add(part.toolCallId)
             }
           }
         }
-        if (msg.role === 'tool' && (msg as any).toolCallId) {
-          toolResultIds.add((msg as any).toolCallId)
+        if (isToolMessageWithId(msg)) {
+          toolResultIds.add(msg.toolCallId)
         }
       }
 

cli/src/__tests__/unit/copy-button.test.ts

@@ -1,3 +1,4 @@
+import { createMockTimers } from '@codebuff/common/testing/mocks/timers'
 import { describe, test, expect, beforeEach, afterEach } from 'bun:test'
 
 import {
@@ -10,6 +11,8 @@ import {
 } from '../../components/copy-button'
 import { initializeThemeStore } from '../../hooks/use-theme'
 
+import type { MockTimers } from '@codebuff/common/testing/mocks/timers'
+
 // Initialize theme before tests
 initializeThemeStore()
 
@@ -101,39 +104,15 @@ describe('CopyButton - exported constants', () => {
 })
 
 describe('CopyButton - copied state reset timing', () => {
-  let originalSetTimeout: typeof setTimeout
-  let originalClearTimeout: typeof clearTimeout
-  let timers: { id: number; ms: number; fn: Function; active: boolean }[]
-  let nextId: number
-
-  const runTimers = () => {
-    for (const t of timers) {
-      if (t.active) t.fn()
-    }
-    timers = []
-  }
+  let mockTimers: MockTimers
 
   beforeEach(() => {
-    timers = []
-    nextId = 1
-    originalSetTimeout = globalThis.setTimeout
-    originalClearTimeout = globalThis.clearTimeout
-
-    globalThis.setTimeout = ((fn: Function, ms?: number) => {
-      const id = nextId++
-      timers.push({ id, ms: Number(ms ?? 0), fn, active: true })
-      return id as any
-    }) as any
-
-    globalThis.clearTimeout = ((id?: any) => {
-      const rec = timers.find((t) => t.id === id)
-      if (rec) rec.active = false
-    }) as any
+    mockTimers = createMockTimers()
+    mockTimers.install()
   })
 
   afterEach(() => {
-    globalThis.setTimeout = originalSetTimeout
-    globalThis.clearTimeout = originalClearTimeout
+    mockTimers.restore()
   })
 
   test('uses the exported COPIED_RESET_DELAY_MS constant (2000ms)', () => {
@@ -150,10 +129,11 @@ describe('CopyButton - copied state reset timing', () => {
 
     handleCopy()
     expect(isCopied).toBe(true)
-    expect(timers.length).toBe(1)
-    expect(timers[0].ms).toBe(COPIED_RESET_DELAY_MS)
+    expect(mockTimers.getPendingCount()).toBe(1)
+    const nextTimer = mockTimers.getNext()
+    expect(nextTimer?.ms).toBe(COPIED_RESET_DELAY_MS)
 
-    runTimers()
+    mockTimers.runAll()
     expect(isCopied).toBe(false)
   })
 
@@ -176,8 +156,7 @@ describe('CopyButton - copied state reset timing', () => {
     handleCopy()
     handleCopy()
 
-    const activeTimers = timers.filter((t) => t.active)
-    expect(activeTimers.length).toBe(1)
+    expect(mockTimers.getPendingCount()).toBe(1)
   })
 })
 

cli/src/hooks/__tests__/use-usage-query.test.ts

@@ -1,3 +1,4 @@
+import { createMockLogger } from '@codebuff/common/testing/mocks/logger'
 import {
   describe,
   test,
@@ -145,47 +146,32 @@ describe('fetchUsageData', () => {
     globalThis.fetch = mock(
       async () => new Response('Error', { status: 500 }),
     ) as unknown as typeof fetch
-    const mockLogger = {
-      error: mock(() => {}),
-      warn: mock(() => {}),
-      info: mock(() => {}),
-      debug: mock(() => {}),
-    }
+    const mockLogger = createMockLogger()
 
     await expect(
-      fetchUsageData({ authToken: 'test-token', logger: mockLogger as any }),
+      fetchUsageData({ authToken: 'test-token', logger: mockLogger }),
     ).rejects.toThrow('Failed to fetch usage: 500')
   })
 
   test('should throw error on 401 unauthorized', async () => {
     globalThis.fetch = mock(
       async () => new Response('Unauthorized', { status: 401 }),
     ) as unknown as typeof fetch
-    const mockLogger = {
-      error: mock(() => {}),
-      warn: mock(() => {}),
-      info: mock(() => {}),
-      debug: mock(() => {}),
-    }
+    const mockLogger = createMockLogger()
 
     await expect(
-      fetchUsageData({ authToken: 'invalid-token', logger: mockLogger as any }),
+      fetchUsageData({ authToken: 'invalid-token', logger: mockLogger }),
     ).rejects.toThrow('Failed to fetch usage: 401')
   })
 
   test('should throw error on 402 payment required', async () => {
     globalThis.fetch = mock(
       async () => new Response('Payment Required', { status: 402 }),
     ) as unknown as typeof fetch
-    const mockLogger = {
-      error: mock(() => {}),
-      warn: mock(() => {}),
-      info: mock(() => {}),
-      debug: mock(() => {}),
-    }
+    const mockLogger = createMockLogger()
 
     await expect(
-      fetchUsageData({ authToken: 'test-token', logger: mockLogger as any }),
+      fetchUsageData({ authToken: 'test-token', logger: mockLogger }),
     ).rejects.toThrow('Failed to fetch usage: 402')
   })
 
@@ -255,19 +241,13 @@ describe('fetchUsageData', () => {
       async () => new Response('Server Error', { status: 503 }),
     ) as unknown as typeof fetch
 
-    const errorMock = mock(() => {})
-    const mockLogger = {
-      error: errorMock,
-      warn: mock(() => {}),
-      info: mock(() => {}),
-      debug: mock(() => {}),
-    }
+    const mockLogger = createMockLogger()
 
     await expect(
-      fetchUsageData({ authToken: 'test-token', logger: mockLogger as any }),
+      fetchUsageData({ authToken: 'test-token', logger: mockLogger }),
     ).rejects.toThrow()
 
-    expect(errorMock).toHaveBeenCalledWith(
+    expect(mockLogger.error).toHaveBeenCalledWith(
       { status: 503 },
       'Failed to fetch usage data from API',
     )