Send rejected event so Leek picks it up (#3374)

Author: mbertrand

main/constants.py

@@ -40,6 +40,7 @@
     "ul",
 }
 ALLOWED_HTML_ATTRIBUTES = {}
+TASK_REJECTED = "task-rejected"
 
 
 class PostHogEvents(Enum):

main/decorators.py

@@ -9,6 +9,8 @@
 from celery.exceptions import Reject
 from django.core.cache import caches
 
+from main.constants import TASK_REJECTED
+
 log = logging.getLogger(__name__)
 
 KEY_PREFIX = "cooldown"
@@ -80,6 +82,12 @@ def wrapper(*args, **kwargs):
                         state=states.REJECTED,
                         meta={"reason": "cooldown", "key": lock_key},
                     )
+                    try:
+                        current_task.send_event(TASK_REJECTED, requeue=False)
+                    except Exception:  # noqa: BLE001
+                        log.warning(
+                            "Failed to emit task-rejected event for %s", lock_key
+                        )
                     reason = "cooldown active"
                     raise Reject(reason, requeue=False)
                 return None

main/decorators_test.py

@@ -3,6 +3,7 @@
 import pytest
 from celery.exceptions import Reject
 
+from main.constants import TASK_REJECTED
 from main.decorators import cooldown_task
 
 
@@ -167,6 +168,7 @@ def my_task():
     mock_task.update_state.assert_called_once()
     assert mock_task.update_state.call_args.kwargs["state"] == "REJECTED"
     assert mock_task.update_state.call_args.kwargs["meta"]["reason"] == "cooldown"
+    mock_task.send_event.assert_called_once_with(TASK_REJECTED, requeue=False)
 
 
 def test_cooldown_task_returns_none_when_called_directly(mock_redis, mocker):
@@ -181,3 +183,4 @@ def my_task():
 
     assert my_task() is None
     mock_task.update_state.assert_not_called()
+    mock_task.send_event.assert_not_called()