darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1580 from GSA/track_costs

Browse Source 
fix test
This commit is contained in:
Kenneth Kehl
2025-03-04 12:29:13 -08:00
committed by GitHub
parent 819dd89854 431a09e3cf
commit ecafaf847f
1 changed files with 9 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tests/app/dao/test_fact_notification_status_dao.py
View File

@@ -36,7 +36,6 @@ def test_fetch_notification_status_for_service_by_month(notify_db_session):
service_2 = create_service(service_name="service_2") service_2 = create_service(service_name="service_2")
create_template(service=service_1) create_template(service=service_1)
create_template(service=service_1, template_type=TemplateType.EMAIL)
# not the service being tested # not the service being tested
create_template(service=service_2) create_template(service=service_2)
@@ -47,18 +46,9 @@ def test_fetch_notification_status_for_service_by_month(notify_db_session):
created_at=datetime(2018, 1, 1, 1, x, 0), created_at=datetime(2018, 1, 1, 1, x, 0),
status=NotificationStatus.DELIVERED, status=NotificationStatus.DELIVERED,
) )
whats_this = create_notification( create_notification(
service_1.templates[0], created_at=datetime(2018, 1, 1, 1, 1, 0) service_1.templates[0], created_at=datetime(2018, 1, 1, 1, 1, 0)
) )
print(f"WTN status = {whats_this.status} type = {whats_this.notification_type}")
questionable_notification = create_notification(
service_1.templates[1],
created_at=datetime(2018, 1, 1, 1, 1, 0),
status=NotificationStatus.DELIVERED,
)
print(
f"QN status = {questionable_notification.status} type = {questionable_notification.notification_type}"
)
create_notification( create_notification(
service_1.templates[0], service_1.templates[0],
@@ -84,30 +74,25 @@ def test_fetch_notification_status_for_service_by_month(notify_db_session):
fetch_notification_status_for_service_by_month( fetch_notification_status_for_service_by_month(
date(2018, 1, 1), date(2018, 2, 28), service_1.id date(2018, 1, 1), date(2018, 2, 28), service_1.id
), ),
key=lambda x: (x.month, x.notification_type, x.notification_status), key=lambda x: (x.month, x.notification_status),
) )
assert len(results) == 4 assert len(results) == 3
assert results[0].month.date() == date(2018, 1, 1) assert results[0].month.date() == date(2018, 1, 1)
assert results[0].notification_type == NotificationType.EMAIL assert results[0].notification_type == NotificationType.SMS
assert results[0].notification_status == NotificationStatus.DELIVERED assert results[0].notification_status == NotificationStatus.CREATED
assert results[0].count == 1 assert results[0].count == 1
assert results[1].month.date() == date(2018, 1, 1) assert results[1].month.date() == date(2018, 1, 1)
assert results[1].notification_type == NotificationType.SMS assert results[1].notification_type == NotificationType.SMS
assert results[1].notification_status == NotificationStatus.CREATED assert results[1].notification_status == NotificationStatus.DELIVERED
assert results[1].count == 1 assert results[1].count == 14
assert results[2].month.date() == date(2018, 1, 1) assert results[2].month.date() == date(2018, 2, 1)
assert results[2].notification_type == NotificationType.SMS assert results[2].notification_type == NotificationType.SMS
assert results[2].notification_status == NotificationStatus.DELIVERED assert results[2].notification_status == NotificationStatus.DELIVERED
assert results[2].count == 14 assert results[2].count == 1
assert results[3].month.date() == date(2018, 2, 1)
assert results[3].notification_type == NotificationType.SMS
assert results[3].notification_status == NotificationStatus.DELIVERED
assert results[3].count == 1
def test_fetch_notification_status_for_service_for_day(notify_db_session): def test_fetch_notification_status_for_service_for_day(notify_db_session):