def test_reset_secret_expired_token_failure(self, mock_get, mock_put): clientId = '0' * 96 clientSecret = '0' * 71 client = Client(clientId, clientSecret) client.token = "ExpiredToken" mock_response = mock.Mock() mock_response.status_code = 401 mock_response.json.return_value = {'message': 'Token expired'} mock_put.return_value = mock_response mock_response2 = mock.Mock() mock_response2.status_code = 200 mock_response2.json.return_value = { 'token': 'newToken', } mock_get.return_value = mock_response2 with self.assertRaises(AuthenticationError): client.reset_secret('1') self.assertEqual(3, mock_put.call_count) self.assertEqual(2, mock_get.call_count) headers = {'Authorization': 'Bearer ExpiredToken'} newheaders = {'Authorization': 'Bearer newToken'} call_list = [ mock.call(self.base_URL + 'apps/1/secret', headers=headers, data=None), mock.call(self.base_URL + 'apps/1/secret', headers=newheaders, data=None), mock.call(self.base_URL + 'apps/1/secret', headers=newheaders, data=None) ] self.assertEqual(mock_put.call_args_list, call_list)
def test_reset_secret_expired_token_success(self, mock_get, mock_put): clientId = '0' * 96 clientSecret = '0' * 71 client = Client(clientId, clientSecret) client.token = "ExpiredToken" mock_response = mock.Mock() mock_response.status_code = 401 mock_response.json.return_value = {'message': 'Token expired'} mock_response3 = mock.Mock() mock_response3.status_code = 200 mock_response3.json.return_value = { 'clientSecret': 'test_clientSecret' } mock_put.side_effect = [mock_response, mock_response3] mock_response2 = mock.Mock() mock_response2.status_code = 200 mock_response2.json.return_value = {'token': 'newToken'} mock_get.return_value = mock_response2 self.assertEqual('test_clientSecret', client.reset_secret('1')) headers = {'Authorization': 'Bearer ExpiredToken'} newheaders = {'Authorization': 'Bearer newToken'} call_list = [ mock.call(self.base_URL + 'apps/1/secret', headers=headers, data=None), mock.call(self.base_URL + 'apps/1/secret', headers=newheaders, data=None) ] self.assertEqual(mock_put.call_args_list, call_list)
def test_reset_secret_invalid_clientid(self, mock_put): clientId = '0' * 96 clientSecret = '0' * 71 client = Client(clientId, clientSecret) client.token = "correctToken" mock_response = mock.Mock() mock_response.status_code = 404 mock_response.json.return_value = { 'message': 'No app exists with given client id.' } mock_put.return_value = mock_response url = Client.base_URL + 'apps/1/secret' headers = {'Authorization': 'Bearer correctToken'} with self.assertRaises(ResourceNotFoundError): client.reset_secret('1') mock_put.assert_called_once_with(url, headers=headers, data=None) self.assertEqual(1, mock_response.json.call_count)
def test_reset_secret(self, mock_put): clientId = '0' * 96 clientSecret = '0' * 71 client = Client(clientId, clientSecret) client.token = "correctToken" mock_response = mock.Mock() mock_response.status_code = 200 mock_response.json.return_value = {'clientSecret': 'test_clientSecret'} mock_put.return_value = mock_response url = Client.base_URL + 'apps/1/secret' headers = {'Authorization': 'Bearer correctToken'} self.assertEqual(client.reset_secret('1'), 'test_clientSecret') mock_put.assert_called_once_with(url, headers=headers, data=None)