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)