1 files changed, 169 insertions, 0 deletions
diff --git a/python/tests/fixtures.py b/python/tests/fixtures.py
new file mode 100644
index 00000000..d3d8c24b
--- /dev/null
+++ b/python/tests/fixtures.py
@@ -0,0 +1,169 @@
+
+import os
+import time
+import json
+import signal
+import pytest
+import fatcat
+import fatcat.sql
+from fatcat.models import *
+
+
+@pytest.fixture
+def full_app():
+    fatcat.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
+    fatcat.app.testing = True
+    fatcat.app.debug = False
+    fatcat.db.session.remove()
+    fatcat.db.drop_all()
+    fatcat.db.create_all()
+    fatcat.sql.populate_db()
+    return fatcat.app
+
+@pytest.fixture
+def app(full_app):
+    return full_app.test_client()
+
+@pytest.fixture
+def rich_app(app):
+    enrichen_test_app(app)
+    return app
+
+
+@pytest.fixture(scope="function")
+def api_client(full_app):
+
+    pid = os.fork()
+    if pid == 0:
+        full_app.testing = False
+        full_app.run(host="localhost", port=8444, debug=False)
+        os._exit(0)
+
+    time.sleep(0.2)
+    yield fatcat.api_client.FatCatApiClient("http://localhost:8444")
+    os.kill(pid, signal.SIGKILL)
+
+
+## Helpers ##################################################################
+
+def enrichen_test_app(app):
+
+    rv = app.post('/v0/editgroup',
+        data=json.dumps(dict(
+            extra=dict(q=1, u="zing"))),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+    obj = json.loads(rv.data.decode('utf-8'))
+    editgroup_id = obj['id']
+
+    rv = app.post('/v0/container',
+        data=json.dumps(dict(
+            name="schmournal",
+            publisher="society of authors",
+            issn="2222-3333",
+            editgroup=editgroup_id,
+            extra=dict(a=2, i="zing"))),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+    obj = json.loads(rv.data.decode('utf-8'))
+    container_id = obj['id']
+
+    rv = app.post('/v0/creator',
+        data=json.dumps(dict(
+            name="anon y. mouse",
+            orcid="0000-0002-1825-0097",
+            editgroup=editgroup_id,
+            extra=dict(w=1, q="zing"))),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+    obj = json.loads(rv.data.decode('utf-8'))
+    creator_id = obj['id']
+
+    rv = app.post('/v0/work',
+        data=json.dumps(dict(
+            title="dummy work",
+            work_type="book",
+            editgroup=editgroup_id,
+            extra=dict(a=3, b="zing"))),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+    obj = json.loads(rv.data.decode('utf-8'))
+    work_id = obj['id']
+
+    # this stub work will be referenced
+    rv = 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 = app.post('/v0/release',
+        data=json.dumps(dict(
+            title="dummy work",
+            work_type="book",
+            work=work_id,
+            container=container_id,
+            creators=[creator_id],
+            doi="10.1234/5678",
+            editgroup=editgroup_id,
+            refs=[
+                dict(stub="some book", target=stub_release_id),
+            ],
+            extra=dict(f=7, b="loopy"))),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+    obj = json.loads(rv.data.decode('utf-8'))
+    release_id = obj['id']
+
+    rv = app.post('/v0/file',
+        data=json.dumps(dict(
+            sha1="deadbeefdeadbeef",
+            size=1234,
+            releases=[release_id],
+            editgroup=editgroup_id,
+            extra=dict(f=4, b="zing"))),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+    obj = json.loads(rv.data.decode('utf-8'))
+    file_id = obj['id']
+
+    rv = app.post('/v0/editgroup/{}/accept'.format(editgroup_id),
+        headers={"content-type": "application/json"})
+    assert rv.status_code == 200
+
+def check_entity_fields(e):
+    for key in ('rev', 'is_live', 'redirect_id'):
+        assert key in e
+    for key in ('id',):
+        assert e[key] is not None
+
+def check_release(e):
+    for key in ('work', 'release_type'):
+        assert key in e
+    for key in ('title', ):
+        assert e[key] is not None
+    for key in ('refs', 'creators'):
+        assert type(e[key]) == list
+
+def check_creator(e):
+    for key in ('name',):
+        assert e[key] is not None
+
+def check_container(e):
+    for key in ('name',):
+        assert e[key] is not None
+
+def check_file(e):
+    for key in ('size', 'sha1'):
+        assert e[key] is not None