GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-08 00:47:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

create ResponseObject, each test result will be associated with a ResponseObject

Browse Source
This commit is contained in:
httprunner
2017-06-28 23:08:17 +08:00
parent 638733d9b6
commit f5705e37f8
3 changed files with 128 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ate/response.py
+22 -15
View File
@@ -1,22 +1,30 @@
from ate import utils, exception
def parse_response_body(resp_obj):
try:
return resp_obj.json()
except ValueError:
return resp_obj.text
class ResponseObject(object):
def parse_response_object(resp_obj):
def __init__(self, resp_obj):
""" initialize with a requests.Response object
@param (requests.Response instance) resp_obj
"""
self.resp_obj = resp_obj
def parse_response_body(self):
try:
return self.resp_obj.json()
except ValueError:
return self.resp_obj.text
def parse_response_object(self):
return {
'status_code': resp_obj.status_code,
'headers': resp_obj.headers,
'body': parse_response_body(resp_obj)
'status_code': self.resp_obj.status_code,
'headers': self.resp_obj.headers,
'body': self.parse_response_body()
}
def diff_response(resp_obj, expected_resp_json):
def diff_response(self, expected_resp_json):
diff_content = {}
resp_info = parse_response_object(resp_obj)
resp_info = self.parse_response_object()
expected_status_code = expected_resp_json.get('status_code', 200)
if resp_info['status_code'] != int(expected_status_code):
@@ -53,9 +61,8 @@ def diff_response(resp_obj, expected_resp_json):
return diff_content
def extract_response(resp_obj, context, delimiter='.'):
def extract_response(self, context, delimiter='.'):
""" extract content from requests.Response, and bind extracted value to context.extractors
@param (requests.Response instance) resp_obj
@param (ate.context.Context instance) context
context.extractors:
{
@@ -74,9 +81,9 @@ def extract_response(resp_obj, context, delimiter='.'):
top_query, sub_query = value.split(delimiter, 1)
if top_query in ["body", "content", "text"]:
json_content = parse_response_body(resp_obj)
json_content = self.parse_response_body()
else:
json_content = getattr(resp_obj, top_query)
json_content = getattr(self.resp_obj, top_query)
if sub_query:
# e.g. key: resp_headers_content_type, sub_query = "content-type"
ate/runner.py
+5 -3
View File
@@ -103,9 +103,11 @@ class TestRunner(object):
except KeyError:
raise exception.ParamsError("URL or METHOD missed!")
resp_obj = self.client.request(url=url, method=method, **req_kwargs)
response.extract_response(resp_obj, self.context)
diff_content = response.diff_response(resp_obj, testcase['response'])
resp = self.client.request(url=url, method=method, **req_kwargs)
resp_obj = response.ResponseObject(resp)
resp_obj.extract_response(self.context)
diff_content = resp_obj.diff_response(testcase['response'])
success = False if diff_content else True
return success, diff_content
test/test_response.py
+43 -28
View File
@@ -7,8 +7,9 @@ class TestResponse(ApiServerUnittest):
def test_parse_response_object_json(self):
url = "http://127.0.0.1:5000/api/users"
resp_obj = requests.get(url)
parse_result = response.parse_response_object(resp_obj)
resp = requests.get(url)
resp_obj = response.ResponseObject(resp)
parse_result = resp_obj.parse_response_object()
self.assertIn('status_code', parse_result)
self.assertIn('headers', parse_result)
self.assertIn('body', parse_result)
@@ -18,8 +19,9 @@ class TestResponse(ApiServerUnittest):
def test_parse_response_object_text(self):
url = "http://127.0.0.1:5000/"
resp_obj = requests.get(url)
parse_result = response.parse_response_object(resp_obj)
resp = requests.get(url)
resp_obj = response.ResponseObject(resp)
parse_result = resp_obj.parse_response_object()
self.assertIn('status_code', parse_result)
self.assertIn('headers', parse_result)
self.assertIn('body', parse_result)
@@ -29,7 +31,7 @@ class TestResponse(ApiServerUnittest):
def test_diff_response_status_code_equal(self):
status_code = random.randint(200, 511)
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'status_code': status_code,
@@ -39,12 +41,13 @@ class TestResponse(ApiServerUnittest):
expected_resp_json = {
'status_code': status_code
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertFalse(diff_content)
def test_diff_response_status_code_not_equal(self):
status_code = random.randint(200, 511)
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'status_code': status_code,
@@ -54,14 +57,15 @@ class TestResponse(ApiServerUnittest):
expected_resp_json = {
'status_code': 512
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertIn('value', diff_content['status_code'])
self.assertIn('expected', diff_content['status_code'])
self.assertEqual(diff_content['status_code']['value'], status_code)
self.assertEqual(diff_content['status_code']['expected'], 512)
def test_diff_response_headers_equal(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
@@ -77,11 +81,12 @@ class TestResponse(ApiServerUnittest):
'def': '456'
}
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertFalse(diff_content)
def test_diff_response_headers_not_equal(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
@@ -99,7 +104,8 @@ class TestResponse(ApiServerUnittest):
'd': 890
}
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertEqual(
diff_content['headers'],
{
@@ -109,7 +115,7 @@ class TestResponse(ApiServerUnittest):
)
def test_diff_response_body_equal(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'body': {
@@ -121,7 +127,8 @@ class TestResponse(ApiServerUnittest):
# expected response body is not specified
expected_resp_json = {}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertFalse(diff_content)
# response body is the same as expected response body
@@ -131,11 +138,12 @@ class TestResponse(ApiServerUnittest):
'count': '10'
}
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertFalse(diff_content)
def test_diff_response_body_not_equal_type_unmatch(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'body': {
@@ -149,7 +157,8 @@ class TestResponse(ApiServerUnittest):
expected_resp_json = {
'body': "ok"
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertEqual(
diff_content['body'],
{
@@ -159,7 +168,7 @@ class TestResponse(ApiServerUnittest):
)
def test_diff_response_body_not_equal_string_unmatch(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'body': "success"
@@ -170,7 +179,8 @@ class TestResponse(ApiServerUnittest):
expected_resp_json = {
'body': "ok"
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertEqual(
diff_content['body'],
{
@@ -180,7 +190,7 @@ class TestResponse(ApiServerUnittest):
)
def test_diff_response_body_not_equal_json_unmatch(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'body': {
@@ -196,7 +206,8 @@ class TestResponse(ApiServerUnittest):
'count': 10
}
}
diff_content = response.diff_response(resp_obj, expected_resp_json)
resp_obj = response.ResponseObject(resp)
diff_content = resp_obj.diff_response(expected_resp_json)
self.assertEqual(
diff_content['body'],
{
@@ -212,7 +223,7 @@ class TestResponse(ApiServerUnittest):
)
def test_extract_response_json(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
@@ -244,7 +255,8 @@ class TestResponse(ApiServerUnittest):
test_context = context.Context()
test_context.bind_extractors(extract_binds)
response.extract_response(resp_obj, test_context)
resp_obj = response.ResponseObject(resp)
resp_obj.extract_response(test_context)
extract_binds_dict = test_context.extractors
self.assertEqual(
@@ -274,7 +286,7 @@ class TestResponse(ApiServerUnittest):
def test_extract_response_fail(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
@@ -300,9 +312,10 @@ class TestResponse(ApiServerUnittest):
test_context = context.Context()
test_context.bind_extractors(extract_binds)
resp_obj = response.ResponseObject(resp)
with self.assertRaises(exception.ParamsError):
response.extract_response(resp_obj, test_context)
resp_obj.extract_response(test_context)
extract_binds = {
"resp_content_list_index_error": "content.person.cities.3"
@@ -310,12 +323,13 @@ class TestResponse(ApiServerUnittest):
test_context = context.Context()
test_context.bind_extractors(extract_binds)
resp_obj = response.ResponseObject(resp)
with self.assertRaises(exception.ParamsError):
response.extract_response(resp_obj, test_context)
resp_obj.extract_response(test_context)
def test_extract_response_json_string(self):
resp_obj = requests.post(
resp = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
@@ -331,7 +345,8 @@ class TestResponse(ApiServerUnittest):
test_context = context.Context()
test_context.bind_extractors(extract_binds)
response.extract_response(resp_obj, test_context)
resp_obj = response.ResponseObject(resp)
resp_obj.extract_response(test_context)
extract_binds_dict = test_context.extractors
self.assertEqual(