使用withRouter和connect，如何在酶中断言调用了history.push？

import { withRouter } from 'react-router-dom';

class YourComponent extends React.Component {
  // ...
}

export default withRouter(YourComponent);

import { shallow } from 'enzyme';
import { createMemoryHistory } from 'history';

// 导入你的组件
import YourComponent from './YourComponent';

describe('YourComponent', () => {
  it('should call history.push when something happens', () => {
    const history = createMemoryHistory();
    const pushMock = jest.fn();
    history.push = pushMock;

    // 使用shallow渲染组件
    const wrapper = shallow(<YourComponent />, {
      context: { router: { history } },
    });

    // 模拟触发某个事件或条件
    // ...

    // 断言history.push被调用
    expect(pushMock).toHaveBeenCalledWith('/your-path');
  });
});