Spaces:
Paused
Paused
| import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; | |
| import type { TemplateContext } from "../templating.js"; | |
| import type { FollowupRun, QueueSettings } from "./queue.js"; | |
| import { createMockTypingController } from "./test-helpers.js"; | |
| const runEmbeddedPiAgentMock = vi.fn(); | |
| const runtimeErrorMock = vi.fn(); | |
| vi.mock("../../agents/model-fallback.js", () => ({ | |
| runWithModelFallback: async ({ | |
| provider, | |
| model, | |
| run, | |
| }: { | |
| provider: string; | |
| model: string; | |
| run: (provider: string, model: string) => Promise<unknown>; | |
| }) => ({ | |
| result: await run(provider, model), | |
| provider, | |
| model, | |
| }), | |
| })); | |
| vi.mock("../../agents/pi-embedded.js", () => ({ | |
| queueEmbeddedPiMessage: vi.fn().mockReturnValue(false), | |
| runEmbeddedPiAgent: (params: unknown) => runEmbeddedPiAgentMock(params), | |
| })); | |
| vi.mock("../../runtime.js", () => ({ | |
| defaultRuntime: { | |
| log: vi.fn(), | |
| error: (...args: unknown[]) => runtimeErrorMock(...args), | |
| exit: vi.fn(), | |
| }, | |
| })); | |
| vi.mock("./queue.js", async () => { | |
| const actual = await vi.importActual<typeof import("./queue.js")>("./queue.js"); | |
| return { | |
| ...actual, | |
| enqueueFollowupRun: vi.fn(), | |
| scheduleFollowupDrain: vi.fn(), | |
| }; | |
| }); | |
| import { runReplyAgent } from "./agent-runner.js"; | |
| describe("runReplyAgent transient HTTP retry", () => { | |
| beforeEach(() => { | |
| runEmbeddedPiAgentMock.mockReset(); | |
| runtimeErrorMock.mockReset(); | |
| vi.useFakeTimers(); | |
| }); | |
| afterEach(() => { | |
| vi.useRealTimers(); | |
| }); | |
| it("retries once after transient 521 HTML failure and then succeeds", async () => { | |
| runEmbeddedPiAgentMock | |
| .mockRejectedValueOnce( | |
| new Error( | |
| `521 <!DOCTYPE html><html lang="en-US"><head><title>Web server is down</title></head><body>Cloudflare</body></html>`, | |
| ), | |
| ) | |
| .mockResolvedValueOnce({ | |
| payloads: [{ text: "Recovered response" }], | |
| meta: {}, | |
| }); | |
| const typing = createMockTypingController(); | |
| const sessionCtx = { | |
| Provider: "telegram", | |
| MessageSid: "msg", | |
| } as unknown as TemplateContext; | |
| const resolvedQueue = { mode: "interrupt" } as unknown as QueueSettings; | |
| const followupRun = { | |
| prompt: "hello", | |
| summaryLine: "hello", | |
| enqueuedAt: Date.now(), | |
| run: { | |
| sessionId: "session", | |
| sessionKey: "main", | |
| messageProvider: "telegram", | |
| sessionFile: "/tmp/session.jsonl", | |
| workspaceDir: "/tmp", | |
| config: {}, | |
| skillsSnapshot: {}, | |
| provider: "anthropic", | |
| model: "claude", | |
| thinkLevel: "low", | |
| verboseLevel: "off", | |
| elevatedLevel: "off", | |
| bashElevated: { | |
| enabled: false, | |
| allowed: false, | |
| defaultLevel: "off", | |
| }, | |
| timeoutMs: 1_000, | |
| blockReplyBreak: "message_end", | |
| }, | |
| } as unknown as FollowupRun; | |
| const runPromise = runReplyAgent({ | |
| commandBody: "hello", | |
| followupRun, | |
| queueKey: "main", | |
| resolvedQueue, | |
| shouldSteer: false, | |
| shouldFollowup: false, | |
| isActive: false, | |
| isStreaming: false, | |
| typing, | |
| sessionCtx, | |
| defaultModel: "anthropic/claude-opus-4-5", | |
| resolvedVerboseLevel: "off", | |
| isNewSession: false, | |
| blockStreamingEnabled: false, | |
| resolvedBlockStreamingBreak: "message_end", | |
| shouldInjectGroupIntro: false, | |
| typingMode: "instant", | |
| }); | |
| await vi.advanceTimersByTimeAsync(2_500); | |
| const result = await runPromise; | |
| expect(runEmbeddedPiAgentMock).toHaveBeenCalledTimes(2); | |
| expect(runtimeErrorMock).toHaveBeenCalledWith( | |
| expect.stringContaining("Transient HTTP provider error before reply"), | |
| ); | |
| const payload = Array.isArray(result) ? result[0] : result; | |
| expect(payload?.text).toContain("Recovered response"); | |
| }); | |
| }); | |