test: isolate recovery hook mocks from full suite

2026-04-03 17:50:24 +09:00
parent ed06428ba3
commit 9fa1ea38e8
2 changed files with 138 additions and 88 deletions
@@ -0,0 +1,113 @@
+import { mock } from "bun:test"
+import type { PluginInput } from "@opencode-ai/plugin"
+import type { OhMyOpenCodeConfig } from "../../config"
+import { createAnthropicContextWindowLimitRecoveryHook } from "./recovery-hook"
+
+type ExecuteCompactFn = typeof import("./executor").executeCompact
+type GetLastAssistantFn = typeof import("./executor").getLastAssistant
+type ParseAnthropicTokenLimitErrorFn = typeof import("./parser").parseAnthropicTokenLimitError
+
+export type MockLastAssistant = {
+  info: {
+    summary?: boolean
+    providerID: string
+    modelID: string
+  }
+  hasContent: boolean
+}
+
+export const executeCompactMock = mock<ExecuteCompactFn>(async () => {})
+export const getLastAssistantMock = mock<GetLastAssistantFn>(async (): Promise<MockLastAssistant> => ({
+  info: {
+    providerID: "anthropic",
+    modelID: "claude-sonnet-4-6",
+  },
+  hasContent: true,
+}))
+export const parseAnthropicTokenLimitErrorMock = mock<ParseAnthropicTokenLimitErrorFn>(() => ({
+  currentTokens: 250000,
+  maxTokens: 200000,
+  errorType: "token_limit_exceeded",
+  providerID: "anthropic",
+  modelID: "claude-sonnet-4-6",
+}))
+
+const pluginConfig = {
+  git_master: {
+    commit_footer: false,
+    include_co_authored_by: false,
+    git_env_prefix: "",
+  },
+} satisfies OhMyOpenCodeConfig
+
+export function createRecoveryHook() {
+  return createAnthropicContextWindowLimitRecoveryHook(
+    createMockContext(),
+    {
+      pluginConfig,
+      dependencies: {
+        executeCompact: executeCompactMock,
+        getLastAssistant: getLastAssistantMock,
+        log: () => {},
+        parseAnthropicTokenLimitError: parseAnthropicTokenLimitErrorMock,
+      },
+    } as never,
+  )
+}
+
+export function createMockContext(): PluginInput {
+  return {
+    client: {
+      session: {
+        messages: mock(() => Promise.resolve({ data: [] })),
+      },
+      tui: {
+        showToast: mock(() => Promise.resolve()),
+      },
+    },
+    project: {} as never,
+    directory: "/tmp",
+    worktree: "/tmp",
+    serverUrl: new URL("http://localhost"),
+    $: {} as never,
+  } as never
+}
+
+export function setupDelayedTimeoutMocks(): {
+  createUntrackedTimeout: () => ReturnType<typeof setTimeout>
+  restore: () => void
+  getClearTimeoutCalls: () => Array<ReturnType<typeof setTimeout>>
+  getScheduledTimeouts: () => Array<ReturnType<typeof setTimeout>>
+} {
+  const originalSetTimeout = globalThis.setTimeout
+  const originalClearTimeout = globalThis.clearTimeout
+  const clearTimeoutCalls: Array<ReturnType<typeof setTimeout>> = []
+  const scheduledTimeouts: Array<ReturnType<typeof setTimeout>> = []
+
+  function createTimeoutHandle(): ReturnType<typeof setTimeout> {
+    const timeoutID = originalSetTimeout(() => {}, 60_000)
+    originalClearTimeout(timeoutID)
+    return timeoutID
+  }
+
+  globalThis.setTimeout = ((_: () => void, _delay?: number) => {
+    const timeoutID = createTimeoutHandle()
+    scheduledTimeouts.push(timeoutID)
+    return timeoutID
+  }) as typeof setTimeout
+
+  globalThis.clearTimeout = ((timeoutID: ReturnType<typeof setTimeout>) => {
+    clearTimeoutCalls.push(timeoutID)
+    originalClearTimeout(timeoutID)
+  }) as typeof clearTimeout
+
+  return {
+    createUntrackedTimeout: createTimeoutHandle,
+    restore: () => {
+      globalThis.setTimeout = originalSetTimeout
+      globalThis.clearTimeout = originalClearTimeout
+    },
+    getClearTimeoutCalls: () => clearTimeoutCalls,
+    getScheduledTimeouts: () => scheduledTimeouts,
+  }
+}