import { Mock } from 'ts-mockery'; import reducer, { getShortUrlVisits, cancelGetShortUrlVisits, GET_SHORT_URL_VISITS_START, GET_SHORT_URL_VISITS_ERROR, GET_SHORT_URL_VISITS, GET_SHORT_URL_VISITS_LARGE, GET_SHORT_URL_VISITS_CANCEL, GET_SHORT_URL_VISITS_PROGRESS_CHANGED, ShortUrlVisits, } from '../../../src/visits/reducers/shortUrlVisits'; import { CREATE_VISITS } from '../../../src/visits/reducers/visitCreation'; import { rangeOf } from '../../../src/utils/utils'; import { Visit } from '../../../src/visits/types'; import { ShlinkVisits } from '../../../src/utils/services/types'; import ShlinkApiClient from '../../../src/utils/services/ShlinkApiClient'; import { ShlinkState } from '../../../src/container/types'; describe('shortUrlVisitsReducer', () => { const visitsMocks = rangeOf(2, () => Mock.all()); describe('reducer', () => { const buildState = (data: Partial) => Mock.of(data); it('returns loading on GET_SHORT_URL_VISITS_START', () => { const state = reducer(buildState({ loading: false }), { type: GET_SHORT_URL_VISITS_START } as any); const { loading } = state; expect(loading).toEqual(true); }); it('returns loadingLarge on GET_SHORT_URL_VISITS_LARGE', () => { const state = reducer(buildState({ loadingLarge: false }), { type: GET_SHORT_URL_VISITS_LARGE } as any); const { loadingLarge } = state; expect(loadingLarge).toEqual(true); }); it('returns cancelLoad on GET_SHORT_URL_VISITS_CANCEL', () => { const state = reducer(buildState({ cancelLoad: false }), { type: GET_SHORT_URL_VISITS_CANCEL } as any); const { cancelLoad } = state; expect(cancelLoad).toEqual(true); }); it('stops loading and returns error on GET_SHORT_URL_VISITS_ERROR', () => { const state = reducer(buildState({ loading: true, error: false }), { type: GET_SHORT_URL_VISITS_ERROR } as any); const { loading, error } = state; expect(loading).toEqual(false); expect(error).toEqual(true); }); it('return visits on GET_SHORT_URL_VISITS', () => { const actionVisits = [{}, {}]; const state = reducer( buildState({ loading: true, error: false }), { type: GET_SHORT_URL_VISITS, visits: actionVisits } as any, ); const { loading, error, visits } = state; expect(loading).toEqual(false); expect(error).toEqual(false); expect(visits).toEqual(actionVisits); }); it.each([ [{ shortCode: 'abc123' }, [ ...visitsMocks, {}]], [{ shortCode: 'def456' }, visitsMocks ], ])('appends a new visit on CREATE_VISIT', (state, expectedVisits) => { const shortUrl = { shortCode: 'abc123', }; const prevState = buildState({ ...state, visits: visitsMocks, }); const { visits } = reducer(prevState, { type: CREATE_VISITS, createdVisits: [{ shortUrl, visit: {} }] } as any); expect(visits).toEqual(expectedVisits); }); it('returns new progress on GET_SHORT_URL_VISITS_PROGRESS_CHANGED', () => { const state = reducer(undefined, { type: GET_SHORT_URL_VISITS_PROGRESS_CHANGED, progress: 85 } as any); expect(state).toEqual(expect.objectContaining({ progress: 85 })); }); }); describe('getShortUrlVisits', () => { type GetVisitsReturn = Promise | ((shortCode: string, query: any) => Promise); const buildApiClientMock = (returned: GetVisitsReturn) => Mock.of({ getShortUrlVisits: jest.fn(typeof returned === 'function' ? returned : async () => returned), }); const dispatchMock = jest.fn(); const getState = () => Mock.of({ shortUrlVisits: Mock.of({ cancelLoad: false }), }); beforeEach(() => dispatchMock.mockReset()); it('dispatches start and error when promise is rejected', async () => { const ShlinkApiClient = buildApiClientMock(Promise.reject({})); await getShortUrlVisits(() => ShlinkApiClient)('abc123')(dispatchMock, getState); expect(dispatchMock).toHaveBeenCalledTimes(2); expect(dispatchMock).toHaveBeenNthCalledWith(1, { type: GET_SHORT_URL_VISITS_START }); expect(dispatchMock).toHaveBeenNthCalledWith(2, { type: GET_SHORT_URL_VISITS_ERROR }); expect(ShlinkApiClient.getShortUrlVisits).toHaveBeenCalledTimes(1); }); it.each([ [ undefined, undefined ], [{}, undefined ], [{ domain: 'foobar.com' }, 'foobar.com' ], ])('dispatches start and success when promise is resolved', async (query, domain) => { const visits = visitsMocks; const shortCode = 'abc123'; const ShlinkApiClient = buildApiClientMock(Promise.resolve({ data: visitsMocks, pagination: { currentPage: 1, pagesCount: 1, totalItems: 1, }, })); await getShortUrlVisits(() => ShlinkApiClient)(shortCode, query)(dispatchMock, getState); expect(dispatchMock).toHaveBeenCalledTimes(2); expect(dispatchMock).toHaveBeenNthCalledWith(1, { type: GET_SHORT_URL_VISITS_START }); expect(dispatchMock).toHaveBeenNthCalledWith(2, { type: GET_SHORT_URL_VISITS, visits, shortCode, domain }); expect(ShlinkApiClient.getShortUrlVisits).toHaveBeenCalledTimes(1); }); it('performs multiple API requests when response contains more pages', async () => { const expectedRequests = 3; const ShlinkApiClient = buildApiClientMock(async (_, { page }) => Promise.resolve({ data: visitsMocks, pagination: { currentPage: page, pagesCount: expectedRequests, totalItems: 1, }, })); await getShortUrlVisits(() => ShlinkApiClient)('abc123')(dispatchMock, getState); expect(ShlinkApiClient.getShortUrlVisits).toHaveBeenCalledTimes(expectedRequests); expect(dispatchMock).toHaveBeenNthCalledWith(3, expect.objectContaining({ visits: [ ...visitsMocks, ...visitsMocks, ...visitsMocks ], })); }); }); describe('cancelGetShortUrlVisits', () => { it('just returns the action with proper type', () => expect(cancelGetShortUrlVisits()).toEqual({ type: GET_SHORT_URL_VISITS_CANCEL })); }); });