open-telemetry
/
opentelemetry-python-contrib
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Protect access to Span implementation (#1188)

This commit is contained in:
Amos Law 2020-10-06 17:44:41 -04:00 committed by GitHub
parent d3d69aed0c
commit 50a1bd59e3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
instrumentation/opentelemetry-instrumentation-celery/tests/test_utils.py
View File

@ -42,7 +42,7 @@ class TestUtils(unittest.TestCase):
"routing_key": "celery", "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) utils.set_attributes_from_context(span, context)
self.assertEqual( self.assertEqual(
@ -78,7 +78,7 @@ class TestUtils(unittest.TestCase):
"retries": 0, "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) utils.set_attributes_from_context(span, context)
self.assertEqual(len(span.attributes), 0) self.assertEqual(len(span.attributes), 0)
@ -99,7 +99,7 @@ class TestUtils(unittest.TestCase):
# propagate and retrieve a Span # propagate and retrieve a Span
task_id = "7c6731af-9533-40c3-83a9-25b58f0d837f" 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) utils.attach_span(fn_task, task_id, span)
span_after = utils.retrieve_span(fn_task, task_id) span_after = utils.retrieve_span(fn_task, task_id)
self.assertIs(span, span_after) self.assertIs(span, span_after)
@ -112,7 +112,7 @@ class TestUtils(unittest.TestCase):
# propagate a Span # propagate a Span
task_id = "7c6731af-9533-40c3-83a9-25b58f0d837f" 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) utils.attach_span(fn_task, task_id, span)
# delete the Span # delete the Span
utils.detach_span(fn_task, task_id) utils.detach_span(fn_task, task_id)