@@ -63,16 +63,14 @@ def dummy_task(x, y):
6363 foo = 42 # noqa
6464 return x / y
6565
66- span_context = Span .start_trace ()
67- with configure_scope () as scope :
68- scope .set_span_context (span_context )
66+ span = Hub .current .start_trace ()
6967
7068 invocation (dummy_task , 1 , 2 )
7169 invocation (dummy_task , 1 , 0 )
7270
7371 event , = events
74- assert event ["contexts" ]["trace" ]["trace_id" ] == span_context .trace_id
75- assert event ["contexts" ]["trace" ]["span_id" ] != span_context .span_id
72+ assert event ["contexts" ]["trace" ]["trace_id" ] == span .trace_id
73+ assert event ["contexts" ]["trace" ]["span_id" ] != span .span_id
7674 assert event ["transaction" ] == "dummy_task"
7775 assert event ["extra" ]["celery-job" ] == dict (
7876 task_name = "dummy_task" , ** expected_context
@@ -92,13 +90,11 @@ def test_simple_no_propagation(capture_events, init_celery):
9290 def dummy_task ():
9391 1 / 0
9492
95- span_context = Span .start_trace ()
96- with configure_scope () as scope :
97- scope .set_span_context (span_context )
93+ span = Hub .current .start_trace ()
9894 dummy_task .delay ()
9995
10096 event , = events
101- assert event ["contexts" ]["trace" ]["trace_id" ] != span_context .trace_id
97+ assert event ["contexts" ]["trace" ]["trace_id" ] != span .trace_id
10298 assert event ["transaction" ] == "dummy_task"
10399 exception , = event ["exception" ]["values" ]
104100 assert exception ["type" ] == "ZeroDivisionError"
0 commit comments