def test_unindex_raises_not_found_exception(self):
exception = ElasticSearchException(error=404,
status=404,
result={'not found': 'not found'})
model = Mock()
model.unindex = Mock(side_effect=exception)
unindex_objects(model, [1, 2, 3], 'foo')
def update_search_index(sender, instance, **kwargs):
if instance.is_complete:
index_objects.delay(sender, [instance.id], public=False)
if instance.is_public_indexable:
index_objects.delay(sender, [instance.id], public_index=True)
else:
unindex_objects(UserProfile, [instance.id], public_index=True)
def test_unindex_objects(self, get_es_mock):
model = MagicMock()
unindex_objects(model, [1, 2, 3], 'foo')
ok_(model.unindex.called)
model.assert_has_calls([
call.unindex(es=get_es_mock(), public_index='foo', id=1),
call.unindex(es=get_es_mock(), public_index='foo', id=2),
call.unindex(es=get_es_mock(), public_index='foo', id=3)])
def test_unindex_objects(self, get_es_mock):
mapping_type = MagicMock()
unindex_objects(mapping_type, [1, 2, 3], 'foo')
ok_(mapping_type.unindex.called)
mapping_type.assert_has_calls([
call.unindex(1, es=get_es_mock(), public_index='foo'),
call.unindex(2, es=get_es_mock(), public_index='foo'),
call.unindex(3, es=get_es_mock(), public_index='foo')])
def test_unindex_objects(self, get_es_mock):
mapping_type = MagicMock()
unindex_objects(mapping_type, [1, 2, 3], 'foo')
ok_(mapping_type.unindex.called)
mapping_type.assert_has_calls([
call.unindex(1, es=get_es_mock(), public_index='foo'),
call.unindex(2, es=get_es_mock(), public_index='foo'),
call.unindex(3, es=get_es_mock(), public_index='foo')])
def test_unindex_objects(self, get_es_mock):
model = MagicMock()
unindex_objects(model, [1, 2, 3], 'foo')
ok_(model.unindex.called)
model.assert_has_calls([
call.unindex(es=get_es_mock(), public_index='foo', id=1),
call.unindex(es=get_es_mock(), public_index='foo', id=2),
call.unindex(es=get_es_mock(), public_index='foo', id=3)
])
def test_unindex_raises_not_found_exception(self):
exception = ElasticSearchException(error=404, status=404, result={"not found": "not found"})
model = Mock()
model.unindex = Mock(side_effect=exception)
unindex_objects(model, [1, 2, 3], "foo")
def test_unindex_raises_not_found_exception(self):
exception = NotFoundError(404, {'not found': 'not found '}, {'foo': 'foo'})
mapping_type = Mock()
mapping_type.unindex(side_effect=exception)
unindex_objects(mapping_type, [1, 2, 3], 'foo')
def remove_from_search_index(sender, instance, **kwargs):
unindex_objects(UserProfile, [instance.id], public_index=False)
unindex_objects(UserProfile, [instance.id], public_index=True)
def test_unindex_raises_not_found_exception(self):
exception = NotFoundError(404, {'not found': 'not found '}, {'foo': 'foo'})
mapping_type = Mock()
mapping_type.unindex(side_effect=exception)
unindex_objects(mapping_type, [1, 2, 3], 'foo')
