orca/mobile/src/session/MobileNativeChatCardHeaderAction.test.tsx

import { createElement } from 'react'
import { act, create, type ReactTestRenderer } from 'react-test-renderer'
import { describe, expect, it, vi } from 'vitest'
import { Keyboard } from 'react-native'
import { MobileNativeChatAsk } from './native-chat/input/MobileNativeChatAsk'
import { MobileNativeChatPermission } from './native-chat/input/MobileNativeChatPermission'

vi.mock('react-native', () => ({
  Keyboard: { dismiss: vi.fn() },
  Pressable: 'Pressable',
  ScrollView: 'ScrollView',
  StyleSheet: { create: (styles: unknown) => styles, hairlineWidth: 1 },
  Text: 'Text',
  TextInput: 'TextInput',
  View: 'View',
  Platform: { OS: 'ios', select: (o: Record<string, unknown>) => o.ios }
}))
vi.mock('lucide-react-native', () => ({
  Check: 'Check',
  ChevronDown: 'ChevronDown',
  ChevronUp: 'ChevronUp',
  ShieldQuestion: 'ShieldQuestion',
  X: 'X'
}))
vi.mock('../components/MobileMarkdown', () => ({ MobileMarkdown: 'Markdown' }))

describe('rendered prompt card collapse controls', () => {
  it('collapses an Ask from its header chevron without cancelling or answering it', async () => {
    const onCollapse = vi.fn()
    const onCancel = vi.fn(async () => true)
    const onAnswer = vi.fn(async () => true)
    let tree: ReactTestRenderer | null = null
    await act(async () => {
      tree = create(
        createElement(MobileNativeChatAsk, {
          prompt: {
            questions: [{ question: 'Pick?', options: [{ label: 'East' }], multiSelect: false }]
          },
          onAnswer,
          onCancel,
          onCollapse
        })
      )
    })
    act(() => tree!.root.findByProps({ accessibilityLabel: 'Collapse' }).props.onPress())
    expect(onCollapse).toHaveBeenCalledOnce()
    expect(Keyboard.dismiss).toHaveBeenCalledOnce()
    expect(onCancel).not.toHaveBeenCalled()
    expect(onAnswer).not.toHaveBeenCalled()
    act(() => tree!.unmount())
  })

  it('shows Collapse in a terminal permission header and keeps Cancel where the lane can cancel', async () => {
    const permission = {
      title: 'Allow Bash?',
      options: [
        { label: 'Allow', send: '1' },
        { label: 'Deny', send: '\x1b' }
      ]
    }
    const onCollapse = vi.fn()
    const onCancel = vi.fn(async () => true)
    let tree: ReactTestRenderer | null = null
    await act(async () => {
      tree = create(
        createElement(MobileNativeChatPermission, {
          permission,
          onRespond: async () => true,
          onCollapse
        })
      )
    })
    expect(tree!.root.findAllByProps({ accessibilityLabel: 'Cancel' })).toHaveLength(0)
    act(() => tree!.root.findByProps({ accessibilityLabel: 'Collapse' }).props.onPress())
    expect(onCollapse).toHaveBeenCalledOnce()
    await act(async () => {
      tree!.update(
        createElement(MobileNativeChatPermission, {
          permission,
          onRespond: async () => true,
          onCancel,
          onCollapse
        })
      )
    })
    expect(tree!.root.findAllByProps({ accessibilityLabel: 'Collapse' })).toHaveLength(0)
    act(() => tree!.root.findByProps({ accessibilityLabel: 'Cancel' }).props.onPress())
    expect(onCancel).toHaveBeenCalledOnce()
    act(() => tree!.unmount())
  })
})