@ai-setting/roy-agent-core 1.3.10 → 1.3.11

package/src/env/task/delegate/delegate-tool.test.ts

@@ -1,498 +0,0 @@
-/**
- * @fileoverview delegate_task Tool Tests
- * 
- * TDD: 红色阶段 - 编写失败的测试
- */
-
-import { describe, test, expect, beforeEach, afterEach, vi } from "vitest";
-import type { TaskComponent } from "../task-component";
-import type { Tool, ToolContext, ToolResult } from "../../tool/types";
-
-// Mock the tool factory
-// Actual implementation will be imported after RED phase
-const createDelegateTool = async (taskComponent: TaskComponent): Promise<Tool> => {
-  const module = await import("./delegate-tool");
-  return module.createDelegateTool(taskComponent);
-};
-
-describe("delegate_task Tool", () => {
-  let mockTaskComponent: any;
-  let tool: Tool;
-  let mockSessionComponent: any;
-  let mockAgentComponent: any;
-
-  beforeEach(async () => {
-    // Mock SessionComponent with proper structure
-    mockSessionComponent = {
-      get: vi.fn(),
-      create: vi.fn(),
-      addMessage: vi.fn(),
-    };
-
-    // Mock AgentComponent with getAgent method (required by ensureSubAgentRegistered)
-    mockAgentComponent = {
-      getAgent: vi.fn((name: string) => {
-        // Return undefined to trigger registration path
-        return undefined;
-      }),
-      registerAgent: vi.fn((name: string, config: any) => ({
-        name,
-        config,
-        status: "idle",
-        plugins: new Map(),
-      })),
-    };
-
-    // Mock TaskComponent
-    mockTaskComponent = {
-      createTask: vi.fn(),
-      updateTask: vi.fn(),
-      createOperation: vi.fn(),
-      getTask: vi.fn(),
-      listTasks: vi.fn(),
-      getTagService: vi.fn().mockReturnValue({
-        listTags: vi.fn().mockResolvedValue([]),
-        findSimilarTasks: vi.fn().mockResolvedValue([]),
-      }),
-      env: {
-        getComponent: vi.fn((name: string) => {
-          if (name === "session") return mockSessionComponent;
-          if (name === "agent") return mockAgentComponent;
-          return { registerTool: vi.fn() };
-        }),
-        handle_query: vi.fn(),
-      },
-    };
-    
-    // Store references for test setup
-    (mockTaskComponent as any).mockSessionComponent = mockSessionComponent;
-    (mockTaskComponent as any).mockAgentComponent = mockAgentComponent;
-    
-    tool = await createDelegateTool(mockTaskComponent);
-  });
-
-  afterEach(() => {
-    vi.clearAllMocks();
-  });
-
-  describe("Tool Definition", () => {
-    test("should have correct tool name", () => {
-      expect(tool.name).toBe("delegate_task");
-    });
-
-    test("should have description with subagent types", () => {
-      expect(tool.description).toContain("delegate");
-      expect(tool.description).toContain("subagent");
-    });
-
-    test("should have parameters schema", () => {
-      expect(tool.parameters).toBeDefined();
-      expect(tool.parameters.shape).toBeDefined();
-    });
-
-    test("should have description field in parameters", () => {
-      const shape = tool.parameters.shape;
-      expect(shape.description).toBeDefined();
-    });
-
-    test("should have prompt field in parameters", () => {
-      const shape = tool.parameters.shape;
-      expect(shape.prompt).toBeDefined();
-    });
-
-    test("should have subagent_type field in parameters", () => {
-      const shape = tool.parameters.shape;
-      expect(shape.subagent_type).toBeDefined();
-    });
-
-    test("should have background field in parameters", () => {
-      const shape = tool.parameters.shape;
-      expect(shape.background).toBeDefined();
-    });
-
-    test("should have task_id field in parameters", () => {
-      const shape = tool.parameters.shape;
-      expect(shape.task_id).toBeDefined();
-    });
-  });
-
-  describe("Synchronous Execution", () => {
-    test("should execute synchronously when background=false", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      // Mock parent session for get
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      // Mock sub-agent session
-      const mockSubSession = {
-        id: "sub-session-456",
-        info: { metadata: { subagent_type: "general" } },
-        addMessage: vi.fn(),
-      };
-      
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      
-      // Mock session component methods
-      mockSessionComponent.get
-        .mockReturnValueOnce(mockParentSession);  // Parent session check
-      
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-      mockTaskComponent.env.handle_query.mockResolvedValue("Task completed successfully");
-
-      const result = await tool.execute(
-        {
-          description: "Test task",
-          prompt: "Do something",
-          subagent_type: "general",
-          background: false,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(true);
-      expect(result.metadata?.sessionId).toBe(mockSubSession.id);
-    });
-
-    test("should return error for unknown subagent_type", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const result = await tool.execute(
-        {
-          description: "Test task",
-          prompt: "Do something",
-          subagent_type: "unknown_type",
-          background: false,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(false);
-      expect(result.error).toContain("Unknown subagent type");
-    });
-
-    test("should use default subagent_type as general", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      const mockSubSession = {
-        id: "sub-session-456",
-        info: { metadata: { subagent_type: "general" } },
-        addMessage: vi.fn(),
-      };
-      
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockReturnValueOnce(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-      mockTaskComponent.env.handle_query.mockResolvedValue("Task completed");
-
-      const result = await tool.execute(
-        {
-          description: "Test task",
-          prompt: "Do something",
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(true);
-    });
-  });
-
-  describe("Background Execution", () => {
-    test("should execute in background when background=true", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      const mockSubSession = {
-        id: "sub-session-456",
-        addMessage: vi.fn(),
-      };
-
-      // Mock for createTask
-      const createdTask = {
-        id: 999,
-        title: "Background task",
-        status: "active",
-      };
-      mockTaskComponent.createTask.mockResolvedValue(createdTask);
-      mockTaskComponent.createOperation.mockResolvedValue({});
-
-      // Mock session component
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockResolvedValue(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-
-      const result = await tool.execute(
-        {
-          description: "Background task",
-          prompt: "Do something in background",
-          subagent_type: "general",
-          background: true,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(true);
-      expect(result.metadata?.background).toBe(true);
-      expect(result.metadata?.status).toBe("accepted");
-    });
-
-    test("should include task_id in metadata for background tasks", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      const mockSubSession = {
-        id: "sub-session-456",
-        addMessage: vi.fn(),
-      };
-
-      // Mock for createTask
-      const createdTask = {
-        id: 123,
-        title: "Background task",
-        status: "active",
-      };
-      mockTaskComponent.createTask.mockResolvedValue(createdTask);
-      mockTaskComponent.createOperation.mockResolvedValue({});
-
-      // Mock session component
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockResolvedValue(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-
-      const result = await tool.execute(
-        {
-          description: "Background task",
-          prompt: "Do something",
-          subagent_type: "general",
-          background: true,
-          task_id: 123,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(true);
-      expect(result.metadata?.task_id).toBe(123);
-    });
-  });
-
-  describe("Task Association", () => {
-    test("should create task if task_id not provided in background mode", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      const mockSubSession = {
-        id: "sub-session-456",
-        addMessage: vi.fn(),
-      };
-
-      // Mock for createTask
-      const createdTask = {
-        id: 999,
-        title: "Background task",
-        status: "active",
-      };
-      mockTaskComponent.createTask.mockResolvedValue(createdTask);
-      mockTaskComponent.createOperation.mockResolvedValue({});
-
-      // Mock session component
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockResolvedValue(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-
-      const result = await tool.execute(
-        {
-          description: "Background task",
-          prompt: "Do something",
-          background: true,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(true);
-      expect(mockTaskComponent.createTask).toHaveBeenCalled();
-    });
-
-    test("should update existing task when task_id provided", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      const mockSubSession = {
-        id: "sub-session-456",
-        addMessage: vi.fn(),
-      };
-
-      const existingTask = {
-        id: 123,
-        title: "Existing task",
-        status: "active",
-      };
-
-      mockTaskComponent.getTask.mockResolvedValue(existingTask);
-      mockTaskComponent.updateTask.mockResolvedValue({
-        ...existingTask,
-        status: "active",
-      });
-      mockTaskComponent.createOperation.mockResolvedValue({});
-
-      // Mock session component
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockResolvedValue(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-
-      const result = await tool.execute(
-        {
-          description: "Existing task",
-          prompt: "Do something",
-          background: true,
-          task_id: 123,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(true);
-      expect(mockTaskComponent.updateTask).toHaveBeenCalled();
-    });
-  });
-
-  describe("Error Handling", () => {
-    test("should handle parent session not found", async () => {
-      const mockContext: ToolContext = {
-        session_id: "non-existent-session",
-      };
-
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockResolvedValue(null);
-
-      const result = await tool.execute(
-        {
-          description: "Test task",
-          prompt: "Do something",
-          background: false,
-        },
-        mockContext
-      );
-
-      expect(result.success).toBe(false);
-      expect(result.error).toContain("Parent session not found");
-    });
-
-    test("should handle execution error", async () => {
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = {
-        id: "test-session-123",
-      };
-
-      const mockSubSession = {
-        id: "sub-session-456",
-        info: { metadata: { subagent_type: "general" } },
-        addMessage: vi.fn(),
-      };
-
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      mockSessionComponent.get.mockResolvedValue(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-      
-      // Mock handle_query to reject with "Execution failed" error
-      mockTaskComponent.env.handle_query.mockRejectedValue(new Error("Execution failed"));
-
-      const result = await tool.execute(
-        {
-          description: "Test task",
-          prompt: "Do something",
-          background: false,
-        },
-        mockContext
-      );
-
-      // The tool should return success: true since it catches the error internally
-      // and returns the error as the result output
-      expect(result.success).toBe(true);
-      expect(result.output).toContain("Execution failed");
-    });
-  });
-
-  describe("BackgroundTaskManager SessionComponent Access", () => {
-    test("should correctly pass env to BackgroundTaskManager", async () => {
-      // This test verifies that the env object is correctly passed to BackgroundTaskManager
-      // so that it can access SessionComponent via env.getComponent("session")
-      
-      const mockContext: ToolContext = {
-        session_id: "test-session-123",
-      };
-
-      const mockParentSession = { id: "test-session-123" };
-      const mockSubSession = {
-        id: "sub-session-456",
-        info: { metadata: { subagent_type: "general" } },
-        addMessage: vi.fn(),
-      };
-
-      const mockSessionComponent = (mockTaskComponent as any).mockSessionComponent;
-      
-      // Mock session component methods
-      mockSessionComponent.get.mockResolvedValue(mockParentSession);
-      mockSessionComponent.create.mockResolvedValue(mockSubSession);
-      mockTaskComponent.env.handle_query.mockResolvedValue("Background task result");
-
-      // Create task
-      const createdTask = {
-        id: 999,
-        title: "Background task",
-        status: "active",
-      };
-      mockTaskComponent.createTask.mockResolvedValue(createdTask);
-      mockTaskComponent.createOperation.mockResolvedValue({});
-
-      const result = await tool.execute(
-        {
-          description: "Background task",
-          prompt: "Do something in background",
-          subagent_type: "general",
-          background: true,
-        },
-        mockContext
-      );
-
-      // Verify that getComponent was called with "session"
-      expect(mockTaskComponent.env.getComponent).toHaveBeenCalledWith("session");
-      
-      // Verify the task was created successfully
-      expect(result.success).toBe(true);
-      expect(result.metadata?.background).toBe(true);
-    });
-  });
-});