diff --git a/app/performance_platform/total_sent_notifications.py b/app/performance_platform/total_sent_notifications.py index 6dc1d039a..324010015 100644 --- a/app/performance_platform/total_sent_notifications.py +++ b/app/performance_platform/total_sent_notifications.py @@ -1,5 +1,6 @@ from app import performance_platform_client from app.dao.fact_notification_status_dao import get_total_sent_notifications_for_day_and_type +from app.utils import get_london_midnight_in_utc def send_total_notifications_sent_for_day_stats(date, notification_type, count): @@ -19,8 +20,10 @@ def get_total_sent_notifications_for_day(day): sms_count = get_total_sent_notifications_for_day_and_type(day, 'sms') letter_count = get_total_sent_notifications_for_day_and_type(day, 'letter') + start_date = get_london_midnight_in_utc(day) + return { - "start_date": day, + "start_date": start_date, "email": { "count": email_count }, diff --git a/tests/app/performance_platform/test_total_sent_notifications.py b/tests/app/performance_platform/test_total_sent_notifications.py index 0945903cc..c291a3794 100644 --- a/tests/app/performance_platform/test_total_sent_notifications.py +++ b/tests/app/performance_platform/test_total_sent_notifications.py @@ -53,15 +53,11 @@ def test_get_total_sent_notifications_yesterday_returns_expected_totals_dict(sam total_count_dict = get_total_sent_notifications_for_day(yesterday) - assert total_count_dict == { - "start_date": yesterday, - "email": { - "count": 3 - }, - "sms": { - "count": 2 - }, - "letter": { - "count": 1 - } - } + assert total_count_dict["email"] == {"count": 3} + assert total_count_dict["sms"] == {"count": 2} + assert total_count_dict["letter"] == {"count": 1} + + # Should return a time around midnight depending on timezones + expected_start = datetime.combine(yesterday, datetime.min.time()) + time_diff = abs(expected_start - total_count_dict["start_date"]) + assert time_diff <= timedelta(minutes=60)