test for refs (api)

1 files changed, 27 insertions, 5 deletions

diff --git a/tests/test_backend.py b/tests/test_backend.py
index 25421560..e6e15f3a 100644
--- a/tests/test_backend.py
+++ b/tests/test_backend.py
@@ -236,6 +236,24 @@ class APITestCase(FatcatTestCase):
         obj = json.loads(rv.data.decode('utf-8'))
         work_id = obj['id']
 
+        # this stub work will be referenced
+        rv = self.app.post('/v0/release',
+            data=json.dumps(dict(
+                title="derivative work",
+                work_type="journal-article",
+                work=work_id,
+                creators=[creator_id],
+                doi="10.1234/58",
+                editgroup=editgroup_id,
+                refs=[
+                    dict(stub="some other journal article"),
+                ],
+                extra=dict(f=7, b="zing"))),
+            headers={"content-type": "application/json"})
+        assert rv.status_code == 200
+        obj = json.loads(rv.data.decode('utf-8'))
+        stub_release_id = obj['id']
+
         rv = self.app.post('/v0/release',
             data=json.dumps(dict(
                 title="dummy work",
@@ -246,9 +264,9 @@ class APITestCase(FatcatTestCase):
                 doi="10.1234/5678",
                 editgroup=editgroup_id,
                 refs=[
-                    dict(stub="some other journal article"),
+                    dict(stub="some book", target=stub_release_id),
                 ],
-                extra=dict(f=7, b="zing"))),
+                extra=dict(f=7, b="loopy"))),
             headers={"content-type": "application/json"})
         assert rv.status_code == 200
         obj = json.loads(rv.data.decode('utf-8'))
@@ -269,9 +287,10 @@ class APITestCase(FatcatTestCase):
         for cls in (WorkIdent, WorkRev, WorkEdit,
                     ContainerIdent, ContainerRev, ContainerEdit,
                     CreatorIdent, CreatorRev, CreatorEdit,
-                    ReleaseIdent, ReleaseRev, ReleaseEdit,
                     FileIdent, FileRev, FileEdit):
             assert cls.query.count() == 1
+        for cls in (ReleaseIdent, ReleaseRev, ReleaseEdit):
+            assert cls.query.count() == 2
 
         for cls in (WorkIdent,
                     ContainerIdent,
@@ -280,6 +299,7 @@ class APITestCase(FatcatTestCase):
                     FileIdent):
             assert cls.query.filter(cls.is_live==True).count() == 0
 
+        assert ChangelogEntry.query.count() == 0
         rv = self.app.post('/v0/editgroup/{}/accept'.format(editgroup_id),
             headers={"content-type": "application/json"})
         assert rv.status_code == 200
@@ -288,16 +308,17 @@ class APITestCase(FatcatTestCase):
         for cls in (WorkIdent, WorkRev, WorkEdit,
                     ContainerIdent, ContainerRev, ContainerEdit,
                     CreatorIdent, CreatorRev, CreatorEdit,
-                    ReleaseIdent, ReleaseRev, ReleaseEdit,
                     FileIdent, FileRev, FileEdit):
             assert cls.query.count() == 1
+        for cls in (ReleaseIdent, ReleaseRev, ReleaseEdit):
+            assert cls.query.count() == 2
 
         for cls in (WorkIdent,
                     ContainerIdent,
                     CreatorIdent,
-                    ReleaseIdent,
                     FileIdent):
             assert cls.query.filter(cls.is_live==True).count() == 1
+        assert ReleaseIdent.query.filter(ReleaseIdent.is_live==True).count() == 2
 
         # Test that foreign key relations worked
         release_rv = json.loads(self.app.get('/v0/release/{}'.format(release_id)).data.decode('utf-8'))
@@ -305,6 +326,7 @@ class APITestCase(FatcatTestCase):
         assert(release_rv['creators'][0]['creator'] == creator_id)
         assert(release_rv['container']['id'] == container_id)
         assert(release_rv['work']['id'] == work_id)
+        assert(release_rv['refs'][0]['target'] == stub_release_id)
 
         file_rv = json.loads(self.app.get('/v0/file/{}'.format(file_id)).data.decode('utf-8'))
         print(file_rv)