File size: 3,150 Bytes
9dc6089 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | /**
* Test: message_sending & message_sent hook wiring
*
* Tests the hook runner methods directly since outbound delivery is deeply integrated.
*/
import { describe, expect, it, vi } from "vitest";
import type { PluginRegistry } from "./registry.js";
import { createHookRunner } from "./hooks.js";
function createMockRegistry(
hooks: Array<{ hookName: string; handler: (...args: unknown[]) => unknown }>,
): PluginRegistry {
return {
hooks: hooks as never[],
typedHooks: hooks.map((h) => ({
pluginId: "test-plugin",
hookName: h.hookName,
handler: h.handler,
priority: 0,
source: "test",
})),
tools: [],
httpHandlers: [],
httpRoutes: [],
channelRegistrations: [],
gatewayHandlers: {},
cliRegistrars: [],
services: [],
providers: [],
commands: [],
} as unknown as PluginRegistry;
}
describe("message_sending hook runner", () => {
it("runMessageSending invokes registered hooks and returns modified content", async () => {
const handler = vi.fn().mockReturnValue({ content: "modified content" });
const registry = createMockRegistry([{ hookName: "message_sending", handler }]);
const runner = createHookRunner(registry);
const result = await runner.runMessageSending(
{ to: "user-123", content: "original content" },
{ channelId: "telegram" },
);
expect(handler).toHaveBeenCalledWith(
{ to: "user-123", content: "original content" },
{ channelId: "telegram" },
);
expect(result?.content).toBe("modified content");
});
it("runMessageSending can cancel message delivery", async () => {
const handler = vi.fn().mockReturnValue({ cancel: true });
const registry = createMockRegistry([{ hookName: "message_sending", handler }]);
const runner = createHookRunner(registry);
const result = await runner.runMessageSending(
{ to: "user-123", content: "blocked" },
{ channelId: "telegram" },
);
expect(result?.cancel).toBe(true);
});
});
describe("message_sent hook runner", () => {
it("runMessageSent invokes registered hooks with success=true", async () => {
const handler = vi.fn();
const registry = createMockRegistry([{ hookName: "message_sent", handler }]);
const runner = createHookRunner(registry);
await runner.runMessageSent(
{ to: "user-123", content: "hello", success: true },
{ channelId: "telegram" },
);
expect(handler).toHaveBeenCalledWith(
{ to: "user-123", content: "hello", success: true },
{ channelId: "telegram" },
);
});
it("runMessageSent invokes registered hooks with error on failure", async () => {
const handler = vi.fn();
const registry = createMockRegistry([{ hookName: "message_sent", handler }]);
const runner = createHookRunner(registry);
await runner.runMessageSent(
{ to: "user-123", content: "hello", success: false, error: "timeout" },
{ channelId: "telegram" },
);
expect(handler).toHaveBeenCalledWith(
{ to: "user-123", content: "hello", success: false, error: "timeout" },
{ channelId: "telegram" },
);
});
});
|