Created a task to update ft_notification_status for the last three days.

tests/app/dao/test_fact_notification_status_dao.py

@@ -0,0 +1,83 @@
+from datetime import timedelta, datetime
+from uuid import UUID
+
+from app.dao.fact_notification_status_dao import update_fact_notification_status, fetch_notification_status_for_day
+from app.models import FactNotificationStatus
+from tests.app.db import create_notification, create_service, create_template
+
+
+def test_update_fact_notification_status(notify_db_session):
+    first_service = create_service(service_name='First Service')
+    first_template = create_template(service=first_service)
+    second_service = create_service(service_name='second Service')
+    second_template = create_template(service=second_service, template_type='email')
+    third_service = create_service(service_name='third Service')
+    third_template = create_template(service=third_service, template_type='letter')
+
+    create_notification(template=first_template, status='delivered')
+    create_notification(template=first_template, created_at=datetime.utcnow() - timedelta(days=1))
+    create_notification(template=second_template, status='temporary-failure')
+    create_notification(template=second_template, created_at=datetime.utcnow() - timedelta(days=1))
+    create_notification(template=third_template, status='created')
+    create_notification(template=third_template, created_at=datetime.utcnow() - timedelta(days=1))
+
+    process_day = datetime.utcnow()
+    data = fetch_notification_status_for_day(process_day=process_day)
+    update_fact_notification_status(data=data, process_day=process_day)
+
+    new_fact_data = FactNotificationStatus.query.order_by(FactNotificationStatus.bst_date,
+                                                          FactNotificationStatus.notification_type
+                                                          ).all()
+
+    assert len(new_fact_data) == 3
+    assert new_fact_data[0].bst_date == process_day.date()
+    assert new_fact_data[0].template_id == second_template.id
+    assert new_fact_data[0].service_id == second_service.id
+    assert new_fact_data[0].job_id == UUID('00000000-0000-0000-0000-000000000000')
+    assert new_fact_data[0].notification_type == 'email'
+    assert new_fact_data[0].notification_status == 'temporary-failure'
+    assert new_fact_data[0].notification_count == 1
+
+    assert new_fact_data[1].bst_date == process_day.date()
+    assert new_fact_data[1].template_id == third_template.id
+    assert new_fact_data[1].service_id == third_service.id
+    assert new_fact_data[1].job_id == UUID('00000000-0000-0000-0000-000000000000')
+    assert new_fact_data[1].notification_type == 'letter'
+    assert new_fact_data[1].notification_status == 'created'
+    assert new_fact_data[1].notification_count == 1
+
+    assert new_fact_data[2].bst_date == process_day.date()
+    assert new_fact_data[2].template_id == first_template.id
+    assert new_fact_data[2].service_id == first_service.id
+    assert new_fact_data[2].job_id == UUID('00000000-0000-0000-0000-000000000000')
+    assert new_fact_data[2].notification_type == 'sms'
+    assert new_fact_data[2].notification_status == 'delivered'
+    assert new_fact_data[2].notification_count == 1
+
+
+def test__update_fact_notification_status_updates_row(notify_db_session):
+    first_service = create_service(service_name='First Service')
+    first_template = create_template(service=first_service)
+    create_notification(template=first_template, status='delivered')
+
+    process_day = datetime.utcnow()
+    data = fetch_notification_status_for_day(process_day=process_day)
+    update_fact_notification_status(data=data, process_day=process_day)
+
+    new_fact_data = FactNotificationStatus.query.order_by(FactNotificationStatus.bst_date,
+                                                          FactNotificationStatus.notification_type
+                                                          ).all()
+    assert len(new_fact_data) == 1
+    assert new_fact_data[0].notification_count == 1
+
+    create_notification(template=first_template, status='delivered')
+
+    data = fetch_notification_status_for_day(process_day=process_day)
+    update_fact_notification_status(data=data, process_day=process_day)
+
+    updated_fact_data = FactNotificationStatus.query.order_by(FactNotificationStatus.bst_date,
+                                                              FactNotificationStatus.notification_type
+                                                              ).all()
+    assert len(updated_fact_data) == 1
+    assert updated_fact_data[0].notification_count == 2
+