def test_import_fill_levels(self, fetch_mock, get_token_mock):
with open(FIX_DIR + "/enevo/fixtures/fill_levels.json") as fill_levels:
fill_levels_json = json.loads(fill_levels.read())
mr = MockResponseAsync()
mr._json = fill_levels_json
# we retry failed urls, so extra side effects needed.
fetch_mock.side_effect = 10 * [mr]
get_token_mock.side_effect = 10 * ['test_token']
slurp.start_import('fill_levels', make_engine=False)
count = session.query(models.EnevoFillLevelRaw).count()
self.assertEqual(count, 10)
slurp.start_import('fill_levels', make_engine=False)
convert_live_raw.main(make_engine=False)
count = session.query(models.EnevoFillLevel).count()
self.assertEqual(count, 20)
def test_import_containers(self, fetch_mock, get_token_mock):
with open(FIX_DIR + "/enevo/fixtures/containers.json") as containers:
containers_json = json.loads(containers.read())
mr = MockResponse()
mr._json = containers_json
fetch_mock.side_effect = [mr]
get_token_mock.side_effect = ['test_token']
slurp.start_import('containers', make_engine=False)
count = session.query(models.EnevoContainer).count()
self.assertEqual(count, 2)
fetch_mock.side_effect = [mr]
get_token_mock.side_effect = ['test_token']
slurp.start_import('containers', make_engine=False)
count = session.query(models.EnevoContainer).count()
self.assertEqual(count, 2)
def test_import_site_content_types(self, fetch_mock, get_token_mock):
path = "/enevo/fixtures/site_content_types.json"
with open(FIX_DIR + path) as site_content_types:
site_content_types_json = json.loads(site_content_types.read())
mr = MockResponse()
mr._json = site_content_types_json
fetch_mock.side_effect = [mr]
get_token_mock.side_effect = ['test_token']
slurp.start_import(endpoint='site_content_types', make_engine=False)
count = session.query(models.EnevoSiteContentType).count()
self.assertEqual(count, 2)
fetch_mock.side_effect = [mr]
get_token_mock.side_effect = ['test_token']
slurp.start_import(endpoint='site_content_types', make_engine=False)
count = session.query(models.EnevoSiteContentType).count()
self.assertEqual(count, 2)