101 lines
3.3 KiB
TypeScript
101 lines
3.3 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
import { captureFullEnv } from "../test-utils/env.js";
|
|
import { SUPERVISOR_HINT_ENV_VARS } from "./supervisor-markers.js";
|
|
|
|
const spawnMock = vi.hoisted(() => vi.fn());
|
|
|
|
vi.mock("node:child_process", () => ({
|
|
spawn: (...args: unknown[]) => spawnMock(...args),
|
|
}));
|
|
|
|
import { restartGatewayProcessWithFreshPid } from "./process-respawn.js";
|
|
|
|
const originalArgv = [...process.argv];
|
|
const originalExecArgv = [...process.execArgv];
|
|
const envSnapshot = captureFullEnv();
|
|
|
|
afterEach(() => {
|
|
envSnapshot.restore();
|
|
process.argv = [...originalArgv];
|
|
process.execArgv = [...originalExecArgv];
|
|
spawnMock.mockClear();
|
|
});
|
|
|
|
function clearSupervisorHints() {
|
|
for (const key of SUPERVISOR_HINT_ENV_VARS) {
|
|
delete process.env[key];
|
|
}
|
|
}
|
|
|
|
describe("restartGatewayProcessWithFreshPid", () => {
|
|
it("returns disabled when OPENCLAW_NO_RESPAWN is set", () => {
|
|
process.env.OPENCLAW_NO_RESPAWN = "1";
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
expect(result.mode).toBe("disabled");
|
|
expect(spawnMock).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("returns supervised when launchd/systemd hints are present", () => {
|
|
process.env.LAUNCH_JOB_LABEL = "ai.openclaw.gateway";
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
expect(result.mode).toBe("supervised");
|
|
expect(spawnMock).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("spawns detached child with current exec argv", () => {
|
|
delete process.env.OPENCLAW_NO_RESPAWN;
|
|
clearSupervisorHints();
|
|
process.execArgv = ["--import", "tsx"];
|
|
process.argv = ["/usr/local/bin/node", "/repo/dist/index.js", "gateway", "run"];
|
|
spawnMock.mockReturnValue({ pid: 4242, unref: vi.fn() });
|
|
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
|
|
expect(result).toEqual({ mode: "spawned", pid: 4242 });
|
|
expect(spawnMock).toHaveBeenCalledWith(
|
|
process.execPath,
|
|
["--import", "tsx", "/repo/dist/index.js", "gateway", "run"],
|
|
expect.objectContaining({
|
|
detached: true,
|
|
stdio: "inherit",
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("returns supervised when OPENCLAW_LAUNCHD_LABEL is set (stock launchd plist)", () => {
|
|
clearSupervisorHints();
|
|
process.env.OPENCLAW_LAUNCHD_LABEL = "ai.openclaw.gateway";
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
expect(result.mode).toBe("supervised");
|
|
expect(spawnMock).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("returns supervised when OPENCLAW_SYSTEMD_UNIT is set", () => {
|
|
clearSupervisorHints();
|
|
process.env.OPENCLAW_SYSTEMD_UNIT = "openclaw-gateway.service";
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
expect(result.mode).toBe("supervised");
|
|
expect(spawnMock).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("returns supervised when OPENCLAW_SERVICE_MARKER is set", () => {
|
|
clearSupervisorHints();
|
|
process.env.OPENCLAW_SERVICE_MARKER = "gateway";
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
expect(result.mode).toBe("supervised");
|
|
expect(spawnMock).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("returns failed when spawn throws", () => {
|
|
delete process.env.OPENCLAW_NO_RESPAWN;
|
|
clearSupervisorHints();
|
|
|
|
spawnMock.mockImplementation(() => {
|
|
throw new Error("spawn failed");
|
|
});
|
|
const result = restartGatewayProcessWithFreshPid();
|
|
expect(result.mode).toBe("failed");
|
|
expect(result.detail).toContain("spawn failed");
|
|
});
|
|
});
|