shlink-web-client/test/common/MenuLayout.test.tsx

import { shallow, ShallowWrapper } from 'enzyme';
import { Route, useParams } from 'react-router-dom';
import { Mock } from 'ts-mockery';
import createMenuLayout from '../../src/common/MenuLayout';
import { NonReachableServer, NotFoundServer, ReachableServer, SelectedServer } from '../../src/servers/data';
import { NoMenuLayout } from '../../src/common/NoMenuLayout';
import { SemVer } from '../../src/utils/helpers/version';

jest.mock('react-router-dom', () => ({
  ...jest.requireActual('react-router-dom'),
  useParams: jest.fn().mockReturnValue({}),
  useLocation: jest.fn().mockReturnValue({}),
}));

describe('<MenuLayout />', () => {
  const ServerError = jest.fn();
  const C = jest.fn();
  const MenuLayout = createMenuLayout(C, C, C, C, C, C, C, C, C, ServerError, C, C, C);
  let wrapper: ShallowWrapper;
  const createWrapper = (selectedServer: SelectedServer) => {
    (useParams as any).mockReturnValue({ serverId: 'abc123' });

    wrapper = shallow(
      <MenuLayout
        sidebarNotPresent={jest.fn()}
        sidebarPresent={jest.fn()}
        selectServer={jest.fn()}
        selectedServer={selectedServer}
      />,
    );

    return wrapper;
  };

  afterEach(jest.clearAllMocks);
  afterEach(() => wrapper?.unmount());

  it.each([
    [null, NoMenuLayout],
    [Mock.of<NotFoundServer>({ serverNotFound: true }), ServerError],
  ])('returns error when server is not found', (selectedServer, ExpectedComp) => {
    const wrapper = createWrapper(selectedServer);
    const comp = wrapper.find(ExpectedComp);

    expect(comp).toHaveLength(1);
  });

  it('returns error if server is not reachable', () => {
    const wrapper = createWrapper(Mock.of<NonReachableServer>()).dive();
    const serverError = wrapper.find(ServerError);

    expect(serverError).toHaveLength(1);
  });

  it.each([
    ['2.6.0' as SemVer, 10],
    ['2.7.0' as SemVer, 10],
    ['2.8.0' as SemVer, 11],
    ['2.10.0' as SemVer, 11],
    ['3.0.0' as SemVer, 12],
    ['3.1.0' as SemVer, 13],
  ])('has expected amount of routes based on selected server\'s version', (version, expectedAmountOfRoutes) => {
    const selectedServer = Mock.of<ReachableServer>({ version });
    const wrapper = createWrapper(selectedServer).dive();
    const routes = wrapper.find(Route);

    expect(routes).toHaveLength(expectedAmountOfRoutes);
    expect(routes.findWhere((element) => element.prop('index'))).toHaveLength(1);
  });
});