packages/testing/lib/api.ts

   1 import noop from '@proton/utils/noop';
   2
   3 type HttpMethod = 'get' | 'post' | 'put' | 'delete';
   4
   5 type ApiMockHandler = (...arg: any[]) => any;
   6
   7 type ApiMockEntry = {
   8     method?: HttpMethod;
   9     handler: (...arg: any[]) => any;
  10 };
  11
  12 type ApiMock = { [url: string]: ApiMockEntry[] | undefined };
  13
  14 export const apiMocksMap: ApiMock = {};
  15
  16 let logging = false;
  17 export const enableMockApiLogging = () => (logging = true);
  18 export const disableMockApiLogging = () => (logging = false);
  19
  20 export const apiMock = jest.fn<Promise<any>, any>(async (args: any) => {
  21     const entryKey = Object.keys(apiMocksMap).find((path) => {
  22         return args.url === path;
  23     });
  24     const entry = apiMocksMap[entryKey || '']?.find(
  25         (entry) => entry.method === undefined || entry.method === args.method
  26     );
  27     if (entry) {
  28         if (logging) {
  29             console.log('apiMock', args);
  30             console.log('apiMock entry', entry);
  31         }
  32         const result = entry.handler({ ...args });
  33         if (logging) {
  34             console.log('apiMock result', result);
  35         }
  36         return result;
  37     }
  38     return {};
  39 });
  40
  41 export const addApiMock = (url: string, handler: ApiMockHandler, method?: HttpMethod) => {
  42     const newEntry = { method, handler };
  43     if (!apiMocksMap[url]) {
  44         apiMocksMap[url] = [newEntry];
  45     } else {
  46         apiMocksMap[url] = apiMocksMap[url]?.filter((entry) => entry.method !== newEntry.method).concat([newEntry]);
  47     }
  48 };
  49
  50 export const addApiResolver = (url: string, method?: HttpMethod) => {
  51     let resolveLastPromise: (result: any) => void = noop;
  52     let rejectLastPromise: (result: any) => void = noop;
  53
  54     const resolve = (value: any) => resolveLastPromise(value);
  55     const reject = (value: any) => rejectLastPromise(value);
  56
  57     const promise = new Promise((resolve, reject) => {
  58         resolveLastPromise = resolve;
  59         rejectLastPromise = reject;
  60     });
  61     addApiMock(url, () => promise, method);
  62     return { resolve, reject };
  63 };
  64
  65 export const clearApiMocks = () => {
  66     Object.keys(apiMocksMap).forEach((key) => delete apiMocksMap[key]);
  67 };