196 lines
6.1 KiB
TypeScript
196 lines
6.1 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import {
|
|
parseApprovalValue,
|
|
getExecApprovalActionIdPrefix,
|
|
SlackExecApprovalHandler,
|
|
type ExecApprovalRequest,
|
|
} from "./exec-approvals.js";
|
|
import type { SlackExecApprovalConfig } from "../../config/types.slack.js";
|
|
|
|
// Helper to encode approval values (mirrors internal implementation)
|
|
function encodeApprovalValue(
|
|
approvalId: string,
|
|
action: "allow-once" | "allow-always" | "deny",
|
|
): string {
|
|
return ["execapproval", encodeURIComponent(approvalId), action].join("|");
|
|
}
|
|
|
|
describe("getExecApprovalActionIdPrefix", () => {
|
|
it("returns the action ID prefix", () => {
|
|
expect(getExecApprovalActionIdPrefix()).toBe("clawdbot_execapproval");
|
|
});
|
|
});
|
|
|
|
describe("parseApprovalValue", () => {
|
|
it("parses valid value", () => {
|
|
const value = encodeApprovalValue("abc-123", "allow-once");
|
|
const result = parseApprovalValue(value);
|
|
expect(result).toEqual({ approvalId: "abc-123", action: "allow-once" });
|
|
});
|
|
|
|
it("parses encoded approval id", () => {
|
|
const value = encodeApprovalValue("abc|123", "allow-always");
|
|
const result = parseApprovalValue(value);
|
|
expect(result).toEqual({ approvalId: "abc|123", action: "allow-always" });
|
|
});
|
|
|
|
it("handles special characters", () => {
|
|
const value = encodeApprovalValue("test=approval&id", "deny");
|
|
const result = parseApprovalValue(value);
|
|
expect(result).toEqual({ approvalId: "test=approval&id", action: "deny" });
|
|
});
|
|
|
|
it("rejects invalid action", () => {
|
|
const value = "execapproval|abc-123|invalid";
|
|
const result = parseApprovalValue(value);
|
|
expect(result).toBeNull();
|
|
});
|
|
|
|
it("rejects missing parts", () => {
|
|
expect(parseApprovalValue("execapproval|abc-123")).toBeNull();
|
|
expect(parseApprovalValue("execapproval")).toBeNull();
|
|
expect(parseApprovalValue("")).toBeNull();
|
|
});
|
|
|
|
it("rejects undefined input", () => {
|
|
expect(parseApprovalValue(undefined)).toBeNull();
|
|
});
|
|
|
|
it("rejects wrong prefix", () => {
|
|
const value = "wrongprefix|abc-123|allow-once";
|
|
const result = parseApprovalValue(value);
|
|
expect(result).toBeNull();
|
|
});
|
|
|
|
it("accepts all valid actions", () => {
|
|
expect(parseApprovalValue(encodeApprovalValue("x", "allow-once"))?.action).toBe("allow-once");
|
|
expect(parseApprovalValue(encodeApprovalValue("x", "allow-always"))?.action).toBe(
|
|
"allow-always",
|
|
);
|
|
expect(parseApprovalValue(encodeApprovalValue("x", "deny"))?.action).toBe("deny");
|
|
});
|
|
});
|
|
|
|
describe("roundtrip encoding", () => {
|
|
it("encodes and decodes correctly", () => {
|
|
const approvalId = "test-approval-with|special&chars";
|
|
const action = "allow-always" as const;
|
|
const encoded = encodeApprovalValue(approvalId, action);
|
|
const result = parseApprovalValue(encoded);
|
|
expect(result).toEqual({ approvalId, action });
|
|
});
|
|
});
|
|
|
|
describe("SlackExecApprovalHandler.shouldHandle", () => {
|
|
function createHandler(config: SlackExecApprovalConfig) {
|
|
// Create a minimal mock WebClient
|
|
const mockClient = {} as any;
|
|
return new SlackExecApprovalHandler({
|
|
client: mockClient,
|
|
accountId: "default",
|
|
config,
|
|
cfg: {},
|
|
});
|
|
}
|
|
|
|
function createRequest(
|
|
overrides: Partial<ExecApprovalRequest["request"]> = {},
|
|
): ExecApprovalRequest {
|
|
return {
|
|
id: "test-id",
|
|
request: {
|
|
command: "echo hello",
|
|
cwd: "/home/user",
|
|
host: "gateway",
|
|
agentId: "test-agent",
|
|
sessionKey: "agent:test-agent:slack:123",
|
|
...overrides,
|
|
},
|
|
createdAtMs: Date.now(),
|
|
expiresAtMs: Date.now() + 60000,
|
|
};
|
|
}
|
|
|
|
it("returns false when disabled", () => {
|
|
const handler = createHandler({ enabled: false, approvers: ["U123"] });
|
|
expect(handler.shouldHandle(createRequest())).toBe(false);
|
|
});
|
|
|
|
it("returns false when no approvers", () => {
|
|
const handler = createHandler({ enabled: true, approvers: [] });
|
|
expect(handler.shouldHandle(createRequest())).toBe(false);
|
|
});
|
|
|
|
it("returns true with minimal config", () => {
|
|
const handler = createHandler({ enabled: true, approvers: ["U123"] });
|
|
expect(handler.shouldHandle(createRequest())).toBe(true);
|
|
});
|
|
|
|
it("filters by agent ID", () => {
|
|
const handler = createHandler({
|
|
enabled: true,
|
|
approvers: ["U123"],
|
|
agentFilter: ["allowed-agent"],
|
|
});
|
|
expect(handler.shouldHandle(createRequest({ agentId: "allowed-agent" }))).toBe(true);
|
|
expect(handler.shouldHandle(createRequest({ agentId: "other-agent" }))).toBe(false);
|
|
expect(handler.shouldHandle(createRequest({ agentId: null }))).toBe(false);
|
|
});
|
|
|
|
it("filters by session key substring", () => {
|
|
const handler = createHandler({
|
|
enabled: true,
|
|
approvers: ["U123"],
|
|
sessionFilter: ["slack"],
|
|
});
|
|
expect(handler.shouldHandle(createRequest({ sessionKey: "agent:test:slack:123" }))).toBe(true);
|
|
expect(handler.shouldHandle(createRequest({ sessionKey: "agent:test:telegram:123" }))).toBe(
|
|
false,
|
|
);
|
|
expect(handler.shouldHandle(createRequest({ sessionKey: null }))).toBe(false);
|
|
});
|
|
|
|
it("filters by session key regex", () => {
|
|
const handler = createHandler({
|
|
enabled: true,
|
|
approvers: ["U123"],
|
|
sessionFilter: ["^agent:.*:slack:"],
|
|
});
|
|
expect(handler.shouldHandle(createRequest({ sessionKey: "agent:test:slack:123" }))).toBe(true);
|
|
expect(handler.shouldHandle(createRequest({ sessionKey: "other:test:slack:123" }))).toBe(false);
|
|
});
|
|
|
|
it("combines agent and session filters", () => {
|
|
const handler = createHandler({
|
|
enabled: true,
|
|
approvers: ["U123"],
|
|
agentFilter: ["my-agent"],
|
|
sessionFilter: ["slack"],
|
|
});
|
|
expect(
|
|
handler.shouldHandle(
|
|
createRequest({
|
|
agentId: "my-agent",
|
|
sessionKey: "agent:my-agent:slack:123",
|
|
}),
|
|
),
|
|
).toBe(true);
|
|
expect(
|
|
handler.shouldHandle(
|
|
createRequest({
|
|
agentId: "other-agent",
|
|
sessionKey: "agent:other:slack:123",
|
|
}),
|
|
),
|
|
).toBe(false);
|
|
expect(
|
|
handler.shouldHandle(
|
|
createRequest({
|
|
agentId: "my-agent",
|
|
sessionKey: "agent:my-agent:telegram:123",
|
|
}),
|
|
),
|
|
).toBe(false);
|
|
});
|
|
});
|