fix #332: display image type

This commit is contained in:
httprunner
2018-08-03 23:36:05 +08:00
parent 8466e4aef4
commit b4efbfcb2a
2 changed files with 6 additions and 8 deletions

View File

@@ -120,9 +120,7 @@ def stringify_data(meta_data, request_or_response):
} }
""" """
headers = meta_data[request_or_response]["headers"] headers = meta_data[request_or_response]["headers"]
request_or_response_dict = meta_data[request_or_response] request_or_response_dict = meta_data[request_or_response]
response_content_type = meta_data["response"]["content_type"]
for key, value in request_or_response_dict.items(): for key, value in request_or_response_dict.items():
@@ -135,11 +133,11 @@ def stringify_data(meta_data, request_or_response):
if not encoding or encoding == "None": if not encoding or encoding == "None":
encoding = "utf-8" encoding = "utf-8"
if request_or_response == "response" and key == "content" and "image" in response_content_type: if request_or_response == "response" and key == "content" \
and "image" in meta_data["response"]["content_type"]:
# display image # display image
meta_data["response"]["content_type"] = "image"
value = "data:{};base64,{}".format( value = "data:{};base64,{}".format(
response_content_type, meta_data["response"]["content_type"],
b64encode(value).decode(encoding) b64encode(value).decode(encoding)
) )
else: else:

View File

@@ -272,7 +272,7 @@
<div style="overflow: auto"> <div style="overflow: auto">
<table> <table>
{% for key, value in record.meta_data.response.items() %} {% for key, value in record.meta_data.response.items() %}
{% if key in ["content", "json", "elapsed_ms", "response_time_ms", "content_size", "content_type"] %} {% if key in ["text", "json", "elapsed_ms", "response_time_ms", "content_size", "content_type"] %}
{% continue %} {% continue %}
{% endif %} {% endif %}
<tr> <tr>
@@ -284,8 +284,8 @@
<strong>{{ header_key }}</strong>: {{ header_value }} <strong>{{ header_key }}</strong>: {{ header_value }}
</div> </div>
{% endfor %} {% endfor %}
{% elif key == "text" %} {% elif key == "content" %}
{% if value == "image" %} {% if "image" in record.meta_data.response.content_type %}
<img src="{{ record.meta_data.response.content }}" /> <img src="{{ record.meta_data.response.content }}" />
{% else %} {% else %}
<pre>{{ record.meta_data.response.text | e }}</pre> <pre>{{ record.meta_data.response.text | e }}</pre>