ai-terminal-assistant/tests/unit/tools/todo/todoread.test.ts

import { describe, it, expect, beforeEach, vi } from 'vitest';

// 使用可变的引用对象来绕过 hoisting 问题
const mockState = {
  isInitialized: vi.fn().mockReturnValue(true),
  getTodos: vi.fn().mockReturnValue([]),
};

vi.mock('../../../../src/tools/todo/todo-manager.js', () => ({
  todoManager: {
    isInitialized: () => mockState.isInitialized(),
    getTodos: () => mockState.getTodos(),
  },
}));

import { todoReadTool } from '../../../../src/tools/todo/todoread.js';

describe('todoReadTool - Todo 读取工具', () => {
  beforeEach(() => {
    vi.clearAllMocks();
    mockState.isInitialized.mockReturnValue(true);
    mockState.getTodos.mockReturnValue([]);
  });

  describe('工具定义', () => {
    it('有正确的名称', () => {
      expect(todoReadTool.name).toBe('todoread');
    });

    it('有正确的元数据', () => {
      expect(todoReadTool.metadata.category).toBe('core');
      expect(todoReadTool.metadata.keywords).toContain('todo');
      expect(todoReadTool.metadata.keywords).toContain('task');
      expect(todoReadTool.metadata.keywords).toContain('list');
    });

    it('无必需参数', () => {
      expect(Object.keys(todoReadTool.parameters)).toHaveLength(0);
    });
  });

  describe('execute - 执行', () => {
    it('成功读取空列表', async () => {
      const result = await todoReadTool.execute({});

      expect(result.success).toBe(true);
      expect(result.output).toBe('[]');
      expect(result.metadata?.totalCount).toBe(0);
      expect(result.metadata?.pendingCount).toBe(0);
    });

    it('成功读取待办列表', async () => {
      const todos = [
        { id: '1', content: '任务1', status: 'pending', createdAt: '2024-01-01', updatedAt: '2024-01-01' },
        { id: '2', content: '任务2', status: 'in_progress', createdAt: '2024-01-01', updatedAt: '2024-01-01' },
        { id: '3', content: '任务3', status: 'completed', createdAt: '2024-01-01', updatedAt: '2024-01-01' },
      ];
      mockState.getTodos.mockReturnValue(todos);

      const result = await todoReadTool.execute({});

      expect(result.success).toBe(true);
      expect(result.metadata?.totalCount).toBe(3);
      expect(result.metadata?.pendingCount).toBe(2); // pending + in_progress
    });

    it('返回 JSON 格式输出', async () => {
      const todos = [
        { id: '1', content: '任务1', status: 'pending', createdAt: '2024-01-01', updatedAt: '2024-01-01' },
      ];
      mockState.getTodos.mockReturnValue(todos);

      const result = await todoReadTool.execute({});

      const parsed = JSON.parse(result.output);
      expect(parsed).toHaveLength(1);
      expect(parsed[0].content).toBe('任务1');
    });

    it('未初始化时返回错误', async () => {
      mockState.isInitialized.mockReturnValue(false);

      const result = await todoReadTool.execute({});

      expect(result.success).toBe(false);
      expect(result.error).toContain('会话管理器未初始化');
    });

    it('返回正确的元数据', async () => {
      const todos = [
        { id: '1', content: '任务1', status: 'pending' },
        { id: '2', content: '任务2', status: 'completed' },
      ];
      mockState.getTodos.mockReturnValue(todos);

      const result = await todoReadTool.execute({});

      expect(result.metadata?.todos).toEqual(todos);
      expect(result.metadata?.pendingCount).toBe(1);
      expect(result.metadata?.totalCount).toBe(2);
    });
  });
});