assertions

untitaker · untitaker · commit 3c90aa0679a2 · 2020-10-22T15:46:36.000+02:00
diff --git a/tests/integrations/celery/test_celery.py b/tests/integrations/celery/test_celery.py
@@ -105,8 +105,11 @@ def test_simple(capture_events, celery, celery_invocation, custom_celery_task_cl
 
     if custom_celery_task_cls:
 
+        custom_calls = []
+
         class CustomTask(celery.Task):
             def __call__(self, *args, **kwargs):
+                custom_calls.append(1)
                 return self.run(*args, **kwargs)
 
         celery.Task = CustomTask
@@ -122,6 +125,8 @@ def dummy_task(x, y):
         celery_invocation(dummy_task, 1, 2)
         _, expected_context = celery_invocation(dummy_task, 1, 0)
 
+    assert not custom_celery_task_cls or custom_calls
+
     (event,) = events
 
     assert event["contexts"]["trace"]["trace_id"] == transaction.trace_id
