add endpoint for 'recent' notifications

This commit is contained in:
Kenneth Kehl
2025-01-30 13:34:44 -08:00
parent 240ec5a712
commit dd9c47e151
2 changed files with 51 additions and 0 deletions

View File

@@ -28,6 +28,7 @@ from app.dao.notifications_dao import (
get_notification_with_personalisation,
get_notifications_for_job,
get_notifications_for_service,
get_recent_notifications_for_job,
get_service_ids_with_notifications_on_date,
notifications_not_yet_sent,
sanitize_successful_notification_by_id,
@@ -692,6 +693,24 @@ def test_get_all_notifications_for_job(sample_job):
assert len(notifications_from_db) == 5
def test_get_recent_notifications_for_job(sample_job):
for x in range(0, 5):
try:
n = create_notification(template=sample_job.template, job=sample_job)
if x == 0:
n.status = NotificationStatus.DELIVERED
elif x in [1, 2]:
n.status = NotificationStatus.FAILED
except IntegrityError:
pass
notifications_from_db = get_recent_notifications_for_job(
sample_job.service.id, sample_job.id
).items
assert len(notifications_from_db) == 3
print(notifications_from_db)
def test_get_all_notifications_for_job_by_status(sample_job):
notifications = partial(
get_notifications_for_job, sample_job.service.id, sample_job.id