如何用HOC编写合适的jest测试？

// withTestLogic.js
import React from 'react';

const withTestLogic = (WrappedComponent) => {
  return class extends React.Component {
    constructor(props) {
      super(props);
      this.state = {
        data: null,
        loading: true,
      };
    }

    componentDidMount() {
      // 模拟数据获取
      setTimeout(() => {
        this.setState({
          data: 'Test Data',
          loading: false,
        });
      }, 1000);
    }

    render() {
      const { data, loading } = this.state;

      // 将props和state传递给被测试组件
      return <WrappedComponent data={data} loading={loading} {...this.props} />;
    }
  };
};

export default withTestLogic;

// MyComponent.test.js
import React from 'react';
import { render, screen } from '@testing-library/react';
import withTestLogic from './withTestLogic';
import MyComponent from './MyComponent';

// 使用HOC包装被测试组件
const WrappedComponent = withTestLogic(MyComponent);

test('renders data correctly', async () => {
  render(<WrappedComponent />);

  // 模拟异步数据获取完成
  await screen.findByText('Test Data');

  // 断言被测试组件是否正确渲染了数据
  expect(screen.getByText('Test Data')).toBeInTheDocument();
});

test('renders loading state correctly', () => {
  render(<WrappedComponent />);

  // 断言被测试组件是否正确渲染了加载状态
  expect(screen.getByText('Loading...')).toBeInTheDocument();
});