whale-code 6.5.10 → 6.6.0

package/dist/server/lib/__tests__/server-agent-loop-abort.test.js

@@ -1,396 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
-
-// ============================================================================
-// Mocks — must be declared before imports
-// ============================================================================
-
-vi.mock("../../../shared/agent-core.js", () => {
-  class MockLoopDetector {
-    resetTurn = vi.fn();
-    getSessionStats = vi.fn(() => ({
-      totalErrors: 0,
-      failedStrategies: 0,
-      consecutiveFailedTurns: 0
-    }));
-  }
-  return {
-    LoopDetector: MockLoopDetector,
-    getContextManagement: vi.fn(() => ({
-      betas: ["context-management-2025-06-27"],
-      config: {
-        edits: []
-      }
-    })),
-    getMaxOutputTokens: vi.fn(() => 16384),
-    getThinkingConfig: vi.fn(() => ({
-      thinking: {
-        type: "disabled"
-      },
-      beta: null
-    })),
-    addPromptCaching: vi.fn((tools, msgs) => ({
-      tools,
-      messages: msgs
-    })),
-    estimateCostUsd: vi.fn(() => 0.001),
-    isRetryableError: vi.fn(() => false),
-    sanitizeError: err => err instanceof Error ? err.message : String(err),
-    routeModel: vi.fn((text, model) => model),
-    resolveToolChoice: vi.fn(() => "auto"),
-    COMPACTION_TRIGGER_TOKENS: 167_000,
-    COMPACTION_TOTAL_BUDGET: 2_000_000,
-    DEFAULT_SESSION_COST_BUDGET_USD: 25,
-    emitCostWarningIfNeeded: vi.fn(),
-    demoteSubagentModel: vi.fn((requested, agentType) => {
-      if (agentType === "explore" || agentType === "research") return "haiku";
-      if (agentType === "general-purpose" || agentType === "plan") {
-        if (requested === "opus") return "opus";
-        if (requested === "sonnet") return "sonnet";
-        return "sonnet";
-      }
-      if (!requested) return "haiku";
-      return requested === "opus" ? "sonnet" : requested === "sonnet" ? "sonnet" : "haiku";
-    }),
-    AGENT_DEFAULTS: {
-      maxTurns: 200,
-      temperature: 1.0,
-      maxConcurrentTools: 7,
-      enableDelegation: true,
-      maxDurationMs: 900_000,
-      maxDurationMsWorkflow: 120_000,
-      enableModelRouting: true,
-      subagentMaxTokens: 32000,
-      subagentMaxTurns: 25,
-      subagentTemperature: 1.0,
-      subagentMaxRetries: 3,
-      maxContinuations: 10,
-      subagentMaxContinuations: 5,
-      maxMessageChars: 100_000,
-      maxHistoryChars: 400_000,
-      maxToolResultChars: 2048,
-      compactionTriggerTokens: 167_000,
-      compactionTotalBudget: 2_000_000,
-      sessionCostBudgetUsd: 25.00,
-      teammateMaxTotalTurns: 50,
-      teamTimeoutMs: 600_000,
-      workerStallMs: 180_000,
-      teamReviewMaxTokens: 1024,
-      apiTimeoutMs: 90_000,
-      toolTimeoutMs: 60_000,
-      toolResultTruncateChars: 500_000
-    }
-  };
-});
-vi.mock("../../../shared/sse-parser.js", () => ({
-  processStreamWithCallbacks: vi.fn(async () => ({
-    text: "Test response",
-    toolUseBlocks: [],
-    thinkingBlocks: [],
-    thinkingTokens: 0,
-    citations: [],
-    usage: {
-      inputTokens: 100,
-      outputTokens: 50,
-      cacheCreationTokens: 0,
-      cacheReadTokens: 0
-    },
-    compactionContent: null,
-    contextManagementApplied: false,
-    stopReason: "end_turn"
-  }))
-}));
-vi.mock("../../../shared/constants.js", () => ({
-  MODELS: {
-    HAIKU: "claude-haiku-4-5-20251001",
-    SONNET: "claude-sonnet-4-6",
-    OPUS: "claude-opus-4-6"
-  },
-  getProvider: vi.fn(() => "anthropic")
-}));
-vi.mock("../../../shared/tool-dispatch.js", () => ({
-  dispatchTools: vi.fn(async (blocks, executor, _opts) => {
-    const results = [];
-    for (const block of blocks) {
-      const res = await executor(block.name, block.input);
-      results.push({
-        type: "tool_result",
-        tool_use_id: block.id,
-        content: res.output,
-        is_error: !res.success
-      });
-    }
-    return {
-      results,
-      bailOut: false,
-      bailMessage: undefined
-    };
-  }),
-  buildAssistantContent: vi.fn(({
-    text,
-    toolUseBlocks
-  }) => {
-    const blocks = [];
-    if (text) blocks.push({
-      type: "text",
-      text
-    });
-    if (toolUseBlocks) for (const t of toolUseBlocks) blocks.push({
-      type: "tool_use",
-      id: t.id,
-      name: t.name,
-      input: t.input
-    });
-    return blocks;
-  })
-}));
-vi.mock("../../tool-router.js", () => ({
-  getCachedToolDefs: vi.fn(tools => tools.map(t => ({
-    name: t.name,
-    description: t.description,
-    input_schema: t.input_schema
-  }))),
-  getFullToolSchemas: vi.fn(() => [])
-}));
-vi.mock("../server-subagent.js", () => ({
-  DELEGATE_TASK_TOOL_DEF: {
-    name: "delegate_task",
-    description: "Delegate to subagent",
-    input_schema: {
-      type: "object",
-      properties: {
-        prompt: {
-          type: "string"
-        }
-      },
-      required: ["prompt"]
-    }
-  },
-  runServerSubagent: vi.fn(async () => ({
-    success: true,
-    output: "Subagent result",
-    turnCount: 1,
-    toolsUsed: [],
-    tokensUsed: {
-      input: 50,
-      output: 25
-    },
-    costUsd: 0.0005
-  }))
-}));
-vi.mock("../../handlers/transcription.js", () => ({
-  handleTranscribe: vi.fn(async () => ({
-    success: true,
-    transcript: "Transcribed text"
-  }))
-}));
-vi.mock("../compaction-service.js", () => ({
-  preCompact: vi.fn(msgs => ({
-    messages: msgs,
-    bytesRemoved: 0
-  }))
-}));
-vi.mock("../provider-failover.js", () => ({
-  providerFailover: {
-    getActiveProvider: vi.fn(model => ({
-      provider: "anthropic",
-      model,
-      failedOver: false,
-      originalProvider: "anthropic"
-    })),
-    recordSuccess: vi.fn(),
-    recordFailure: vi.fn()
-  }
-}));
-vi.mock("../session-checkpoint.js", () => ({
-  saveCheckpoint: vi.fn(async () => {})
-}));
-vi.mock("../logger.js", () => ({
-  createLogger: vi.fn(() => ({
-    info: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    debug: vi.fn()
-  }))
-}));
-
-// ============================================================================
-// Imports — after all mocks
-// ============================================================================
-
-import { runServerAgentLoop } from "../server-agent-loop.js";
-import { processStreamWithCallbacks } from "../../../shared/sse-parser.js";
-import { estimateCostUsd, isRetryableError, routeModel } from "../../../shared/agent-core.js";
-import { preCompact } from "../compaction-service.js";
-import { providerFailover } from "../provider-failover.js";
-
-// ============================================================================
-// Helpers
-// ============================================================================
-
-function makeTool(name) {
-  return {
-    name,
-    description: `Tool: ${name}`,
-    input_schema: {
-      type: "object",
-      properties: {}
-    }
-  };
-}
-function makeAnthropicMock() {
-  return {
-    beta: {
-      messages: {
-        create: vi.fn(async () => async function* () {}())
-      }
-    }
-  };
-}
-function makeSupabaseMock() {
-  return {
-    from: vi.fn(() => ({
-      insert: vi.fn(async () => ({
-        error: null
-      })),
-      select: vi.fn(() => ({
-        data: [],
-        error: null
-      }))
-    }))
-  };
-}
-function makeOpts(overrides = {}) {
-  return {
-    anthropic: makeAnthropicMock(),
-    supabase: makeSupabaseMock(),
-    model: "claude-sonnet-4-6",
-    systemPrompt: "You are a helpful assistant.",
-    messages: [{
-      role: "user",
-      content: "Hello"
-    }],
-    tools: [makeTool("test_tool")],
-    maxTurns: 10,
-    temperature: 0.7,
-    executeTool: vi.fn(async () => ({
-      success: true,
-      data: {
-        result: "ok"
-      }
-    })),
-    enableStreaming: true,
-    ...overrides
-  };
-}
-
-// ============================================================================
-// TESTS — abort, max turns, compaction, continuation, retry, routing, failover
-// ============================================================================
-
-describe("server-agent-loop — abort & retry", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-    vi.mocked(processStreamWithCallbacks).mockReset();
-    vi.mocked(processStreamWithCallbacks).mockResolvedValue({
-      text: "Test response",
-      toolUseBlocks: [],
-      thinkingBlocks: [],
-      thinkingTokens: 0,
-      citations: [],
-      usage: {
-        inputTokens: 100,
-        outputTokens: 50,
-        cacheCreationTokens: 0,
-        cacheReadTokens: 0
-      },
-      compactionContent: null,
-      contextManagementApplied: false,
-      stopReason: "end_turn"
-    });
-    vi.mocked(estimateCostUsd).mockReturnValue(0.001);
-    vi.mocked(isRetryableError).mockReturnValue(false);
-    vi.mocked(routeModel).mockImplementation((_text, model) => model);
-    vi.mocked(preCompact).mockImplementation(msgs => ({
-      messages: msgs,
-      bytesRemoved: 0
-    }));
-    vi.mocked(providerFailover.getActiveProvider).mockImplementation(model => ({
-      provider: "anthropic",
-      model,
-      failedOver: false,
-      originalProvider: "anthropic"
-    }));
-  });
-  describe("abort conditions", () => {
-    it("stops when client disconnects", async () => {
-      const result = await runServerAgentLoop(makeOpts({
-        clientDisconnected: {
-          value: true
-        }
-      }));
-      expect(result.turnCount).toBe(0);
-      expect(result.finalText).toBe("");
-    });
-    it("stops when duration exceeds max", async () => {
-      const onText = vi.fn();
-      const result = await runServerAgentLoop(makeOpts({
-        startedAt: Date.now() - 400_000,
-        maxDurationMs: 300_000,
-        onText
-      }));
-      expect(result.turnCount).toBe(0);
-      expect(onText).toHaveBeenCalledWith(expect.stringContaining("timeout"));
-    });
-    it("stops when cost budget exceeded", async () => {
-      vi.mocked(estimateCostUsd).mockReturnValue(10.0);
-      let callCount = 0;
-      vi.mocked(processStreamWithCallbacks).mockImplementation(async () => {
-        callCount++;
-        if (callCount === 1) {
-          return {
-            text: "",
-            toolUseBlocks: [{
-              id: "tu-1",
-              name: "test_tool",
-              input: {}
-            }],
-            thinkingBlocks: [],
-            thinkingTokens: 0,
-            citations: [],
-            usage: {
-              inputTokens: 100,
-              outputTokens: 50,
-              cacheCreationTokens: 0,
-              cacheReadTokens: 0
-            },
-            compactionContent: null,
-            contextManagementApplied: false,
-            stopReason: "tool_use"
-          };
-        }
-        return {
-          text: "Done",
-          toolUseBlocks: [],
-          thinkingBlocks: [],
-          thinkingTokens: 0,
-          citations: [],
-          usage: {
-            inputTokens: 50,
-            outputTokens: 20,
-            cacheCreationTokens: 0,
-            cacheReadTokens: 0
-          },
-          compactionContent: null,
-          contextManagementApplied: false,
-          stopReason: "end_turn"
-        };
-      });
-      const onText = vi.fn();
-      await runServerAgentLoop(makeOpts({
-        maxCostUsd: 0.0001,
-        onText
-      }));
-      expect(onText).toHaveBeenCalledWith(expect.stringContaining("cost budget"));
-    });
-  });
-});