From 6de1b226d0cd01429f6d461537a275c9fc22481e Mon Sep 17 00:00:00 2001 From: Kenneth Kehl <@kkehl@flexion.us> Date: Fri, 13 Dec 2024 10:15:47 -0800 Subject: [PATCH] cleanup --- app/delivery/send_to_providers.py | 5 ++++- tests/app/celery/test_provider_tasks.py | 1 - 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/app/delivery/send_to_providers.py b/app/delivery/send_to_providers.py index 4cc57fa6a..e41062b41 100644 --- a/app/delivery/send_to_providers.py +++ b/app/delivery/send_to_providers.py @@ -16,7 +16,10 @@ from app import ( from app.aws.s3 import get_personalisation_from_s3, get_phone_number_from_s3 from app.celery.test_key_tasks import send_email_response, send_sms_response from app.dao.email_branding_dao import dao_get_email_branding_by_id -from app.dao.notifications_dao import dao_update_notification, update_notification_message_id +from app.dao.notifications_dao import ( + dao_update_notification, + update_notification_message_id, +) from app.dao.provider_details_dao import get_provider_details_by_notification_type from app.dao.service_sms_sender_dao import dao_get_sms_senders_by_service_id from app.enums import BrandType, KeyType, NotificationStatus, NotificationType diff --git a/tests/app/celery/test_provider_tasks.py b/tests/app/celery/test_provider_tasks.py index 7923f9498..80d5a0d7e 100644 --- a/tests/app/celery/test_provider_tasks.py +++ b/tests/app/celery/test_provider_tasks.py @@ -131,7 +131,6 @@ def test_should_call_send_sms_to_provider_from_deliver_sms_task( ): mocker.patch("app.delivery.send_to_providers.send_sms_to_provider") mocker.patch("app.celery.provider_tasks.check_sms_delivery_receipt") - mocker.patch("app.celery.provider_tasks.update_notification_message_id") deliver_sms(sample_notification.id) app.delivery.send_to_providers.send_sms_to_provider.assert_called_with(