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))