def setUp(self):
self.context = open_dependency_context()
requests_stub = EndlessFake(pattern_obj=requests)
session_stub = EndlessFake(pattern_obj=requests.sessions.Session())
self.send_spy = FunctionSpy(return_value=EndlessFake())
session_stub.send = self.send_spy
requests_stub.Session = lambda *a, **k: session_stub
self.context.inject(requests, requests_stub)
def test_sends_configured_timeout_for_session_request(self):
planted = 37
self.context.set_env(HTTP_CLIENT_SOCKET_TIMEOUT=planted)
session_stub = EndlessFake()
self.requests_stub.Session = lambda: session_stub
send_spy = FunctionSpy(return_value=EndlessFake())
session_stub.send = send_spy
client = HttpClient()
client.enable_cookies()
client.get("http://truffles")
expect(send_spy["timeout"]).to(equal(planted))