当axios调用中的状态正在更新时，如何在单元测试(React JS)中断言状态？

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

jest.mock('axios'); // 使用jest的mock函数来模拟axios

test('should update state when axios call is in progress', async () => {
  const mockData = { message: 'Success' };
  axios.get.mockResolvedValue({ data: mockData }); // 模拟axios的返回结果

  const { getByTestId } = render(<MyComponent />);
  const button = getByTestId('my-button');

  fireEvent.click(button); // 触发按钮点击事件

  expect(getByTestId('loading-indicator')).toBeInTheDocument(); // 断言loading指示器显示

  await waitFor(() => {
    expect(getByTestId('message')).toHaveTextContent(mockData.message); // 断言状态更新后的消息显示
  });
});