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

Added a test to make sure the loop exits.

Browse Source
This commit is contained in:
Rebecca Law
2018-12-20 17:02:49 +00:00
parent 89923eab71
commit d8a0a3f5ab
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/app/dao/notification_dao/test_notification_dao_delete_notifications.py
View File

@@ -186,6 +186,20 @@ def test_delete_notifications_does_try_to_delete_from_s3_when_letter_has_not_bee
mock_get_s3.assert_not_called() mock_get_s3.assert_not_called()
def test_delete_notifications_calls_subquery(
notify_db_session, mocker
):
service = create_service()
sms_template = create_template(service=service)
create_notification(template=sms_template, created_at=datetime.now() - timedelta(days=8))
create_notification(template=sms_template, created_at=datetime.now() - timedelta(days=8))
create_notification(template=sms_template, created_at=datetime.now() - timedelta(days=8))
assert Notification.query.count() == 3
delete_notifications_created_more_than_a_week_ago_by_type('sms', qry_limit=1)
assert Notification.query.count() == 0
def _create_templates(sample_service): def _create_templates(sample_service):
email_template = create_template(service=sample_service, template_type='email') email_template = create_template(service=sample_service, template_type='email')
sms_template = create_template(service=sample_service) sms_template = create_template(service=sample_service)