def test_catch_index_error_not_index_error(self):
conn = ESHttpConnection()
try:
conn._catch_index_error((
1, 2,
'{"errors":true, "items": [{"foo": "bar"}]}'))
except:
raise Exception('Unexpected exeption')
def test_perform_request_debug(self, mock_log, mock_catch):
mock_log.level = logging.DEBUG
conn = ESHttpConnection()
conn.pool = Mock()
conn.pool.urlopen.return_value = Mock(
data=six.b('foo'), status=200)
conn.perform_request('POST', 'http://localhost:9200')
mock_log.debug.assert_called_once_with(
"('POST', 'http://localhost:9200')")
conn.perform_request('POST', 'http://localhost:9200'*200)
assert mock_catch.called
assert mock_log.debug.call_count == 2
def test_perform_request_exception(self):
conn = ESHttpConnection()
conn.pool = Mock()
conn.pool.urlopen.side_effect = TransportError('N/A', '')
with pytest.raises(JHTTPBadRequest):
conn.perform_request('POST', 'http://localhost:9200')
def test_catch_index_error(self):
conn = ESHttpConnection()
with pytest.raises(JHTTPBadRequest):
conn._catch_index_error((
1, 2,
'{"errors":true, "items": [{"index": {"error": "FOO"}}]}'))
def test_catch_index_error_no_errors(self):
conn = ESHttpConnection()
try:
conn._catch_index_error((1, 2, '{"errors":false}'))
except:
raise Exception('Unexpected exeption')
def test_catch_index_error_no_data_loaded(self):
conn = ESHttpConnection()
try:
conn._catch_index_error((1, 2, '[]'))
except:
raise Exception('Unexpected exeption')