darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-21 07:51:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Set filter on query

Browse Source
This commit is contained in:
Rebecca Law
2018-04-25 16:17:56 +01:00
parent e32debbca1
commit cff731c52d
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/dao/fact_billing_dao.py
View File

@@ -94,7 +94,7 @@ def fetch_billing_data_for_day(process_day, service_id=None):
Service Service
) )
if service_id: if service_id:
transit_data.filter(Notification.service_id == service_id) transit_data = transit_data.filter(Notification.service_id == service_id)
return transit_data.all() return transit_data.all()

3
tests/app/dao/test_ft_billing_dao.py
View File

@@ -165,7 +165,8 @@ def test_get_rate(notify_db_session):
create_rate(start_date=datetime.utcnow(), value=3.3, notification_type='email') create_rate(start_date=datetime.utcnow(), value=3.3, notification_type='email')
create_letter_rate(start_date=datetime.utcnow(), rate=4.4) create_letter_rate(start_date=datetime.utcnow(), rate=4.4)
non_letter_rates, letter_rates = get_rates_for_billing() non_letter_rates, letter_rates = get_rates_for_billing()
rate = get_rate(non_letter_rates=non_letter_rates, letter_rates=letter_rates, notification_type='sms', date=datetime.utcnow()) rate = get_rate(non_letter_rates=non_letter_rates, letter_rates=letter_rates, notification_type='sms',
date=datetime.utcnow())
letter_rate = get_rate(non_letter_rates=non_letter_rates, letter_rates=letter_rates, letter_rate = get_rate(non_letter_rates=non_letter_rates, letter_rates=letter_rates,
notification_type='letter', notification_type='letter',
crown=True, crown=True,