def test_perform_request_error(self, mock_response):
mock_response.return_value.status = 500
connection = AwsHttpConnection(aws_access_key_id='access_key', aws_secret_access_key='secret')
with self.assertRaises(ElasticsearchException):
with patch('elasticsearch.connection.base.logger.debug') as mock_logger:
connection.perform_request('get', 'http://example.com')
self.assertGreater(mock_logger.call_count, 0)
def test_perform_request_success(self, mock_response):
mock_response.return_value.status = 200
connection = AwsHttpConnection(aws_access_key_id='access_key', aws_secret_access_key='secret')
with patch('elasticsearch.connection.base.logger.info') as mock_logger:
status, _header, _data = connection.perform_request('get', 'http://example.com')
self.assertEqual(status, 200)
self.assertGreater(mock_logger.call_count, 0)
def test_boto_service_unavailable(self, mock_make_request):
connection = AwsHttpConnection(aws_access_key_id='access_key', aws_secret_access_key='secret')
mock_make_request.side_effect = BotoServerError(503, 'Service Unavailable')
try:
connection.perform_request('get', 'http://example.com')
except TransportError as transport_error:
self.assertEqual(transport_error.status_code, 503)
else:
self.fail('Expected a transport error to be raised.')
def test_perform_request_error(self, mock_response):
mock_response.return_value.status = 500
connection = AwsHttpConnection(aws_access_key_id='access_key',
aws_secret_access_key='secret')
with self.assertRaises(ElasticsearchException):
with patch('elasticsearch.connection.base.logger.debug'
) as mock_logger:
connection.perform_request('get', 'http://example.com')
self.assertGreater(mock_logger.call_count, 0)
def test_perform_request_success(self, mock_response):
mock_response.return_value.status = 200
connection = AwsHttpConnection(aws_access_key_id='access_key',
aws_secret_access_key='secret')
with patch('elasticsearch.connection.base.logger.info') as mock_logger:
status, _header, _data = connection.perform_request(
'get', 'http://example.com')
self.assertEqual(status, 200)
self.assertGreater(mock_logger.call_count, 0)
def test_boto_service_unavailable(self, mock_make_request):
connection = AwsHttpConnection(aws_access_key_id='access_key',
aws_secret_access_key='secret')
mock_make_request.side_effect = BotoServerError(
503, 'Service Unavailable')
try:
connection.perform_request('get', 'http://example.com')
except TransportError as transport_error:
self.assertEqual(transport_error.status_code, 503)
else:
self.fail('Expected a transport error to be raised.')
