iNaturalistReactNative/tests/unit/components/SharedComponents/ObsDetails/HeaderEditIcon.test.js

import { fireEvent, screen } from "@testing-library/react-native";
import HeaderEditIcon from "components/SharedComponents/ObsDetails/HeaderEditIcon";
import React from "react";
import useStore from "stores/useStore";
import factory from "tests/factory";
import { renderComponent } from "tests/helpers/render";

const mockNavigate = jest.fn();
const mockSetOptions = jest.fn();
jest.mock( "@react-navigation/native", () => ( {
  ...jest.requireActual( "@react-navigation/native" ),
  useNavigation: () => ( {
    navigate: mockNavigate,
    setOptions: mockSetOptions,
  } ),
} ) );

describe( "HeaderEditIcon", () => {
  const mockObservation = factory( "LocalObservation" );

  beforeEach( () => {
    jest.clearAllMocks();
  } );

  it( "sets navigation header options", () => {
    renderComponent( <HeaderEditIcon observation={mockObservation} /> );

    expect( mockSetOptions ).toHaveBeenCalledWith(
      expect.objectContaining( {
        headerRight: expect.any( Function ),
      } ),
    );
  } );

  it( "renders edit button and calls relevant functions on press", () => {
    const prepareObsEdit = jest.fn();
    const setMyObsOffsetToRestore = jest.fn();

    useStore.setState( {
      prepareObsEdit,
      setMyObsOffsetToRestore,
    } );

    renderComponent( <HeaderEditIcon observation={mockObservation} /> );

    const headerRightCall = mockSetOptions.mock.calls[0][0];
    const HeaderRightComponent = headerRightCall.headerRight;

    renderComponent( <HeaderRightComponent /> );

    const editButton = screen.getByTestId( "ObsEditIcon" );
    expect( editButton ).toBeVisible();

    fireEvent.press( editButton );

    expect( prepareObsEdit ).toHaveBeenCalledWith( mockObservation );
    expect( mockNavigate ).toHaveBeenCalledWith( "ObsEdit", undefined );
    expect( setMyObsOffsetToRestore ).toHaveBeenCalled();
  } );
} );