Function mockQueryWithReply

src/ink/replSubmitAssistantTurn.test.tsx:38–65

(content: string)

const ORIGINAL_API_KEY = process.env.ANTHROPIC_API_KEY

async function mockQueryWithReply(content: string): Promise<{
  readonly getCallCount: () => number
}> {
  let callCount = 0
  const queryPaths = [
    import.meta.resolve('../query.ts'),
    import.meta.resolve('../query.js'),
  ]
  const { createAssistantMessage } = await import('../utils/messages.js')
  const actualQueryModule = await import(import.meta.resolve('../query.ts'))

  for (const path of queryPaths) {
    mock.module(path, () => ({
      ...actualQueryModule,
      query: async function* () {
        callCount += 1
        yield createAssistantMessage({
          content,
        })
        return { reason: 'completed' } as never
      },
    }))
  }

  return {
    getCallCount: () => callCount,
  }
}

beforeEach(() => {
  process.env.CLAUDE_CODE_NO_FLICKER = '1'