avoid forward lyrebird header in response (#888)

* avoid forward lyrebird header in response

---------

Co-authored-by: noO0ob <wujiasheng03@meituan.com>

lyrebird/mock/blueprints/core.py

@@ -52,13 +52,13 @@ def index(path=''):
         resp = Response(
             gen(),
             status=req_context.flow['response']['code'],
-            headers=req_context.flow['response']['headers']
+            headers=req_context.get_response_headers()
         )
     elif req_context.flow['response']:
         resp = Response(
             req_context.flow['response'].get('data', ''),
             status=req_context.flow['response'].get('code', 200),
-            headers=req_context.flow['response'].get('headers', {})
+            headers=req_context.get_response_headers()
         )
     else:
         path_not_found_handler.handle(req_context)

lyrebird/mock/handlers/handler_context.py

@@ -17,6 +17,7 @@
 
 logger = get_logger()
 proxy_handler = ProxyHandler()
+lyrebird_response_headers = None
 
 
 class HandlerContext:
@@ -213,7 +214,20 @@ def get_request_headers(self):
                 continue
             headers[name] = value
         return headers
-
+
+    # Before response returns, remove the Lyrebird internal headers
+    def get_response_headers(self):
+        global lyrebird_response_headers
+        if not lyrebird_response_headers:
+            lyrebird_response_headers = application.config.get('proxy.response.delete_headers', [])
+
+        headers = self.flow['response'].get('headers', {})
+        lyrebird_response_headers = []
+        for key in lyrebird_response_headers:
+            if key in headers:
+                del headers[key]
+        return headers
+
     def get_request_cookies(self, in_request_handler=True):
         if in_request_handler:
             return self.request.cookies