feat: accepts promises for change config and change response

Author: igorjacauna

src/runtime/handler.ts

@@ -90,10 +90,10 @@ export default class Handler {
     const pathConfig = this.path;
     const paramsConfig = this.params;
     const interceptorId = this.scope.axios.interceptors.request.use(
-      requestConfig => {
+      async requestConfig => {
         if (matchRequest(verbConfig, pathConfig, requestConfig, paramsConfig)) {
           if (once) ejectFromRequest(this.scope.axios, interceptorId);
-          const result = configChanger(requestConfig);
+          const result = await Promise.resolve(configChanger(requestConfig));
           return result;
         }
         return requestConfig;
@@ -127,10 +127,10 @@ export default class Handler {
     const pathConfig = this.path;
     const paramsConfig = this.params;
     const interceptorId = this.scope.axios.interceptors.response.use(
-      (response: AxiosResponse<T>) => {
+      async (response: AxiosResponse<T>) => {
         if (matchResponse(verbConfig, pathConfig, response, paramsConfig)) {
           if (once) ejectFromResponse(this.scope.axios, interceptorId);
-          response.data = responseChanger(response.data);
+          response.data = await Promise.resolve(responseChanger(response.data));
         }
         return response;
       },

src/types.ts

@@ -6,7 +6,7 @@ export type RouteConfig = (
 
 export type RequestConfigChanger = (
   config: InternalAxiosRequestConfig,
-) => InternalAxiosRequestConfig;
+) => InternalAxiosRequestConfig | Promise<InternalAxiosRequestConfig>;
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-export type ResponseChanger<T = any> = (data: T) => T;
+export type ResponseChanger<T = any> = (data: T) => T | Promise<T>;

test/index.test.ts

@@ -244,6 +244,28 @@ describe('axios-dev-proxy tests', () => {
       expect(response.data).toEqual({ data: 2, xpto: 2 });
       expect(response.status).toEqual(200);
     });
+
+    it('should change original response with promise', async () => {
+      server.get('/').reply(200, { data: 1, xpto: 2 });
+
+      proxy.onGet('/').changeResponseDataOnce<{
+        data: number;
+        xpto: number;
+      }>(
+        data =>
+          new Promise(resolve => {
+            resolve({
+              ...data,
+              data: 2,
+            });
+          }),
+      );
+
+      const response = await api.get('/');
+
+      expect(response.data).toEqual({ data: 2, xpto: 2 });
+      expect(response.status).toEqual(200);
+    });
   });
 
   describe('always GET configs', () => {
@@ -448,5 +470,24 @@ describe('axios-dev-proxy tests', () => {
       expect(server.pendingMocks()).toHaveLength(1);
       expect(server2.pendingMocks()).toHaveLength(1);
     });
+
+    it('should change the request config once with a promise', async () => {
+      const server2 = nock('https://api-2.com.br');
+      server2.get('/config').reply(200, { data: 2 });
+      server.get('/config').reply(200, { data: 1 });
+      proxy.onGet('/config').changeRequestOnce(
+        async config =>
+          new Promise(resolve => {
+            config.baseURL = 'https://api-2.com.br';
+            resolve(config);
+          }),
+      );
+
+      await api.get('/config');
+      await api.get('/config');
+
+      expect(server.isDone()).toBe(true);
+      expect(server2.isDone()).toBe(true);
+    });
   });
 });