oh-my-opencode/src/hooks/anthropic-context-window-limit-recovery/recovery-hook.test-support.ts

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>
  runScheduledTimeout: (index: number) => void
  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>> = []
  const scheduledCallbacks: Array<() => void> = []

  function createTimeoutHandle(): ReturnType<typeof setTimeout> {
    const timeoutID = originalSetTimeout(() => {}, 60_000)
    originalClearTimeout(timeoutID)
    return timeoutID
  }

  globalThis.setTimeout = ((callback: () => void, _delay?: number) => {
    const timeoutID = createTimeoutHandle()
    scheduledTimeouts.push(timeoutID)
    scheduledCallbacks.push(callback)
    return timeoutID
  }) as typeof setTimeout

  globalThis.clearTimeout = ((timeoutID: ReturnType<typeof setTimeout>) => {
    clearTimeoutCalls.push(timeoutID)
    originalClearTimeout(timeoutID)
  }) as typeof clearTimeout

  return {
    createUntrackedTimeout: createTimeoutHandle,
    runScheduledTimeout: (index: number) => {
      scheduledCallbacks[index]?.()
    },
    restore: () => {
      globalThis.setTimeout = originalSetTimeout
      globalThis.clearTimeout = originalClearTimeout
    },
    getClearTimeoutCalls: () => clearTimeoutCalls,
    getScheduledTimeouts: () => scheduledTimeouts,
  }
}