File size: 4,156 Bytes
f778c12 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | // Safe gateway restart tests cover operator-facing acknowledgement copy.
import { beforeEach, describe, expect, it, vi } from "vitest";
const callGatewayCli = vi.hoisted(() => vi.fn());
const refreshLegacySystemdServiceMetadata = vi.hoisted(() =>
vi.fn<(_env: NodeJS.ProcessEnv, timeoutMs?: number) => Promise<boolean>>(async () => false),
);
const resolveGatewayServiceMutationError = vi.hoisted(() => vi.fn<() => Error | null>(() => null));
const appendGatewayLifecycleAudit = vi.hoisted(() => vi.fn());
const runtimeError = vi.hoisted(() => vi.fn());
const runtimeLog = vi.hoisted(() => vi.fn());
const runtimeWriteJson = vi.hoisted(() => vi.fn());
vi.mock("../../gateway/call.js", () => ({
callGatewayCli,
}));
vi.mock("../../daemon/systemd.js", () => ({
refreshLegacySystemdServiceMetadata,
}));
vi.mock("../../infra/gateway-supervision.js", () => ({
resolveGatewayServiceMutationError,
}));
vi.mock("../../runtime.js", () => ({
defaultRuntime: {
error: runtimeError,
log: runtimeLog,
writeJson: runtimeWriteJson,
},
writeRuntimeJson: (_runtime: unknown, payload: unknown) => runtimeWriteJson(payload),
}));
vi.mock("./lifecycle-audit.js", () => ({
appendGatewayLifecycleAudit,
}));
describe("runSafeGatewayRestart", () => {
beforeEach(() => {
callGatewayCli.mockReset();
appendGatewayLifecycleAudit.mockReset();
refreshLegacySystemdServiceMetadata.mockReset().mockResolvedValue(false);
resolveGatewayServiceMutationError.mockReset().mockReturnValue(null);
runtimeError.mockReset();
runtimeLog.mockReset();
runtimeWriteJson.mockReset();
});
it("keeps skip-deferral RPC behavior when bounded metadata cleanup fails", async () => {
vi.spyOn(process, "platform", "get").mockReturnValue("linux");
const { runSafeGatewayRestart } = await import("./lifecycle-safe-restart.js");
refreshLegacySystemdServiceMetadata.mockRejectedValueOnce(
new Error("systemd cleanup timed out"),
);
callGatewayCli.mockResolvedValueOnce({
status: "scheduled",
preflight: {
safe: false,
activeWork: {
queueSize: 1,
runningTasks: 0,
activeRequests: 0,
activeAgentRuns: 0,
pendingReplies: 2,
totalActive: 3,
},
blockers: [{ kind: "pending-replies", count: 2, message: "2 pending reply(ies)" }],
summary: "restart deferred: 2 pending reply(ies)",
},
restart: { pid: 123 },
});
await expect(
runSafeGatewayRestart({ json: true, safe: true, skipDeferral: true }),
).resolves.toBe(true);
expect(callGatewayCli).toHaveBeenCalledWith({
method: "gateway.restart.request",
params: { reason: "gateway.restart.safe", skipDeferral: true },
timeoutMs: 10_000,
});
expect(refreshLegacySystemdServiceMetadata).toHaveBeenCalledWith(process.env, 5_000);
expect(runtimeError).toHaveBeenCalledWith(expect.stringContaining("systemd cleanup timed out"));
expect(runtimeWriteJson).toHaveBeenCalledWith(
expect.objectContaining({
message:
"safe restart requested; gateway bypassing active-work deferral; " +
"shutdown may still wait for pending replies to drain",
result: "scheduled",
}),
);
});
it("keeps the RPC restart while skipping ineligible service mutation", async () => {
vi.spyOn(process, "platform", "get").mockReturnValue("linux");
resolveGatewayServiceMutationError.mockReturnValueOnce(
new Error("gateway lifecycle is managed by an external supervisor"),
);
const { runSafeGatewayRestart } = await import("./lifecycle-safe-restart.js");
callGatewayCli.mockResolvedValueOnce({
status: "scheduled",
preflight: { blockers: [], summary: "safe to restart now" },
restart: { pid: 123 },
});
await expect(runSafeGatewayRestart({ json: true, safe: true })).resolves.toBe(true);
expect(refreshLegacySystemdServiceMetadata).not.toHaveBeenCalled();
expect(callGatewayCli).toHaveBeenCalledOnce();
expect(runtimeError).toHaveBeenCalledWith(expect.stringContaining("external supervisor"));
});
});
|