mirror of https://github.com/openclaw/openclaw.git
95 lines
3.3 KiB
TypeScript
95 lines
3.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import type { ClawdbotConfig } from "../runtime-api.js";
|
|
import {
|
|
buildBroadcastSessionKey,
|
|
buildFeishuAgentBody,
|
|
resolveBroadcastAgents,
|
|
toMessageResourceType,
|
|
} from "./bot.js";
|
|
|
|
describe("buildFeishuAgentBody", () => {
|
|
it("builds message id, speaker, quoted content, mentions, and permission notice in order", () => {
|
|
const body = buildFeishuAgentBody({
|
|
ctx: {
|
|
content: "hello world",
|
|
senderName: "Sender Name",
|
|
senderOpenId: "ou-sender",
|
|
messageId: "msg-42",
|
|
mentionTargets: [{ openId: "ou-target", name: "Target User", key: "@_user_1" }],
|
|
},
|
|
quotedContent: "previous message",
|
|
permissionErrorForAgent: {
|
|
code: 99991672,
|
|
message: "permission denied",
|
|
grantUrl: "https://open.feishu.cn/app/cli_test",
|
|
},
|
|
});
|
|
|
|
expect(body).toBe(
|
|
'[message_id: msg-42]\nSender Name: [Replying to: "previous message"]\n\nhello world\n\n[System: Your reply will automatically @mention: Target User. Do not write @xxx yourself.]\n\n[System: The bot encountered a Feishu API permission error. Please inform the user about this issue and provide the permission grant URL for the admin to authorize. Permission grant URL: https://open.feishu.cn/app/cli_test]',
|
|
);
|
|
});
|
|
});
|
|
|
|
describe("toMessageResourceType", () => {
|
|
it("maps image to image", () => {
|
|
expect(toMessageResourceType("image")).toBe("image");
|
|
});
|
|
|
|
it("maps audio to file", () => {
|
|
expect(toMessageResourceType("audio")).toBe("file");
|
|
});
|
|
|
|
it("maps video/file/sticker to file", () => {
|
|
expect(toMessageResourceType("video")).toBe("file");
|
|
expect(toMessageResourceType("file")).toBe("file");
|
|
expect(toMessageResourceType("sticker")).toBe("file");
|
|
});
|
|
});
|
|
|
|
describe("resolveBroadcastAgents", () => {
|
|
it("returns agent list when broadcast config has the peerId", () => {
|
|
const cfg: ClawdbotConfig = { broadcast: { oc_group123: ["susan", "main"] } };
|
|
expect(resolveBroadcastAgents(cfg, "oc_group123")).toEqual(["susan", "main"]);
|
|
});
|
|
|
|
it("returns null when no broadcast config", () => {
|
|
const cfg = {} as ClawdbotConfig;
|
|
expect(resolveBroadcastAgents(cfg, "oc_group123")).toBeNull();
|
|
});
|
|
|
|
it("returns null when peerId not in broadcast", () => {
|
|
const cfg: ClawdbotConfig = { broadcast: { oc_other: ["susan"] } };
|
|
expect(resolveBroadcastAgents(cfg, "oc_group123")).toBeNull();
|
|
});
|
|
|
|
it("returns null when agent list is empty", () => {
|
|
const cfg: ClawdbotConfig = { broadcast: { oc_group123: [] } };
|
|
expect(resolveBroadcastAgents(cfg, "oc_group123")).toBeNull();
|
|
});
|
|
});
|
|
|
|
describe("buildBroadcastSessionKey", () => {
|
|
it("replaces agent ID prefix in session key", () => {
|
|
expect(buildBroadcastSessionKey("agent:main:feishu:group:oc_group123", "main", "susan")).toBe(
|
|
"agent:susan:feishu:group:oc_group123",
|
|
);
|
|
});
|
|
|
|
it("handles compound peer IDs", () => {
|
|
expect(
|
|
buildBroadcastSessionKey(
|
|
"agent:main:feishu:group:oc_group123:sender:ou_user1",
|
|
"main",
|
|
"susan",
|
|
),
|
|
).toBe("agent:susan:feishu:group:oc_group123:sender:ou_user1");
|
|
});
|
|
|
|
it("returns base key unchanged when prefix does not match", () => {
|
|
expect(buildBroadcastSessionKey("custom:key:format", "main", "susan")).toBe(
|
|
"custom:key:format",
|
|
);
|
|
});
|
|
});
|