test(ci): isolate runtime and rules dependencies

src/hooks/runtime-fallback/dispose.test.ts

@@ -1,4 +1,6 @@
-import { afterAll, afterEach, beforeEach, describe, expect, mock, test } from "bun:test"
+import { afterEach, beforeEach, describe, expect, mock, test } from "bun:test"
+import type { AutoRetryHelpers } from "./auto-retry"
+import { createRuntimeFallbackHook } from "./hook"
 import type { HookDeps, RuntimeFallbackPluginInput } from "./types"
 
 let capturedDeps: HookDeps | undefined
@@ -16,31 +18,18 @@ const mockCreateAutoRetryHelpers = mock((deps: HookDeps) => {
   }
 })
 
-const mockCreateEventHandler = mock(() => async () => {})
-const mockCreateMessageUpdateHandler = mock(() => async () => {})
-const mockCreateChatMessageHandler = mock(() => async () => {})
+const mockCreateEventHandler = mock((_deps: HookDeps, _helpers: AutoRetryHelpers) => async () => {})
+const mockCreateMessageUpdateHandler = mock((_deps: HookDeps, _helpers: AutoRetryHelpers) => async () => {})
+const mockCreateChatMessageHandler = mock((_deps: HookDeps) => async () => {})
 
-mock.module("./auto-retry", () => ({
-  createAutoRetryHelpers: mockCreateAutoRetryHelpers,
-}))
-
-mock.module("./event-handler", () => ({
-  createEventHandler: mockCreateEventHandler,
-}))
-
-mock.module("./message-update-handler", () => ({
-  createMessageUpdateHandler: mockCreateMessageUpdateHandler,
-}))
-
-mock.module("./chat-message-handler", () => ({
-  createChatMessageHandler: mockCreateChatMessageHandler,
-}))
-
-afterAll(() => {
-  mock.restore()
-})
-
-const { createRuntimeFallbackHook } = await import("./hook")
+function createHookWithMocks() {
+  return createRuntimeFallbackHook(createMockContext(), { pluginConfig: {} }, {
+    createAutoRetryHelpers: mockCreateAutoRetryHelpers,
+    createEventHandler: mockCreateEventHandler,
+    createMessageUpdateHandler: mockCreateMessageUpdateHandler,
+    createChatMessageHandler: mockCreateChatMessageHandler,
+  })
+}
 
 function createMockContext(): RuntimeFallbackPluginInput {
   return {
@@ -109,7 +98,7 @@ describe("createRuntimeFallbackHook dispose", () => {
 
   test("#given runtime-fallback hook handles its first event #when dispose() is called #then cleanup interval is cleared", async () => {
     // given
-    const hook = createRuntimeFallbackHook(createMockContext(), { pluginConfig: {} })
+    const hook = createHookWithMocks()
     await hook.event({ event: { type: "session.created", properties: {} } })
 
     // when
@@ -122,7 +111,7 @@ describe("createRuntimeFallbackHook dispose", () => {
 
   test("#given hook with session state data #when dispose() is called #then all Maps and Sets are empty", () => {
     // given
-    const hook = createRuntimeFallbackHook(createMockContext(), { pluginConfig: {} })
+    const hook = createHookWithMocks()
     const fallbackTimeout = setTimeout(() => {}, 60_000)
 
     capturedDeps?.sessionStates.set("session-1", {
@@ -150,7 +139,7 @@ describe("createRuntimeFallbackHook dispose", () => {
 
   test("#given hook with pending fallback timeouts #when dispose() is called #then timeouts are cleared before Map is emptied", () => {
     // given
-    const hook = createRuntimeFallbackHook(createMockContext(), { pluginConfig: {} })
+    const hook = createHookWithMocks()
     const fallbackTimeout = setTimeout(() => {}, 60_000)
     capturedDeps?.sessionFallbackTimeouts.set("session-1", fallbackTimeout)
 

src/hooks/runtime-fallback/hook.ts

@@ -1,18 +1,37 @@
-import type { HookDeps, RuntimeFallbackHook, RuntimeFallbackInterval, RuntimeFallbackOptions, RuntimeFallbackPluginInput, RuntimeFallbackTimeout } from "./types"
-import { DEFAULT_CONFIG } from "./constants"
 import { createAutoRetryHelpers } from "./auto-retry"
+import { createChatMessageHandler } from "./chat-message-handler"
+import { DEFAULT_CONFIG } from "./constants"
 import { createEventHandler } from "./event-handler"
 import { createMessageUpdateHandler } from "./message-update-handler"
-import { createChatMessageHandler } from "./chat-message-handler"
+import type { HookDeps, RuntimeFallbackHook, RuntimeFallbackInterval, RuntimeFallbackOptions, RuntimeFallbackPluginInput, RuntimeFallbackTimeout } from "./types"
 
 declare function setInterval(callback: () => void, delay?: number): RuntimeFallbackInterval
 declare function clearInterval(interval: RuntimeFallbackInterval): void
 declare function clearTimeout(timeout: RuntimeFallbackTimeout): void
 
+type RuntimeFallbackHookFactories = {
+  createAutoRetryHelpers: typeof createAutoRetryHelpers
+  createEventHandler: typeof createEventHandler
+  createMessageUpdateHandler: typeof createMessageUpdateHandler
+  createChatMessageHandler: typeof createChatMessageHandler
+}
+
+const defaultRuntimeFallbackHookFactories: RuntimeFallbackHookFactories = {
+  createAutoRetryHelpers,
+  createEventHandler,
+  createMessageUpdateHandler,
+  createChatMessageHandler,
+}
+
 export function createRuntimeFallbackHook(
   ctx: RuntimeFallbackPluginInput,
-  options?: RuntimeFallbackOptions
+  options?: RuntimeFallbackOptions,
+  factoryOverrides: Partial<RuntimeFallbackHookFactories> = {},
 ): RuntimeFallbackHook {
+  const factories = {
+    ...defaultRuntimeFallbackHookFactories,
+    ...factoryOverrides,
+  }
   const config = {
     enabled: options?.config?.enabled ?? DEFAULT_CONFIG.enabled,
     retry_on_errors: options?.config?.retry_on_errors ?? DEFAULT_CONFIG.retry_on_errors,
@@ -35,10 +54,10 @@ export function createRuntimeFallbackHook(
     sessionStatusRetryKeys: new Map(),
   }
 
-  const helpers = createAutoRetryHelpers(deps)
-  const baseEventHandler = createEventHandler(deps, helpers)
-  const messageUpdateHandler = createMessageUpdateHandler(deps, helpers)
-  const chatMessageHandler = createChatMessageHandler(deps)
+  const helpers = factories.createAutoRetryHelpers(deps)
+  const baseEventHandler = factories.createEventHandler(deps, helpers)
+  const messageUpdateHandler = factories.createMessageUpdateHandler(deps, helpers)
+  const chatMessageHandler = factories.createChatMessageHandler(deps)
 
   let cleanupInterval: RuntimeFallbackInterval | null = null
   let intervalStarted = false