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_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_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.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_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_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_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
