darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-20 15:31:15 -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
)
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()

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_letter_rate(start_date=datetime.utcnow(), rate=4.4)
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,
notification_type='letter',
crown=True,