@renxqoo/renx-code 0.0.9 → 0.0.12

package/src/agent/runtime/runtime.test.ts

@@ -1,296 +0,0 @@
-import * as path from 'node:path';
-import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
-
-vi.mock('./tool-confirmation', () => ({
-  resolveToolConfirmDecision: vi.fn().mockResolvedValue({ approved: true, message: 'Approved' }),
-}));
-
-vi.mock('./source-modules', () => ({
-  getSourceModules: vi.fn(),
-  resolveWorkspaceRoot: vi.fn(),
-}));
-
-import {
-  disposeAgentRuntime,
-  getAgentModelId,
-  getAgentModelLabel,
-  listAgentModels,
-  runAgentPrompt,
-  switchAgentModel,
-} from './runtime';
-import * as sourceModules from './source-modules';
-import type { AgentEventHandlers } from './types';
-
-const buildMockModules = () => {
-  class FakeToolManager {
-    private readonly tools: unknown[] = [];
-
-    registerTool = vi.fn((tool: unknown) => {
-      this.tools.push(tool);
-    });
-
-    getTools = vi.fn(() => this.tools);
-  }
-
-  const createNamedTool = (name: string) =>
-    class NamedFakeTool {
-      toToolSchema() {
-        return {
-          type: 'function',
-          function: {
-            name,
-          },
-        };
-      }
-    };
-
-  class FakeAgent {
-    on = vi.fn();
-    off = vi.fn();
-  }
-
-  class FakeAppService {
-    static lastRequest: unknown;
-
-    async listContextMessages() {
-      return [];
-    }
-
-    async runForeground(request: unknown) {
-      FakeAppService.lastRequest = request;
-      return {
-        executionId: 'exec_runtime',
-        conversationId: 'conv_runtime',
-        messages: [
-          {
-            messageId: 'msg_assistant',
-            role: 'assistant',
-            type: 'assistant-text',
-            content: 'done',
-          },
-        ],
-        finishReason: 'stop' as const,
-        steps: 1,
-        run: {},
-      };
-    }
-  }
-
-  class FakeAppStore {
-    static lastDbPath: string | undefined;
-
-    constructor(dbPath: string) {
-      FakeAppStore.lastDbPath = dbPath;
-    }
-
-    close = vi.fn().mockResolvedValue(undefined);
-  }
-
-  return {
-    ProviderRegistry: {
-      getModelIds: () => ['glm-5', 'claude-3.5-sonnet'],
-      getModelConfig: (modelId: string) => ({
-        name: modelId === 'glm-5' ? 'GLM-5' : 'Claude 3.5 Sonnet',
-        envApiKey: 'TEST_API_KEY',
-        provider: modelId === 'glm-5' ? 'zhipu' : 'anthropic',
-        model: modelId,
-      }),
-      createFromEnv: () => ({}),
-    },
-    loadEnvFiles: vi.fn().mockResolvedValue([]),
-    loadConfigToEnv: vi.fn().mockReturnValue([]),
-    buildSystemPrompt: vi.fn(() => 'Test system prompt'),
-    resolveRenxDatabasePath: vi.fn((env?: NodeJS.ProcessEnv) =>
-      path.join(env?.RENX_HOME ?? renxHome, 'data.db')
-    ),
-    resolveRenxTaskDir: vi.fn((env?: NodeJS.ProcessEnv) =>
-      path.join(env?.RENX_HOME ?? renxHome, 'task')
-    ),
-    createLoggerFromEnv: vi.fn(() => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn() })),
-    createAgentLoggerAdapter: vi.fn((logger: Record<string, unknown>) => ({
-      info: typeof logger.info === 'function' ? logger.info.bind(logger) : undefined,
-      warn: typeof logger.warn === 'function' ? logger.warn.bind(logger) : undefined,
-      error: typeof logger.error === 'function' ? logger.error.bind(logger) : undefined,
-    })),
-    StatelessAgent: FakeAgent,
-    AgentAppService: FakeAppService,
-    createSqliteAgentAppStore: (dbPath: string) => new FakeAppStore(dbPath),
-    DefaultToolManager: FakeToolManager,
-    BashTool: createNamedTool('bash'),
-    WriteFileTool: createNamedTool('write_file'),
-    FileReadTool: createNamedTool('file_read'),
-    FileEditTool: createNamedTool('file_edit'),
-    FileHistoryListTool: createNamedTool('file_history_list'),
-    FileHistoryRestoreTool: createNamedTool('file_history_restore'),
-    GlobTool: createNamedTool('glob'),
-    GrepTool: createNamedTool('grep'),
-    SkillTool: createNamedTool('skill'),
-    TaskTool: createNamedTool('agent'),
-    TaskCreateTool: createNamedTool('task_create'),
-    TaskGetTool: createNamedTool('task_get'),
-    TaskListTool: createNamedTool('task_list'),
-    TaskUpdateTool: createNamedTool('task_update'),
-    TaskStopTool: createNamedTool('task_stop'),
-    TaskOutputTool: createNamedTool('task_output'),
-    TaskStore: class {},
-    RealSubagentRunnerAdapter: class {},
-  };
-};
-
-describe('runtime', () => {
-  const mockGetSourceModules = sourceModules.getSourceModules as unknown as ReturnType<
-    typeof vi.fn
-  >;
-  const mockResolveWorkspaceRoot = sourceModules.resolveWorkspaceRoot as unknown as ReturnType<
-    typeof vi.fn
-  >;
-  const originalApiKey = process.env.TEST_API_KEY;
-  const originalRenxHome = process.env.RENX_HOME;
-  const originalPromptCacheKey = process.env.AGENT_PROMPT_CACHE_KEY;
-  const originalPromptCacheRetention = process.env.AGENT_PROMPT_CACHE_RETENTION;
-  const renxHome = path.join(process.cwd(), '.tmp-renx-home');
-
-  beforeEach(() => {
-    vi.clearAllMocks();
-    process.env.TEST_API_KEY = 'test-key';
-    process.env.RENX_HOME = renxHome;
-    delete process.env.AGENT_PROMPT_CACHE_KEY;
-    delete process.env.AGENT_PROMPT_CACHE_RETENTION;
-    mockResolveWorkspaceRoot.mockReturnValue('/test/workspace');
-    mockGetSourceModules.mockResolvedValue(buildMockModules());
-  });
-
-  afterEach(async () => {
-    await disposeAgentRuntime();
-    if (originalApiKey === undefined) {
-      delete process.env.TEST_API_KEY;
-    } else {
-      process.env.TEST_API_KEY = originalApiKey;
-    }
-    if (originalRenxHome === undefined) {
-      delete process.env.RENX_HOME;
-    } else {
-      process.env.RENX_HOME = originalRenxHome;
-    }
-    if (originalPromptCacheKey === undefined) {
-      delete process.env.AGENT_PROMPT_CACHE_KEY;
-    } else {
-      process.env.AGENT_PROMPT_CACHE_KEY = originalPromptCacheKey;
-    }
-    if (originalPromptCacheRetention === undefined) {
-      delete process.env.AGENT_PROMPT_CACHE_RETENTION;
-    } else {
-      process.env.AGENT_PROMPT_CACHE_RETENTION = originalPromptCacheRetention;
-    }
-  });
-
-  it('returns the active model label', async () => {
-    await expect(getAgentModelLabel()).resolves.toBe('GLM-5');
-  });
-
-  it('returns the active model id', async () => {
-    await expect(getAgentModelId()).resolves.toBe('glm-5');
-  });
-
-  it('lists models with current selection', async () => {
-    await expect(listAgentModels()).resolves.toEqual([
-      {
-        id: 'claude-3.5-sonnet',
-        name: 'Claude 3.5 Sonnet',
-        provider: 'anthropic',
-        apiKeyEnv: 'TEST_API_KEY',
-        configured: true,
-        current: false,
-      },
-      {
-        id: 'glm-5',
-        name: 'GLM-5',
-        provider: 'zhipu',
-        apiKeyEnv: 'TEST_API_KEY',
-        configured: true,
-        current: true,
-      },
-    ]);
-  });
-
-  it('switches model when the target is configured', async () => {
-    await expect(switchAgentModel('claude-3.5-sonnet')).resolves.toEqual({
-      modelId: 'claude-3.5-sonnet',
-      modelLabel: 'Claude 3.5 Sonnet',
-    });
-  });
-
-  it('runs a prompt and returns the assembled result', async () => {
-    const handlers: AgentEventHandlers = {
-      onTextDelta: vi.fn(),
-      onTextComplete: vi.fn(),
-      onToolUse: vi.fn(),
-      onToolStream: vi.fn(),
-      onToolResult: vi.fn(),
-      onToolConfirm: vi.fn(),
-      onUsage: vi.fn(),
-    };
-
-    await expect(runAgentPrompt('Test prompt', handlers)).resolves.toEqual(
-      expect.objectContaining({
-        text: 'done',
-        completionReason: 'stop',
-        modelLabel: 'GLM-5',
-      })
-    );
-  });
-
-  it('stores app state under RENX_HOME by default', async () => {
-    const modules = buildMockModules();
-    const appStoreClass = modules.createSqliteAgentAppStore('/ignore').constructor as {
-      lastDbPath?: string;
-    };
-    mockGetSourceModules.mockResolvedValue(
-      modules as unknown as Awaited<ReturnType<typeof sourceModules.getSourceModules>>
-    );
-
-    await disposeAgentRuntime();
-    await runAgentPrompt('Test prompt', {});
-
-    expect(appStoreClass.lastDbPath).toBe(path.join(renxHome, 'data.db'));
-  });
-
-  it('hides file history tools from the parent agent tool list', async () => {
-    const modules = buildMockModules();
-    const appServiceClass = modules.AgentAppService as {
-      lastRequest?: { tools?: Array<{ function?: { name?: string } }> };
-    };
-    mockGetSourceModules.mockResolvedValue(
-      modules as unknown as Awaited<ReturnType<typeof sourceModules.getSourceModules>>
-    );
-
-    await runAgentPrompt('Test prompt', {});
-
-    const toolNames =
-      appServiceClass.lastRequest?.tools?.map(tool => tool.function?.name).filter(Boolean) || [];
-
-    expect(toolNames).toContain('file_read');
-    expect(toolNames).toContain('file_edit');
-    expect(toolNames).not.toContain('file_history_list');
-    expect(toolNames).not.toContain('file_history_restore');
-  });
-
-  it('forwards prompt cache config from env to the app request', async () => {
-    const modules = buildMockModules();
-    const appServiceClass = modules.AgentAppService as {
-      lastRequest?: { config?: Record<string, unknown> };
-    };
-    mockGetSourceModules.mockResolvedValue(
-      modules as unknown as Awaited<ReturnType<typeof sourceModules.getSourceModules>>
-    );
-    process.env.AGENT_PROMPT_CACHE_KEY = 'cache-{conversationId}';
-    process.env.AGENT_PROMPT_CACHE_RETENTION = '24h';
-
-    await runAgentPrompt('Test prompt', {});
-
-    expect(appServiceClass.lastRequest?.config).toEqual({
-      prompt_cache_key: expect.stringMatching(/^cache-opentui-/),
-      prompt_cache_retention: '24h',
-    });
-  });
-});