如何在jest react测试中模拟此函数？

import { render, fireEvent } from '@testing-library/react';
import MyComponent from './MyComponent';

// 模拟函数
const mockFunction = jest.fn();

test('测试模拟函数', () => {
  // 渲染组件
  const { getByTestId } = render(<MyComponent onClick={mockFunction} />);
  
  // 触发事件
  fireEvent.click(getByTestId('my-button'));
  
  // 断言模拟函数被调用
  expect(mockFunction).toHaveBeenCalled();
});