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_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_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_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_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 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_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 add_annotation(id_): annotation = storage.fetch_annotation(celery.request.db, id_) if annotation: index(celery.request.es, annotation, celery.request)