Тестирование функционального компонента React с Jest+Enzyme

Есть вот такой функциональный компонент (условно):

 const BackToOperationLoadButton = ({
      setIsInOperationAction,
      setStepAction,
    }) => {
      const [isBackCaptionActive, setBackCaptionActive] = useState(false);
    
  const applyButtonHandler = () => {
    setIsInOperationAction(false);
    setBackCaptionActive(false);
  };
  return (
    <div className='back-to-operation-load-button'>

      <div className='back-to-operation-load-button__modeless-dialog'>
        <Dialog
          isActive={isBackCaptionActive}
          onApply={applyButtonHandler}
          message='Example'
        />
      </div>
    </div>
  );
};

const actionCreator = {
  setIsInOperationAction: actions.app.openOperation,
};

export default connect(null, actionCreator)(BackToOperationLoadButton);

Я пытаюсь протестировать applyButtonHandler:

const setIsInOperationAction = jest.fn();

<...>

 wrapper.find(Dialog).prop('onApply')();
 expect(setIsInOperationAction).toHaveBeenCalled();

Но тесты валятся - вызовов не было.Хотя если поставить лог в тестируемую функцию в коде компонента, то лог вызывается. Пробовала асинхронно тестить, задержки ставить, но не помогло.