const isGatewaySigusr1RestartExternallyAllowed = vi.fn(() => false);

const markGatewaySigusr1RestartHandled = vi.fn();

const peekGatewaySigusr1RestartReason = vi.fn<() => string | undefined>(() => undefined);

const resetGatewayRestartStateForInProcessRestart = vi.fn();

const scheduleGatewaySigusr1Restart = vi.fn((_opts?: { delayMs?: number; reason?: string }) => ({

ok: true,

pid: process.pid,

const markGatewayDraining = vi.fn();

const waitForActiveTasks = vi.fn(async (_timeoutMs?: number) => ({ drained: true }));

const resetAllLanes = vi.fn();

const reloadTaskRegistryFromStore = vi.fn();

const getActiveBundledRuntimeDepsInstallCount = vi.fn(() => 0);

const waitForBundledRuntimeDepsInstallIdle = vi.fn(async (_timeoutMs?: number) => ({

drained: true,

isGatewaySigusr1RestartExternallyAllowed: () => isGatewaySigusr1RestartExternallyAllowed(),

markGatewaySigusr1RestartHandled: () => markGatewaySigusr1RestartHandled(),

peekGatewaySigusr1RestartReason: () => peekGatewaySigusr1RestartReason(),

resetGatewayRestartStateForInProcessRestart: () => resetGatewayRestartStateForInProcessRestart(),

scheduleGatewaySigusr1Restart: (opts?: { delayMs?: number; reason?: string }) =>

scheduleGatewaySigusr1Restart(opts),

}));

resetAllLanes: () => resetAllLanes(),

}));

vi.mock("../../tasks/runtime-internal.js", () => ({

reloadTaskRegistryFromStore: () => reloadTaskRegistryFromStore(),

}));

vi.mock("../../plugins/bundled-runtime-deps-activity.js", () => ({

getActiveBundledRuntimeDepsInstallCount: () => getActiveBundledRuntimeDepsInstallCount(),

waitForBundledRuntimeDepsInstallIdle: (timeoutMs?: number) =>

});

});

it("restarts after SIGUSR1 even when drain times out, and resets lanes for the new iteration", async () => {

it("restarts after SIGUSR1 even when drain times out, and resets runtime state for the new iteration", async () => {

vi.clearAllMocks();

loadConfig.mockReturnValue({

gateway: {

});

expect(markGatewaySigusr1RestartHandled).toHaveBeenCalledTimes(1);

expect(resetAllLanes).toHaveBeenCalledTimes(1);

expect(resetGatewayRestartStateForInProcessRestart).toHaveBeenCalledTimes(1);

expect(reloadTaskRegistryFromStore).toHaveBeenCalledTimes(1);

sigusr1();

expect(markGatewaySigusr1RestartHandled).toHaveBeenCalledTimes(2);

expect(markGatewayDraining).toHaveBeenCalledTimes(2);

expect(resetAllLanes).toHaveBeenCalledTimes(2);

expect(resetGatewayRestartStateForInProcessRestart).toHaveBeenCalledTimes(2);

expect(reloadTaskRegistryFromStore).toHaveBeenCalledTimes(2);

expect(acquireGatewayLock).toHaveBeenCalledTimes(3);

sigterm();