import { Mock } from 'ts-mockery'; import { addDays, formatISO, subDays } from 'date-fns'; import { getNonOrphanVisits as getNonOrphanVisitsCreator, nonOrphanVisitsReducerCreator, } from '../../../src/visits/reducers/nonOrphanVisits'; import { rangeOf } from '../../../src/utils/utils'; import { Visit } from '../../../src/visits/types'; import { ShlinkVisits } from '../../../src/api/types'; import { ShlinkApiClient } from '../../../src/api/services/ShlinkApiClient'; import { ShlinkState } from '../../../src/container/types'; import { formatIsoDate } from '../../../src/utils/helpers/date'; import { DateInterval } from '../../../src/utils/helpers/dateIntervals'; import { createNewVisits } from '../../../src/visits/reducers/visitCreation'; import { VisitsInfo } from '../../../src/visits/reducers/types'; describe('nonOrphanVisitsReducer', () => { const now = new Date(); const visitsMocks = rangeOf(2, () => Mock.all()); const getNonOrphanVisitsCall = jest.fn(); const buildShlinkApiClient = () => Mock.of({ getNonOrphanVisits: getNonOrphanVisitsCall }); const creator = getNonOrphanVisitsCreator(buildShlinkApiClient); const { asyncThunk: getNonOrphanVisits, progressChangedAction, largeAction, fallbackToIntervalAction } = creator; const { reducer, cancelGetVisits: cancelGetNonOrphanVisits } = nonOrphanVisitsReducerCreator(creator); beforeEach(jest.clearAllMocks); describe('reducer', () => { const buildState = (data: Partial) => Mock.of(data); it('returns loading on GET_NON_ORPHAN_VISITS_START', () => { const { loading } = reducer(buildState({ loading: false }), { type: getNonOrphanVisits.pending.toString() }); expect(loading).toEqual(true); }); it('returns loadingLarge on GET_NON_ORPHAN_VISITS_LARGE', () => { const { loadingLarge } = reducer(buildState({ loadingLarge: false }), { type: largeAction.toString() }); expect(loadingLarge).toEqual(true); }); it('returns cancelLoad on GET_NON_ORPHAN_VISITS_CANCEL', () => { const { cancelLoad } = reducer(buildState({ cancelLoad: false }), { type: cancelGetNonOrphanVisits.toString() }); expect(cancelLoad).toEqual(true); }); it('stops loading and returns error on GET_NON_ORPHAN_VISITS_ERROR', () => { const { loading, error } = reducer( buildState({ loading: true, error: false }), { type: getNonOrphanVisits.rejected.toString() }, ); expect(loading).toEqual(false); expect(error).toEqual(true); }); it('return visits on GET_NON_ORPHAN_VISITS', () => { const actionVisits = [{}, {}]; const { loading, error, visits } = reducer(buildState({ loading: true, error: false }), { type: getNonOrphanVisits.fulfilled.toString(), payload: { visits: actionVisits }, }); expect(loading).toEqual(false); expect(error).toEqual(false); expect(visits).toEqual(actionVisits); }); it.each([ [{}, visitsMocks.length + 2], [ Mock.of({ query: { endDate: formatIsoDate(subDays(now, 1)) ?? undefined }, }), visitsMocks.length, ], [ Mock.of({ query: { startDate: formatIsoDate(addDays(now, 1)) ?? undefined }, }), visitsMocks.length, ], [ Mock.of({ query: { startDate: formatIsoDate(subDays(now, 5)) ?? undefined, endDate: formatIsoDate(subDays(now, 2)) ?? undefined, }, }), visitsMocks.length, ], [ Mock.of({ query: { startDate: formatIsoDate(subDays(now, 5)) ?? undefined, endDate: formatIsoDate(addDays(now, 3)) ?? undefined, }, }), visitsMocks.length + 2, ], ])('prepends new visits on CREATE_VISIT', (state, expectedVisits) => { const prevState = buildState({ ...state, visits: visitsMocks }); const visit = Mock.of({ date: formatIsoDate(now) ?? undefined }); const { visits } = reducer(prevState, { type: createNewVisits.toString(), payload: { createdVisits: [{ visit }, { visit }] }, }); expect(visits).toHaveLength(expectedVisits); }); it('returns new progress on GET_NON_ORPHAN_VISITS_PROGRESS_CHANGED', () => { const state = reducer(undefined, { type: progressChangedAction.toString(), payload: 85 }); expect(state).toEqual(expect.objectContaining({ progress: 85 })); }); it('returns fallbackInterval on GET_NON_ORPHAN_VISITS_FALLBACK_TO_INTERVAL', () => { const fallbackInterval: DateInterval = 'last30Days'; const state = reducer( undefined, { type: fallbackToIntervalAction.toString(), payload: fallbackInterval }, ); expect(state).toEqual(expect.objectContaining({ fallbackInterval })); }); }); describe('getNonOrphanVisits', () => { const dispatchMock = jest.fn(); const getState = () => Mock.of({ orphanVisits: { cancelLoad: false }, }); beforeEach(jest.resetAllMocks); it('dispatches start and error when promise is rejected', async () => { getNonOrphanVisitsCall.mockRejectedValue({}); await getNonOrphanVisits({})(dispatchMock, getState, {}); expect(dispatchMock).toHaveBeenCalledTimes(2); expect(dispatchMock).toHaveBeenNthCalledWith(1, expect.objectContaining({ type: getNonOrphanVisits.pending.toString(), })); expect(dispatchMock).toHaveBeenNthCalledWith(2, expect.objectContaining({ type: getNonOrphanVisits.rejected.toString(), })); expect(getNonOrphanVisitsCall).toHaveBeenCalledTimes(1); }); it.each([ [undefined], [{}], ])('dispatches start and success when promise is resolved', async (query) => { const visits = visitsMocks.map((visit) => ({ ...visit, visitedUrl: '' })); getNonOrphanVisitsCall.mockResolvedValue({ data: visits, pagination: { currentPage: 1, pagesCount: 1, totalItems: 1, }, }); await getNonOrphanVisits({ query })(dispatchMock, getState, {}); expect(dispatchMock).toHaveBeenCalledTimes(2); expect(dispatchMock).toHaveBeenNthCalledWith(1, expect.objectContaining( { type: getNonOrphanVisits.pending.toString() }, )); expect(dispatchMock).toHaveBeenNthCalledWith(2, expect.objectContaining({ type: getNonOrphanVisits.fulfilled.toString(), payload: { visits, query: query ?? {} }, })); expect(getNonOrphanVisitsCall).toHaveBeenCalledTimes(1); }); it.each([ [ [Mock.of({ date: formatISO(subDays(new Date(), 5)) })], { type: fallbackToIntervalAction.toString(), payload: 'last7Days' }, 3, ], [ [Mock.of({ date: formatISO(subDays(new Date(), 200)) })], { type: fallbackToIntervalAction.toString(), payload: 'last365Days' }, 3, ], [[], expect.objectContaining({ type: getNonOrphanVisits.fulfilled.toString() }), 2], ])('dispatches fallback interval when the list of visits is empty', async ( lastVisits, expectedSecondDispatch, expectedAmountOfDispatches, ) => { const buildVisitsResult = (data: Visit[] = []): ShlinkVisits => ({ data, pagination: { currentPage: 1, pagesCount: 1, totalItems: 1, }, }); getNonOrphanVisitsCall .mockResolvedValueOnce(buildVisitsResult()) .mockResolvedValueOnce(buildVisitsResult(lastVisits)); await getNonOrphanVisits({ doIntervalFallback: true })(dispatchMock, getState, {}); expect(dispatchMock).toHaveBeenCalledTimes(expectedAmountOfDispatches); expect(dispatchMock).toHaveBeenNthCalledWith(1, expect.objectContaining({ type: getNonOrphanVisits.pending.toString(), })); expect(dispatchMock).toHaveBeenNthCalledWith(2, expectedSecondDispatch); expect(getNonOrphanVisitsCall).toHaveBeenCalledTimes(2); }); }); describe('cancelGetNonOrphanVisits', () => { it('just returns the action with proper type', () => expect(cancelGetNonOrphanVisits()).toEqual({ type: cancelGetNonOrphanVisits.toString() })); }); });