diff --git a/src/agents/bash-tools.test.ts b/src/agents/bash-tools.test.ts index 4a94ce46d..825a4cc20 100644 --- a/src/agents/bash-tools.test.ts +++ b/src/agents/bash-tools.test.ts @@ -10,6 +10,7 @@ import { sanitizeBinaryOutput } from "./shell-utils.js"; const isWin = process.platform === "win32"; const shortDelayCmd = isWin ? "ping -n 2 127.0.0.1 > nul" : "sleep 0.05"; +const yieldDelayCmd = isWin ? "ping -n 3 127.0.0.1 > nul" : "sleep 0.2"; const longDelayCmd = isWin ? "ping -n 4 127.0.0.1 > nul" : "sleep 2"; const joinCommands = (commands: string[]) => commands.join(isWin ? " & " : "; "); @@ -51,7 +52,7 @@ beforeEach(() => { describe("bash tool backgrounding", () => { it("backgrounds after yield and can be polled", async () => { const result = await bashTool.execute("call1", { - command: echoAfterDelay("done"), + command: joinCommands([yieldDelayCmd, "echo done"]), yieldMs: 10, }); diff --git a/src/commands/doctor.test.ts b/src/commands/doctor.test.ts index 395ee993c..1ba4f2c5e 100644 --- a/src/commands/doctor.test.ts +++ b/src/commands/doctor.test.ts @@ -94,6 +94,7 @@ const serviceIsLoaded = vi.fn().mockResolvedValue(false); const serviceStop = vi.fn().mockResolvedValue(undefined); const serviceRestart = vi.fn().mockResolvedValue(undefined); const serviceUninstall = vi.fn().mockResolvedValue(undefined); +const callGateway = vi.fn().mockRejectedValue(new Error("gateway closed")); vi.mock("@clack/prompts", () => ({ confirm, @@ -133,6 +134,14 @@ vi.mock("../daemon/program-args.js", () => ({ resolveGatewayProgramArguments, })); +vi.mock("../gateway/call.js", async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + callGateway, + }; +}); + vi.mock("../process/exec.js", () => ({ runExec, runCommandWithTimeout,