import type { OpenClawConfig } from "../config/config.js";

import { loggingState } from "../logging/state.js";

import type { RuntimeEnv } from "../runtime.js";

import { agentCliCommand } from "./agent-via-gateway.js";

import { __testing as agentViaGatewayTesting, agentCliCommand } from "./agent-via-gateway.js";

import type { agentCommand as AgentCommand } from "./agent.js";

const loadConfig = vi.hoisted(() => vi.fn());

);

const agentCommand = vi.hoisted(() => vi.fn());

const agentModuleLoadCount = vi.hoisted(() => vi.fn());

const loadAgentSessionModuleMock = vi.hoisted(() => vi.fn());

const runtime: RuntimeEnv = {

log: vi.fn(),

expect(agentCommand).toHaveBeenCalledTimes(expectedCalls);

}

async function waitForGatewayCall(expectedCalls = 1) {

for (

let attempt = 0;

attempt < 50 && callGateway.mock.calls.length < expectedCalls;

attempt += 1

) {

await new Promise<void>((resolve) => setTimeout(resolve, 0));

}

expect(callGateway).toHaveBeenCalledTimes(expectedCalls);

}

function mockMessages(mock: unknown): string[] {

const calls = (mock as { mock?: { calls?: unknown[][] } }).mock?.calls ?? [];

return calls.map(([message]) => String(message));

beforeEach(() => {

vi.clearAllMocks();

agentViaGatewayTesting.resetLazyImportsForTests();

loadAgentSessionModuleMock.mockImplementation(async () => await import("./agent/session.js"));

agentViaGatewayTesting.setAgentSessionModuleLoaderForTests(loadAgentSessionModuleMock);

originalForceConsoleToStderr = loggingState.forceConsoleToStderr;

loggingState.forceConsoleToStderr = false;

});

expect(params.sessionId).toBeUndefined();

expect(params.to).toBeUndefined();

expect(agentCommand).not.toHaveBeenCalled();

expect(loadAgentSessionModuleMock).not.toHaveBeenCalled();

});

});

const run = agentCliCommand({ message: "hi", to: "+1555" }, runtime, {

process: signals.processLike,

});

await Promise.resolve();

await waitForGatewayCall();

signals.emit(signalName);

expect(signals.listenerCount("SIGTERM")).toBe(0);

expect(signals.listenerCount("SIGINT")).toBe(0);

process: signals.processLike,

},

);

await Promise.resolve();

await waitForGatewayCall();

signals.emit("SIGTERM");

await run;

const run = agentCliCommand({ message: "hi", to: "+1555" }, runtime, {

process: signals.processLike,

});

await Promise.resolve();

await waitForGatewayCall();

signals.emit("SIGTERM");

await run;

process: signals.processLike,

},

);

await Promise.resolve();

await waitForGatewayCall();

signals.emit("SIGTERM");

await run;

process: signals.processLike,

},

);

await Promise.resolve();

await waitForGatewayCall();

signals.emit("SIGTERM");

await run;

process: signals.processLike,

},

);

await Promise.resolve();

await waitForGatewayCall();

signals.emit("SIGTERM");

await run;

process: signals.processLike,

},

);

await Promise.resolve();

await waitForGatewayCall();

signals.emit("SIGTERM");

await run;