From c40111a57609080c8013e4473dd1009e0cac357a Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Tue, 2 May 2017 11:20:01 +0100 Subject: [PATCH] Split up a test --- tests/app/dao/test_notification_usage_dao.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/app/dao/test_notification_usage_dao.py b/tests/app/dao/test_notification_usage_dao.py index efeb85d51..738510719 100644 --- a/tests/app/dao/test_notification_usage_dao.py +++ b/tests/app/dao/test_notification_usage_dao.py @@ -18,6 +18,12 @@ def test_get_rates_for_year(notify_db, notify_db_session): assert rates[0].rate == 1.50 assert datetime.strftime(rates[1].valid_from, '%Y-%m-%d %H:%M:%S') == "2016-09-01 00:00:00" assert rates[1].rate == 1.6 + + +def test_get_rates_for_year_returns_correct_rates(notify_db, notify_db_session): + set_up_rate(notify_db, datetime(2016, 4, 1), 1.50) + set_up_rate(notify_db, datetime(2016, 9, 1), 1.60) + set_up_rate(notify_db, datetime(2017, 6, 1), 1.75) start_date, end_date = get_financial_year(2017) rates_2017 = get_rates_for_year(start_date, end_date, 'sms') assert len(rates_2017) == 2