From 2538743c6b182b92811181b5b9b251a4656a1858 Mon Sep 17 00:00:00 2001 From: debugtalk Date: Fri, 28 Dec 2018 16:15:18 +0800 Subject: [PATCH] update omitting request & response length: 1, request: only omit upload file type; 2, response: keep raw content for extracting; omit in summary --- httprunner/built_in.py | 1 - httprunner/client.py | 9 ++++++++- httprunner/response.py | 3 --- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/httprunner/built_in.py b/httprunner/built_in.py index cbbc85d1..e3e0d1bd 100644 --- a/httprunner/built_in.py +++ b/httprunner/built_in.py @@ -14,7 +14,6 @@ import time from httprunner.compat import basestring, builtin_str, integer_types, str from httprunner.exceptions import ParamsError -from httprunner.utils import convert_dict_to_params, lower_dict_keys from requests_toolbelt import MultipartEncoder diff --git a/httprunner/client.py b/httprunner/client.py index de060ff3..a31031c2 100644 --- a/httprunner/client.py +++ b/httprunner/client.py @@ -86,7 +86,14 @@ class HttpSession(requests.Session): request_body = resp_obj.request.body if request_body: - req_resp_dict["request"]["body"] = omit_long_data(request_body) + request_content_type = lower_dict_keys( + req_resp_dict["request"]["headers"] + ).get("content-type") + if request_content_type and "multipart/form-data" in request_content_type: + # upload file type + req_resp_dict["request"]["body"] = "upload file stream (OMITTED)" + else: + req_resp_dict["request"]["body"] = request_body # log request details in debug mode log_print(req_resp_dict, "request") diff --git a/httprunner/response.py b/httprunner/response.py index f13e2bc5..aef4e9a1 100644 --- a/httprunner/response.py +++ b/httprunner/response.py @@ -31,9 +31,6 @@ class ResponseObject(object): else: value = getattr(self.resp_obj, key) - if key in ["text", "content"]: - value = utils.omit_long_data(value) - self.__dict__[key] = value return value except AttributeError: