From 8aeaee2a0a8eead6a1aad4c0c05ec67a636f202f Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Sat, 14 Apr 2018 10:59:07 -0700 Subject: refactor dummy generation --- tests/test_backend.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'tests/test_backend.py') diff --git a/tests/test_backend.py b/tests/test_backend.py index 93e3f11a..7a139ef7 100644 --- a/tests/test_backend.py +++ b/tests/test_backend.py @@ -28,6 +28,7 @@ class FatcatTestCase(unittest.TestCase): fatcat.db.session.remove() fatcat.db.drop_all() fatcat.db.create_all() + fatcat.sql.populate_db() def test_health(self): rv = self.app.get('/health') @@ -35,7 +36,7 @@ class FatcatTestCase(unittest.TestCase): assert obj['ok'] def test_works(self): - fatcat.sql.populate_db() + fatcat.dummy.insert_example_works() # Invalid Id rv = self.app.get('/v0/work/_') @@ -61,8 +62,11 @@ class FatcatTestCase(unittest.TestCase): def test_populate(self): fatcat.sql.populate_db() - def test_populate_complex(self): - fatcat.sql.populate_complex_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: @@ -71,9 +75,9 @@ class FatcatTestCase(unittest.TestCase): fatcat.sql.add_crossref(obj) def test_hydrate_work(self): - fatcat.sql.populate_complex_db() + fatcat.dummy.insert_random_works() fatcat.sql.hydrate_work(1) def test_hydrate_release(self): - fatcat.sql.populate_complex_db() + fatcat.dummy.insert_random_works() fatcat.sql.hydrate_release(1) -- cgit v1.2.3