def it_raises_updating_token_after_3_tries(self, response, token_response): returns = [token_response, token_response, token_response, response] h = RequestHandler("") h._send_request = Mock(side_effect=lambda *args: returns.pop(0)) with patch("ubersmith.api.time") as time: time.sleep = lambda x: None with pytest.raises(UpdatingTokenResponse): h.process_request("uber.method_list")
def it_raises_updating_token_after_3_tries(self, response, token_response): returns = [ token_response, token_response, token_response, response, ] h = RequestHandler('') h._send_request = Mock(side_effect=lambda *args: returns.pop(0)) with patch('ubersmith.api.time') as time: time.sleep = lambda x: None with pytest.raises(UpdatingTokenResponse): h.process_request('uber.method_list')
def it_handles_updating_token_with_charset(self, response, token_response_with_charset): returns = [token_response_with_charset, token_response_with_charset, response] h = RequestHandler("") h._send_request = Mock(side_effect=lambda *args: returns.pop(0)) with patch("ubersmith.api.time") as time: time.sleep = lambda x: None assert self.test_data == h.process_request("uber.method_list").data
def it_handles_updating_token(self, response, token_response): returns = [ token_response, token_response, response, ] h = RequestHandler('') h._send_request = Mock(side_effect=lambda *args: returns.pop(0)) with patch('ubersmith.api.time') as time: time.sleep = lambda x: None assert self.test_data == h.process_request('uber.method_list').data
def it_validates_bad_methods(self): h = RequestHandler('') with pytest.raises(RequestError) as e: h.process_request('boop') assert str(e.value) == "Requested method is not valid."
def it_can_disable_ssl_validation(self, response): h = RequestHandler('', verify=False) with patch('ubersmith.api.RequestHandler.session') as session: session.post.return_value = response h.process_request('uber.method_list') assert session.post.call_args[1]['verify'] is False
def it_validates_ssl(self, response): h = RequestHandler('') with patch('ubersmith.api.RequestHandler.session') as session: session.post.return_value = response h.process_request('uber.method_list') assert session.post.call_args[1]['verify'] is True
def it_raises_maintenance_response(self, maintenance_response): h = RequestHandler('') h._send_request = Mock(return_value=maintenance_response) with pytest.raises(MaintenanceResponse): h.process_request('uber.method_list')
def it_can_disable_ssl_validation(self, response): h = RequestHandler("", verify=False) with patch("ubersmith.api.RequestHandler.session") as session: session.post.return_value = response h.process_request("uber.method_list") assert session.post.call_args[1]["verify"] is False
def it_can_disable_ssl_validation(self, response): h = RequestHandler('', verify=False) with patch('ubersmith.api.requests') as requests: requests.post.return_value = response h.process_request('uber.method_list') assert requests.post.call_args[1]['verify'] is False
def it_validates_ssl(self, response): h = RequestHandler('') with patch('ubersmith.api.requests') as requests: requests.post.return_value = response h.process_request('uber.method_list') assert requests.post.call_args[1]['verify'] is True
def it_handles_normal_responses(self, response): h = RequestHandler('') h._send_request = Mock(return_value=response) assert self.test_data == h.process_request('uber.method_list').data
def it_validates_ssl(self, response): h = RequestHandler("") with patch("ubersmith.api.RequestHandler.session") as session: session.post.return_value = response h.process_request("uber.method_list") assert session.post.call_args[1]["verify"] is True