shlink-web-client/test/visits/ShortUrlVisits.test.js

import React from 'react';
import { shallow } from 'enzyme';
import { identity } from 'ramda';
import { Card } from 'reactstrap';
import createShortUrlVisits from '../../src/visits/ShortUrlVisits';
import Message from '../../src/utils/Message';
import GraphCard from '../../src/visits/GraphCard';
import SortableBarGraph from '../../src/visits/SortableBarGraph';
import DateRangeRow from '../../src/utils/DateRangeRow';

describe('<ShortUrlVisits />', () => {
  let wrapper;
  const processStatsFromVisits = () => (
    { os: {}, browsers: {}, referrers: {}, countries: {}, cities: {}, citiesForMap: {} }
  );
  const getShortUrlVisitsMock = jest.fn();
  const match = {
    params: { shortCode: 'abc123' },
  };
  const location = { search: '' };

  const createComponent = (shortUrlVisits) => {
    const ShortUrlVisits = createShortUrlVisits({ processStatsFromVisits }, () => '');

    wrapper = shallow(
      <ShortUrlVisits
        getShortUrlDetail={identity}
        getShortUrlVisits={getShortUrlVisitsMock}
        match={match}
        location={location}
        shortUrlVisits={shortUrlVisits}
        shortUrlDetail={{}}
        cancelGetShortUrlVisits={identity}
        matchMedia={() => ({ matches: false })}
      />
    );

    return wrapper;
  };

  afterEach(() => {
    getShortUrlVisitsMock.mockReset();
    wrapper && wrapper.unmount();
  });

  it('renders a preloader when visits are loading', () => {
    const wrapper = createComponent({ loading: true, visits: [] });
    const loadingMessage = wrapper.find(Message);

    expect(loadingMessage).toHaveLength(1);
    expect(loadingMessage.html()).toContain('Loading...');
  });

  it('renders a warning when loading large amounts of visits', () => {
    const wrapper = createComponent({ loading: true, loadingLarge: true, visits: [] });
    const loadingMessage = wrapper.find(Message);

    expect(loadingMessage).toHaveLength(1);
    expect(loadingMessage.html()).toContain('This is going to take a while... :S');
  });

  it('renders an error message when visits could not be loaded', () => {
    const wrapper = createComponent({ loading: false, error: true, visits: [] });
    const errorMessage = wrapper.find(Card);

    expect(errorMessage).toHaveLength(1);
    expect(errorMessage.html()).toContain('An error occurred while loading visits :(');
  });

  it('renders a message when visits are loaded but the list is empty', () => {
    const wrapper = createComponent({ loading: false, error: false, visits: [] });
    const message = wrapper.find(Message);

    expect(message).toHaveLength(1);
    expect(message.html()).toContain('There are no visits matching current filter  :(');
  });

  it('renders all graphics when visits are properly loaded', () => {
    const wrapper = createComponent({ loading: false, error: false, visits: [{}, {}, {}] });
    const graphs = wrapper.find(GraphCard);
    const sortableBarGraphs = wrapper.find(SortableBarGraph);

    expect(graphs.length + sortableBarGraphs.length).toEqual(5);
  });

  it('reloads visits when selected dates change', () => {
    const wrapper = createComponent({ loading: false, error: false, visits: [{}, {}, {}] });
    const dateRange = wrapper.find(DateRangeRow);

    dateRange.simulate('startDateChange', '2016-01-01T00:00:00+01:00');
    dateRange.simulate('endDateChange', '2016-01-02T00:00:00+01:00');
    dateRange.simulate('endDateChange', '2016-01-03T00:00:00+01:00');

    expect(getShortUrlVisitsMock).toHaveBeenCalledTimes(4);
    expect(wrapper.state('startDate')).toEqual('2016-01-01T00:00:00+01:00');
    expect(wrapper.state('endDate')).toEqual('2016-01-03T00:00:00+01:00');
  });

  it('holds the map button content generator on cities graph extraHeaderContent', () => {
    const wrapper = createComponent({ loading: false, error: false, visits: [{}, {}, {}] });
    const citiesGraph = wrapper.find(SortableBarGraph).find('[title="Cities"]');
    const extraHeaderContent = citiesGraph.prop('extraHeaderContent');

    expect(extraHeaderContent).toHaveLength(1);
    expect(typeof extraHeaderContent).toEqual('function');
  });
});