import { join } from "node:path"; import { afterEach, describe, expect, it, vi } from "vitest"; import { normalizeTestText } from "../../test/helpers/normalize-text.js"; import { withTempHome as withTempHomeBase } from "../../test/helpers/temp-home.js"; vi.mock("../agents/pi-embedded.js", () => ({ abortEmbeddedPiRun: vi.fn().mockReturnValue(false), compactEmbeddedPiSession: vi.fn(), runEmbeddedPiAgent: vi.fn(), queueEmbeddedPiMessage: vi.fn().mockReturnValue(false), resolveEmbeddedSessionLane: (key: string) => `session:${key.trim() || "main"}`, isEmbeddedPiRunActive: vi.fn().mockReturnValue(false), isEmbeddedPiRunStreaming: vi.fn().mockReturnValue(false), })); const usageMocks = vi.hoisted(() => ({ loadProviderUsageSummary: vi.fn().mockResolvedValue({ updatedAt: 0, providers: [], }), formatUsageSummaryLine: vi.fn().mockReturnValue("📊 Usage: Claude 80% left"), resolveUsageProviderId: vi.fn((provider: string) => provider.split("/")[0]), })); vi.mock("../infra/provider-usage.js", () => usageMocks); const modelCatalogMocks = vi.hoisted(() => ({ loadModelCatalog: vi.fn().mockResolvedValue([ { provider: "anthropic", id: "claude-opus-4-5", name: "Claude Opus 4.5", contextWindow: 200000, }, { provider: "openrouter", id: "anthropic/claude-opus-4-5", name: "Claude Opus 4.5 (OpenRouter)", contextWindow: 200000, }, { provider: "openai", id: "gpt-4.1-mini", name: "GPT-4.1 mini" }, { provider: "openai", id: "gpt-5.2", name: "GPT-5.2" }, { provider: "openai-codex", id: "gpt-5.2", name: "GPT-5.2 (Codex)" }, { provider: "minimax", id: "MiniMax-M2.1", name: "MiniMax M2.1" }, ]), resetModelCatalogCacheForTest: vi.fn(), })); vi.mock("../agents/model-catalog.js", () => modelCatalogMocks); import { abortEmbeddedPiRun, runEmbeddedPiAgent } from "../agents/pi-embedded.js"; import { getReplyFromConfig } from "./reply.js"; const _MAIN_SESSION_KEY = "agent:main:main"; const webMocks = vi.hoisted(() => ({ webAuthExists: vi.fn().mockResolvedValue(true), getWebAuthAgeMs: vi.fn().mockReturnValue(120_000), readWebSelfId: vi.fn().mockReturnValue({ e164: "+1999" }), })); vi.mock("../web/session.js", () => webMocks); async function withTempHome(fn: (home: string) => Promise): Promise { return withTempHomeBase( async (home) => { vi.mocked(runEmbeddedPiAgent).mockClear(); vi.mocked(abortEmbeddedPiRun).mockClear(); return await fn(home); }, { prefix: "clawdbot-triggers-" }, ); } function makeCfg(home: string) { return { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: join(home, "clawd"), }, }, channels: { whatsapp: { allowFrom: ["*"], }, }, session: { store: join(home, "sessions.json") }, }; } afterEach(() => { vi.restoreAllMocks(); }); describe("trigger handling", () => { it("filters usage summary to the current model provider", async () => { await withTempHome(async (home) => { usageMocks.loadProviderUsageSummary.mockClear(); const res = await getReplyFromConfig( { Body: "/status", From: "+1000", To: "+2000", Provider: "whatsapp", SenderE164: "+1000", }, {}, makeCfg(home), ); const text = Array.isArray(res) ? res[0]?.text : res?.text; expect(normalizeTestText(text ?? "")).toContain("Usage: Claude 80% left"); expect(usageMocks.loadProviderUsageSummary).toHaveBeenCalledWith( expect.objectContaining({ providers: ["anthropic"] }), ); }); }); it("emits /status once (no duplicate inline + final)", async () => { await withTempHome(async (home) => { const blockReplies: Array<{ text?: string }> = []; const res = await getReplyFromConfig( { Body: "/status", From: "+1000", To: "+2000", Provider: "whatsapp", SenderE164: "+1000", }, { onBlockReply: async (payload) => { blockReplies.push(payload); }, }, makeCfg(home), ); const replies = res ? (Array.isArray(res) ? res : [res]) : []; expect(blockReplies.length).toBe(0); expect(replies.length).toBe(1); expect(String(replies[0]?.text ?? "")).toContain("Model:"); }); }); it("emits /usage once (alias of /status)", async () => { await withTempHome(async (home) => { const blockReplies: Array<{ text?: string }> = []; const res = await getReplyFromConfig( { Body: "/usage", From: "+1000", To: "+2000", Provider: "whatsapp", SenderE164: "+1000", }, { onBlockReply: async (payload) => { blockReplies.push(payload); }, }, makeCfg(home), ); const replies = res ? (Array.isArray(res) ? res : [res]) : []; expect(blockReplies.length).toBe(0); expect(replies.length).toBe(1); expect(String(replies[0]?.text ?? "")).toContain("Model:"); }); }); it("sends one inline status and still returns agent reply for mixed text", async () => { await withTempHome(async (home) => { vi.mocked(runEmbeddedPiAgent).mockResolvedValue({ payloads: [{ text: "agent says hi" }], meta: { durationMs: 1, agentMeta: { sessionId: "s", provider: "p", model: "m" }, }, }); const blockReplies: Array<{ text?: string }> = []; const res = await getReplyFromConfig( { Body: "here we go /status now", From: "+1002", To: "+2000", Provider: "whatsapp", SenderE164: "+1002", }, { onBlockReply: async (payload) => { blockReplies.push(payload); }, }, makeCfg(home), ); const replies = res ? (Array.isArray(res) ? res : [res]) : []; expect(blockReplies.length).toBe(1); expect(String(blockReplies[0]?.text ?? "")).toContain("Model:"); expect(replies.length).toBe(1); expect(replies[0]?.text).toBe("agent says hi"); const prompt = vi.mocked(runEmbeddedPiAgent).mock.calls[0]?.[0]?.prompt ?? ""; expect(prompt).not.toContain("/status"); }); }); it("aborts even with timestamp prefix", async () => { await withTempHome(async (home) => { const res = await getReplyFromConfig( { Body: "[Dec 5 10:00] stop", From: "+1000", To: "+2000", }, {}, makeCfg(home), ); const text = Array.isArray(res) ? res[0]?.text : res?.text; expect(text).toBe("⚙️ Agent was aborted."); expect(runEmbeddedPiAgent).not.toHaveBeenCalled(); }); }); it("handles /stop without invoking the agent", async () => { await withTempHome(async (home) => { const res = await getReplyFromConfig( { Body: "/stop", From: "+1003", To: "+2000", }, {}, makeCfg(home), ); const text = Array.isArray(res) ? res[0]?.text : res?.text; expect(text).toBe("⚙️ Agent was aborted."); expect(runEmbeddedPiAgent).not.toHaveBeenCalled(); }); }); });