feat: response return url (#27)

vbenjs · Jun 4, 2021 · de9ed27 · de9ed27
1 parent b3454dd
commit de9ed27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/createMockServer.ts b/src/createMockServer.ts
@@ -80,7 +80,7 @@ export async function requestMiddleware(opt: ViteMockOptions) {
       } else {
         const body = await parseJson(req);
         const mockResponse = isFunction(response)
-          ? response({ body, query, headers: req.headers })
+          ? response({ url: req.url, body, query, headers: req.headers })
           : response;
         res.setHeader('Content-Type', 'application/json');
         res.statusCode = statusCode || 200;

diff --git a/src/types.ts b/src/types.ts
@@ -22,7 +22,7 @@ export declare interface MockMethod {
   method?: MethodType;
   timeout?: number;
   statusCode?: number;
-  response?: ((opt: { body: Recordable; query: Recordable; headers: Recordable }) => any) | any;
+  response?: ((opt: { url: Recordable, body: Recordable; query: Recordable; headers: Recordable }) => any) | any;
   rawResponse?: (req: IncomingMessage, res: ServerResponse) => void;
 }