-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrequests_mock_assertion.py
85 lines (68 loc) · 2.7 KB
/
requests_mock_assertion.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
from json import JSONDecodeError
from pathlib import Path
from bx_py_utils.test_utils.assertion import assert_equal
from bx_py_utils.test_utils.snapshot import assert_snapshot
def build_requests_mock_history(mock, only_json=True):
history = []
for request in mock.request_history:
request_info = {
'request': f'{request.method} {request.url}'
}
if request.body:
try:
json_data = request.json()
except JSONDecodeError as err:
if only_json:
raise AssertionError(f'{request.method} {request.url} without valid JSON: {err} in:\n{err.doc!r}')
request_info['text'] = request.text
else:
request_info['json'] = json_data
history.append(request_info)
return history
def assert_json_requests_mock(mock, data):
"""
Check the requests mock history. In this case all requests must be JSON.
e.g.:
with requests_mock.mock() as m:
m.post('http://test.tld', text='resp')
requests.post('http://test.tld', json={'foo': 'bar'})
assert_json_requests_mock(mock=m, data=[{
'request': 'POST http://test.tld/',
'json': {'foo': 'bar'},
}])
"""
history = build_requests_mock_history(mock, only_json=True)
assert_equal(history, data, msg='Request history are not equal:')
def assert_json_requests_mock_snapshot(mock):
"""
Check requests mock history via snapshot. Accepts only JSON requests.
:param mock:
:return:
"""
history = build_requests_mock_history(mock, only_json=True)
assert_snapshot(got=history, self_file_path=Path(__file__))
def assert_requests_mock(mock, data):
"""
Check the requests mock history. Accept mixed "text" and "JSON".
e.g.:
with requests_mock.mock() as m:
m.get('http://test.tld', text='foo')
m.post('http://test.tld', text='bar')
requests.post('http://test.tld', data={'foo': 'one'})
requests.post('http://test.tld', json={'foo': 'two'})
assert_requests_mock(mock=m, data=[{
'request': 'POST http://test.tld/',
'text': 'foo=one',
}, {
'request': 'POST http://test.tld/',
'json': {'foo': 'two'},
}])
"""
history = build_requests_mock_history(mock, only_json=False)
assert_equal(history, data, msg='Request history are not equal:')
def assert_requests_mock_snapshot(mock):
"""
Check requests mock history via snapshot. Accept mixed "text" and "JSON".
"""
history = build_requests_mock_history(mock, only_json=False)
assert_snapshot(got=history, self_file_path=Path(__file__))