From edf4c0a9ae2a396957697c0075ee047a40d38706 Mon Sep 17 00:00:00 2001 From: Daniel Szoke Date: Thu, 25 Apr 2024 12:21:13 +0200 Subject: [PATCH] test(celery): Add test for nonzero retry count Add a test to verify that retry count gets set correctly, also for nonzero values. --- tests/integrations/celery/test_celery.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/tests/integrations/celery/test_celery.py b/tests/integrations/celery/test_celery.py index bb95a315a2..3a0ce746d3 100644 --- a/tests/integrations/celery/test_celery.py +++ b/tests/integrations/celery/test_celery.py @@ -618,7 +618,7 @@ def example_task(): assert result.get() == "success" -def test_retry_count(init_celery, capture_events): +def test_retry_count_zero(init_celery, capture_events): celery = init_celery(enable_tracing=True) events = capture_events() @@ -630,3 +630,20 @@ def task(): ... (event,) = events (span,) = event["spans"] assert span["data"]["messaging.message.retry.count"] == 0 + + +@mock.patch("celery.app.task.Task.request") +def test_retry_count_nonzero(mock_request, init_celery, capture_events): + mock_request.retries = 3 + + celery = init_celery(enable_tracing=True) + events = capture_events() + + @celery.task() + def task(): ... + + task.apply_async() + + (event,) = events + (span,) = event["spans"] + assert span["data"]["messaging.message.retry.count"] == 3