summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2018-04-14 14:15:26 -0700
committerBryan Newbold <bnewbold@robocracy.org>2018-04-14 14:15:26 -0700
commit3c9e30a5b029cb607fc78a3451536d0dc65dcecd (patch)
treedf59be59e4d829963d0f7ff8c8047dfac6effcb4 /tests
parent8fdd0b859047cf7e1dabc9f0f88ec6a5201e1f74 (diff)
downloadfatcat-3c9e30a5b029cb607fc78a3451536d0dc65dcecd.tar.gz
fatcat-3c9e30a5b029cb607fc78a3451536d0dc65dcecd.zip
more mashmallow fixes
Diffstat (limited to 'tests')
-rw-r--r--tests/test_backend.py75
1 files changed, 69 insertions, 6 deletions
diff --git a/tests/test_backend.py b/tests/test_backend.py
index 4cd474cc..1b470429 100644
--- a/tests/test_backend.py
+++ b/tests/test_backend.py
@@ -4,6 +4,7 @@ import json
import pytest
import fatcat
import fatcat.sql
+from fatcat.models import *
import unittest
import tempfile
@@ -18,24 +19,22 @@ def check_entity_fields(e):
assert e[key] is not None
def check_release(e):
- check_entity_fields(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):
- check_entity_fields(e)
- for key in ('name'):
+ for key in ('name',):
assert e[key] is not None
def check_container(e):
- check_entity_fields(e)
- for key in ('name'):
+ for key in ('name',):
assert e[key] is not None
def check_file(e):
- check_entity_fields(e)
for key in ('size', 'sha1'):
assert e[key] is not None
@@ -78,6 +77,70 @@ class ModelTestCase(FatcatTestCase):
fatcat.dummy.insert_random_works()
fatcat.sql.hydrate_release(1)
+ def test_schema_release_rev(self):
+ assert ReleaseRev.query.count() == 0
+ e = {
+ "title": "Bogus title",
+ "release_type": "book",
+ "creators": [],
+ "refs": [],
+ }
+ model = release_rev_schema.load(e)
+ fatcat.db.session.add(model.data)
+ fatcat.db.session.commit()
+ assert ReleaseRev.query.count() == 1
+ model_after = ReleaseRev.query.first()
+ serial = release_rev_schema.dump(model_after).data
+ #check_release(serial)
+ for k in e.keys():
+ assert e[k] == serial[k]
+
+ def test_schema_creator_rev(self):
+ assert ReleaseRev.query.count() == 0
+ e = {
+ "name": "Robin (Batman)",
+ }
+ model = creator_rev_schema.load(e)
+ fatcat.db.session.add(model.data)
+ fatcat.db.session.commit()
+ assert CreatorRev.query.count() == 1
+ model_after = CreatorRev.query.first()
+ serial = creator_rev_schema.dump(model_after).data
+ check_creator(serial)
+ for k in e.keys():
+ assert e[k] == serial[k]
+
+ def test_schema_container_rev(self):
+ assert ReleaseRev.query.count() == 0
+ e = {
+ "name": "Papers Monthly",
+ }
+ model = container_rev_schema.load(e)
+ fatcat.db.session.add(model.data)
+ fatcat.db.session.commit()
+ assert ContainerRev.query.count() == 1
+ model_after = ContainerRev.query.first()
+ serial = container_rev_schema.dump(model_after).data
+ check_container(serial)
+ for k in e.keys():
+ assert e[k] == serial[k]
+
+ def test_schema_file_rev(self):
+ assert ReleaseRev.query.count() == 0
+ e = {
+ "sha1": "asdf",
+ "size": 6,
+ }
+ model = file_rev_schema.load(e)
+ print(model)
+ fatcat.db.session.add(model.data)
+ fatcat.db.session.commit()
+ assert FileRev.query.count() == 1
+ model_after = FileRev.query.first()
+ serial = file_rev_schema.dump(model_after).data
+ check_file(serial)
+ for k in e.keys():
+ assert e[k] == serial[k]
## API Tests #################################################################