keep original request body when the request is mocked and diff_mode i… (

#800) * keep original request body when the request is mocked and diff_mode is enabled * use flow['origin_request'] --------- Co-authored-by: yangfan100 <[email protected]>
Meituan-Dianping · Nov 9, 2023 · afcae22 · afcae22
1 parent c37d86d
commit afcae22
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lyrebird/mock/handlers/handler_context.py b/lyrebird/mock/handlers/handler_context.py
@@ -189,6 +189,9 @@ def get_request_body(self, in_request_handler=True):
         else:
             if in_request_handler:
                 _data = self.request.data or self.request.form or None
+            # When origin_request is not saved, the original data cannot be obtained when diff-mode is enabled.
+            elif self.flow.get('origin_request', {}).get('data'):
+                _data = self.flow.get('origin_request', {}).get('data')
             else:
                 _data = DataHelper.flow2origin(self.flow['request'])
         return _data

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