applications/drive/src/app/store/_links/useLinksQueue.test.tsx

   1 import { act, renderHook } from '@testing-library/react-hooks';
   2
   3 import { useLinksQueue } from './useLinksQueue';
   4
   5 const mockedLoadLinksMeta = jest.fn();
   6 jest.mock('./useLinksListing', () => ({
   7     useLinksListing: () => ({
   8         loadLinksMeta: mockedLoadLinksMeta,
   9     }),
  10 }));
  11
  12 const mockedGetLink = jest.fn();
  13 jest.mock('./useLinksState', () => ({
  14     __esModule: true,
  15     default: () => ({
  16         getLink: mockedGetLink,
  17     }),
  18 }));
  19
  20 jest.useFakeTimers();
  21 jest.spyOn(global, 'setTimeout');
  22
  23 const mockedAbort = jest.fn();
  24 let mockedAbortSignal = { aborted: false };
  25 // @ts-ignore - not mocking the entire AbortSignal, sorry =)
  26 global.AbortController = jest.fn(() => ({
  27     signal: mockedAbortSignal,
  28     abort: mockedAbort,
  29 }));
  30
  31 const SHARE_ID = 'shareId';
  32
  33 const getLinkIds = (count: number) => [...Array(count).keys()].map((id) => `linkId-${id}`);
  34 const getFakeRef: <T>(value: T) => React.MutableRefObject<T> = (value) => ({ current: value });
  35
  36 describe('useLinksQueue', () => {
  37     let hook: {
  38         current: ReturnType<typeof useLinksQueue>;
  39     };
  40     let unmountHook: () => void;
  41
  42     beforeEach(() => {
  43         const { result, unmount } = renderHook(() => useLinksQueue());
  44
  45         hook = result;
  46         unmountHook = unmount;
  47
  48         jest.clearAllTimers();
  49         jest.clearAllMocks();
  50         mockedGetLink.mockReset();
  51         mockedLoadLinksMeta.mockReset();
  52
  53         mockedAbort.mockReset();
  54         mockedAbortSignal.aborted = false;
  55
  56         mockedGetLink.mockImplementation(() => undefined);
  57     });
  58
  59     it('should not add to queue if link is already in state', async () => {
  60         mockedGetLink.mockImplementation((shareId, linkId) => ({
  61             shareId,
  62             linkId,
  63         }));
  64
  65         await act(async () => {
  66             hook.current.addToQueue(SHARE_ID, 'linkId');
  67         });
  68
  69         expect(setTimeout).not.toHaveBeenCalled();
  70     });
  71
  72     it('should not add to queue if linkId is already in queue', async () => {
  73         await act(async () => {
  74             hook.current.addToQueue(SHARE_ID, 'linkId');
  75             hook.current.addToQueue(SHARE_ID, 'linkId');
  76         });
  77
  78         expect(setTimeout).toHaveBeenCalledTimes(1);
  79     });
  80
  81     it('should debounce processing to wait for the queue to fill up', async () => {
  82         const items = getLinkIds(7);
  83
  84         await act(async () => {
  85             items.forEach((item) => hook.current.addToQueue(SHARE_ID, item));
  86         });
  87
  88         jest.runAllTimers();
  89
  90         expect(setTimeout).toHaveBeenCalledTimes(items.length);
  91         expect(mockedLoadLinksMeta).toHaveBeenCalledTimes(1);
  92     });
  93
  94     it('should not load links if the domRef is null', async () => {
  95         await act(async () => {
  96             hook.current.addToQueue(SHARE_ID, 'linkId', getFakeRef(null));
  97         });
  98
  99         jest.runAllTimers();
 100
 101         expect(setTimeout).toHaveBeenCalledTimes(1);
 102         expect(mockedLoadLinksMeta).not.toHaveBeenCalled();
 103     });
 104
 105     it('should abort when the hook is unmounted', async () => {
 106         await act(async () => {
 107             hook.current.addToQueue(SHARE_ID, 'linkId');
 108         });
 109
 110         jest.runAllTimers();
 111         unmountHook();
 112
 113         expect(mockedAbort).toHaveBeenCalled();
 114     });
 115
 116     it('should not load links if aborted', async () => {
 117         mockedAbortSignal.aborted = true;
 118
 119         await act(async () => {
 120             hook.current.addToQueue(SHARE_ID, 'linkId');
 121         });
 122
 123         jest.runAllTimers();
 124
 125         expect(setTimeout).toHaveBeenCalledTimes(1);
 126         expect(mockedLoadLinksMeta).not.toHaveBeenCalled();
 127     });
 128
 129     it('should not infinite loop if loadLinksMeta fails', async () => {
 130         // Silence console errors
 131         const consoleErrorMock = jest.spyOn(console, 'error').mockImplementation(() => {});
 132
 133         mockedLoadLinksMeta.mockRejectedValue(new Error('oh no'));
 134
 135         await act(async () => {
 136             hook.current.addToQueue(SHARE_ID, 'linkId');
 137         });
 138
 139         await jest.runAllTimersAsync();
 140
 141         expect(setTimeout).toHaveBeenCalledTimes(1);
 142         expect(mockedLoadLinksMeta).toHaveBeenCalledTimes(1);
 143         expect(consoleErrorMock).toHaveBeenCalledTimes(1);
 144
 145         consoleErrorMock.mockRestore();
 146     }, 1000);
 147 });