feat: 添加完整的单元测试套件

- 新增 vitest 测试框架配置 - 添加 54 个测试文件，共 951 个测试用例 - 覆盖核心模块： - Agent: executor, registry, config-loader, permission-merger - Context: manager, compaction, prune, token-counter - Permission: manager, bash/file/git/web checkers, wildcard - Session: manager, storage - Tools: filesystem (12个), git (10个), web, shell, todo, task - LSP: client, server, language - Utils: config, diff - UI: terminal
2025-12-11 14:45:24 +08:00
parent f4df6483a6
commit 729fb2d42a
58 changed files with 14320 additions and 3 deletions
@@ -0,0 +1,224 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+
+// Mock child_process
+const mockExec = vi.fn();
+vi.mock('child_process', () => ({
+  exec: (cmd: string, opts: any, cb?: Function) => {
+    if (typeof opts === 'function') {
+      cb = opts;
+    }
+    setImmediate(() => {
+      const result = mockExec(cmd);
+      if (result.error) {
+        const err = result.error;
+        err.stdout = result.stdout || '';
+        err.stderr = result.stderr || '';
+        cb?.(err, result.stdout || '', result.stderr || '');
+      } else {
+        cb?.(null, result.stdout || '', result.stderr || '');
+      }
+    });
+  },
+}));
+
+// Mock permission manager
+vi.mock('../../../../src/permission/index.js', () => ({
+  getPermissionManager: vi.fn(() => ({
+    checkGitPermission: vi.fn().mockResolvedValue({
+      allowed: true,
+      action: 'allow',
+    }),
+  })),
+}));
+
+// Mock loadDescription
+vi.mock('../../../../src/tools/load_description.js', () => ({
+  loadDescription: vi.fn(() => '暂存工作区变更'),
+}));
+
+import { gitStashTool } from '../../../../src/tools/git/git_stash.js';
+import { getPermissionManager } from '../../../../src/permission/index.js';
+
+describe('gitStashTool - Git Stash 工具', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    mockExec.mockReturnValue({ stdout: 'Saved working directory', stderr: '' });
+  });
+
+  describe('工具定义', () => {
+    it('有正确的名称', () => {
+      expect(gitStashTool.name).toBe('git_stash');
+    });
+
+    it('有正确的元数据', () => {
+      expect(gitStashTool.metadata.category).toBe('git');
+      expect(gitStashTool.metadata.keywords).toContain('stash');
+      expect(gitStashTool.metadata.keywords).toContain('save');
+    });
+
+    it('所有参数都是可选的', () => {
+      expect(gitStashTool.parameters.action.required).toBe(false);
+      expect(gitStashTool.parameters.message.required).toBe(false);
+      expect(gitStashTool.parameters.index.required).toBe(false);
+      expect(gitStashTool.parameters.include_untracked.required).toBe(false);
+    });
+  });
+
+  describe('execute - 执行', () => {
+    it('默认 push 操作暂存变更', async () => {
+      const result = await gitStashTool.execute({});
+
+      expect(result.success).toBe(true);
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash push'));
+    });
+
+    it('带消息暂存', async () => {
+      const result = await gitStashTool.execute({
+        message: 'WIP: feature',
+      });
+
+      expect(result.success).toBe(true);
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('-m'));
+    });
+
+    it('包含未跟踪文件', async () => {
+      const result = await gitStashTool.execute({
+        include_untracked: true,
+      });
+
+      expect(result.success).toBe(true);
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('-u'));
+    });
+
+    it('列出暂存', async () => {
+      mockExec.mockReturnValue({
+        stdout: 'stash@{0}: WIP on main\nstash@{1}: feature',
+        stderr: '',
+      });
+
+      const result = await gitStashTool.execute({ action: 'list' });
+
+      expect(result.success).toBe(true);
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash list'));
+    });
+
+    it('恢复暂存 (pop)', async () => {
+      mockExec.mockReturnValue({ stdout: '', stderr: '' });
+
+      const result = await gitStashTool.execute({ action: 'pop' });
+
+      expect(result.success).toBe(true);
+      expect(result.output).toContain('暂存已恢复并删除');
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash pop'));
+    });
+
+    it('应用暂存 (apply)', async () => {
+      mockExec.mockReturnValue({ stdout: '', stderr: '' });
+
+      const result = await gitStashTool.execute({ action: 'apply' });
+
+      expect(result.success).toBe(true);
+      expect(result.output).toContain('暂存已恢复');
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash apply'));
+    });
+
+    it('删除暂存 (drop)', async () => {
+      mockExec.mockReturnValue({ stdout: '', stderr: '' });
+
+      const result = await gitStashTool.execute({ action: 'drop', index: 1 });
+
+      expect(result.success).toBe(true);
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash drop stash@{1}'));
+    });
+
+    it('清除所有暂存', async () => {
+      mockExec.mockReturnValue({ stdout: '', stderr: '' });
+
+      const result = await gitStashTool.execute({ action: 'clear' });
+
+      expect(result.success).toBe(true);
+      expect(result.output).toContain('所有暂存已清除');
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash clear'));
+    });
+
+    it('显示暂存内容', async () => {
+      mockExec.mockReturnValue({ stdout: 'diff --git a/file.ts', stderr: '' });
+
+      const result = await gitStashTool.execute({ action: 'show' });
+
+      expect(result.success).toBe(true);
+      expect(mockExec).toHaveBeenCalledWith(expect.stringContaining('git stash show -p'));
+    });
+
+    it('未知操作返回错误', async () => {
+      const result = await gitStashTool.execute({ action: 'unknown' });
+
+      expect(result.success).toBe(false);
+      expect(result.error).toContain('未知操作');
+    });
+
+    it('没有变更时返回友好错误', async () => {
+      // 确保权限通过
+      vi.mocked(getPermissionManager).mockReturnValue({
+        checkGitPermission: vi.fn().mockResolvedValue({ allowed: true }),
+      } as any);
+      mockExec.mockReturnValue({
+        error: new Error('Command failed'),
+        stdout: '',
+        stderr: 'No local changes to save',
+      });
+
+      const result = await gitStashTool.execute({});
+
+      expect(result.success).toBe(false);
+      expect(result.error).toContain('没有需要暂存的变更');
+    });
+
+    it('恢复冲突时返回友好错误', async () => {
+      // 确保权限通过
+      vi.mocked(getPermissionManager).mockReturnValue({
+        checkGitPermission: vi.fn().mockResolvedValue({ allowed: true }),
+      } as any);
+      mockExec.mockReturnValue({
+        error: new Error('Command failed'),
+        stdout: '',
+        stderr: 'CONFLICT (content): Merge conflict',
+      });
+
+      const result = await gitStashTool.execute({ action: 'pop' });
+
+      expect(result.success).toBe(false);
+      expect(result.error).toContain('冲突');
+    });
+
+    it('权限被拒绝时返回错误', async () => {
+      vi.mocked(getPermissionManager).mockReturnValue({
+        checkGitPermission: vi.fn().mockResolvedValue({
+          allowed: false,
+          action: 'deny',
+          reason: '不允许暂存操作',
+        }),
+      } as any);
+
+      const result = await gitStashTool.execute({});
+
+      expect(result.success).toBe(false);
+      expect(result.error).toContain('权限被拒绝');
+    });
+
+    it('需要确认时返回提示', async () => {
+      vi.mocked(getPermissionManager).mockReturnValue({
+        checkGitPermission: vi.fn().mockResolvedValue({
+          allowed: false,
+          action: 'ask',
+          needsConfirmation: true,
+        }),
+      } as any);
+
+      const result = await gitStashTool.execute({});
+
+      expect(result.success).toBe(false);
+      expect(result.error).toContain('需要用户确认');
+    });
+  });
+});