def test_it_calls_partial(self, partial):
request = self.mock_request()
storage.legacy_create_annotation(request, self.annotation_data())
partial.assert_called_once_with(
storage.fetch_annotation, request, _postgres=False)
def test_it_calls_notify(self, AnnotationBeforeSaveEvent):
request = self.mock_request()
storage.legacy_create_annotation(request, self.annotation_data())
request.registry.notify.assert_called_once_with(
AnnotationBeforeSaveEvent.return_value)
def test_it_calls_notify(self, AnnotationBeforeSaveEvent):
request = self.mock_request()
storage.legacy_create_annotation(request, self.annotation_data())
request.registry.notify.assert_called_once_with(
AnnotationBeforeSaveEvent.return_value)
def test_it_calls_partial(self, partial):
request = self.mock_request()
storage.legacy_create_annotation(request, self.annotation_data())
partial.assert_called_once_with(
storage.fetch_annotation, request, _postgres=False)
def test_it_inits_AnnotationBeforeSaveEvent(self,
AnnotationBeforeSaveEvent,
models):
request = self.mock_request()
storage.legacy_create_annotation(request, self.annotation_data())
AnnotationBeforeSaveEvent.assert_called_once_with(
request, models.elastic.Annotation.return_value)
def test_it_inits_AnnotationBeforeSaveEvent(self,
AnnotationBeforeSaveEvent,
models):
request = self.mock_request()
storage.legacy_create_annotation(request, self.annotation_data())
AnnotationBeforeSaveEvent.assert_called_once_with(
request, models.elastic.Annotation.return_value)
def create(request):
"""Create an annotation from the POST payload."""
json_payload = _json_payload(request)
# Validate the annotation for, and create the annotation in, PostgreSQL.
if request.feature('postgres'):
schema = schemas.CreateAnnotationSchema(request)
appstruct = schema.validate(copy.deepcopy(json_payload))
annotation = storage.create_annotation(request, appstruct)
# Validate the annotation for, and create the annotation in, Elasticsearch.
legacy_schema = schemas.LegacyCreateAnnotationSchema(request)
legacy_appstruct = legacy_schema.validate(copy.deepcopy(json_payload))
# When 'postgres' is on make sure that annotations in the legacy
# Elasticsearch database use the same IDs as the PostgreSQL ones.
if request.feature('postgres'):
assert annotation.id
legacy_appstruct['id'] = annotation.id
legacy_annotation = storage.legacy_create_annotation(
request, legacy_appstruct)
if request.feature('postgres'):
_publish_annotation_event(request, annotation, 'create')
return AnnotationJSONPresenter(request, annotation).asdict()
_publish_annotation_event(request, legacy_annotation, 'create')
return AnnotationJSONPresenter(request, legacy_annotation).asdict()
def create(request):
"""Create an annotation from the POST payload."""
json_payload = _json_payload(request)
# Validate the annotation for, and create the annotation in, PostgreSQL.
if request.feature('postgres'):
schema = schemas.CreateAnnotationSchema(request)
appstruct = schema.validate(copy.deepcopy(json_payload))
annotation = storage.create_annotation(request, appstruct)
# Validate the annotation for, and create the annotation in, Elasticsearch.
legacy_schema = schemas.LegacyCreateAnnotationSchema(request)
legacy_appstruct = legacy_schema.validate(copy.deepcopy(json_payload))
# When 'postgres' is on make sure that annotations in the legacy
# Elasticsearch database use the same IDs as the PostgreSQL ones.
if request.feature('postgres'):
assert annotation.id
legacy_appstruct['id'] = annotation.id
legacy_annotation = storage.legacy_create_annotation(request,
legacy_appstruct)
if request.feature('postgres'):
annotation_dict = (
AnnotationJSONPresenter(request, annotation).asdict())
else:
annotation_dict = (
AnnotationJSONPresenter(request, legacy_annotation).asdict())
_publish_annotation_event(request, annotation_dict, 'create')
return annotation_dict
def test_it_returns_the_annotation(self, models):
result = storage.legacy_create_annotation(self.mock_request(),
self.annotation_data())
assert result == models.elastic.Annotation.return_value
def test_it_calls_annotation_save(self, models):
storage.legacy_create_annotation(self.mock_request(),
self.annotation_data())
models.elastic.Annotation.return_value.save.assert_called_once_with()
def test_it_calls_prepare(self, models, partial, transform):
storage.legacy_create_annotation(self.mock_request(),
self.annotation_data())
transform.prepare.assert_called_once_with(
models.elastic.Annotation.return_value, partial.return_value)
def test_it_inits_an_elastic_annotation_model(self, models):
data = self.annotation_data()
storage.legacy_create_annotation(self.mock_request(), data)
models.elastic.Annotation.assert_called_once_with(data)
def test_it_returns_the_annotation(self, models):
result = storage.legacy_create_annotation(self.mock_request(),
self.annotation_data())
assert result == models.elastic.Annotation.return_value
def test_it_calls_annotation_save(self, models):
storage.legacy_create_annotation(self.mock_request(),
self.annotation_data())
models.elastic.Annotation.return_value.save.assert_called_once_with()
def test_it_calls_prepare(self, models, partial, transform):
storage.legacy_create_annotation(self.mock_request(),
self.annotation_data())
transform.prepare.assert_called_once_with(
models.elastic.Annotation.return_value, partial.return_value)
def test_it_inits_an_elastic_annotation_model(self, models):
data = self.annotation_data()
storage.legacy_create_annotation(self.mock_request(), data)
models.elastic.Annotation.assert_called_once_with(data)