def test_get_by_ids_no_index_raise(self, mock_mget):
obj = es.ES('Foo', 'foondex')
documents = [{'_id': 1, '_type': 'Story'}]
mock_mget.side_effect = es.IndexNotFoundException()
with pytest.raises(JHTTPNotFound) as ex:
obj.get_by_ids(documents, _raise_on_empty=True)
assert 'resource not found (Index does not exist)' in str(ex.value)
def test_get_item_no_index_not_raise(self, mock_get):
obj = es.ES('Foo', 'foondex')
mock_get.side_effect = es.IndexNotFoundException()
try:
obj.get_item(name='foo', _raise_on_empty=False)
except JHTTPNotFound:
raise Exception('Unexpected error')
def test_get_collection_no_index_raise(self, mock_search):
obj = es.ES('Foo', 'foondex')
mock_search.side_effect = es.IndexNotFoundException()
with pytest.raises(JHTTPNotFound) as ex:
obj.get_collection(body={'foo': 'bar'},
_raise_on_empty=True,
from_=0)
assert 'resource not found (Index does not exist)' in str(ex.value)
def test_get_by_ids_no_index_not_raise(self, mock_mget):
obj = es.ES('Foo', 'foondex')
documents = [{'_id': 1, '_type': 'Story'}]
mock_mget.side_effect = es.IndexNotFoundException()
try:
docs = obj.get_by_ids(documents, _raise_on_empty=False)
except JHTTPNotFound:
raise Exception('Unexpected error')
assert len(docs) == 0
def test_get_collection_no_index_not_raise(self, mock_search):
obj = es.ES('Foo', 'foondex')
mock_search.side_effect = es.IndexNotFoundException()
try:
docs = obj.get_collection(body={'foo': 'bar'},
_raise_on_empty=False,
from_=0)
except JHTTPNotFound:
raise Exception('Unexpected error')
assert len(docs) == 0
def test_do_count_no_index(self, mock_count):
obj = es.ES('Foo', 'foondex')
mock_count.side_effect = es.IndexNotFoundException()
val = obj.do_count({
'foo': 1,
'size': 2,
'from_': 0,
'sort': 'foo:asc'
})
assert val == 0
mock_count.assert_called_once_with(foo=1)
def test_aggregation_index_not_exists(self, mock_search, mock_build):
mock_search.side_effect = es.IndexNotFoundException()
mock_build.return_value = {
'size': 1,
'from_': 2,
'sort': 3,
'body': {
'query': 'query1'
}
}
obj = es.ES('Foo', 'foondex')
with pytest.raises(JHTTPNotFound) as ex:
obj.aggregate(_aggregations_params={'zoo': 5},
param1=6,
_raise_on_empty=True)
assert 'Aggregation failed: Index does not exist' in str(ex.value)
def test_index_missing_documents_no_index(self, mock_mget, mock_bulk):
obj = es.ES('Foo', 'foondex')
documents = [
{
'_pk': 1,
'name': 'foo'
},
]
mock_mget.side_effect = es.IndexNotFoundException()
obj.index_missing_documents(documents)
mock_mget.assert_called_once_with(index='foondex',
doc_type='Foo',
fields=['_id'],
body={'ids': [1]})
mock_bulk.assert_called_once_with('index', [{
'_pk': 1,
'name': 'foo'
}], None)
def test_get_item_no_index_raise(self, mock_get):
obj = es.ES('Foo', 'foondex')
mock_get.side_effect = es.IndexNotFoundException()
with pytest.raises(JHTTPNotFound) as ex:
obj.get_item(name='foo')
assert 'resource not found (Index does not exist)' in str(ex.value)
