87 lines
2.1 KiB
TypeScript
87 lines
2.1 KiB
TypeScript
import { EventEmitter } from "node:events";
|
|
import type * as childProcess from "node:child_process";
|
|
import { PassThrough } from "node:stream";
|
|
import { afterEach, expect, test, vi } from "vitest";
|
|
|
|
const spawnMock = vi.fn();
|
|
|
|
vi.mock("node:child_process", async () => {
|
|
const actual = await vi.importActual<typeof import("node:child_process")>("node:child_process");
|
|
return {
|
|
...actual,
|
|
spawn: spawnMock,
|
|
};
|
|
});
|
|
|
|
const { spawnLocalAgent, stopLocalAgent } = await import("../agents-manager");
|
|
|
|
const flushMicrotasks = async () => {
|
|
await Promise.resolve();
|
|
};
|
|
|
|
type FakeChildProcess = EventEmitter & {
|
|
stdout: PassThrough;
|
|
stderr: PassThrough;
|
|
exitCode: number | null;
|
|
signalCode: NodeJS.Signals | null;
|
|
kill: ReturnType<typeof vi.fn>;
|
|
};
|
|
|
|
const createFakeChild = () => {
|
|
const child = new EventEmitter() as FakeChildProcess;
|
|
|
|
child.stdout = new PassThrough();
|
|
child.stderr = new PassThrough();
|
|
child.exitCode = null;
|
|
child.signalCode = null;
|
|
child.kill = vi.fn(() => {
|
|
child.exitCode = 0;
|
|
child.emit("exit", 0, null);
|
|
return true;
|
|
});
|
|
|
|
return child;
|
|
};
|
|
|
|
afterEach(async () => {
|
|
await stopLocalAgent();
|
|
spawnMock.mockReset();
|
|
vi.restoreAllMocks();
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
test("respawns the local agent after an unexpected exit", async () => {
|
|
vi.useFakeTimers();
|
|
|
|
const firstChild = createFakeChild();
|
|
const secondChild = createFakeChild();
|
|
spawnMock
|
|
.mockReturnValueOnce(firstChild as unknown as ReturnType<typeof childProcess.spawn>)
|
|
.mockReturnValueOnce(secondChild as unknown as ReturnType<typeof childProcess.spawn>);
|
|
|
|
await spawnLocalAgent();
|
|
|
|
firstChild.exitCode = 1;
|
|
firstChild.emit("exit", 1, null);
|
|
|
|
vi.advanceTimersByTime(1_000);
|
|
await flushMicrotasks();
|
|
|
|
expect(spawnMock).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
test("does not respawn the local agent after an intentional stop", async () => {
|
|
vi.useFakeTimers();
|
|
|
|
const child = createFakeChild();
|
|
spawnMock.mockReturnValue(child as unknown as ReturnType<typeof childProcess.spawn>);
|
|
|
|
await spawnLocalAgent();
|
|
await stopLocalAgent();
|
|
|
|
vi.advanceTimersByTime(1_000);
|
|
await flushMicrotasks();
|
|
|
|
expect(spawnMock).toHaveBeenCalledTimes(1);
|
|
expect(child.kill).toHaveBeenCalledTimes(1);
|
|
});
|