File size: 2,318 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 | /**
* Test: session_start & session_end hook wiring
*
* Tests the hook runner methods directly since session init 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("session hook runner methods", () => {
it("runSessionStart invokes registered session_start hooks", async () => {
const handler = vi.fn();
const registry = createMockRegistry([{ hookName: "session_start", handler }]);
const runner = createHookRunner(registry);
await runner.runSessionStart(
{ sessionId: "abc-123", resumedFrom: "old-session" },
{ sessionId: "abc-123", agentId: "main" },
);
expect(handler).toHaveBeenCalledWith(
{ sessionId: "abc-123", resumedFrom: "old-session" },
{ sessionId: "abc-123", agentId: "main" },
);
});
it("runSessionEnd invokes registered session_end hooks", async () => {
const handler = vi.fn();
const registry = createMockRegistry([{ hookName: "session_end", handler }]);
const runner = createHookRunner(registry);
await runner.runSessionEnd(
{ sessionId: "abc-123", messageCount: 42 },
{ sessionId: "abc-123", agentId: "main" },
);
expect(handler).toHaveBeenCalledWith(
{ sessionId: "abc-123", messageCount: 42 },
{ sessionId: "abc-123", agentId: "main" },
);
});
it("hasHooks returns true for registered session hooks", () => {
const registry = createMockRegistry([{ hookName: "session_start", handler: vi.fn() }]);
const runner = createHookRunner(registry);
expect(runner.hasHooks("session_start")).toBe(true);
expect(runner.hasHooks("session_end")).toBe(false);
});
});
|