From 5d6e2afaeb901ca1b2f3a9fb4d6257557fa96d03 Mon Sep 17 00:00:00 2001 From: noO0oOo0ob <38344038+noO0oOo0ob@users.noreply.github.com> Date: Fri, 8 Mar 2024 13:16:07 +0800 Subject: [PATCH] fix mock handler not change keep-request-body flag when set ssr (#832) --- lyrebird/mock/handlers/flow_editor_handler.py | 2 +- lyrebird/mock/handlers/handler_context.py | 3 ++- lyrebird/version.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/lyrebird/mock/handlers/flow_editor_handler.py b/lyrebird/mock/handlers/flow_editor_handler.py index 641bb474b..99d3f2373 100644 --- 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 index 14208643b..a7cc50e09 100644 --- 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 index a30493389..e11292d0f 100644 --- 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