diff --git a/tests/app/celery/test_provider_tasks.py b/tests/app/celery/test_provider_tasks.py index 5f1a4c925..ad875d339 100644 --- a/tests/app/celery/test_provider_tasks.py +++ b/tests/app/celery/test_provider_tasks.py @@ -69,6 +69,31 @@ def test_should_check_delivery_receipts_failure(sample_notification, mocker): mock_update.assert_called_once() +def test_should_check_delivery_receipts_client_error(sample_notification, mocker): + mocker.patch("app.delivery.send_to_providers.send_sms_to_provider") + mocker.patch( + "app.celery.provider_tasks.aws_cloudwatch_client.is_localstack", + return_value=False, + ) + mock_update = mocker.patch( + "app.celery.provider_tasks.update_notification_status_by_id" + ) + error_response = {"Error": {"Code": "SomeCode", "Message": "Some Message"}} + operation_name = "SomeOperation" + mocker.patch( + "app.celery.provider_tasks.aws_cloudwatch_client.check_sms", + side_effect=ClientError(error_response, operation_name), + ) + mock_sanitize = mocker.patch( + "app.celery.provider_tasks.sanitize_successful_notification_by_id" + ) + check_sms_delivery_receipt( + "message_id", sample_notification.id, "2024-10-20 00:00:00+0:00" + ) + mock_sanitize.assert_not_called() + mock_update.assert_called_once() + + def test_should_call_send_sms_to_provider_from_deliver_sms_task( sample_notification, mocker ):