如何在功能性React组件中测试一个函数是否被称为onChange？

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

test('should call onChange function when input changes', () => {
  // 准备测试环境
  const onChange = jest.fn(); // 使用jest.fn()创建一个mock函数
  const { getByTestId } = render(<ComponentName onChange={onChange} />);
  const input = getByTestId('input'); // 假设输入框有一个data-testid为'input'

  // 模拟用户操作
  fireEvent.change(input, { target: { value: 'test' } });

  // 断言函数被调用
  expect(onChange).toHaveBeenCalledTimes(1);
});