From ed68634b9d9523064327d55d8343fa45e95e08b6 Mon Sep 17 00:00:00 2001 From: Amos Law Date: Tue, 6 Oct 2020 17:44:41 -0400 Subject: [PATCH] Protect access to Span implementation (#1188) --- .../tests/test_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/instrumentation/opentelemetry-instrumentation-celery/tests/test_utils.py b/instrumentation/opentelemetry-instrumentation-celery/tests/test_utils.py index 0842890e8..3df0f445b 100644 --- a/instrumentation/opentelemetry-instrumentation-celery/tests/test_utils.py +++ b/instrumentation/opentelemetry-instrumentation-celery/tests/test_utils.py @@ -42,7 +42,7 @@ class TestUtils(unittest.TestCase): "routing_key": "celery", } - span = trace.Span("name", mock.Mock(spec=trace_api.SpanContext)) + span = trace._Span("name", mock.Mock(spec=trace_api.SpanContext)) utils.set_attributes_from_context(span, context) self.assertEqual( @@ -78,7 +78,7 @@ class TestUtils(unittest.TestCase): "retries": 0, } - span = trace.Span("name", mock.Mock(spec=trace_api.SpanContext)) + span = trace._Span("name", mock.Mock(spec=trace_api.SpanContext)) utils.set_attributes_from_context(span, context) self.assertEqual(len(span.attributes), 0) @@ -99,7 +99,7 @@ class TestUtils(unittest.TestCase): # propagate and retrieve a Span task_id = "7c6731af-9533-40c3-83a9-25b58f0d837f" - span = trace.Span("name", mock.Mock(spec=trace_api.SpanContext)) + span = trace._Span("name", mock.Mock(spec=trace_api.SpanContext)) utils.attach_span(fn_task, task_id, span) span_after = utils.retrieve_span(fn_task, task_id) self.assertIs(span, span_after) @@ -112,7 +112,7 @@ class TestUtils(unittest.TestCase): # propagate a Span task_id = "7c6731af-9533-40c3-83a9-25b58f0d837f" - span = trace.Span("name", mock.Mock(spec=trace_api.SpanContext)) + span = trace._Span("name", mock.Mock(spec=trace_api.SpanContext)) utils.attach_span(fn_task, task_id, span) # delete the Span utils.detach_span(fn_task, task_id)