当react中的本地useState发生变化时，如何测试组件渲染子组件？

import React, { useState } from 'react';
import { mount } from 'enzyme';

const ParentComponent = () => {
  const [count, setCount] = useState(0);

  return (
    <div>
      <p>Count: {count}</p>
      <ChildComponent count={count} />
    </div>
  );
};

const ChildComponent = ({ count }) => {
  return <p>Child Count: {count}</p>;
};

describe('ParentComponent', () => {
  it('should render ChildComponent with correct count', () => {
    const wrapper = mount(<ParentComponent />);
    const countElement = wrapper.find('p').at(0);
    const childCountElement = wrapper.find('p').at(1);

    expect(countElement.text()).toBe('Count: 0');
    expect(childCountElement.text()).toBe('Child Count: 0');

    // 模拟状态变化
    wrapper.find('button').simulate('click');

    expect(countElement.text()).toBe('Count: 1');
    expect(childCountElement.text()).toBe('Child Count: 1');
  });
});