在一个组件中多次模拟useSelector的正确方法是什么？

import { render, screen } from '@testing-library/react';
import { useSelector } from 'react-redux';
import { createStore } from 'redux';
import { Provider } from 'react-redux';
import reducer from '../path/to/reducer';

const mockState = {
  selector1: 'value1',
  selector2: 'value2',
  // 其他selector的返回值
};

jest.mock('react-redux', () => ({
  ...jest.requireActual('react-redux'),
  useSelector: jest.fn(),
}));

const store = createStore(reducer, mockState);

render(
  <Provider store={store}>
    <YourComponent />
  </Provider>
);

useSelector.mockImplementation((selector) => {
  if (selector === selector1) {
    return mockState.selector1;
  }
  if (selector === selector2) {
    return mockState.selector2;
  }
  // 其他selector的模拟返回值
});

expect(screen.getByText(mockState.selector1)).toBeInTheDocument();
expect(screen.getByText(mockState.selector2)).toBeInTheDocument();
// 其他断言和验证