From 2acd0a87b9667423086026f437165caf2bb258b2 Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Fri, 10 Jan 2025 15:02:43 -0800
Subject: [PATCH] fix uuid

---
 tests/app/celery/test_tasks.py                             | 7 +++----
 .../service/send_notification/test_send_notification.py    | 4 ++--
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/tests/app/celery/test_tasks.py b/tests/app/celery/test_tasks.py
index 12c15d334..ccc19bc8d 100644
--- a/tests/app/celery/test_tasks.py
+++ b/tests/app/celery/test_tasks.py
@@ -435,7 +435,7 @@ def test_should_send_template_to_correct_sms_task_and_persist(
     assert persisted_notification.personalisation == {}
     assert persisted_notification.notification_type == NotificationType.SMS
     mocked_deliver_sms.assert_called_once_with(
-        [str(persisted_notification.id)], queue="send-sms-tasks"
+        [str(persisted_notification.id)], queue="send-sms-tasks", countdown=30
     )
 
 
@@ -471,7 +471,7 @@ def test_should_save_sms_if_restricted_service_and_valid_number(
     assert not persisted_notification.personalisation
     assert persisted_notification.notification_type == NotificationType.SMS
     provider_tasks.deliver_sms.apply_async.assert_called_once_with(
-        [str(persisted_notification.id)], queue="send-sms-tasks"
+        [str(persisted_notification.id)], queue="send-sms-tasks", countdown=30
     )
 
 
@@ -670,7 +670,7 @@ def test_should_use_email_template_and_persist(
     assert persisted_notification.notification_type == NotificationType.EMAIL
 
     provider_tasks.deliver_email.apply_async.assert_called_once_with(
-        [str(persisted_notification.id)], queue="send-email-tasks", countdown=30
+        [str(persisted_notification.id)], queue="send-email-tasks"
     )
 
 
@@ -946,7 +946,6 @@ def test_save_sms_uses_sms_sender_reply_to_text(mocker, notify_db_session):
         encryption.encrypt(notification),
     )
 
-    batch_insert_notifications()
     persisted_notification = Notification.query.one()
     assert persisted_notification.reply_to_text == "+12028675309"
 
diff --git a/tests/app/service/send_notification/test_send_notification.py b/tests/app/service/send_notification/test_send_notification.py
index 0ec50428e..d0b49a982 100644
--- a/tests/app/service/send_notification/test_send_notification.py
+++ b/tests/app/service/send_notification/test_send_notification.py
@@ -150,7 +150,7 @@ def test_send_notification_with_placeholders_replaced(
                 {"template_version": sample_email_template_with_placeholders.version}
             )
 
-            mocked.assert_called_once_with([notification_id], queue="send-email-tasks")
+            mocked.assert_called_once_with([notification_id], queue="send-email-tasks", countdown=30)
             assert response.status_code == 201
             assert response_data["body"] == "Hello Jo\nThis is an email from GOV.UK"
             assert response_data["subject"] == "Jo"
@@ -1185,7 +1185,7 @@ def test_should_allow_store_original_number_on_sms_notification(
     response_data = json.loads(response.data)["data"]
     notification_id = response_data["notification"]["id"]
 
-    mocked.assert_called_once_with([notification_id], queue="send-sms-tasks")
+    mocked.assert_called_once_with([notification_id], queue="send-sms-tasks", countdown=30)
     assert response.status_code == 201
     assert notification_id
     notifications = Notification.query.all()