def test_get_token_all_correct_Internal_error(self, mock_get):
clientId = '0' * 96
clientSecret = '0' * 71
client = Client(clientId, clientSecret)
mock_response = mock.Mock()
mock_response.status_code = 500
mock_get.return_value = mock_response
url = self.base_URL + 'auth/?clientID=' + clientId + '&clientSecret=' + clientSecret
with self.assertRaises(InternalError):
client.request_token()
mock_get.assert_called_once_with(url)
self.assertEqual(0, mock_response.json.call_count)
def test_get_token_incorrect_Secret(self, mock_get,
mock_http_error_handler):
clientId = '0' * 96
clientSecret = '0' * 71
client = Client(clientId, clientSecret)
mock_response = mock.Mock()
expected_dict = {"message": "Bad client secret"}
mock_response.json.return_value = expected_dict
mock_response.status_code = 401
mock_get.return_value = mock_response
url = self.base_URL + 'auth/?clientID=' + clientId + '&clientSecret=' + clientSecret
with self.assertRaises(AuthenticationError):
client.request_token()
mock_get.assert_called_once_with(url)
self.assertEqual(1, mock_response.json.call_count)
return mock_http_error_handler
def test_get_token_all_correct(self, mock_get):
clientId = '0' * 96
clientSecret = '0' * 71
client = Client(clientId, clientSecret)
mock_response = mock.Mock()
expected_dict = {"token": "correctToken"}
mock_response.json.return_value = expected_dict
mock_response.status_code = 200
mock_get.return_value = mock_response
response_dict = client.request_token()
mock_get.assert_called_once_with(self.base_URL + 'auth/?clientID=' +
clientId + '&clientSecret=' +
clientSecret)
self.assertEqual(1, mock_response.json.call_count)
self.assertEqual(client.token, 'correctToken',
'token not stored in the client object.')
return response_dict
