mirror of
https://github.com/moltbot/moltbot.git
synced 2026-05-05 04:48:17 +00:00
196 lines
6.7 KiB
TypeScript
196 lines
6.7 KiB
TypeScript
import "./reply.directive.directive-behavior.e2e-mocks.js";
|
|
import { describe, expect, it, vi } from "vitest";
|
|
import type { OpenClawConfig } from "../config/config.js";
|
|
import { loadSessionStore } from "../config/sessions.js";
|
|
import {
|
|
AUTHORIZED_WHATSAPP_COMMAND,
|
|
assertElevatedOffStatusReply,
|
|
installDirectiveBehaviorE2EHooks,
|
|
makeElevatedDirectiveConfig,
|
|
makeRestrictedElevatedDisabledConfig,
|
|
makeWhatsAppDirectiveConfig,
|
|
replyText,
|
|
runEmbeddedPiAgent,
|
|
sessionStorePath,
|
|
withTempHome,
|
|
} from "./reply.directive.directive-behavior.e2e-harness.js";
|
|
import { getReplyFromConfig } from "./reply.js";
|
|
|
|
const COMMAND_MESSAGE_BASE = {
|
|
From: "+1222",
|
|
To: "+1222",
|
|
CommandAuthorized: true,
|
|
} as const;
|
|
|
|
async function runCommand(
|
|
home: string,
|
|
body: string,
|
|
options: { defaults?: Record<string, unknown>; extra?: Record<string, unknown> } = {},
|
|
) {
|
|
const res = await getReplyFromConfig(
|
|
{ ...COMMAND_MESSAGE_BASE, Body: body },
|
|
{},
|
|
makeWhatsAppDirectiveConfig(
|
|
home,
|
|
{
|
|
model: "anthropic/claude-opus-4-5",
|
|
...options.defaults,
|
|
},
|
|
options.extra ?? {},
|
|
),
|
|
);
|
|
return replyText(res);
|
|
}
|
|
|
|
async function runElevatedCommand(home: string, body: string) {
|
|
return getReplyFromConfig(
|
|
{ ...AUTHORIZED_WHATSAPP_COMMAND, Body: body },
|
|
{},
|
|
makeElevatedDirectiveConfig(home),
|
|
);
|
|
}
|
|
|
|
describe("directive behavior", () => {
|
|
installDirectiveBehaviorE2EHooks();
|
|
|
|
it("shows current verbose level when /verbose has no argument", async () => {
|
|
await withTempHome(async (home) => {
|
|
const text = await runCommand(home, "/verbose", { defaults: { verboseDefault: "on" } });
|
|
expect(text).toContain("Current verbose level: on");
|
|
expect(text).toContain("Options: on, full, off.");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("shows current reasoning level when /reasoning has no argument", async () => {
|
|
await withTempHome(async (home) => {
|
|
const text = await runCommand(home, "/reasoning");
|
|
expect(text).toContain("Current reasoning level: off");
|
|
expect(text).toContain("Options: on, off, stream.");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("shows current elevated level when /elevated has no argument", async () => {
|
|
await withTempHome(async (home) => {
|
|
const res = await runElevatedCommand(home, "/elevated");
|
|
const text = replyText(res);
|
|
expect(text).toContain("Current elevated level: on");
|
|
expect(text).toContain("Options: on, off, ask, full.");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("shows current exec defaults when /exec has no argument", async () => {
|
|
await withTempHome(async (home) => {
|
|
const text = await runCommand(home, "/exec", {
|
|
extra: {
|
|
tools: {
|
|
exec: {
|
|
host: "gateway",
|
|
security: "allowlist",
|
|
ask: "always",
|
|
node: "mac-1",
|
|
},
|
|
},
|
|
},
|
|
});
|
|
expect(text).toContain(
|
|
"Current exec defaults: host=gateway, security=allowlist, ask=always, node=mac-1.",
|
|
);
|
|
expect(text).toContain(
|
|
"Options: host=sandbox|gateway|node, security=deny|allowlist|full, ask=off|on-miss|always, node=<id>.",
|
|
);
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("persists elevated off and reflects it in /status (even when default is on)", async () => {
|
|
await withTempHome(async (home) => {
|
|
const storePath = sessionStorePath(home);
|
|
const res = await runElevatedCommand(home, "/elevated off\n/status");
|
|
const text = replyText(res);
|
|
assertElevatedOffStatusReply(text);
|
|
|
|
const store = loadSessionStore(storePath);
|
|
expect(store["agent:main:main"]?.elevatedLevel).toBe("off");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("shows current elevated level as off after toggling it off", async () => {
|
|
await withTempHome(async (home) => {
|
|
await runElevatedCommand(home, "/elevated off");
|
|
const res = await runElevatedCommand(home, "/elevated");
|
|
const text = replyText(res);
|
|
expect(text).toContain("Current elevated level: off");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("can toggle elevated off then back on (status reflects on)", async () => {
|
|
await withTempHome(async (home) => {
|
|
const storePath = sessionStorePath(home);
|
|
await runElevatedCommand(home, "/elevated off");
|
|
await runElevatedCommand(home, "/elevated on");
|
|
const res = await runElevatedCommand(home, "/status");
|
|
const text = replyText(res);
|
|
const optionsLine = text?.split("\n").find((line) => line.trim().startsWith("⚙️"));
|
|
expect(optionsLine).toBeTruthy();
|
|
expect(optionsLine).toContain("elevated");
|
|
|
|
const store = loadSessionStore(storePath);
|
|
expect(store["agent:main:main"]?.elevatedLevel).toBe("on");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("rejects per-agent elevated when disabled", async () => {
|
|
await withTempHome(async (home) => {
|
|
const res = await getReplyFromConfig(
|
|
{
|
|
Body: "/elevated on",
|
|
From: "+1222",
|
|
To: "+1222",
|
|
Provider: "whatsapp",
|
|
SenderE164: "+1222",
|
|
SessionKey: "agent:restricted:main",
|
|
CommandAuthorized: true,
|
|
},
|
|
{},
|
|
makeRestrictedElevatedDisabledConfig(home) as unknown as OpenClawConfig,
|
|
);
|
|
|
|
const text = replyText(res);
|
|
expect(text).toContain("agents.list[].tools.elevated.enabled");
|
|
expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
it("strips inline elevated directives from the user text (does not persist session override)", async () => {
|
|
await withTempHome(async (home) => {
|
|
vi.mocked(runEmbeddedPiAgent).mockResolvedValue({
|
|
payloads: [{ text: "ok" }],
|
|
meta: {
|
|
durationMs: 1,
|
|
agentMeta: { sessionId: "s", provider: "p", model: "m" },
|
|
},
|
|
});
|
|
const storePath = sessionStorePath(home);
|
|
|
|
await getReplyFromConfig(
|
|
{
|
|
Body: "hello there /elevated off",
|
|
From: "+1222",
|
|
To: "+1222",
|
|
Provider: "whatsapp",
|
|
SenderE164: "+1222",
|
|
},
|
|
{},
|
|
makeElevatedDirectiveConfig(home),
|
|
);
|
|
|
|
const store = loadSessionStore(storePath);
|
|
expect(store["agent:main:main"]?.elevatedLevel).toBeUndefined();
|
|
|
|
const calls = vi.mocked(runEmbeddedPiAgent).mock.calls;
|
|
expect(calls.length).toBeGreaterThan(0);
|
|
const call = calls[0]?.[0];
|
|
expect(call?.prompt).toContain("hello there");
|
|
expect(call?.prompt).not.toContain("/elevated");
|
|
});
|
|
});
|
|
});
|