whale-code 6.5.10 → 6.6.0

package/dist/cli/services/__tests__/team-lead-comms-messaging.test.js

@@ -1,250 +0,0 @@
-/**
- * Tests for team-lead.ts — sendToTeammate, broadcast, getMessages, getProgress
- */
-
-import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
-// ============================================================================
-// MOCKS
-// ============================================================================
-
-const mockCreateTeam = vi.fn();
-const mockLoadTeam = vi.fn();
-const mockSaveTeam = vi.fn();
-const mockAddTeammate = vi.fn();
-const mockAddTask = vi.fn();
-const mockGetTeamProgress = vi.fn();
-const mockSetTeamStatus = vi.fn();
-const mockSendMessage = vi.fn();
-const mockGetUnreadMessages = vi.fn();
-const mockMarkMessagesRead = vi.fn();
-const mockFailTask = vi.fn();
-const mockUpdateTeammate = vi.fn();
-vi.mock("../team-state.js", () => ({
-  createTeam: (...args) => mockCreateTeam(...args),
-  loadTeam: (...args) => mockLoadTeam(...args),
-  saveTeam: (...args) => mockSaveTeam(...args),
-  addTeammate: (...args) => mockAddTeammate(...args),
-  addTask: (...args) => mockAddTask(...args),
-  getTeamProgress: (...args) => mockGetTeamProgress(...args),
-  setTeamStatus: (...args) => mockSetTeamStatus(...args),
-  sendMessage: (...args) => mockSendMessage(...args),
-  getUnreadMessages: (...args) => mockGetUnreadMessages(...args),
-  markMessagesRead: (...args) => mockMarkMessagesRead(...args),
-  failTask: (...args) => mockFailTask(...args),
-  updateTeammate: (...args) => mockUpdateTeammate(...args)
-}));
-const mockSpawnTeammate = vi.fn();
-vi.mock("../teammate.js", () => ({
-  spawnTeammate: (...args) => mockSpawnTeammate(...args)
-}));
-vi.mock("../telemetry.js", () => ({
-  logSpan: vi.fn(),
-  generateTraceId: vi.fn(() => "trace-123"),
-  generateSpanId: vi.fn(() => "span-123"),
-  getConversationId: vi.fn(() => "conv-123"),
-  PKG_VERSION: "0.0.0-test"
-}));
-vi.mock("../config-store.js", () => ({
-  resolveConfig: vi.fn(() => ({
-    storeId: "store-1"
-  }))
-}));
-vi.mock("../agent-events.js", () => {
-  const emitter = {
-    emitTeamStart: vi.fn(),
-    emitTeamProgress: vi.fn(),
-    emitTeamTask: vi.fn(),
-    emitTeamDone: vi.fn()
-  };
-  return {
-    getGlobalEmitter: vi.fn(() => emitter)
-  };
-});
-vi.mock("../agent-loop.js", () => ({
-  getModelShortName: vi.fn(() => "sonnet")
-}));
-vi.mock("../agent-config.js", () => ({
-  loadCLIAgentConfig: vi.fn(() => Promise.resolve(null))
-}));
-vi.mock("../server-tools.js", () => ({
-  loadServerToolDefinitions: vi.fn(() => Promise.resolve([]))
-}));
-import { TeamLead } from "../team-lead.js";
-// ============================================================================
-// HELPERS
-// ============================================================================
-
-function makeTeamState(overrides = {}) {
-  return {
-    id: "team-123",
-    name: "Test Team",
-    leadId: "lead",
-    teammates: [],
-    tasks: [],
-    messages: [],
-    createdAt: "2026-01-01T00:00:00.000Z",
-    updatedAt: "2026-01-01T00:00:00.000Z",
-    status: "active",
-    ...overrides
-  };
-}
-
-// ============================================================================
-// TESTS
-// ============================================================================
-
-beforeEach(() => {
-  vi.clearAllMocks();
-  vi.useFakeTimers({
-    shouldAdvanceTime: false
-  });
-});
-afterEach(() => {
-  vi.useRealTimers();
-});
-describe("sendToTeammate", () => {
-  it("should return false if no team created", async () => {
-    const lead = new TeamLead();
-    const result = await lead.sendToTeammate("tm-1", "Hello");
-    expect(result).toBe(false);
-  });
-  it("should send message via team-state sendMessage", async () => {
-    const team = makeTeamState();
-    mockCreateTeam.mockReturnValue(team);
-    mockLoadTeam.mockReturnValue(team);
-    mockSaveTeam.mockImplementation(() => {});
-    mockAddTask.mockResolvedValue({
-      id: "t-1",
-      description: "Task"
-    });
-    mockAddTeammate.mockResolvedValue({
-      id: "tm-1"
-    });
-    mockSendMessage.mockResolvedValue({
-      id: "msg-1"
-    });
-    const lead = new TeamLead();
-    await lead.createTeam({
-      name: "Test",
-      teammateCount: 1,
-      tasks: [{
-        description: "Task"
-      }]
-    });
-    const result = await lead.sendToTeammate("tm-1", "Hello");
-    expect(result).toBe(true);
-    expect(mockSendMessage).toHaveBeenCalledWith("team-123", "lead", "tm-1", "Hello");
-  });
-});
-describe("broadcast", () => {
-  it("should return false if no team", async () => {
-    const lead = new TeamLead();
-    const result = await lead.broadcast("Hello");
-    expect(result).toBe(false);
-  });
-  it("should broadcast to all", async () => {
-    const team = makeTeamState();
-    mockCreateTeam.mockReturnValue(team);
-    mockLoadTeam.mockReturnValue(team);
-    mockSaveTeam.mockImplementation(() => {});
-    mockAddTask.mockResolvedValue({
-      id: "t-1",
-      description: "Task"
-    });
-    mockAddTeammate.mockResolvedValue({
-      id: "tm-1"
-    });
-    mockSendMessage.mockResolvedValue({
-      id: "msg-1"
-    });
-    const lead = new TeamLead();
-    await lead.createTeam({
-      name: "Test",
-      teammateCount: 1,
-      tasks: [{
-        description: "Task"
-      }]
-    });
-    const result = await lead.broadcast("Hello all");
-    expect(mockSendMessage).toHaveBeenCalledWith("team-123", "lead", "all", "Hello all");
-  });
-});
-describe("getMessages", () => {
-  it("should return empty array if no team", async () => {
-    const lead = new TeamLead();
-    const messages = await lead.getMessages();
-    expect(messages).toEqual([]);
-  });
-  it("should get and mark messages as read", async () => {
-    const team = makeTeamState();
-    mockCreateTeam.mockReturnValue(team);
-    mockLoadTeam.mockReturnValue(team);
-    mockSaveTeam.mockImplementation(() => {});
-    mockAddTask.mockResolvedValue({
-      id: "t-1",
-      description: "Task"
-    });
-    mockAddTeammate.mockResolvedValue({
-      id: "tm-1"
-    });
-    mockGetUnreadMessages.mockResolvedValue([{
-      id: "msg-1",
-      from: "tm-1",
-      content: "Hi",
-      to: "lead",
-      timestamp: "",
-      read: false
-    }]);
-    mockMarkMessagesRead.mockResolvedValue(true);
-    const lead = new TeamLead();
-    await lead.createTeam({
-      name: "Test",
-      teammateCount: 1,
-      tasks: [{
-        description: "Task"
-      }]
-    });
-    const messages = await lead.getMessages();
-    expect(messages.length).toBe(1);
-    expect(messages[0].content).toBe("Hi");
-    expect(mockMarkMessagesRead).toHaveBeenCalledWith("team-123", ["msg-1"]);
-  });
-});
-describe("getProgress", () => {
-  it("should return null if no team", () => {
-    const lead = new TeamLead();
-    expect(lead.getProgress()).toBeNull();
-  });
-  it("should return progress from team state", async () => {
-    const team = makeTeamState();
-    mockCreateTeam.mockReturnValue(team);
-    mockLoadTeam.mockReturnValue(team);
-    mockSaveTeam.mockImplementation(() => {});
-    mockAddTask.mockResolvedValue({
-      id: "t-1",
-      description: "Task"
-    });
-    mockAddTeammate.mockResolvedValue({
-      id: "tm-1"
-    });
-    mockGetTeamProgress.mockReturnValue({
-      total: 3,
-      completed: 1,
-      percentComplete: 33
-    });
-    const lead = new TeamLead();
-    await lead.createTeam({
-      name: "Test",
-      teammateCount: 1,
-      tasks: [{
-        description: "Task"
-      }]
-    });
-    const progress = lead.getProgress();
-    expect(progress).toEqual({
-      total: 3,
-      completed: 1,
-      percentComplete: 33
-    });
-  });
-});

package/dist/cli/services/__tests__/team-lead-comms-result.test.js

@@ -1,232 +0,0 @@
-/**
- * Tests for team-lead.ts — buildResult via runTeam completion
- */
-
-import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
-import { EventEmitter } from "events";
-
-// ============================================================================
-// MOCKS
-// ============================================================================
-
-const mockCreateTeam = vi.fn();
-const mockLoadTeam = vi.fn();
-const mockSaveTeam = vi.fn();
-const mockAddTeammate = vi.fn();
-const mockAddTask = vi.fn();
-const mockGetTeamProgress = vi.fn();
-const mockSetTeamStatus = vi.fn();
-const mockSendMessage = vi.fn();
-const mockGetUnreadMessages = vi.fn();
-const mockMarkMessagesRead = vi.fn();
-const mockFailTask = vi.fn();
-const mockUpdateTeammate = vi.fn();
-vi.mock("../team-state.js", () => ({
-  createTeam: (...args) => mockCreateTeam(...args),
-  loadTeam: (...args) => mockLoadTeam(...args),
-  saveTeam: (...args) => mockSaveTeam(...args),
-  addTeammate: (...args) => mockAddTeammate(...args),
-  addTask: (...args) => mockAddTask(...args),
-  getTeamProgress: (...args) => mockGetTeamProgress(...args),
-  setTeamStatus: (...args) => mockSetTeamStatus(...args),
-  sendMessage: (...args) => mockSendMessage(...args),
-  getUnreadMessages: (...args) => mockGetUnreadMessages(...args),
-  markMessagesRead: (...args) => mockMarkMessagesRead(...args),
-  failTask: (...args) => mockFailTask(...args),
-  updateTeammate: (...args) => mockUpdateTeammate(...args)
-}));
-const mockSpawnTeammate = vi.fn();
-vi.mock("../teammate.js", () => ({
-  spawnTeammate: (...args) => mockSpawnTeammate(...args)
-}));
-vi.mock("../telemetry.js", () => ({
-  logSpan: vi.fn(),
-  generateTraceId: vi.fn(() => "trace-123"),
-  generateSpanId: vi.fn(() => "span-123"),
-  getConversationId: vi.fn(() => "conv-123"),
-  PKG_VERSION: "0.0.0-test"
-}));
-vi.mock("../config-store.js", () => ({
-  resolveConfig: vi.fn(() => ({
-    storeId: "store-1"
-  }))
-}));
-vi.mock("../agent-events.js", () => {
-  const emitter = {
-    emitTeamStart: vi.fn(),
-    emitTeamProgress: vi.fn(),
-    emitTeamTask: vi.fn(),
-    emitTeamDone: vi.fn()
-  };
-  return {
-    getGlobalEmitter: vi.fn(() => emitter)
-  };
-});
-vi.mock("../agent-loop.js", () => ({
-  getModelShortName: vi.fn(() => "sonnet")
-}));
-vi.mock("../agent-config.js", () => ({
-  loadCLIAgentConfig: vi.fn(() => Promise.resolve(null))
-}));
-vi.mock("../server-tools.js", () => ({
-  loadServerToolDefinitions: vi.fn(() => Promise.resolve([]))
-}));
-import { TeamLead } from "../team-lead.js";
-// ============================================================================
-// HELPERS
-// ============================================================================
-
-function makeTeamState(overrides = {}) {
-  return {
-    id: "team-123",
-    name: "Test Team",
-    leadId: "lead",
-    teammates: [],
-    tasks: [],
-    messages: [],
-    createdAt: "2026-01-01T00:00:00.000Z",
-    updatedAt: "2026-01-01T00:00:00.000Z",
-    status: "active",
-    ...overrides
-  };
-}
-function makeChildProcessMock() {
-  const child = new EventEmitter();
-  child.kill = vi.fn();
-  child.send = vi.fn();
-  child.pid = Math.floor(Math.random() * 10000);
-  return child;
-}
-beforeEach(() => {
-  vi.clearAllMocks();
-  vi.useFakeTimers({
-    shouldAdvanceTime: false
-  });
-});
-afterEach(() => {
-  vi.useRealTimers();
-});
-
-// ============================================================================
-// TESTS — buildResult
-// ============================================================================
-
-describe("buildResult (via runTeam completion)", () => {
-  it("should build result with success when all tasks completed", async () => {
-    const worker = makeChildProcessMock();
-    const team = makeTeamState({
-      teammates: [{
-        id: "tm-1",
-        name: "A",
-        model: "sonnet",
-        status: "done",
-        tokensUsed: {
-          input: 100,
-          output: 50
-        },
-        startedAt: ""
-      }],
-      tasks: [{
-        id: "t-1",
-        description: "Task 1",
-        status: "completed",
-        result: "Done",
-        createdAt: "",
-        updatedAt: ""
-      }]
-    });
-    mockCreateTeam.mockReturnValue(team);
-    mockLoadTeam.mockReturnValue(team);
-    mockSaveTeam.mockImplementation(() => {});
-    mockAddTask.mockResolvedValue({
-      id: "t-1",
-      description: "Task 1"
-    });
-    mockAddTeammate.mockResolvedValue({
-      id: "tm-1"
-    });
-    mockSpawnTeammate.mockResolvedValue(worker);
-    mockGetTeamProgress.mockReturnValue({
-      total: 1,
-      completed: 1,
-      blocked: 0,
-      percentComplete: 100,
-      inProgress: 0,
-      pending: 0
-    });
-    mockSetTeamStatus.mockResolvedValue(true);
-    const lead = new TeamLead();
-    await lead.createTeam({
-      name: "Test",
-      teammateCount: 1,
-      tasks: [{
-        description: "Task 1"
-      }]
-    });
-    const resultPromise = lead.runTeam();
-    await vi.advanceTimersByTimeAsync(10);
-    worker.emit("exit", 0);
-    const result = await resultPromise;
-    expect(result.success).toBe(true);
-    expect(result.teamId).toBe("team-123");
-    expect(result.tokensUsed.input).toBe(100);
-    expect(result.tokensUsed.output).toBe(50);
-  });
-  it("should build result with failure when tasks are blocked", async () => {
-    const worker = makeChildProcessMock();
-    const team = makeTeamState({
-      teammates: [{
-        id: "tm-1",
-        name: "A",
-        model: "sonnet",
-        status: "done",
-        tokensUsed: {
-          input: 0,
-          output: 0
-        },
-        startedAt: ""
-      }],
-      tasks: [{
-        id: "t-1",
-        description: "Task 1",
-        status: "blocked",
-        error: "Failed",
-        createdAt: "",
-        updatedAt: ""
-      }]
-    });
-    mockCreateTeam.mockReturnValue(team);
-    mockLoadTeam.mockReturnValue(team);
-    mockSaveTeam.mockImplementation(() => {});
-    mockAddTask.mockResolvedValue({
-      id: "t-1",
-      description: "Task 1"
-    });
-    mockAddTeammate.mockResolvedValue({
-      id: "tm-1"
-    });
-    mockSpawnTeammate.mockResolvedValue(worker);
-    mockGetTeamProgress.mockReturnValue({
-      total: 1,
-      completed: 0,
-      blocked: 1,
-      percentComplete: 0,
-      inProgress: 0,
-      pending: 0
-    });
-    mockSetTeamStatus.mockResolvedValue(true);
-    const lead = new TeamLead();
-    await lead.createTeam({
-      name: "Test",
-      teammateCount: 1,
-      tasks: [{
-        description: "Task 1"
-      }]
-    });
-    const resultPromise = lead.runTeam();
-    await vi.advanceTimersByTimeAsync(10);
-    worker.emit("exit", 0);
-    const result = await resultPromise;
-    expect(result.success).toBe(false);
-  });
-});