@alumbwe/anvil 1.0.20 → 1.0.22

package/vendor/@anvil/common/src/testing/fixtures/agent-runtime.ts

@@ -0,0 +1,334 @@
+/**
+ * Test fixtures for agent runtime testing.
+ *
+ * Provides pre-built test fixtures and factory functions for
+ * testing agent runtime components without needing to set up
+ * all the dependencies manually.
+ *
+ * @example
+ * ```typescript
+ * import {
+ *   createTestAgentRuntimeParams,
+ *   createTestAgentRuntimeDeps,
+ *   mockFileContext,
+ * } from '@anvil/common/testing/fixtures/agent-runtime'
+ *
+ * const params = createTestAgentRuntimeParams()
+ * const { agentTemplate, localAgentTemplates } = params
+ * ```
+ */
+
+import { mock } from 'bun:test'
+
+import { promptSuccess } from '../../util/error'
+
+import type { ProjectFileContext } from '../../util/file'
+
+export const mockFileContext: ProjectFileContext = {
+  projectRoot: '/test',
+  cwd: '/test',
+  fileTree: [],
+  fileTokenScores: {},
+  knowledgeFiles: {},
+  userKnowledgeFiles: {},
+  agentTemplates: {},
+  customToolDefinitions: {},
+  gitChanges: {
+    status: '',
+    diff: '',
+    diffCached: '',
+    lastCommitMessages: '',
+  },
+  changesSinceLastChat: {},
+  shellConfigFiles: {},
+  systemInfo: {
+    platform: 'test',
+    shell: 'test',
+    nodeVersion: 'test',
+    arch: 'test',
+    homedir: '/home/test',
+    cpus: 1,
+    chromeAvailable: false,
+  },
+}
+
+/** @deprecated Use mockFileContext */
+export const testFileContext: ProjectFileContext = mockFileContext
+
+export const testLogger = {
+  debug: () => {},
+  error: () => {},
+  info: () => {},
+  warn: () => {},
+}
+
+export const testFetch = Object.assign(
+  async () => {
+    throw new Error('fetch not implemented in test runtime')
+  },
+  {
+    preconnect: async () => {
+      throw new Error('fetch.preconnect not implemented in test runtime')
+    },
+  },
+)
+
+export const testClientEnv = {
+  NEXT_PUBLIC_CB_ENVIRONMENT: 'test' as const,
+  NEXT_PUBLIC_ANVIL_APP_URL: 'https://test.codebuff.com',
+  NEXT_PUBLIC_SUPPORT_EMAIL: 'support@codebuff.test',
+  NEXT_PUBLIC_POSTHOG_API_KEY: 'test-posthog-key',
+  NEXT_PUBLIC_POSTHOG_HOST_URL: 'https://test.posthog.com',
+  NEXT_PUBLIC_STRIPE_PUBLISHABLE_KEY: 'pk_test_123',
+  NEXT_PUBLIC_STRIPE_CUSTOMER_PORTAL: 'https://test.stripe.com/portal',
+  NEXT_PUBLIC_GOOGLE_SITE_VERIFICATION_ID: undefined,
+  NEXT_PUBLIC_WEB_PORT: 3000,
+}
+
+export const testCiEnv = {
+  CI: undefined,
+  GITHUB_ACTIONS: undefined,
+  RENDER: undefined,
+  IS_PULL_REQUEST: undefined,
+  ANVIL_GITHUB_TOKEN: undefined,
+  ANVIL_API_KEY: 'test-api-key',
+}
+
+/** @deprecated Use createTestAgentRuntimeParams() */
+export const TEST_AGENT_RUNTIME_IMPL = Object.freeze({
+  clientEnv: testClientEnv,
+  ciEnv: testCiEnv,
+  trackEvent: () => {},
+  logger: testLogger,
+  fetch: testFetch,
+  getUserInfoFromApiKey: async <T extends string>({
+    fields,
+  }: {
+    apiKey: string
+    fields: readonly T[]
+  }) => {
+    const user = {
+      id: 'test-user-id',
+      email: 'test@example.com',
+      discord_id: 'test-discord-id',
+      stripe_customer_id: null,
+      banned: false,
+      created_at: new Date('2024-01-01T00:00:00Z'),
+    } as const
+    return Object.fromEntries(
+      fields.map((field) => [field, user[field as keyof typeof user]]),
+    ) as {
+      [K in T]: (typeof user)[K & keyof typeof user]
+    }
+  },
+  fetchAgentFromDatabase: async () => null,
+  startAgentRun: async () => 'test-agent-run-id',
+  finishAgentRun: async () => {},
+  addAgentStep: async () => 'test-agent-step-id',
+  consumeCreditsWithFallback: async () => {
+    throw new Error(
+      'consumeCreditsWithFallback not implemented in test runtime',
+    )
+  },
+  promptAiSdkStream: async function* () {
+    throw new Error('promptAiSdkStream not implemented in test runtime')
+  },
+  promptAiSdk: async function () {
+    throw new Error('promptAiSdk not implemented in test runtime')
+  },
+  promptAiSdkStructured: async function () {
+    throw new Error('promptAiSdkStructured not implemented in test runtime')
+  },
+  databaseAgentCache: new Map(),
+  handleStepsLogChunk: () => {
+    throw new Error('handleStepsLogChunk not implemented in test runtime')
+  },
+  requestToolCall: () => {
+    throw new Error('requestToolCall not implemented in test runtime')
+  },
+  requestMcpToolData: () => {
+    throw new Error('requestMcpToolData not implemented in test runtime')
+  },
+  requestFiles: () => {
+    throw new Error('requestFiles not implemented in test runtime')
+  },
+  requestOptionalFile: () => {
+    throw new Error('requestOptionalFile not implemented in test runtime')
+  },
+  sendSubagentChunk: () => {
+    throw new Error('sendSubagentChunk not implemented in test runtime')
+  },
+  sendAction: () => {
+    throw new Error('sendAction not implemented in test runtime')
+  },
+  apiKey: 'test-api-key',
+})
+
+export interface TestAgentRuntimeParams {
+  agentTemplate: {
+    id: string
+    displayName: string
+    model: string
+    inputSchema: Record<string, unknown>
+    outputMode: string
+    includeMessageHistory: boolean
+    inheritParentSystemPrompt: boolean
+    mcpServers: Record<string, unknown>
+    toolNames: string[]
+    spawnableAgents: string[]
+    systemPrompt: string
+    instructionsPrompt: string
+    stepPrompt: string
+  }
+  localAgentTemplates: Record<string, TestAgentRuntimeParams['agentTemplate']>
+  sendAction: ReturnType<typeof mock>
+  requestFiles: ReturnType<typeof mock>
+  requestToolCall: ReturnType<typeof mock>
+  onResponseChunk: ReturnType<typeof mock>
+  fileContext: ProjectFileContext
+  promptAiSdkStream: ReturnType<typeof mock>
+  promptAiSdk: ReturnType<typeof mock>
+  promptAiSdkStructured: ReturnType<typeof mock>
+  requestMcpToolData: ReturnType<typeof mock>
+  startAgentRun: ReturnType<typeof mock>
+  finishAgentRun: ReturnType<typeof mock>
+  addAgentStep: ReturnType<typeof mock>
+  logger: typeof testLogger
+  trackEvent: ReturnType<typeof mock>
+  clientEnv: typeof testClientEnv
+  ciEnv: typeof testCiEnv
+  apiKey: string
+  fetch: typeof testFetch
+  fetchAgentFromDatabase: ReturnType<typeof mock>
+  databaseAgentCache: Map<string, null>
+  consumeCreditsWithFallback: ReturnType<typeof mock>
+  getUserInfoFromApiKey: ReturnType<typeof mock>
+  handleStepsLogChunk: ReturnType<typeof mock>
+  requestOptionalFile: ReturnType<typeof mock>
+  sendSubagentChunk: ReturnType<typeof mock>
+}
+
+export function createTestAgentRuntimeParams(
+  overrides: Partial<TestAgentRuntimeParams> = {},
+): TestAgentRuntimeParams {
+  const defaultTemplate: TestAgentRuntimeParams['agentTemplate'] = {
+    id: 'test-agent',
+    displayName: 'Test Agent',
+    model: 'claude-3-5-sonnet-20241022',
+    inputSchema: {},
+    outputMode: 'last_message',
+    includeMessageHistory: true,
+    inheritParentSystemPrompt: false,
+    mcpServers: {},
+    toolNames: ['read_files', 'write_file', 'end_turn'],
+    spawnableAgents: [],
+    systemPrompt: 'You are a test agent.',
+    instructionsPrompt: 'Help the user with testing.',
+    stepPrompt: '',
+  }
+
+  const agentTemplate = overrides.agentTemplate ?? defaultTemplate
+
+  return {
+    agentTemplate,
+    localAgentTemplates: overrides.localAgentTemplates ?? {
+      'test-agent': agentTemplate,
+    },
+    sendAction: overrides.sendAction ?? mock(() => {}),
+    requestFiles: overrides.requestFiles ?? mock(async () => ({})),
+    requestToolCall:
+      overrides.requestToolCall ??
+      mock(async () => ({ success: true, result: 'mock result' })),
+    onResponseChunk: overrides.onResponseChunk ?? mock(() => {}),
+    fileContext: overrides.fileContext ?? mockFileContext,
+    promptAiSdkStream:
+      overrides.promptAiSdkStream ??
+      mock(async function* () {
+        yield { type: 'text' as const, text: 'Mock response\n\n' }
+        yield {
+          type: 'tool-call' as const,
+          toolName: 'end_turn',
+          toolCallId: 'mock-id',
+          input: {},
+        }
+        return promptSuccess('mock-message-id')
+      }),
+    promptAiSdk: overrides.promptAiSdk ?? mock(async () => promptSuccess('Mock response')),
+    promptAiSdkStructured:
+      overrides.promptAiSdkStructured ?? mock(async () => promptSuccess({})),
+    requestMcpToolData: overrides.requestMcpToolData ?? mock(async () => ({})),
+    startAgentRun: overrides.startAgentRun ?? mock(async () => 'test-run-id'),
+    finishAgentRun: overrides.finishAgentRun ?? mock(async () => {}),
+    addAgentStep: overrides.addAgentStep ?? mock(async () => 'test-step-id'),
+    logger: overrides.logger ?? testLogger,
+    trackEvent: overrides.trackEvent ?? mock(() => {}),
+    clientEnv: overrides.clientEnv ?? testClientEnv,
+    ciEnv: overrides.ciEnv ?? testCiEnv,
+    apiKey: overrides.apiKey ?? 'test-api-key',
+    fetch: overrides.fetch ?? testFetch,
+    fetchAgentFromDatabase:
+      overrides.fetchAgentFromDatabase ?? mock(async () => null),
+    databaseAgentCache: overrides.databaseAgentCache ?? new Map<string, null>(),
+    consumeCreditsWithFallback:
+      overrides.consumeCreditsWithFallback ?? mock(async () => {}),
+    getUserInfoFromApiKey:
+      overrides.getUserInfoFromApiKey ??
+      mock(async () => ({
+        id: 'test-user-id',
+        email: 'test@example.com',
+      })),
+    handleStepsLogChunk: overrides.handleStepsLogChunk ?? mock(() => {}),
+    requestOptionalFile:
+      overrides.requestOptionalFile ?? mock(async () => null),
+    sendSubagentChunk: overrides.sendSubagentChunk ?? mock(() => {}),
+    ...overrides,
+  }
+}
+
+export function createTestAgentRuntimeDeps(): Omit<
+  TestAgentRuntimeParams,
+  'agentTemplate' | 'localAgentTemplates'
+> {
+  return {
+    sendAction: mock(() => {}),
+    requestFiles: mock(async () => ({})),
+    requestToolCall: mock(async () => ({
+      success: true,
+      result: 'mock result',
+    })),
+    onResponseChunk: mock(() => {}),
+    fileContext: mockFileContext,
+    promptAiSdkStream: mock(async function* () {
+      yield { type: 'text' as const, text: 'Mock response\n\n' }
+      yield {
+        type: 'tool-call' as const,
+        toolName: 'end_turn',
+        toolCallId: 'mock-id',
+        input: {},
+      }
+      return promptSuccess('mock-message-id')
+    }),
+    promptAiSdk: mock(async () => promptSuccess('Mock response')),
+    promptAiSdkStructured: mock(async () => promptSuccess({})),
+    requestMcpToolData: mock(async () => ({})),
+    startAgentRun: mock(async () => 'test-run-id'),
+    finishAgentRun: mock(async () => {}),
+    addAgentStep: mock(async () => 'test-step-id'),
+    logger: testLogger,
+    trackEvent: mock(() => {}),
+    clientEnv: testClientEnv,
+    ciEnv: testCiEnv,
+    apiKey: 'test-api-key',
+    fetch: testFetch,
+    fetchAgentFromDatabase: mock(async () => null),
+    databaseAgentCache: new Map<string, null>(),
+    consumeCreditsWithFallback: mock(async () => {}),
+    getUserInfoFromApiKey: mock(async () => ({
+      id: 'test-user-id',
+      email: 'test@example.com',
+    })),
+    handleStepsLogChunk: mock(() => {}),
+    requestOptionalFile: mock(async () => null),
+    sendSubagentChunk: mock(() => {}),
+  }
+}

package/vendor/@anvil/common/src/testing/impl/agent-runtime.ts

@@ -0,0 +1,6 @@
+/**
+ * @deprecated Use `@anvil/common/testing/fixtures/agent-runtime` instead.
+ */
+
+export * from '../fixtures/agent-runtime'
+export type { TestAgentRuntimeParams } from '../fixtures/agent-runtime'

package/vendor/@anvil/common/src/testing/index.ts

@@ -0,0 +1,84 @@
+/**
+ * Consolidated testing utilities for Anvil.
+ *
+ * This module re-exports all testing utilities from a single entry point,
+ * making it easy to import everything you need for testing.
+ *
+ * ## Module Overview
+ *
+ * - **mocks**: Typed mock factories for logger, analytics, database, crypto, and streams
+ * - **fixtures**: Pre-built test fixtures for agent runtime and other components
+ * - **errors**: Typed error creators for testing error handling
+ * - **mock-modules**: Dynamic module mocking utilities
+ * - **env**: Test environment helpers
+ *
+ * @example
+ * ```typescript
+ * import {
+ *   // Mock factories
+ *   createMockLogger,
+ *   createMockAnalytics,
+ *   setupDbSpies,
+ *   setupCryptoMocks,
+ *
+ *   // Fixtures
+ *   createTestAgentRuntimeParams,
+ *
+ *   // Errors
+ *   createNodeError,
+ *
+ *   // Module mocking
+ *   mockModule,
+ *   clearMockedModules,
+ * } from '@anvil/common/testing'
+ * ```
+ *
+ * @module testing
+ */
+
+// ============================================================================
+// Mock Factories
+// ============================================================================
+
+export * from './mocks'
+
+// ============================================================================
+// Fixtures
+// ============================================================================
+
+export {
+  createTestAgentRuntimeParams,
+  createTestAgentRuntimeDeps,
+  mockFileContext,
+} from './fixtures/agent-runtime'
+export type { TestAgentRuntimeParams } from './fixtures/agent-runtime'
+
+// ============================================================================
+// Error Utilities
+// ============================================================================
+
+export { createNodeError, createPostgresError } from './errors'
+export type { NodeError, PostgresError } from './errors'
+
+// ============================================================================
+// Module Mocking
+// ============================================================================
+
+export { mockModule, clearMockedModules } from './mock-modules'
+
+// ============================================================================
+// Test Setup Utilities
+// ============================================================================
+
+export { createTestSetup, sleep, waitFor, captureCallArgs } from './setup'
+export type { CreateTestSetupOptions, TestSetupResult } from './setup'
+
+// ============================================================================
+// Environment Helpers (re-exported from sibling modules)
+// ============================================================================
+
+// Note: These are in separate files for historical reasons but are commonly
+// used together with other testing utilities.
+// Import directly from their modules if you need only env helpers:
+//   - '@anvil/common/testing-env-process' for process env
+//   - '@anvil/common/testing-env-ci' for CI env

package/vendor/@anvil/common/src/testing/mock-modules.ts

@@ -0,0 +1,53 @@
+import { mock } from 'bun:test'
+
+export type MockResult = {
+  clear: () => void
+}
+
+const originalModuleCache: Record<string, any> = {}
+let mockModuleCache: Record<string, MockResult> = {}
+
+/**
+ *
+ * @param modulePath - the path starting from this files' path.
+ * @param renderMocks - function to generate mocks (by their named or default exports)
+ * @returns an object
+ */
+export async function mockModule(
+  modulePath: string,
+  renderMocks: () => Record<string, any>,
+): Promise<MockResult> {
+  let original = originalModuleCache[modulePath]
+  if (!original) {
+    const moduleExports = await import(modulePath)
+    original = {
+      ...moduleExports,
+    }
+    originalModuleCache[modulePath] = original
+  }
+  let mocks = renderMocks()
+  let result = {
+    ...original,
+    ...mocks,
+  }
+  mock.module(modulePath, () => result)
+  let num = 0
+  let key = modulePath
+  while (key in mockModuleCache) {
+    num++
+    key = `${modulePath}\n${num}`
+  }
+  const mocked: MockResult = {
+    clear: () => {
+      mock.module(modulePath, () => original)
+      delete mockModuleCache[key]
+    },
+  }
+  mockModuleCache[key] = mocked
+  return mocked
+}
+
+export const clearMockedModules = () => {
+  Object.values(mockModuleCache).forEach((mockResult) => mockResult.clear())
+  mockModuleCache = {}
+}

package/vendor/@anvil/common/src/testing/mock-types.ts

@@ -0,0 +1,123 @@
+/**
+ * Shared mock types for testing.
+ *
+ * This module provides common mock types and factory functions that are
+ * frequently used across test files. Using these shared types ensures
+ * consistency and reduces duplication in test code.
+ *
+ * @example
+ * ```typescript
+ * import {
+ *   createMockLogger,
+ *   type MockUserInfo,
+ *   type MockCreditResult,
+ * } from '@anvil/common/testing/mock-types'
+ *
+ * const logger = createMockLogger()
+ * const userInfo: MockUserInfo = { id: 'user-123' }
+ * ```
+ */
+
+import { mock } from 'bun:test'
+
+import type { Logger } from '../types/contracts/logger'
+
+/**
+ * Mock user info returned by API key lookup functions.
+ * Contains the minimal user identification data needed for testing.
+ */
+export interface MockUserInfo {
+  id: string
+}
+
+/**
+ * Mock result from credit consumption operations.
+ * Used when testing billing-related functionality.
+ */
+export interface MockCreditResult {
+  success: boolean
+  value: { chargedToOrganization: boolean }
+}
+
+/**
+ * Mock file stat result for filesystem operations.
+ * Provides typed methods for checking file type.
+ */
+export interface MockStatResult {
+  isDirectory: () => boolean
+  isFile: () => boolean
+}
+
+/**
+ * Typed mock logger where each method is a Bun test mock.
+ * Useful for verifying that specific log methods were called.
+ */
+export type MockLogger = {
+  [K in keyof Logger]: ReturnType<typeof mock> & Logger[K]
+}
+
+/**
+ * Creates a mock logger with all methods as Bun test mocks.
+ * Each method can be inspected for calls using mock.calls.
+ *
+ * @example
+ * ```typescript
+ * const logger = createMockLogger()
+ * someFunction({ logger })
+ * expect(logger.error.mock.calls.length).toBe(1)
+ * ```
+ */
+export function createMockLogger(): MockLogger {
+  return {
+    info: mock(() => {}) as ReturnType<typeof mock> & Logger['info'],
+    error: mock(() => {}) as ReturnType<typeof mock> & Logger['error'],
+    warn: mock(() => {}) as ReturnType<typeof mock> & Logger['warn'],
+    debug: mock(() => {}) as ReturnType<typeof mock> & Logger['debug'],
+  }
+}
+
+/**
+ * Creates a mock stat result for filesystem testing.
+ *
+ * @param options - Configure whether the mock represents a directory or file
+ * @returns A MockStatResult with the specified behavior
+ *
+ * @example
+ * ```typescript
+ * const dirStat = createMockStatResult({ isDirectory: true })
+ * const fileStat = createMockStatResult({ isFile: true })
+ * ```
+ */
+export function createMockStatResult(options: {
+  isDirectory?: boolean
+  isFile?: boolean
+}): MockStatResult {
+  return {
+    isDirectory: () => options.isDirectory ?? false,
+    isFile: () => options.isFile ?? false,
+  }
+}
+
+/**
+ * Creates a mock credit result for billing-related tests.
+ *
+ * @param options - Configure the success state and organization charging
+ * @returns A MockCreditResult with the specified values
+ *
+ * @example
+ * ```typescript
+ * const successResult = createMockCreditResult({ success: true })
+ * const orgResult = createMockCreditResult({ success: true, chargedToOrganization: true })
+ * ```
+ */
+export function createMockCreditResult(
+  options: {
+    success?: boolean
+    chargedToOrganization?: boolean
+  } = {},
+): MockCreditResult {
+  return {
+    success: options.success ?? true,
+    value: { chargedToOrganization: options.chargedToOrganization ?? false },
+  }
+}