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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
try:
from http import HTTPStatus
HTTP_STATUS_FOUND = HTTPStatus.FOUND
except ImportError:
from httplib import FOUND as HTTP_STATUS_FOUND
import pytest
import requests
import requests_mock
def test_simple(requests_mock):
requests_mock.get('https://httpbin.org/get', text='data')
assert 'data' == requests.get('https://httpbin.org/get').text
def test_redirect_and_nesting():
url_inner = "inner-mock://example.test/"
url_middle = "middle-mock://example.test/"
url_outer = "outer-mock://example.test/"
url_base = "https://www.example.com/"
text_middle = 'middle' + url_middle
text_outer = 'outer' + url_outer
text_base = 'outer' + url_base
with requests_mock.Mocker() as outer_mock:
outer_mock.get(url_base, text=text_base)
outer_mock.get(url_outer, text=text_outer)
with requests_mock.Mocker(real_http=True) as middle_mock:
middle_mock.get(url_middle, text=text_middle)
with requests_mock.Mocker() as inner_mock:
inner_mock.post(url_inner,
status_code=HTTP_STATUS_FOUND,
headers={'location': url_base})
inner_mock.get(url_base, real_http=True)
assert text_base == requests.post(url_inner).text # nosec
with pytest.raises(requests_mock.NoMockAddress):
requests.get(url_middle)
with pytest.raises(requests_mock.NoMockAddress):
requests.get(url_outer)
# back to middle mock
with pytest.raises(requests_mock.NoMockAddress):
requests.post(url_inner)
assert text_middle == requests.get(url_middle).text # nosec
assert text_outer == requests.get(url_outer).text # nosec
# back to outter mock
with pytest.raises(requests_mock.NoMockAddress):
requests.post(url_inner)
with pytest.raises(requests_mock.NoMockAddress):
requests.get(url_middle)
assert text_outer == requests.get(url_outer).text # nosec
def test_mixed_mocks():
url = 'mock://example.test/'
with requests_mock.Mocker() as global_mock:
global_mock.get(url, text='global')
session = requests.Session()
text = session.get(url).text
assert text == 'global' # nosec
with requests_mock.Mocker(session=session) as session_mock:
session_mock.get(url, real_http=True)
text = session.get(url).text
assert text == 'global' # nosec
def test_threaded_sessions():
"""
When using requests_futures.FuturesSession() with a ThreadPoolExecutor
there is a race condition where one threaded request removes the
monkeypatched get_adapter() method from the Session before another threaded
request is finished using it.
"""
from requests_futures.sessions import FuturesSession
url1 = 'http://www.example.com/requests-mock-fake-url1'
url2 = 'http://www.example.com/requests-mock-fake-url2'
with requests_mock.Mocker() as m:
# respond with 204 so we know its us
m.get(url1, status_code=204)
m.get(url2, status_code=204)
# NOTE(phodge): just firing off two .get() requests right after each
# other was a pretty reliable way to reproduce the race condition on my
# intel Macbook Pro but YMMV. Guaranteeing the race condition to
# reappear might require replacing the Session.send() with a wrapper
# that delays kicking off the request for url1 until the request for
# url2 has restored the original session.get_adapter(), but replacing
# Session.send() could be difficult because the requests_mock.Mocker()
# context manager has *already* monkeypatched this method.
session = FuturesSession()
future1 = session.get(url1)
future2 = session.get(url2)
# verify both requests were handled by the mock dispatcher
assert future1.result().status_code == 204
assert future2.result().status_code == 204
class TestClass(object):
def configure(self, requests_mock):
requests_mock.get('https://httpbin.org/get', text='data')
def test_one(self, requests_mock):
self.configure(requests_mock)
assert 'data' == requests.get('https://httpbin.org/get').text
|
