88 lines
2.6 KiB
TypeScript
88 lines
2.6 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
|
|
import { createClawdbotTools } from "./clawdbot-tools.js";
|
|
|
|
vi.mock("./tools/gateway.js", () => ({
|
|
callGatewayTool: vi.fn(async () => ({ ok: true })),
|
|
}));
|
|
|
|
describe("gateway tool", () => {
|
|
it("schedules SIGUSR1 restart", async () => {
|
|
vi.useFakeTimers();
|
|
const kill = vi.spyOn(process, "kill").mockImplementation(() => true);
|
|
|
|
try {
|
|
const tool = createClawdbotTools({
|
|
config: { commands: { restart: true } },
|
|
}).find((candidate) => candidate.name === "gateway");
|
|
expect(tool).toBeDefined();
|
|
if (!tool) throw new Error("missing gateway tool");
|
|
|
|
const result = await tool.execute("call1", {
|
|
action: "restart",
|
|
delayMs: 0,
|
|
});
|
|
expect(result.details).toMatchObject({
|
|
ok: true,
|
|
pid: process.pid,
|
|
signal: "SIGUSR1",
|
|
delayMs: 0,
|
|
});
|
|
|
|
expect(kill).not.toHaveBeenCalled();
|
|
await vi.runAllTimersAsync();
|
|
expect(kill).toHaveBeenCalledWith(process.pid, "SIGUSR1");
|
|
} finally {
|
|
kill.mockRestore();
|
|
vi.useRealTimers();
|
|
}
|
|
});
|
|
|
|
it("passes config.apply through gateway call", async () => {
|
|
const { callGatewayTool } = await import("./tools/gateway.js");
|
|
const tool = createClawdbotTools({
|
|
agentSessionKey: "agent:main:whatsapp:dm:+15555550123",
|
|
}).find((candidate) => candidate.name === "gateway");
|
|
expect(tool).toBeDefined();
|
|
if (!tool) throw new Error("missing gateway tool");
|
|
|
|
const raw = '{\n agents: { defaults: { workspace: "~/clawd" } }\n}\n';
|
|
await tool.execute("call2", {
|
|
action: "config.apply",
|
|
raw,
|
|
});
|
|
|
|
expect(callGatewayTool).toHaveBeenCalledWith(
|
|
"config.apply",
|
|
expect.any(Object),
|
|
expect.objectContaining({
|
|
raw: raw.trim(),
|
|
sessionKey: "agent:main:whatsapp:dm:+15555550123",
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("passes update.run through gateway call", async () => {
|
|
const { callGatewayTool } = await import("./tools/gateway.js");
|
|
const tool = createClawdbotTools({
|
|
agentSessionKey: "agent:main:whatsapp:dm:+15555550123",
|
|
}).find((candidate) => candidate.name === "gateway");
|
|
expect(tool).toBeDefined();
|
|
if (!tool) throw new Error("missing gateway tool");
|
|
|
|
await tool.execute("call3", {
|
|
action: "update.run",
|
|
note: "test update",
|
|
});
|
|
|
|
expect(callGatewayTool).toHaveBeenCalledWith(
|
|
"update.run",
|
|
expect.any(Object),
|
|
expect.objectContaining({
|
|
note: "test update",
|
|
sessionKey: "agent:main:whatsapp:dm:+15555550123",
|
|
}),
|
|
);
|
|
});
|
|
});
|