Skip to content

Commit

Permalink
tests
Browse files Browse the repository at this point in the history
  • Loading branch information
aaxelb committed Sep 23, 2024
1 parent 5a7dd86 commit 101c7a8
Show file tree
Hide file tree
Showing 8 changed files with 369 additions and 19 deletions.
8 changes: 7 additions & 1 deletion how-to/use-the-api.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,18 @@
`GET /trove/index-value-search`: search values for specific properties on index-cards

## Posting index-cards
> NOTE: currently used only by other COS projects, not yet for public use
> NOTE: currently used only by other COS projects, not yet for public use, authorization required
`POST /trove/ingest?focus_iri=...&record_identifier=...`:

currently supports only `Content-Type: text/turtle`

query params:
- `focus_iri` (required): full iri of the focus resource, exactly as used in the request body
- `record_identifier` (required): a source-specific identifier for the metadata record (no format restrictions) -- sending another record with the same `record_identifier` is considered a full update (only the most recent is used)
- `nonurgent`: if present (regardless of value), ingestion may be given a lower priority -- recommended for bulk or background operations
- `is_supplementary`: if present (regardless of value), this record's metadata will be added to all pre-existing index-cards from the same user with the same `focus_iri` (if any), but will not get an index-card of its own nor affect the last-updated timestamp (e.g. in OAI-PMH) of the index-cards it supplements

## Deleting index-cards

`DELETE /trove/ingest?record_identifier=...`: request
Expand Down
67 changes: 67 additions & 0 deletions tests/trove/digestive_tract/test_derive.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
import json

from django.test import TestCase
from primitive_metadata import primitive_rdf as rdf

from tests import factories
from trove import digestive_tract
from trove import models as trove_db
from trove.vocab.namespaces import TROVE
from trove.util.iris import get_sufficiently_unique_iri


_BLARG = rdf.IriNamespace('https://blarg.example/')


class TestDigestiveTractDerive(TestCase):
@classmethod
def setUpTestData(cls):
cls.focus_iri = _BLARG.this
_focus_ident = trove_db.ResourceIdentifier.objects.get_or_create_for_iri(cls.focus_iri)
_raw = factories.RawDatumFactory()
cls.indexcard = trove_db.Indexcard.objects.create(source_record_suid=_raw.suid)
cls.indexcard.focus_identifier_set.add(_focus_ident)
cls.latest_rdf = trove_db.LatestIndexcardRdf.objects.create(
indexcard=cls.indexcard,
from_raw_datum=_raw,
focus_iri=cls.focus_iri,
rdf_as_turtle='''@prefix blarg: <https://blarg.example/> .
blarg:this
a blarg:Thing ;
blarg:like blarg:that .
''',
)

def test_derive(self):
(_derived,) = digestive_tract.derive(self.indexcard)
self.assertEqual(_derived.upriver_indexcard_id, self.indexcard.id)
self.assertEqual(_derived.deriver_identifier.sufficiently_unique_iri, get_sufficiently_unique_iri(TROVE['derive/osfmap_json']))
self.assertEqual(json.loads(_derived.derived_text), {
'@id': _BLARG.this,
'resourceType': [{'@id': _BLARG.Thing}],
_BLARG.like: [{'@id': _BLARG.that}],
})

def test_derive_with_supplementary(self):
_supp_raw = factories.RawDatumFactory(
suid=factories.SourceUniqueIdentifierFactory(is_supplementary=True),
)
trove_db.SupplementaryIndexcardRdf.objects.create(
indexcard=self.indexcard,
from_raw_datum=_supp_raw,
supplementary_suid=_supp_raw.suid,
focus_iri=self.focus_iri,
rdf_as_turtle='''@prefix blarg: <https://blarg.example/> .
blarg:this blarg:like blarg:another ;
blarg:unlike blarg:nonthing .
''',
)
(_derived,) = digestive_tract.derive(self.indexcard)
self.assertEqual(_derived.upriver_indexcard_id, self.indexcard.id)
self.assertEqual(_derived.deriver_identifier.sufficiently_unique_iri, get_sufficiently_unique_iri(TROVE['derive/osfmap_json']))
self.assertEqual(json.loads(_derived.derived_text), {
'@id': _BLARG.this,
'resourceType': [{'@id': _BLARG.Thing}],
_BLARG.like: [{'@id': _BLARG.that}, {'@id': _BLARG.another}],
_BLARG.unlike: [{'@id': _BLARG.nonthing}],
})
91 changes: 91 additions & 0 deletions tests/trove/digestive_tract/test_extract.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,91 @@
from django.test import TestCase
from primitive_metadata import primitive_rdf as rdf

from tests import factories
from trove import digestive_tract
from trove import models as trove_db


_BLARG = rdf.IriNamespace('https://blarg.example/')


class TestDigestiveTractExtract(TestCase):
@classmethod
def setUpTestData(cls):
_focus_ident = trove_db.ResourceIdentifier.objects.get_or_create_for_iri(_BLARG.this)
cls.raw = factories.RawDatumFactory(
mediatype='text/turtle',
datum='''@prefix blarg: <https://blarg.example/> .
blarg:this
a blarg:Thing ;
blarg:like blarg:that .
''',
suid__focus_identifier=_focus_ident,
)
cls.supplementary_raw = factories.RawDatumFactory(
mediatype='text/turtle',
datum='''@prefix blarg: <https://blarg.example/> .
blarg:this blarg:like blarg:another ;
blarg:unlike blarg:nonthing .
''',
suid=factories.SourceUniqueIdentifierFactory(
source_config=cls.raw.suid.source_config,
focus_identifier=cls.raw.suid.focus_identifier,
is_supplementary=True,
),
)

def test_setup(self):
self.assertEqual(trove_db.Indexcard.objects.all().count(), 0)
self.assertEqual(trove_db.LatestIndexcardRdf.objects.all().count(), 0)
self.assertEqual(trove_db.ArchivedIndexcardRdf.objects.all().count(), 0)
self.assertEqual(trove_db.SupplementaryIndexcardRdf.objects.all().count(), 0)

def test_extract(self):
(_indexcard,) = digestive_tract.extract(self.raw)
self.assertEqual(_indexcard.source_record_suid_id, self.raw.suid_id)
_focus_idents = list(
_indexcard.focus_identifier_set.values_list('sufficiently_unique_iri', flat=True),
)
self.assertEqual(_focus_idents, ['://blarg.example/this'])
_focustype_idents = list(
_indexcard.focustype_identifier_set.values_list('sufficiently_unique_iri', flat=True),
)
self.assertEqual(_focustype_idents, ['://blarg.example/Thing'])
self.assertEqual(list(_indexcard.supplementary_rdf_set.all()), [])
_latest_rdf = _indexcard.latest_rdf
self.assertEqual(_latest_rdf.from_raw_datum_id, self.raw.id)
self.assertEqual(_latest_rdf.indexcard_id, _indexcard.id)
self.assertEqual(_latest_rdf.focus_iri, _BLARG.this)
self.assertEqual(_latest_rdf.as_rdf_tripledict(), {
_BLARG.this: {
rdf.RDF.type: {_BLARG.Thing},
_BLARG.like: {_BLARG.that},
},
})

def test_extract_supplementary_without_prior(self):
_cards = digestive_tract.extract(self.supplementary_raw)
self.assertEqual(_cards, [])
self.assertEqual(trove_db.Indexcard.objects.all().count(), 0)
self.assertEqual(trove_db.LatestIndexcardRdf.objects.all().count(), 0)
self.assertEqual(trove_db.ArchivedIndexcardRdf.objects.all().count(), 0)
self.assertEqual(trove_db.SupplementaryIndexcardRdf.objects.all().count(), 0)

def test_extract_supplementary(self):
(_orig_indexcard,) = digestive_tract.extract(self.raw)
_orig_timestamp = _orig_indexcard.latest_rdf.modified
(_indexcard,) = digestive_tract.extract(self.supplementary_raw)
self.assertEqual(_orig_indexcard.id, _indexcard.id)
self.assertEqual(_indexcard.source_record_suid_id, self.raw.suid_id)
(_supp_rdf,) = _indexcard.supplementary_rdf_set.all()
self.assertEqual(_supp_rdf.from_raw_datum_id, self.supplementary_raw.id)
self.assertEqual(_supp_rdf.indexcard_id, _indexcard.id)
self.assertEqual(_supp_rdf.focus_iri, _BLARG.this)
self.assertEqual(_supp_rdf.as_rdf_tripledict(), {
_BLARG.this: {
_BLARG.like: {_BLARG.another},
_BLARG.unlike: {_BLARG.nonthing},
},
})
self.assertEqual(_indexcard.latest_rdf.modified, _orig_timestamp)
77 changes: 77 additions & 0 deletions tests/trove/digestive_tract/test_swallow.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,77 @@
from unittest import mock
from django.test import TestCase

from tests import factories
from trove import digestive_tract
from share import models as share_db


class TestDigestiveTractSwallow(TestCase):
@classmethod
def setUpTestData(cls):
cls.user = factories.ShareUserFactory()
cls.turtle = '''
@prefix blarg: <https://blarg.example/> .
blarg:this
a blarg:Thing ;
blarg:like blarg:that .
'''

def test_setup(self):
self.assertEqual(share_db.RawDatum.objects.all().count(), 0)

def test_swallow(self):
with mock.patch('trove.digestive_tract.task__extract_and_derive') as _mock_task:
digestive_tract.swallow(
from_user=self.user,
record=self.turtle,
record_identifier='blarg',
record_mediatype='text/turtle',
focus_iri='https://blarg.example/this',
)
(_raw,) = share_db.RawDatum.objects.all()
self.assertEqual(_raw.datum, self.turtle)
self.assertEqual(_raw.mediatype, 'text/turtle')
self.assertEqual(_raw.suid.identifier, 'blarg')
self.assertEqual(_raw.suid.focus_identifier.sufficiently_unique_iri, '://blarg.example/this')
self.assertEqual(_raw.suid.source_config.source.user_id, self.user.id)
self.assertFalse(_raw.suid.is_supplementary)
_mock_task.delay.assert_called_once_with(_raw.id, urgent=False)

def test_swallow_urgent(self):
with mock.patch('trove.digestive_tract.task__extract_and_derive') as _mock_task:
digestive_tract.swallow(
from_user=self.user,
record=self.turtle,
record_identifier='blarg',
record_mediatype='text/turtle',
focus_iri='https://blarg.example/this',
urgent=True
)
(_raw,) = share_db.RawDatum.objects.all()
self.assertEqual(_raw.datum, self.turtle)
self.assertEqual(_raw.mediatype, 'text/turtle')
self.assertEqual(_raw.suid.identifier, 'blarg')
self.assertEqual(_raw.suid.focus_identifier.sufficiently_unique_iri, '://blarg.example/this')
self.assertEqual(_raw.suid.source_config.source.user_id, self.user.id)
self.assertFalse(_raw.suid.is_supplementary)
_mock_task.delay.assert_called_once_with(_raw.id, urgent=True)

def test_swallow_supplementary(self):
with mock.patch('trove.digestive_tract.task__extract_and_derive') as _mock_task:
digestive_tract.swallow(
from_user=self.user,
record=self.turtle,
record_identifier='blarg',
record_mediatype='text/turtle',
focus_iri='https://blarg.example/this',
is_supplementary=True,
)
(_raw,) = share_db.RawDatum.objects.all()
self.assertEqual(_raw.datum, self.turtle)
self.assertEqual(_raw.mediatype, 'text/turtle')
self.assertEqual(_raw.suid.identifier, 'blarg')
self.assertEqual(_raw.suid.focus_identifier.sufficiently_unique_iri, '://blarg.example/this')
self.assertEqual(_raw.suid.source_config.source.user_id, self.user.id)
self.assertTrue(_raw.suid.is_supplementary)
_mock_task.delay.assert_called_once_with(_raw.id, urgent=False)
96 changes: 94 additions & 2 deletions tests/trove/views/test_ingest.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,98 @@
from http import HTTPStatus
from unittest import mock

from django.test import TestCase

from tests import factories


class TestIngest(TestCase):
def test_simple_ingest(self):
pass # TODO
@classmethod
def setUpTestData(cls):
cls.user = factories.ShareUserFactory()

def test_post(self):
with mock.patch('trove.views.ingest.digestive_tract') as _mock_tract:
_resp = self.client.post(
'/trove/ingest',
query_params={
'focus_iri': 'https://foo.example/blarg',
'record_identifier': 'blarg',
},
content_type='text/turtle',
data='turtleturtleturtle',
headers={'Authorization': self.user.authorization()},
)
self.assertEqual(_resp.status_code, HTTPStatus.CREATED)
_mock_tract.swallow.assert_called_once_with(
from_user=self.user,
record='turtleturtleturtle',
record_identifier='blarg',
record_mediatype='text/turtle',
focus_iri='https://foo.example/blarg',
urgent=True,
is_supplementary=False,
)

def test_post_nonurgent(self):
with mock.patch('trove.views.ingest.digestive_tract') as _mock_tract:
_resp = self.client.post(
'/trove/ingest',
query_params={
'focus_iri': 'https://foo.example/blarg',
'record_identifier': 'blarg',
'nonurgent': '',
},
content_type='text/turtle',
data='turtleturtleturtle',
headers={'Authorization': self.user.authorization()},
)
self.assertEqual(_resp.status_code, HTTPStatus.CREATED)
_mock_tract.swallow.assert_called_once_with(
from_user=self.user,
record='turtleturtleturtle',
record_identifier='blarg',
record_mediatype='text/turtle',
focus_iri='https://foo.example/blarg',
urgent=False,
is_supplementary=False,
)

def test_post_supplementary(self):
with mock.patch('trove.views.ingest.digestive_tract') as _mock_tract:
_resp = self.client.post(
'/trove/ingest',
query_params={
'focus_iri': 'https://foo.example/blarg',
'record_identifier': 'blarg',
'is_supplementary': '',
},
content_type='text/turtle',
data='turtleturtleturtle',
headers={'Authorization': self.user.authorization()},
)
self.assertEqual(_resp.status_code, HTTPStatus.CREATED)
_mock_tract.swallow.assert_called_once_with(
from_user=self.user,
record='turtleturtleturtle',
record_identifier='blarg',
record_mediatype='text/turtle',
focus_iri='https://foo.example/blarg',
urgent=True,
is_supplementary=True,
)

def test_delete(self):
with mock.patch('trove.views.ingest.digestive_tract') as _mock_tract:
_resp = self.client.delete(
'/trove/ingest',
query_params={
'record_identifier': 'blarg',
},
headers={'Authorization': self.user.authorization()},
)
self.assertEqual(_resp.status_code, HTTPStatus.OK)
_mock_tract.expel.assert_called_once_with(
from_user=self.user,
record_identifier='blarg',
)
Loading

0 comments on commit 101c7a8

Please sign in to comment.