UNPKG

@rocketmakers/api-swr

Version:

Rocketmakers front-end library for parsing a generated Typescript API client into a set of configurable React hooks for fetching and mutating data.

77 lines (76 loc) 3.34 kB
import { act, renderHook } from '@testing-library/react'; import * as React from 'react'; import useSwr from 'swr'; import * as useClientFetchModule from './useClientFetch'; import { useQuery } from './useQuery'; // Mock SWR hook jest.mock('swr', () => ({ __esModule: true, default: jest.fn((key, innerFetcher) => { const [data, setData] = React.useState(); React.useEffect(() => { if (key) { void innerFetcher().then((fetchedData) => act(() => setData(fetchedData))); } }, [innerFetcher, key]); return { data }; }), })); // Define test data const fetcher = jest.fn().mockReturnValue(Promise.resolve({ test: 'OK' })); const endpointId = 'controllerKey.endpointKey'; const hookConfig = { cacheKey: 'id', params: { id: 'test-cache-key-value' }, fetchConfig: { headers: { Authorization: 'Bearer token' } }, swrConfig: { revalidateOnFocus: false }, }; // Mock client fetch const mockUseClientFetch = jest.fn((incomingEndpointId, mode, conf, innerFetcher, params) => { return { validationErrors: [], clientFetch: () => innerFetcher(params, conf), data: undefined, error: undefined, isLoading: false, }; }); jest.mock('./useClientFetch.ts'); useClientFetchModule.useClientFetch.mockImplementation(mockUseClientFetch); describe('useQuery', () => { beforeEach(() => { jest.clearAllMocks(); }); it('should call useSwr with the correct parameters when used', () => { const { result } = renderHook(() => useQuery(endpointId, fetcher, hookConfig)); expect(result.current).toBeTruthy(); expect(useSwr).toHaveBeenCalledWith('controllerKey.endpointKey.test-cache-key-value', expect.any(Function), { revalidateOnFocus: false, }); }); it('should combine hook level and global SWR configs when used', () => { const { result } = renderHook(() => useQuery(endpointId, fetcher, hookConfig, undefined, undefined, { errorRetryCount: 5 })); expect(result.current).toBeTruthy(); expect(useSwr).toHaveBeenCalledWith('controllerKey.endpointKey.test-cache-key-value', expect.any(Function), { errorRetryCount: 5, revalidateOnFocus: false, }); }); it('should call useClientFetch with the correct parameters when used', () => { const { result } = renderHook(() => useQuery(endpointId, fetcher, hookConfig)); expect(result.current).toBeTruthy(); expect(mockUseClientFetch).toHaveBeenCalledWith(endpointId, 'query', hookConfig.fetchConfig, fetcher, hookConfig.params, undefined, undefined, undefined); }); it('should not call fetcher if waitFor is false', () => { renderHook(() => useQuery(endpointId, fetcher, Object.assign(Object.assign({}, hookConfig), { waitFor: false }))); expect(fetcher).not.toHaveBeenCalled(); }); it('should call fetcher if waitFor is undefined', () => { renderHook(() => useQuery(endpointId, fetcher, hookConfig)); expect(fetcher).toHaveBeenCalled(); }); it('should call fetcher if waitFor is true', () => { renderHook(() => useQuery(endpointId, fetcher, Object.assign(Object.assign({}, hookConfig), { waitFor: true }))); expect(fetcher).toHaveBeenCalled(); }); });