Migrated more tests to react testing lib

test/visits/ShortUrlVisits.test.tsx

@@ -1,49 +1,54 @@
-import { shallow, ShallowWrapper } from 'enzyme';
+import { render, screen } from '@testing-library/react';
+import userEvent from '@testing-library/user-event';
+import { MemoryRouter } from 'react-router-dom';
 import { identity } from 'ramda';
 import { Mock } from 'ts-mockery';
+import { formatISO } from 'date-fns';
 import { ShortUrlVisits as createShortUrlVisits, ShortUrlVisitsProps } from '../../src/visits/ShortUrlVisits';
-import { ShortUrlVisitsHeader } from '../../src/visits/ShortUrlVisitsHeader';
 import { ShortUrlVisits as ShortUrlVisitsState } from '../../src/visits/reducers/shortUrlVisits';
 import { ShortUrlDetail } from '../../src/short-urls/reducers/shortUrlDetail';
-import { VisitsStats } from '../../src/visits/VisitsStats';
 import { MercureBoundProps } from '../../src/mercure/helpers/boundToMercureHub';
 import { ReportExporter } from '../../src/common/services/ReportExporter';
-
-jest.mock('react-router-dom', () => ({
-  ...jest.requireActual('react-router-dom'),
-  useNavigate: jest.fn().mockReturnValue(jest.fn()),
-  useLocation: jest.fn().mockReturnValue({ search: '' }),
-  useParams: jest.fn().mockReturnValue({ shortCode: 'abc123' }),
-}));
+import { Visit } from '../../src/visits/types';
+import { Settings } from '../../src/settings/reducers/settings';
 
 describe('<ShortUrlVisits />', () => {
-  let wrapper: ShallowWrapper;
   const getShortUrlVisitsMock = jest.fn();
-  const ShortUrlVisits = createShortUrlVisits(Mock.all<ReportExporter>());
-
-  beforeEach(() => {
-    wrapper = shallow(
-      <ShortUrlVisits
-        {...Mock.all<ShortUrlVisitsProps>()}
-        {...Mock.of<MercureBoundProps>({ mercureInfo: {} })}
-        getShortUrlDetail={identity}
-        getShortUrlVisits={getShortUrlVisitsMock}
-        shortUrlVisits={Mock.of<ShortUrlVisitsState>({ loading: true, visits: [] })}
-        shortUrlDetail={Mock.all<ShortUrlDetail>()}
-        cancelGetShortUrlVisits={() => {}}
-      />,
-    ).dive(); // Dive is needed as this component is wrapped in a HOC
+  const exportVisits = jest.fn();
+  const shortUrlVisits = Mock.of<ShortUrlVisitsState>({ visits: [Mock.of<Visit>({ date: formatISO(new Date()) })] });
+  const ShortUrlVisits = createShortUrlVisits(Mock.of<ReportExporter>({ exportVisits }));
+  const setUp = () => ({
+    user: userEvent.setup(),
+    ...render(
+      <MemoryRouter>
+        <ShortUrlVisits
+          {...Mock.all<ShortUrlVisitsProps>()}
+          {...Mock.of<MercureBoundProps>({ mercureInfo: {} })}
+          getShortUrlDetail={identity}
+          getShortUrlVisits={getShortUrlVisitsMock}
+          shortUrlVisits={shortUrlVisits}
+          shortUrlDetail={Mock.all<ShortUrlDetail>()}
+          settings={Mock.all<Settings>()}
+          cancelGetShortUrlVisits={() => {}}
+        />
+      </MemoryRouter>,
+    ),
   });
 
-  afterEach(jest.clearAllMocks);
-  afterEach(() => wrapper.unmount());
+  it('wraps visits stats and header', () => {
+    setUp();
+    expect(screen.getAllByRole('heading')[0]).toHaveTextContent('Visits for');
+    expect(getShortUrlVisitsMock).toHaveBeenCalled();
+  });
 
-  it('renders visit stats and visits header', () => {
-    const visitStats = wrapper.find(VisitsStats);
-    const visitHeader = wrapper.find(ShortUrlVisitsHeader);
+  it('exports visits when clicking the button', async () => {
+    const { user } = setUp();
+    const btn = screen.getByRole('button', { name: 'Export (1)' });
 
-    expect(visitStats).toHaveLength(1);
-    expect(visitStats.prop('isOrphanVisits')).not.toBeDefined();
-    expect(visitHeader).toHaveLength(1);
+    expect(exportVisits).not.toHaveBeenCalled();
+    expect(btn).toBeInTheDocument();
+
+    await user.click(btn);
+    expect(exportVisits).toHaveBeenCalledWith('short-url_undefined_visits.csv', expect.anything());
   });
 });