def test_sus_list_return_all_cases_paginated(self):
DataSus().save(self.db.session,
id=1,
region='region',
state='state',
date=datetime.date(2020, 3, 29),
newcases=1,
totalcases=2,
newdeaths=1,
totaldeaths=1,
update=datetime.date(2020, 3, 29))
DataSus().save(self.db.session,
id=2,
region='region2',
state='state2',
date=datetime.date(2020, 3, 29),
newcases=1,
totalcases=2,
newdeaths=1,
totaldeaths=1,
update=datetime.date(2020, 3, 29))
self.db.session.commit()
response = datasus_services.get_sus_list(1)
self.assertEqual(len(response.get('sus_list')), 2)
def test_sus_graphs_total_cases(self):
yesterday = datetime.date.today() - datetime.timedelta(days=1)
DataSus().save(self.db.session,
id=1,
region='region_1',
state='state',
city='city',
coduf=1,
codmun=2,
population=100000,
date=datetime.date.today(),
newcases=1,
totalcases=2,
newdeaths=1,
totaldeaths=1,
update=datetime.date.today())
DataSus().save(self.db.session,
id=2,
region='region_2',
state='state',
city='city',
coduf=1,
codmun=2,
population=100000,
date=yesterday,
newcases=1,
totalcases=2,
newdeaths=1,
totaldeaths=1,
update=yesterday)
DataSus().save(self.db.session,
id=3,
region='region_2',
state='state',
city='city',
coduf=76,
codmun=2,
population=100000,
date=datetime.date.today(),
newcases=1,
totalcases=2,
newdeaths=1,
totaldeaths=1,
update=datetime.date.today())
self.db.session.commit()
response = datasus_services.get_graph_total_cases()
self.assertEqual(len(response), 1)
for current_date in response:
self.assertIn('region', current_date.keys())
self.assertIn('date', current_date.keys())
self.assertIn('totalCases', current_date.keys())
self.assertIn('totalDeaths', current_date.keys())
def get_sus_list(page):
try:
if page:
sus_list, pagination = DataSus().fetch_paginated(db.session, page)
else:
sus_list = DataSus().fetch_all(db.session)
pagination = {}
return {'sus_list': sus_list, 'pagination': pagination}
finally:
db.session.close()
def test_sus_graphs_last_30_days(self):
# Return only the last 30 elements
for days_amount in range(31):
current_day = datetime.date.today() - datetime.timedelta(
days=days_amount)
DataSus().save(self.db.session,
id=days_amount,
region='Brasil',
state='state',
date=current_day,
newcases=1,
totalcases=2,
newdeaths=1,
totaldeaths=1,
update=current_day)
self.db.session.commit()
response = datasus_services.get_graph_last_30_days()
self.assertEqual(len(response), 30)
for current_date in response:
self.assertIn('date', current_date.keys())
self.assertIn('totalCases', current_date.keys())
self.assertIn('totalDeaths', current_date.keys())
self.assertIn('newCases', current_date.keys())
self.assertIn('newDeaths', current_date.keys())
def test_if_save_method_saves_datasus_on_database(self):
DataSus().save(self.db.session,
region='Sudeste',
state='SP',
coduf=1,
city="São Paulo",
date=datetime.date.today(),
newcases=1,
totalcases=10,
newdeaths=1,
totaldeaths=2,
update=datetime.datetime.now(),
codmun=1,
codhealthregion=100,
namehealthregion='Região',
epiweek=20,
population=10000000,
recoverednew=5,
accompanyingnew=3)
self.db.session.commit()
_model = self.db.session.query(DataSus).filter_by(state='SP').first()
self.assertIsNotNone(_model)