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

diff http response: headers

Browse Source
This commit is contained in:
httprunner
2017-06-21 15:56:01 +08:00
parent 1bd2281be6
commit 5e03acba5f
2 changed files with 73 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ate/utils.py
+13
View File
@@ -44,4 +44,17 @@ def diff_response(resp_obj, expected_resp_json):
'expected': expected_status_code 'expected': expected_status_code
} }
expected_headers = expected_resp_json.get('headers', {})
for header_key, expected_header_value in expected_headers.items():
header_value = resp_info['headers'].get(header_key, None)
if str(header_value) != str(expected_header_value):
if 'headers' not in diff_content:
diff_content['headers'] = {}
diff_content['headers'][header_key] = {
'value': header_value,
'expected': str(expected_header_value)
}
return diff_content return diff_content
test/test_utils.py
+60 -11
View File
@@ -56,11 +56,12 @@ class TestUtils(ApiServerUnittest):
def test_diff_response_status_code_equal(self): def test_diff_response_status_code_equal(self):
status_code = random.randint(200, 511) status_code = random.randint(200, 511)
url = "http://127.0.0.1:5000/customize-response" resp_obj = requests.post(
response_dict = { url="http://127.0.0.1:5000/customize-response",
'status_code': status_code, json={
} 'status_code': status_code,
resp_obj = requests.post(url, json=response_dict) }
)
expected_resp_json = { expected_resp_json = {
'status_code': status_code 'status_code': status_code
@@ -70,18 +71,66 @@ class TestUtils(ApiServerUnittest):
def test_diff_response_status_code_not_equal(self): def test_diff_response_status_code_not_equal(self):
status_code = random.randint(200, 511) status_code = random.randint(200, 511)
url = "http://127.0.0.1:5000/customize-response" resp_obj = requests.post(
response_dict = { url="http://127.0.0.1:5000/customize-response",
'status_code': status_code, json={
} 'status_code': status_code,
resp_obj = requests.post(url, json=response_dict) }
)
expected_resp_json = { expected_resp_json = {
'status_code': 512 'status_code': 512
} }
diff_content = utils.diff_response(resp_obj, expected_resp_json) diff_content = utils.diff_response(resp_obj, expected_resp_json)
print('diff_content', diff_content)
self.assertIn('value', diff_content['status_code']) self.assertIn('value', diff_content['status_code'])
self.assertIn('expected', 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']['value'], status_code)
self.assertEqual(diff_content['status_code']['expected'], 512) self.assertEqual(diff_content['status_code']['expected'], 512)
def test_diff_response_headers_equal(self):
resp_obj = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
'abc': 123,
'def': 456
}
}
)
expected_resp_json = {
'headers': {
'abc': 123,
'def': '456'
}
}
diff_content = utils.diff_response(resp_obj, expected_resp_json)
self.assertFalse(diff_content)
def test_diff_response_headers_not_equal(self):
resp_obj = requests.post(
url="http://127.0.0.1:5000/customize-response",
json={
'headers': {
'a': 123,
'b': '456',
'c': '789'
}
}
)
expected_resp_json = {
'headers': {
'a': '123',
'b': '457',
'd': 890
}
}
diff_content = utils.diff_response(resp_obj, expected_resp_json)
self.assertEqual(
diff_content['headers'],
{
'b': {'expected': '457', 'value': '456'},
'd': {'expected': '890', 'value': None}
}
)