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

@@ -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