154 lines
4.8 KiB
TypeScript
154 lines
4.8 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
import type { OpenClawConfig } from "../../config/config.js";
|
|
import type { MsgContext } from "../templating.js";
|
|
import { callGateway } from "../../gateway/call.js";
|
|
import { buildCommandContext, handleCommands } from "./commands.js";
|
|
import { parseInlineDirectives } from "./directive-handling.js";
|
|
|
|
vi.mock("../../gateway/call.js", () => ({
|
|
callGateway: vi.fn(),
|
|
}));
|
|
|
|
function buildParams(commandBody: string, cfg: OpenClawConfig, ctxOverrides?: Partial<MsgContext>) {
|
|
const ctx = {
|
|
Body: commandBody,
|
|
CommandBody: commandBody,
|
|
CommandSource: "text",
|
|
CommandAuthorized: true,
|
|
Provider: "whatsapp",
|
|
Surface: "whatsapp",
|
|
...ctxOverrides,
|
|
} as MsgContext;
|
|
|
|
const command = buildCommandContext({
|
|
ctx,
|
|
cfg,
|
|
isGroup: false,
|
|
triggerBodyNormalized: commandBody.trim().toLowerCase(),
|
|
commandAuthorized: true,
|
|
});
|
|
|
|
return {
|
|
ctx,
|
|
cfg,
|
|
command,
|
|
directives: parseInlineDirectives(commandBody),
|
|
elevated: { enabled: true, allowed: true, failures: [] },
|
|
sessionKey: "agent:main:main",
|
|
workspaceDir: "/tmp",
|
|
defaultGroupActivation: () => "mention",
|
|
resolvedVerboseLevel: "off" as const,
|
|
resolvedReasoningLevel: "off" as const,
|
|
resolveDefaultThinkingLevel: async () => undefined,
|
|
provider: "whatsapp",
|
|
model: "test-model",
|
|
contextTokens: 0,
|
|
isGroup: false,
|
|
};
|
|
}
|
|
|
|
describe("/approve command", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
it("rejects invalid usage", async () => {
|
|
const cfg = {
|
|
commands: { text: true },
|
|
channels: { whatsapp: { allowFrom: ["*"] } },
|
|
} as OpenClawConfig;
|
|
const params = buildParams("/approve", cfg);
|
|
const result = await handleCommands(params);
|
|
expect(result.shouldContinue).toBe(false);
|
|
expect(result.reply?.text).toContain("Usage: /approve");
|
|
});
|
|
|
|
it("submits approval", async () => {
|
|
const cfg = {
|
|
commands: { text: true },
|
|
channels: { whatsapp: { allowFrom: ["*"] } },
|
|
} as OpenClawConfig;
|
|
const params = buildParams("/approve abc allow-once", cfg, { SenderId: "123" });
|
|
|
|
const mockCallGateway = vi.mocked(callGateway);
|
|
mockCallGateway.mockResolvedValueOnce({ ok: true });
|
|
|
|
const result = await handleCommands(params);
|
|
expect(result.shouldContinue).toBe(false);
|
|
expect(result.reply?.text).toContain("Exec approval allow-once submitted");
|
|
expect(mockCallGateway).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
method: "exec.approval.resolve",
|
|
params: { id: "abc", decision: "allow-once" },
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("rejects gateway clients without approvals scope", async () => {
|
|
const cfg = {
|
|
commands: { text: true },
|
|
} as OpenClawConfig;
|
|
const params = buildParams("/approve abc allow-once", cfg, {
|
|
Provider: "webchat",
|
|
Surface: "webchat",
|
|
GatewayClientScopes: ["operator.write"],
|
|
});
|
|
|
|
const mockCallGateway = vi.mocked(callGateway);
|
|
mockCallGateway.mockResolvedValueOnce({ ok: true });
|
|
|
|
const result = await handleCommands(params);
|
|
expect(result.shouldContinue).toBe(false);
|
|
expect(result.reply?.text).toContain("requires operator.approvals");
|
|
expect(mockCallGateway).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("allows gateway clients with approvals scope", async () => {
|
|
const cfg = {
|
|
commands: { text: true },
|
|
} as OpenClawConfig;
|
|
const params = buildParams("/approve abc allow-once", cfg, {
|
|
Provider: "webchat",
|
|
Surface: "webchat",
|
|
GatewayClientScopes: ["operator.approvals"],
|
|
});
|
|
|
|
const mockCallGateway = vi.mocked(callGateway);
|
|
mockCallGateway.mockResolvedValueOnce({ ok: true });
|
|
|
|
const result = await handleCommands(params);
|
|
expect(result.shouldContinue).toBe(false);
|
|
expect(result.reply?.text).toContain("Exec approval allow-once submitted");
|
|
expect(mockCallGateway).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
method: "exec.approval.resolve",
|
|
params: { id: "abc", decision: "allow-once" },
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("allows gateway clients with admin scope", async () => {
|
|
const cfg = {
|
|
commands: { text: true },
|
|
} as OpenClawConfig;
|
|
const params = buildParams("/approve abc allow-once", cfg, {
|
|
Provider: "webchat",
|
|
Surface: "webchat",
|
|
GatewayClientScopes: ["operator.admin"],
|
|
});
|
|
|
|
const mockCallGateway = vi.mocked(callGateway);
|
|
mockCallGateway.mockResolvedValueOnce({ ok: true });
|
|
|
|
const result = await handleCommands(params);
|
|
expect(result.shouldContinue).toBe(false);
|
|
expect(result.reply?.text).toContain("Exec approval allow-once submitted");
|
|
expect(mockCallGateway).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
method: "exec.approval.resolve",
|
|
params: { id: "abc", decision: "allow-once" },
|
|
}),
|
|
);
|
|
});
|
|
});
|