From 97d1116d72b6c11c4606d4ff5fc61f5c6b8a7fda Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Sun, 1 Dec 2024 19:36:25 -0800
Subject: [PATCH] Cover retry

---
 src/documents/tests/test_workflows.py | 30 +++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/documents/tests/test_workflows.py b/src/documents/tests/test_workflows.py
index ddad49588..b861672b5 100644
--- a/src/documents/tests/test_workflows.py
+++ b/src/documents/tests/test_workflows.py
@@ -10,6 +10,7 @@ from django.utils import timezone
 from guardian.shortcuts import assign_perm
 from guardian.shortcuts import get_groups_with_perms
 from guardian.shortcuts import get_users_with_perms
+from httpx import HTTPStatusError
 from rest_framework.test import APITestCase
 
 from documents.signals.handlers import run_workflows
@@ -18,6 +19,8 @@ from documents.signals.handlers import send_webhook
 if TYPE_CHECKING:
     from django.db.models import QuerySet
 
+from celery.exceptions import MaxRetriesExceededError
+
 from documents import tasks
 from documents.data_models import ConsumableDocument
 from documents.data_models import DocumentSource
@@ -2534,3 +2537,30 @@ class TestWorkflows(
 
             expected_str = "Webhook sent to http://paperless-ngx.com"
             self.assertIn(expected_str, cm.output[0])
+
+    @mock.patch("httpx.post")
+    @mock.patch("documents.signals.handlers.send_webhook.retry")
+    def test_workflow_webhook_send_webhook_retry(self, mock_retry, mock_http):
+        mock_http.return_value.raise_for_status = mock.Mock(
+            side_effect=HTTPStatusError(
+                "Error",
+                request=mock.Mock(),
+                response=mock.Mock(),
+            ),
+        )
+        mock_retry.side_effect = MaxRetriesExceededError("Max retries exceeded")
+
+        with self.assertLogs("paperless.handlers") as cm:
+            send_webhook(
+                url="http://paperless-ngx.com",
+                data="Test message",
+                headers={},
+                files=None,
+            )
+
+            self.assertEqual(mock_retry.call_count, 1)
+
+            expected_str = "Failed sending webhook to http://paperless-ngx.com"
+            expected_str2 = "Max retries exceeded"
+            self.assertIn(expected_str, cm.output[0])
+            self.assertIn(expected_str2, cm.output[1])