ewilliams28/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-08-27 12:56:01 -06:00
Code Issues Packages Projects Releases Wiki Activity

test: dedupe gateway tool mock reads

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-13 02:23:20 +01:00
parent 4aedf91028
commit f537b06e8d
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/agents/tools/gateway-tool.test.ts
+10 -11
View File
@@ -67,7 +67,8 @@ vi.mock("./gateway.js", () => ({
}));
function requireRestartSentinelPayload(): RestartSentinelPayload {
const payload = writeRestartSentinelMock.mock.calls.at(-1)?.[0];
const calls = writeRestartSentinelMock.mock.calls;
const payload = calls[calls.length - 1]?.[0];
if (!payload) {
throw new Error("expected restart sentinel payload");
}
@@ -75,7 +76,8 @@ function requireRestartSentinelPayload(): RestartSentinelPayload {
}
function requireScheduledRestartArgs(): NonNullable<ScheduleGatewayRestartArgs> {
const args = scheduleGatewaySigusr1RestartMock.mock.calls.at(-1)?.[0];
const calls = scheduleGatewaySigusr1RestartMock.mock.calls;
const args = calls[calls.length - 1]?.[0];
if (!args) {
throw new Error("expected scheduled restart args");
}
@@ -138,8 +140,7 @@ describe("gateway tool restart continuation", () => {
});
expect(writeRestartSentinelMock).not.toHaveBeenCalled();
const scheduledArgs = scheduleGatewaySigusr1RestartMock.mock.calls.at(-1)?.[0];
await scheduledArgs?.emitHooks?.beforeEmit?.();
await requireScheduledRestartArgs().emitHooks?.beforeEmit?.();
const payload = requireRestartSentinelPayload();
expect(payload.kind).toBe("restart");
@@ -176,8 +177,7 @@ describe("gateway tool restart continuation", () => {
continuationMessage: "Reply after restart",
});
const scheduledArgs = scheduleGatewaySigusr1RestartMock.mock.calls.at(-1)?.[0];
await scheduledArgs?.emitHooks?.beforeEmit?.();
await requireScheduledRestartArgs().emitHooks?.beforeEmit?.();
expect(requireRestartSentinelPayload().continuation).toEqual({
kind: "agentTurn",
@@ -199,8 +199,7 @@ describe("gateway tool restart continuation", () => {
reason: "restart requested",
});
const scheduledArgs = scheduleGatewaySigusr1RestartMock.mock.calls.at(-1)?.[0];
await scheduledArgs?.emitHooks?.beforeEmit?.();
await requireScheduledRestartArgs().emitHooks?.beforeEmit?.();
const payload = requireRestartSentinelPayload();
expect(payload.sessionKey).toBe("agent:main:main");
@@ -220,9 +219,9 @@ describe("gateway tool restart continuation", () => {
action: "restart",
});
const scheduledArgs = scheduleGatewaySigusr1RestartMock.mock.calls.at(-1)?.[0];
await scheduledArgs?.emitHooks?.beforeEmit?.();
await scheduledArgs?.emitHooks?.afterEmitRejected?.();
const scheduledArgs = requireScheduledRestartArgs();
await scheduledArgs.emitHooks?.beforeEmit?.();
await scheduledArgs.emitHooks?.afterEmitRejected?.();
expect(removeRestartSentinelFileMock).toHaveBeenCalledWith("/tmp/restart");
});