mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-01 07:35:34 -05:00
add endpoint for 'recent' notifications
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user