def test_it_presents_the_annotation(self, es, presenters, pyramid_request):
annotation = mock.Mock()
index.index(es, annotation, pyramid_request)
presenters.AnnotationSearchIndexPresenter.assert_called_once_with(
annotation)
def test_it_notifies_before_save_event(self, AnnotationTransformEvent, es,
notify, presenters,
pyramid_request):
index.index(es, mock.Mock(), pyramid_request)
event = AnnotationTransformEvent.return_value
notify.assert_called_once_with(event)
def test_it_allows_to_override_target_index(self, es, presenters,
pyramid_request):
index.index(es,
mock.Mock(),
pyramid_request,
target_index='custom-index')
_, kwargs = es.conn.index.call_args
assert kwargs['index'] == 'custom-index'
def test_it_indexes_the_annotation(self, es, presenters, pyramid_request):
index.index(es, mock.Mock(), pyramid_request)
es.conn.index.assert_called_once_with(
index='hypothesis',
doc_type='annotation',
body=presenters.AnnotationSearchIndexPresenter.return_value.asdict.return_value,
id='test_annotation_id',
)
def test_it_indexes_the_annotation(self, es, presenters, pyramid_request):
index.index(es, mock.Mock(), pyramid_request)
es.conn.index.assert_called_once_with(
index='hypothesis',
doc_type='annotation',
body=presenters.AnnotationSearchIndexPresenter.return_value.asdict.return_value,
id='test_annotation_id',
)
def test_it_creates_an_annotation_before_save_event(
self, AnnotationTransformEvent, es, presenters, pyramid_request):
presented = presenters.AnnotationSearchIndexPresenter.return_value.asdict(
)
index.index(es, mock.Mock(), pyramid_request)
AnnotationTransformEvent.assert_called_once_with(
pyramid_request, presented)
def test_it_notifies_before_save_event(self,
AnnotationTransformEvent,
es,
notify,
presenters,
pyramid_request):
index.index(es, mock.Mock(), pyramid_request)
event = AnnotationTransformEvent.return_value
notify.assert_called_once_with(event)
def add_annotation(id_):
annotation = storage.fetch_annotation(celery.request.db, id_)
if annotation:
index(celery.request.es, annotation, celery.request)
# If a reindex is running at the moment, add annotation to the new index
# as well.
future_index = _current_reindex_new_name(celery.request)
if future_index is not None:
index(celery.request.es, annotation, celery.request,
target_index=future_index)
def test_it_creates_an_annotation_before_save_event(self,
AnnotationTransformEvent,
es,
presenters,
pyramid_request):
annotation = mock.Mock()
presented = presenters.AnnotationSearchIndexPresenter.return_value.asdict()
index.index(es, annotation, pyramid_request)
AnnotationTransformEvent.assert_called_once_with(pyramid_request, annotation, presented)
def add_annotation(id_):
annotation = storage.fetch_annotation(celery.request.db, id_)
if annotation:
index(celery.request.es, annotation, celery.request)
# If a reindex is running at the moment, add annotation to the new index
# as well.
future_index = _current_reindex_new_name(celery.request)
if future_index is not None:
index(celery.request.es, annotation, celery.request,
target_index=future_index)
if annotation.is_reply:
add_annotation.delay(annotation.thread_root_id)
def add_annotation(id_):
annotation = storage.fetch_annotation(celery.request.db, id_)
if annotation:
index(celery.request.es, annotation, celery.request)
if celery.request.feature('index_es6'):
index(celery.request.es6, annotation, celery.request)
# If a reindex is running at the moment, add annotation to the new index
# as well.
future_index = _current_reindex_new_name(celery.request,
'reindex.new_index')
if future_index is not None:
index(celery.request.es,
annotation,
celery.request,
target_index=future_index)
future_es6_index = _current_reindex_new_name(celery.request,
'reindex.new_es6_index')
if future_es6_index is not None:
index(celery.request.es6,
annotation,
celery.request,
target_index=future_es6_index)
if annotation.is_reply:
add_annotation.delay(annotation.thread_root_id)
def _index(annotation):
"""Index the given annotation into Elasticsearch."""
index.index(es_client, annotation, pyramid_request)
def test_it_allows_to_override_target_index(self, es, presenters, pyramid_request):
index.index(es, mock.Mock(), pyramid_request, target_index='custom-index')
_, kwargs = es.conn.index.call_args
assert kwargs['index'] == 'custom-index'
def test_it_presents_the_annotation(self, es, presenters, pyramid_request):
annotation = mock.Mock()
index.index(es, annotation, pyramid_request)
presenters.AnnotationSearchIndexPresenter.assert_called_once_with(annotation)
