Skip to content

Commit 89a6509

Browse files
committed
This is also the enable_tracing init option.
1 parent 96110f7 commit 89a6509

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

tests/integrations/celery/test_celery.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,7 @@ def example_task():
611611
def test_messaging_destination_name_default_exchange(
612612
mock_request, routing_key, init_celery, capture_events
613613
):
614-
celery_app = init_celery(enable_tracing=True)
614+
celery_app = init_celery(traces_sample_rate=1.0)
615615
events = capture_events()
616616
mock_request.delivery_info = {"routing_key": routing_key, "exchange": ""}
617617

@@ -635,7 +635,7 @@ def test_messaging_destination_name_nondefault_exchange(
635635
that the routing key is the queue name. Other exchanges may not guarantee this
636636
behavior.
637637
"""
638-
celery_app = init_celery(enable_tracing=True)
638+
celery_app = init_celery(traces_sample_rate=1.0)
639639
events = capture_events()
640640
mock_request.delivery_info = {"routing_key": "celery", "exchange": "custom"}
641641

@@ -650,7 +650,7 @@ def task(): ...
650650

651651

652652
def test_messaging_id(init_celery, capture_events):
653-
celery = init_celery(enable_tracing=True)
653+
celery = init_celery(traces_sample_rate=1.0)
654654
events = capture_events()
655655

656656
@celery.task
@@ -664,7 +664,7 @@ def example_task(): ...
664664

665665

666666
def test_retry_count_zero(init_celery, capture_events):
667-
celery = init_celery(enable_tracing=True)
667+
celery = init_celery(traces_sample_rate=1.0)
668668
events = capture_events()
669669

670670
@celery.task()
@@ -681,7 +681,7 @@ def task(): ...
681681
def test_retry_count_nonzero(mock_request, init_celery, capture_events):
682682
mock_request.retries = 3
683683

684-
celery = init_celery(enable_tracing=True)
684+
celery = init_celery(traces_sample_rate=1.0)
685685
events = capture_events()
686686

687687
@celery.task()
@@ -696,7 +696,7 @@ def task(): ...
696696

697697
@pytest.mark.parametrize("system", ("redis", "amqp"))
698698
def test_messaging_system(system, init_celery, capture_events):
699-
celery = init_celery(enable_tracing=True)
699+
celery = init_celery(traces_sample_rate=1.0)
700700
events = capture_events()
701701

702702
# Does not need to be a real URL, since we use always eager
@@ -721,7 +721,7 @@ def publish(*args, **kwargs):
721721

722722
monkeypatch.setattr(kombu.messaging.Producer, "_publish", publish)
723723

724-
sentry_init(integrations=[CeleryIntegration()], enable_tracing=True)
724+
sentry_init(integrations=[CeleryIntegration()], traces_sample_rate=1.0)
725725
celery = Celery(__name__, broker=f"{system}://example.com") # noqa: E231
726726
events = capture_events()
727727

@@ -759,7 +759,7 @@ def task(): ...
759759

760760

761761
def tests_span_origin_consumer(init_celery, capture_events):
762-
celery = init_celery(enable_tracing=True)
762+
celery = init_celery(traces_sample_rate=1.0)
763763
celery.conf.broker_url = "redis://example.com" # noqa: E231
764764

765765
events = capture_events()
@@ -783,7 +783,7 @@ def publish(*args, **kwargs):
783783

784784
monkeypatch.setattr(kombu.messaging.Producer, "_publish", publish)
785785

786-
sentry_init(integrations=[CeleryIntegration()], enable_tracing=True)
786+
sentry_init(integrations=[CeleryIntegration()], traces_sample_rate=1.0)
787787
celery = Celery(__name__, broker="redis://example.com") # noqa: E231
788788

789789
events = capture_events()
@@ -812,7 +812,7 @@ def test_send_task_wrapped(
812812
capture_events,
813813
reset_integrations,
814814
):
815-
sentry_init(integrations=[CeleryIntegration()], enable_tracing=True)
815+
sentry_init(integrations=[CeleryIntegration()], traces_sample_rate=1.0)
816816
celery = Celery(__name__, broker="redis://example.com") # noqa: E231
817817

818818
events = capture_events()

0 commit comments

Comments
 (0)