import { beforeEach, describe, expect, it, vi } from "vitest"; const messageCommandMock = vi.fn(async () => {}); vi.mock("../../../commands/message.js", () => ({ messageCommand: messageCommandMock, })); vi.mock("../../../globals.js", () => ({ danger: (s: string) => s, setVerbose: vi.fn(), })); vi.mock("../../plugin-registry.js", () => ({ ensurePluginRegistryLoaded: vi.fn(), })); const hasHooksMock = vi.fn((_hookName: string) => false); const runGatewayStopMock = vi.fn( async (_event: { reason?: string }, _ctx: Record) => {}, ); const runGlobalGatewayStopSafelyMock = vi.fn( async (params: { event: { reason?: string }; ctx: Record; onError?: (err: unknown) => void; }) => { if (!hasHooksMock("gateway_stop")) { return; } try { await runGatewayStopMock(params.event, params.ctx); } catch (err) { params.onError?.(err); } }, ); vi.mock("../../../plugins/hook-runner-global.js", () => ({ runGlobalGatewayStopSafely: runGlobalGatewayStopSafelyMock, })); const exitMock = vi.fn((): never => { throw new Error("exit"); }); const errorMock = vi.fn(); const runtimeMock = { log: vi.fn(), error: errorMock, exit: exitMock }; vi.mock("../../../runtime.js", () => ({ defaultRuntime: runtimeMock, })); vi.mock("../../deps.js", () => ({ createDefaultDeps: () => ({}), })); const { createMessageCliHelpers } = await import("./helpers.js"); const baseSendOptions = { channel: "discord", target: "123", message: "hi", }; function createRunMessageAction() { const fakeCommand = { help: vi.fn() } as never; return createMessageCliHelpers(fakeCommand, "discord").runMessageAction; } async function runSendAction(opts: Record = {}) { const runMessageAction = createRunMessageAction(); await expect(runMessageAction("send", { ...baseSendOptions, ...opts })).rejects.toThrow("exit"); } function expectNoAccountFieldInPassedOptions() { const passedOpts = ( messageCommandMock.mock.calls as unknown as Array<[Record]> )?.[0]?.[0]; expect(passedOpts).toBeTruthy(); if (!passedOpts) { throw new Error("expected message command call"); } expect(passedOpts).not.toHaveProperty("account"); } describe("runMessageAction", () => { beforeEach(() => { vi.clearAllMocks(); messageCommandMock.mockClear().mockResolvedValue(undefined); hasHooksMock.mockClear().mockReturnValue(false); runGatewayStopMock.mockClear().mockResolvedValue(undefined); runGlobalGatewayStopSafelyMock.mockClear(); exitMock.mockClear().mockImplementation((): never => { throw new Error("exit"); }); }); it("calls exit(0) after successful message delivery", async () => { await runSendAction(); expect(exitMock).toHaveBeenCalledOnce(); expect(exitMock).toHaveBeenCalledWith(0); }); it("runs gateway_stop hooks before exit when registered", async () => { hasHooksMock.mockReturnValueOnce(true); await runSendAction(); expect(runGatewayStopMock).toHaveBeenCalledWith({ reason: "cli message action complete" }, {}); expect(exitMock).toHaveBeenCalledWith(0); }); it("calls exit(1) when message delivery fails", async () => { messageCommandMock.mockRejectedValueOnce(new Error("send failed")); await runSendAction(); expect(errorMock).toHaveBeenCalledWith("Error: send failed"); expect(exitMock).toHaveBeenCalledOnce(); expect(exitMock).toHaveBeenCalledWith(1); }); it("runs gateway_stop hooks on failure before exit(1)", async () => { hasHooksMock.mockReturnValueOnce(true); messageCommandMock.mockRejectedValueOnce(new Error("send failed")); await runSendAction(); expect(runGatewayStopMock).toHaveBeenCalledWith({ reason: "cli message action complete" }, {}); expect(exitMock).toHaveBeenCalledWith(1); }); it("logs gateway_stop failure and still exits with success code", async () => { hasHooksMock.mockReturnValueOnce(true); runGatewayStopMock.mockRejectedValueOnce(new Error("hook failed")); await runSendAction(); expect(errorMock).toHaveBeenCalledWith("gateway_stop hook failed: Error: hook failed"); expect(exitMock).toHaveBeenCalledWith(0); }); it("logs gateway_stop failure and preserves failure exit code when send fails", async () => { hasHooksMock.mockReturnValueOnce(true); messageCommandMock.mockRejectedValueOnce(new Error("send failed")); runGatewayStopMock.mockRejectedValueOnce(new Error("hook failed")); await runSendAction(); expect(errorMock).toHaveBeenNthCalledWith(1, "Error: send failed"); expect(errorMock).toHaveBeenNthCalledWith(2, "gateway_stop hook failed: Error: hook failed"); expect(exitMock).toHaveBeenCalledWith(1); }); it("does not call exit(0) when the action throws", async () => { messageCommandMock.mockRejectedValueOnce(new Error("boom")); await runSendAction(); // exit should only be called once with code 1, never with 0 expect(exitMock).toHaveBeenCalledOnce(); expect(exitMock).not.toHaveBeenCalledWith(0); }); it("does not call exit(0) if the error path returns", async () => { messageCommandMock.mockRejectedValueOnce(new Error("boom")); exitMock.mockClear().mockImplementation(() => undefined as never); const runMessageAction = createRunMessageAction(); await expect(runMessageAction("send", baseSendOptions)).resolves.toBeUndefined(); expect(errorMock).toHaveBeenCalledWith("Error: boom"); expect(exitMock).toHaveBeenCalledOnce(); expect(exitMock).toHaveBeenCalledWith(1); expect(exitMock).not.toHaveBeenCalledWith(0); }); it("passes action and maps account to accountId", async () => { const fakeCommand = { help: vi.fn() } as never; const { runMessageAction } = createMessageCliHelpers(fakeCommand, "discord"); await expect( runMessageAction("poll", { channel: "discord", target: "456", account: "acct-1", message: "hi", }), ).rejects.toThrow("exit"); expect(messageCommandMock).toHaveBeenCalledWith( expect.objectContaining({ action: "poll", channel: "discord", target: "456", accountId: "acct-1", message: "hi", }), expect.anything(), expect.anything(), ); // account key should be stripped in favor of accountId expectNoAccountFieldInPassedOptions(); }); it("strips non-string account values instead of passing accountId", async () => { const runMessageAction = createRunMessageAction(); await expect( runMessageAction("send", { channel: "discord", target: "789", account: 42, message: "hi", }), ).rejects.toThrow("exit"); expect(messageCommandMock).toHaveBeenCalledWith( expect.objectContaining({ action: "send", channel: "discord", target: "789", accountId: undefined, }), expect.anything(), expect.anything(), ); expectNoAccountFieldInPassedOptions(); }); });