test: move cli local suites out of e2e
This commit is contained in:
Peter Steinberger
257
src/cli/gateway-cli.coverage.test.ts
Normal file
257
src/cli/gateway-cli.coverage.test.ts
Normal file
@@ -0,0 +1,257 @@
|
||||
import { Command } from "commander";
|
||||
import { describe, expect, it, vi } from "vitest";
|
||||
import { withEnvOverride } from "../config/test-helpers.js";
|
||||
import { createCliRuntimeCapture } from "./test-runtime-capture.js";
|
||||
|
||||
type DiscoveredBeacon = Awaited<
|
||||
ReturnType<typeof import("../infra/bonjour-discovery.js").discoverGatewayBeacons>
|
||||
>[number];
|
||||
|
||||
const callGateway = vi.fn<(opts: unknown) => Promise<{ ok: true }>>(async () => ({ ok: true }));
|
||||
const startGatewayServer = vi.fn<
|
||||
(port: number, opts?: unknown) => Promise<{ close: () => Promise<void> }>
|
||||
>(async () => ({
|
||||
close: vi.fn(async () => {}),
|
||||
}));
|
||||
const setVerbose = vi.fn();
|
||||
const forceFreePortAndWait = vi.fn<
|
||||
(port: number) => Promise<{ killed: unknown[]; waitedMs: number; escalatedToSigkill: boolean }>
|
||||
>(async () => ({
|
||||
killed: [],
|
||||
waitedMs: 0,
|
||||
escalatedToSigkill: false,
|
||||
}));
|
||||
const serviceIsLoaded = vi.fn().mockResolvedValue(true);
|
||||
const discoverGatewayBeacons = vi.fn<(opts: unknown) => Promise<DiscoveredBeacon[]>>(
|
||||
async () => [],
|
||||
);
|
||||
const gatewayStatusCommand = vi.fn<(opts: unknown) => Promise<void>>(async () => {});
|
||||
|
||||
const { runtimeLogs, runtimeErrors, defaultRuntime, resetRuntimeCapture } =
|
||||
createCliRuntimeCapture();
|
||||
|
||||
vi.mock(
|
||||
new URL("../../gateway/call.ts", new URL("./gateway-cli/call.ts", import.meta.url)).href,
|
||||
() => ({
|
||||
callGateway: (opts: unknown) => callGateway(opts),
|
||||
randomIdempotencyKey: () => "rk_test",
|
||||
}),
|
||||
);
|
||||
|
||||
vi.mock("../gateway/server.js", () => ({
|
||||
startGatewayServer: (port: number, opts?: unknown) => startGatewayServer(port, opts),
|
||||
}));
|
||||
|
||||
vi.mock("../globals.js", () => ({
|
||||
info: (msg: string) => msg,
|
||||
isVerbose: () => false,
|
||||
setVerbose: (enabled: boolean) => setVerbose(enabled),
|
||||
}));
|
||||
|
||||
vi.mock("../runtime.js", () => ({
|
||||
defaultRuntime,
|
||||
}));
|
||||
|
||||
vi.mock("./ports.js", () => ({
|
||||
forceFreePortAndWait: (port: number) => forceFreePortAndWait(port),
|
||||
}));
|
||||
|
||||
vi.mock("../daemon/service.js", () => ({
|
||||
resolveGatewayService: () => ({
|
||||
label: "LaunchAgent",
|
||||
loadedText: "loaded",
|
||||
notLoadedText: "not loaded",
|
||||
install: vi.fn(),
|
||||
uninstall: vi.fn(),
|
||||
stop: vi.fn(),
|
||||
restart: vi.fn(),
|
||||
isLoaded: serviceIsLoaded,
|
||||
readCommand: vi.fn(),
|
||||
readRuntime: vi.fn().mockResolvedValue({ status: "running" }),
|
||||
}),
|
||||
}));
|
||||
|
||||
vi.mock("../daemon/program-args.js", () => ({
|
||||
resolveGatewayProgramArguments: async () => ({
|
||||
programArguments: ["/bin/node", "cli", "gateway", "--port", "18789"],
|
||||
}),
|
||||
}));
|
||||
|
||||
vi.mock("../infra/bonjour-discovery.js", () => ({
|
||||
discoverGatewayBeacons: (opts: unknown) => discoverGatewayBeacons(opts),
|
||||
}));
|
||||
|
||||
vi.mock("../commands/gateway-status.js", () => ({
|
||||
gatewayStatusCommand: (opts: unknown) => gatewayStatusCommand(opts),
|
||||
}));
|
||||
|
||||
const { registerGatewayCli } = await import("./gateway-cli.js");
|
||||
|
||||
function createGatewayProgram() {
|
||||
const program = new Command();
|
||||
program.exitOverride();
|
||||
registerGatewayCli(program);
|
||||
return program;
|
||||
}
|
||||
|
||||
async function runGatewayCommand(args: string[]) {
|
||||
const program = createGatewayProgram();
|
||||
await program.parseAsync(args, { from: "user" });
|
||||
}
|
||||
|
||||
async function expectGatewayExit(args: string[]) {
|
||||
await expect(runGatewayCommand(args)).rejects.toThrow("__exit__:1");
|
||||
}
|
||||
|
||||
describe("gateway-cli coverage", () => {
|
||||
it("registers call/health commands and routes to callGateway", async () => {
|
||||
resetRuntimeCapture();
|
||||
callGateway.mockClear();
|
||||
|
||||
await runGatewayCommand(["gateway", "call", "health", "--params", '{"x":1}', "--json"]);
|
||||
|
||||
expect(callGateway).toHaveBeenCalledTimes(1);
|
||||
expect(runtimeLogs.join("\n")).toContain('"ok": true');
|
||||
}, 60_000);
|
||||
|
||||
it("registers gateway probe and routes to gatewayStatusCommand", async () => {
|
||||
resetRuntimeCapture();
|
||||
gatewayStatusCommand.mockClear();
|
||||
|
||||
await runGatewayCommand(["gateway", "probe", "--json"]);
|
||||
|
||||
expect(gatewayStatusCommand).toHaveBeenCalledTimes(1);
|
||||
}, 60_000);
|
||||
|
||||
it.each([
|
||||
{
|
||||
label: "json output",
|
||||
args: ["gateway", "discover", "--json"],
|
||||
expectedOutput: ['"beacons"', '"wsUrl"', "ws://"],
|
||||
},
|
||||
{
|
||||
label: "human output",
|
||||
args: ["gateway", "discover", "--timeout", "1"],
|
||||
expectedOutput: [
|
||||
"Gateway Discovery",
|
||||
"Found 1 gateway(s)",
|
||||
"- Studio openclaw.internal.",
|
||||
" tailnet: studio.tailnet.ts.net",
|
||||
" host: studio.openclaw.internal",
|
||||
" ws: ws://studio.openclaw.internal:18789",
|
||||
],
|
||||
},
|
||||
])("registers gateway discover and prints $label", async ({ args, expectedOutput }) => {
|
||||
resetRuntimeCapture();
|
||||
discoverGatewayBeacons.mockClear();
|
||||
discoverGatewayBeacons.mockResolvedValueOnce([
|
||||
{
|
||||
instanceName: "Studio (OpenClaw)",
|
||||
displayName: "Studio",
|
||||
domain: "openclaw.internal.",
|
||||
host: "studio.openclaw.internal",
|
||||
lanHost: "studio.local",
|
||||
tailnetDns: "studio.tailnet.ts.net",
|
||||
gatewayPort: 18789,
|
||||
sshPort: 22,
|
||||
},
|
||||
]);
|
||||
|
||||
await runGatewayCommand(args);
|
||||
|
||||
expect(discoverGatewayBeacons).toHaveBeenCalledTimes(1);
|
||||
const out = runtimeLogs.join("\n");
|
||||
for (const text of expectedOutput) {
|
||||
expect(out).toContain(text);
|
||||
}
|
||||
});
|
||||
|
||||
it("validates gateway discover timeout", async () => {
|
||||
resetRuntimeCapture();
|
||||
discoverGatewayBeacons.mockClear();
|
||||
await expectGatewayExit(["gateway", "discover", "--timeout", "0"]);
|
||||
|
||||
expect(runtimeErrors.join("\n")).toContain("gateway discover failed:");
|
||||
expect(discoverGatewayBeacons).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("fails gateway call on invalid params JSON", async () => {
|
||||
resetRuntimeCapture();
|
||||
callGateway.mockClear();
|
||||
await expectGatewayExit(["gateway", "call", "status", "--params", "not-json"]);
|
||||
|
||||
expect(callGateway).not.toHaveBeenCalled();
|
||||
expect(runtimeErrors.join("\n")).toContain("Gateway call failed:");
|
||||
});
|
||||
|
||||
it("validates gateway ports and handles force/start errors", async () => {
|
||||
resetRuntimeCapture();
|
||||
|
||||
// Invalid port
|
||||
await expectGatewayExit(["gateway", "--port", "0", "--token", "test-token"]);
|
||||
|
||||
// Force free failure
|
||||
forceFreePortAndWait.mockImplementationOnce(async () => {
|
||||
throw new Error("boom");
|
||||
});
|
||||
await expectGatewayExit([
|
||||
"gateway",
|
||||
"--port",
|
||||
"18789",
|
||||
"--token",
|
||||
"test-token",
|
||||
"--force",
|
||||
"--allow-unconfigured",
|
||||
]);
|
||||
|
||||
// Start failure (generic)
|
||||
startGatewayServer.mockRejectedValueOnce(new Error("nope"));
|
||||
const beforeSigterm = new Set(process.listeners("SIGTERM"));
|
||||
const beforeSigint = new Set(process.listeners("SIGINT"));
|
||||
await expectGatewayExit([
|
||||
"gateway",
|
||||
"--port",
|
||||
"18789",
|
||||
"--token",
|
||||
"test-token",
|
||||
"--allow-unconfigured",
|
||||
]);
|
||||
for (const listener of process.listeners("SIGTERM")) {
|
||||
if (!beforeSigterm.has(listener)) {
|
||||
process.removeListener("SIGTERM", listener);
|
||||
}
|
||||
}
|
||||
for (const listener of process.listeners("SIGINT")) {
|
||||
if (!beforeSigint.has(listener)) {
|
||||
process.removeListener("SIGINT", listener);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it("prints stop hints on GatewayLockError when service is loaded", async () => {
|
||||
resetRuntimeCapture();
|
||||
serviceIsLoaded.mockResolvedValue(true);
|
||||
|
||||
const { GatewayLockError } = await import("../infra/gateway-lock.js");
|
||||
startGatewayServer.mockRejectedValueOnce(
|
||||
new GatewayLockError("another gateway instance is already listening"),
|
||||
);
|
||||
await expectGatewayExit(["gateway", "--token", "test-token", "--allow-unconfigured"]);
|
||||
|
||||
expect(startGatewayServer).toHaveBeenCalled();
|
||||
expect(runtimeErrors.join("\n")).toContain("Gateway failed to start:");
|
||||
expect(runtimeErrors.join("\n")).toContain("gateway stop");
|
||||
});
|
||||
|
||||
it("uses env/config port when --port is omitted", async () => {
|
||||
await withEnvOverride({ OPENCLAW_GATEWAY_PORT: "19001" }, async () => {
|
||||
resetRuntimeCapture();
|
||||
startGatewayServer.mockClear();
|
||||
|
||||
startGatewayServer.mockRejectedValueOnce(new Error("nope"));
|
||||
await expectGatewayExit(["gateway", "--token", "test-token", "--allow-unconfigured"]);
|
||||
|
||||
expect(startGatewayServer).toHaveBeenCalledWith(19001, expect.anything());
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user