fix mock handler not change keep-request-body flag when set ssr (#832)

Meituan-Dianping · Mar 8, 2024 · 5d6e2af · 5d6e2af
1 parent b5a523f
commit 5d6e2af
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/lyrebird/mock/handlers/flow_editor_handler.py b/lyrebird/mock/handlers/flow_editor_handler.py
@@ -16,7 +16,7 @@ def on_request_handler(self, handler_context):
             return
 
         self.script_executor(matched_funcs, handler_context.flow)
-        handler_context.set_request_edited()
+        handler_context.set_request_edited(handler_context.flow.get('keep_origin_request_body', False))
         handler_context.flow['request']['headers']['lyrebird_modified'] = 'modified'
 
     def on_request_upstream_handler(self, handler_context):

diff --git a/lyrebird/mock/handlers/handler_context.py b/lyrebird/mock/handlers/handler_context.py
@@ -171,7 +171,8 @@ def _read_origin_request_info_from_header(self, headers=None):
             path=self.request.path[len(self.MOCK_PATH_PREFIX):]
         )
 
-    def set_request_edited(self):
+    def set_request_edited(self, keep_origin_request_body=False):
+        self.flow['keep_origin_request_body'] = self.flow.get('keep_origin_request_body', False) and keep_origin_request_body
         self.is_request_edited = True
 
     def set_response_edited(self):

diff --git a/lyrebird/version.py b/lyrebird/version.py
@@ -1,3 +1,3 @@
-IVERSION = (2, 25, 2)
+IVERSION = (2, 25, 3)
 VERSION = ".".join(str(i) for i in IVERSION)
 LYREBIRD = "Lyrebird " + VERSION