mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-10 22:54:31 +00:00
test: move command status and health suites out of e2e
This commit is contained in:
189
src/commands/health.test.ts
Normal file
189
src/commands/health.test.ts
Normal file
@@ -0,0 +1,189 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { stripAnsi } from "../terminal/ansi.js";
|
||||
import { formatHealthCheckFailure } from "./health-format.js";
|
||||
import type { HealthSummary } from "./health.js";
|
||||
import { formatHealthChannelLines, healthCommand } from "./health.js";
|
||||
|
||||
const runtime = {
|
||||
log: vi.fn(),
|
||||
error: vi.fn(),
|
||||
exit: vi.fn(),
|
||||
};
|
||||
|
||||
const defaultSessions: HealthSummary["sessions"] = {
|
||||
path: "/tmp/sessions.json",
|
||||
count: 0,
|
||||
recent: [],
|
||||
};
|
||||
|
||||
const createMainAgentSummary = (sessions = defaultSessions) => ({
|
||||
agentId: "main",
|
||||
isDefault: true,
|
||||
heartbeat: {
|
||||
enabled: true,
|
||||
every: "1m",
|
||||
everyMs: 60_000,
|
||||
prompt: "hi",
|
||||
target: "last",
|
||||
ackMaxChars: 160,
|
||||
},
|
||||
sessions,
|
||||
});
|
||||
|
||||
const createHealthSummary = (params: {
|
||||
channels: HealthSummary["channels"];
|
||||
channelOrder: string[];
|
||||
channelLabels: HealthSummary["channelLabels"];
|
||||
sessions?: HealthSummary["sessions"];
|
||||
}): HealthSummary => {
|
||||
const sessions = params.sessions ?? defaultSessions;
|
||||
return {
|
||||
ok: true,
|
||||
ts: Date.now(),
|
||||
durationMs: 5,
|
||||
channels: params.channels,
|
||||
channelOrder: params.channelOrder,
|
||||
channelLabels: params.channelLabels,
|
||||
heartbeatSeconds: 60,
|
||||
defaultAgentId: "main",
|
||||
agents: [createMainAgentSummary(sessions)],
|
||||
sessions,
|
||||
};
|
||||
};
|
||||
|
||||
const callGatewayMock = vi.fn();
|
||||
vi.mock("../gateway/call.js", () => ({
|
||||
callGateway: (...args: unknown[]) => callGatewayMock(...args),
|
||||
}));
|
||||
|
||||
describe("healthCommand", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("outputs JSON from gateway", async () => {
|
||||
const agentSessions = {
|
||||
path: "/tmp/sessions.json",
|
||||
count: 1,
|
||||
recent: [{ key: "+1555", updatedAt: Date.now(), age: 0 }],
|
||||
};
|
||||
const snapshot = createHealthSummary({
|
||||
channels: {
|
||||
whatsapp: { accountId: "default", linked: true, authAgeMs: 5000 },
|
||||
telegram: {
|
||||
accountId: "default",
|
||||
configured: true,
|
||||
probe: { ok: true, elapsedMs: 1 },
|
||||
},
|
||||
discord: { accountId: "default", configured: false },
|
||||
},
|
||||
channelOrder: ["whatsapp", "telegram", "discord"],
|
||||
channelLabels: {
|
||||
whatsapp: "WhatsApp",
|
||||
telegram: "Telegram",
|
||||
discord: "Discord",
|
||||
},
|
||||
sessions: agentSessions,
|
||||
});
|
||||
callGatewayMock.mockResolvedValueOnce(snapshot);
|
||||
|
||||
await healthCommand({ json: true, timeoutMs: 5000 }, runtime as never);
|
||||
|
||||
expect(runtime.exit).not.toHaveBeenCalled();
|
||||
const logged = runtime.log.mock.calls[0]?.[0] as string;
|
||||
const parsed = JSON.parse(logged) as HealthSummary;
|
||||
expect(parsed.channels.whatsapp?.linked).toBe(true);
|
||||
expect(parsed.channels.telegram?.configured).toBe(true);
|
||||
expect(parsed.sessions.count).toBe(1);
|
||||
});
|
||||
|
||||
it("prints text summary when not json", async () => {
|
||||
callGatewayMock.mockResolvedValueOnce(
|
||||
createHealthSummary({
|
||||
channels: {
|
||||
whatsapp: { accountId: "default", linked: false, authAgeMs: null },
|
||||
telegram: { accountId: "default", configured: false },
|
||||
discord: { accountId: "default", configured: false },
|
||||
},
|
||||
channelOrder: ["whatsapp", "telegram", "discord"],
|
||||
channelLabels: {
|
||||
whatsapp: "WhatsApp",
|
||||
telegram: "Telegram",
|
||||
discord: "Discord",
|
||||
},
|
||||
}),
|
||||
);
|
||||
|
||||
await healthCommand({ json: false }, runtime as never);
|
||||
|
||||
expect(runtime.exit).not.toHaveBeenCalled();
|
||||
expect(runtime.log).toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("formats per-account probe timings", () => {
|
||||
const summary = createHealthSummary({
|
||||
channels: {
|
||||
telegram: {
|
||||
accountId: "main",
|
||||
configured: true,
|
||||
probe: { ok: true, elapsedMs: 196, bot: { username: "pinguini_ugi_bot" } },
|
||||
accounts: {
|
||||
main: {
|
||||
accountId: "main",
|
||||
configured: true,
|
||||
probe: { ok: true, elapsedMs: 196, bot: { username: "pinguini_ugi_bot" } },
|
||||
},
|
||||
flurry: {
|
||||
accountId: "flurry",
|
||||
configured: true,
|
||||
probe: { ok: true, elapsedMs: 190, bot: { username: "flurry_ugi_bot" } },
|
||||
},
|
||||
poe: {
|
||||
accountId: "poe",
|
||||
configured: true,
|
||||
probe: { ok: true, elapsedMs: 188, bot: { username: "poe_ugi_bot" } },
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
channelOrder: ["telegram"],
|
||||
channelLabels: { telegram: "Telegram" },
|
||||
});
|
||||
|
||||
const lines = formatHealthChannelLines(summary, { accountMode: "all" });
|
||||
expect(lines).toContain(
|
||||
"Telegram: ok (@pinguini_ugi_bot:main:196ms, @flurry_ugi_bot:flurry:190ms, @poe_ugi_bot:poe:188ms)",
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
describe("formatHealthCheckFailure", () => {
|
||||
it("keeps non-rich output stable", () => {
|
||||
const err = new Error("gateway closed (1006 abnormal closure): no close reason");
|
||||
expect(formatHealthCheckFailure(err, { rich: false })).toBe(
|
||||
`Health check failed: ${String(err)}`,
|
||||
);
|
||||
});
|
||||
|
||||
it("formats gateway connection details as indented key/value lines", () => {
|
||||
const err = new Error(
|
||||
[
|
||||
"gateway closed (1006 abnormal closure (no close frame)): no close reason",
|
||||
"Gateway target: ws://127.0.0.1:19001",
|
||||
"Source: local loopback",
|
||||
"Config: /Users/steipete/.openclaw-dev/openclaw.json",
|
||||
"Bind: loopback",
|
||||
].join("\n"),
|
||||
);
|
||||
|
||||
expect(stripAnsi(formatHealthCheckFailure(err, { rich: true }))).toBe(
|
||||
[
|
||||
"Health check failed: gateway closed (1006 abnormal closure (no close frame)): no close reason",
|
||||
" Gateway target: ws://127.0.0.1:19001",
|
||||
" Source: local loopback",
|
||||
" Config: /Users/steipete/.openclaw-dev/openclaw.json",
|
||||
" Bind: loopback",
|
||||
].join("\n"),
|
||||
);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user