From 198a619ffbb185ccc8995ae1de61772e35c3d822 Mon Sep 17 00:00:00 2001
From: Andrew Leith <andrew.leith@cds-snc.ca>
Date: Wed, 18 Dec 2024 16:24:14 +0000
Subject: [PATCH 1/4] fix(start_job): add exception handling in case API throws
 an annual limits error at this point; update the expected error message
 strings

---
 app/main/views/send.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/app/main/views/send.py b/app/main/views/send.py
index 53acc9ba63..eaf88d5252 100644
--- a/app/main/views/send.py
+++ b/app/main/views/send.py
@@ -888,8 +888,12 @@ def check_notification_preview(service_id, template_id, filetype):
 @main.route("/services/<service_id>/start-job/<upload_id>", methods=["POST"])
 @user_has_permissions("send_messages", restrict_admin_usage=True)
 def start_job(service_id, upload_id):
-    job_api_client.create_job(upload_id, service_id, scheduled_for=request.form.get("scheduled_for", ""))
-
+    try:
+        job_api_client.create_job(upload_id, service_id, scheduled_for=request.form.get("scheduled_for", ""))
+    except HTTPError as exception:
+        return render_template(
+            "views/notifications/check.html", **(get_template_error_dict(exception) if exception else {}), template=None
+        )
     session.pop("sender_id", None)
 
     return redirect(
@@ -1107,11 +1111,12 @@ def get_template_error_dict(exception):
         error = "too-many-sms-messages"
     elif "Content for template has a character count greater than the limit of" in exception.message:
         error = "message-too-long"
-    elif "Exceeded annual email sending limit" in exception.message:
+    elif "Exceeded annual email sending" in exception.message:
         error = "too-many-email-annual"
-    elif "Exceeded annual SMS sending limit" in exception.message:
+    elif "Exceeded annual SMS sending" in exception.message:
         error = "too-many-sms-annual"
     else:
+        current_app.logger.error("Unhandled exception from API: {}".format(exception))
         raise exception
 
     return {

From 60d6426c16ee75810d6e898f6f3b0cf195b4106a Mon Sep 17 00:00:00 2001
From: Andrew Leith <andrew.leith@cds-snc.ca>
Date: Wed, 18 Dec 2024 16:24:36 +0000
Subject: [PATCH 2/4] fix(check.html): don't render the template preview in the
 case of an API exception

---
 app/templates/views/notifications/check.html | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/templates/views/notifications/check.html b/app/templates/views/notifications/check.html
index 0161a74f17..e9d61b7f94 100644
--- a/app/templates/views/notifications/check.html
+++ b/app/templates/views/notifications/check.html
@@ -88,7 +88,9 @@ <h2 class="heading-medium">{{_('You cannot send this email message today') }}</h
     ) }}
   {% endif %}
 
-  {{ template|string|translate_preview_template }}
+  {% if template %}
+    {{ template|string|translate_preview_template }}
+  {% endif %}
 
   <div class="js-stick-at-bottom-when-scrolling">
     <form method="post" enctype="multipart/form-data" action="{{url_for(

From 318052a29bbf071547812a49f42b9a4e025dec53 Mon Sep 17 00:00:00 2001
From: Andrew Leith <andrew.leith@cds-snc.ca>
Date: Wed, 18 Dec 2024 16:24:44 +0000
Subject: [PATCH 3/4] chore: add tests

---
 tests/app/main/views/test_send.py | 98 +++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)

diff --git a/tests/app/main/views/test_send.py b/tests/app/main/views/test_send.py
index a5aeb2db72..706abb69ad 100644
--- a/tests/app/main/views/test_send.py
+++ b/tests/app/main/views/test_send.py
@@ -3679,3 +3679,101 @@ def test_correct_error_displayed(
             elif error_shown == "none":
                 assert page.find(attrs={"data-testid": "exceeds-annual"}) is None
                 assert page.find(attrs={"data-testid": "exceeds-daily"}) is None
+
+    @pytest.mark.parametrize(
+        "notification_type, exception_msg_api, expected_error_msg_admin",
+        [
+            ("email", "Exceeded annual email sending", "These messages exceed the annual limit"),
+            ("sms", "Exceeded annual SMS sending", "These messages exceed the annual limit"),
+        ],
+    )
+    def test_error_msgs_from_api_for_one_off(
+        self,
+        client_request,
+        service_one,
+        fake_uuid,
+        mocker,
+        mock_get_service_template_with_placeholders,
+        mock_get_template_statistics,
+        notification_type,
+        exception_msg_api,
+        expected_error_msg_admin,
+    ):
+        class MockHTTPError(HTTPError):
+            message = exception_msg_api
+
+        mocker.patch(
+            "app.notification_api_client.send_notification",
+            side_effect=MockHTTPError(),
+        )
+
+        if notification_type == "sms":
+            with client_request.session_transaction() as session:
+                session["recipient"] = "6502532223"
+                session["placeholders"] = {"name": "a" * 600}
+        elif notification_type == "email":
+            with client_request.session_transaction() as session:
+                session["recipient"] = "test@example.com"
+                session["placeholders"] = {"name": "a" * 600}
+
+        page = client_request.post(
+            "main.send_notification",
+            service_id=service_one["id"],
+            template_id=fake_uuid,
+            _expected_status=200,
+        )
+
+        assert normalize_spaces(page.select("h1")[0].text) == expected_error_msg_admin
+
+    @pytest.mark.parametrize(
+        "exception_msg_api, expected_error_msg_admin",
+        [
+            # ("email","Exceeded annual email sending", "These messages exceed the annual limit"),
+            ("Exceeded annual SMS sending", "These messages exceed the annual limit")
+        ],
+    )
+    def test_error_msgs_from_api_for_bulk(
+        self,
+        client_request,
+        mock_create_job,
+        mock_get_job,
+        mock_get_notifications,
+        mock_get_service_template,
+        mock_get_service_data_retention,
+        mocker,
+        fake_uuid,
+        exception_msg_api,
+        expected_error_msg_admin,
+    ):
+        class MockHTTPError(HTTPError):
+            message = exception_msg_api
+
+        data = mock_get_job(SERVICE_ONE_ID, fake_uuid)["data"]
+        job_id = data["id"]
+        original_file_name = data["original_file_name"]
+        template_id = data["template"]
+        notification_count = data["notification_count"]
+        with client_request.session_transaction() as session:
+            session["file_uploads"] = {
+                fake_uuid: {
+                    "template_id": template_id,
+                    "notification_count": notification_count,
+                    "valid": True,
+                }
+            }
+
+        mocker.patch(
+            "app.job_api_client.create_job",
+            side_effect=MockHTTPError(),
+        )
+        page = client_request.post(
+            "main.start_job",
+            service_id=SERVICE_ONE_ID,
+            upload_id=job_id,
+            original_file_name=original_file_name,
+            _data={},
+            _follow_redirects=True,
+            _expected_status=200,
+        )
+
+        assert normalize_spaces(page.select("h1")[0].text) == expected_error_msg_admin

From 5757c1c4fcd173c103d37aa7c5bede6fe8cde274 Mon Sep 17 00:00:00 2001
From: Andrew Leith <andrew.leith@cds-snc.ca>
Date: Wed, 18 Dec 2024 17:09:42 +0000
Subject: [PATCH 4/4] fix: pass time_to_reset into the view

---
 app/main/views/send.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/main/views/send.py b/app/main/views/send.py
index eaf88d5252..f0e4e86406 100644
--- a/app/main/views/send.py
+++ b/app/main/views/send.py
@@ -892,7 +892,10 @@ def start_job(service_id, upload_id):
         job_api_client.create_job(upload_id, service_id, scheduled_for=request.form.get("scheduled_for", ""))
     except HTTPError as exception:
         return render_template(
-            "views/notifications/check.html", **(get_template_error_dict(exception) if exception else {}), template=None
+            "views/notifications/check.html",
+            time_to_reset=get_limit_reset_time_et(),
+            **(get_template_error_dict(exception) if exception else {}),
+            template=None,
         )
     session.pop("sender_id", None)