From d23303a826d0b02963c9c9c6bdb6dd37156fc7f9 Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Tue, 1 Jul 2025 12:43:23 -0700
Subject: [PATCH] more tests

---
 tests/app/celery/test_tasks.py | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/tests/app/celery/test_tasks.py b/tests/app/celery/test_tasks.py
index fac121047..8e1a94c5b 100644
--- a/tests/app/celery/test_tasks.py
+++ b/tests/app/celery/test_tasks.py
@@ -1,12 +1,13 @@
 import json
 import uuid
 from datetime import datetime, timedelta
-from unittest.mock import ANY, MagicMock, Mock, call
+from unittest.mock import ANY, MagicMock, Mock, call, patch
 
 import pytest
 import requests_mock
 from celery.exceptions import Retry
 from freezegun import freeze_time
+from psycopg2 import IntegrityError
 from requests import RequestException
 from sqlalchemy import func, select
 from sqlalchemy.exc import SQLAlchemyError
@@ -22,6 +23,7 @@ from app.celery.tasks import (
     process_row,
     s3,
     save_api_email,
+    save_api_email_or_sms,
     save_api_sms,
     save_email,
     save_sms,
@@ -1721,3 +1723,32 @@ def test_total_sending_limits_exceeded(mocker):
     assert mock_job.job_status == "sending limits exceeded"
     assert mock_job.processing_finished == datetime(2024, 11, 10, 12, 0, 0)
     mock_dao_update_job.assert_called_once_with(mock_job)
+
+
+def test_save_api_email_or_sms_integrity_error():
+    encrypted = MagicMock()
+    decrypted = {
+        "id": "notif-id",
+        "service_id": "service-id",
+        "notification_type": "email",
+        "template_id": "template-id",
+        "template_version": 1,
+        "to": "test@example.com",
+        "client_reference": None,
+        "created_at": "2025-01-01T00:00:00",
+        "reply_to_text": None,
+        "status": "created",
+        "document_download_count": 0,
+    }
+
+    with patch("app.tasks.encyrption.decrypt", return_value=decrypted), patch(
+        "app.tasks.SerialisedService.from_id"
+    ), patch("app.tasks.get_notification", return_value=None), patch(
+        "app.tasks.persist_notification", side_effect=IntegrityError("msg", None, None)
+    ), patch(
+        "app.tasks.current_app.logger.warning"
+    ) as mock_log:
+
+        save_api_email_or_sms(encrypted)
+        mock_log.assert_called_once()
+        assert "already exists" in mock_log.call_args[0][0]