mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-17 04:58:50 -04:00
Refactor code to make it clearer
- Added code in `delete_notifications_created_more_than_a_week_ago_by_type` to remove notifications to email_reply_to older than 7 days - Added transactional to `delete_notifications_created_more_than_a_week_ago_by_type`
This commit is contained in:
@@ -126,12 +126,9 @@ def test_should_call_delete_sms_notifications_more_than_week_in_task(notify_api,
|
||||
|
||||
|
||||
def test_should_call_delete_email_notifications_more_than_week_in_task(notify_api, mocker):
|
||||
mocked_delete_noti_to_reply_to = mocker.patch(
|
||||
'app.celery.scheduled_tasks.delete_notification_to_email_reply_to_more_than_a_week_ago')
|
||||
mocked_notifications = mocker.patch(
|
||||
'app.celery.scheduled_tasks.delete_notifications_created_more_than_a_week_ago_by_type')
|
||||
delete_email_notifications_older_than_seven_days()
|
||||
mocked_delete_noti_to_reply_to.assert_called_once_with()
|
||||
mocked_notifications.assert_called_once_with('email')
|
||||
|
||||
|
||||
|
||||
@@ -41,7 +41,6 @@ from app.dao.notifications_dao import (
|
||||
dao_update_notification,
|
||||
dao_update_notifications_for_job_to_sent_to_dvla,
|
||||
delete_notifications_created_more_than_a_week_ago_by_type,
|
||||
delete_notification_to_email_reply_to_more_than_a_week_ago,
|
||||
get_notification_by_id,
|
||||
get_notification_for_job,
|
||||
get_notification_with_personalisation,
|
||||
@@ -1027,7 +1026,6 @@ def test_should_delete_notification_to_email_reply_to_after_seven_days(
|
||||
assert len(all_notification_email_reply_to) == 10
|
||||
|
||||
# Records before 3rd should be deleted
|
||||
delete_notification_to_email_reply_to_more_than_a_week_ago()
|
||||
delete_notifications_created_more_than_a_week_ago_by_type(EMAIL_TYPE)
|
||||
remaining_email_notifications = Notification.query.filter_by(notification_type=EMAIL_TYPE).all()
|
||||
remaining_notification_to_email_reply_to = NotificationEmailReplyTo.query.filter_by().all()
|
||||
|
||||
Reference in New Issue
Block a user