diff options
| -rw-r--r-- | tests/test_backend.py | 82 |
1 files changed, 56 insertions, 26 deletions
diff --git a/tests/test_backend.py b/tests/test_backend.py index 4c9169f3..4cd474cc 100644 --- a/tests/test_backend.py +++ b/tests/test_backend.py @@ -12,12 +12,32 @@ import tempfile ## Helpers ################################################################## def check_entity_fields(e): - for key in ('id', 'rev', 'is_live', 'redirect_id'): + for key in ('rev', 'is_live', 'redirect_id'): assert key in e for key in ('id',): assert e[key] is not None -## API Tests ################################################################ +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 + +def check_creator(e): + check_entity_fields(e) + for key in ('name'): + assert e[key] is not None + +def check_container(e): + check_entity_fields(e) + 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 class FatcatTestCase(unittest.TestCase): @@ -30,12 +50,45 @@ class FatcatTestCase(unittest.TestCase): fatcat.db.create_all() fatcat.sql.populate_db() + +## Model Tests ############################################################### + +class ModelTestCase(FatcatTestCase): + + def test_populate(self): + fatcat.sql.populate_db() + + def test_example_works(self): + fatcat.dummy.insert_example_works() + + def test_random_works(self): + fatcat.dummy.insert_random_works() + + def test_load_crossref(self): + with open('./tests/files/crossref-works.2018-01-21.badsample.json', 'r') as f: + raw = [json.loads(l) for l in f.readlines() if len(l) > 3] + for obj in raw: + fatcat.sql.add_crossref(obj) + + def test_hydrate_work(self): + fatcat.dummy.insert_random_works() + fatcat.sql.hydrate_work(1) + + def test_hydrate_release(self): + fatcat.dummy.insert_random_works() + fatcat.sql.hydrate_release(1) + + +## API Tests ################################################################# + +class APITestCase(FatcatTestCase): + def test_health(self): rv = self.app.get('/health') obj = json.loads(rv.data.decode('utf-8')) assert obj['ok'] - def test_works(self): + def test_api_work(self): fatcat.dummy.insert_example_works() # Invalid Id @@ -58,26 +111,3 @@ class FatcatTestCase(unittest.TestCase): # Missing Id rv = self.app.get('/v0/work/r3zga5b9cd7ef8gh084714iljk') assert rv.status_code == 404 - - def test_populate(self): - fatcat.sql.populate_db() - - def test_example_works(self): - fatcat.dummy.insert_example_works() - - def test_random_works(self): - fatcat.dummy.insert_random_works() - - def test_load_crossref(self): - with open('./tests/files/crossref-works.2018-01-21.badsample.json', 'r') as f: - raw = [json.loads(l) for l in f.readlines() if len(l) > 3] - for obj in raw: - fatcat.sql.add_crossref(obj) - - def test_hydrate_work(self): - fatcat.dummy.insert_random_works() - fatcat.sql.hydrate_work(1) - - def test_hydrate_release(self): - fatcat.dummy.insert_random_works() - fatcat.sql.hydrate_release(1) |