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 b571b41ab5
commit e27718a076
2 changed files with 9 additions and 9 deletions

View File

@ -188,7 +188,7 @@ class TestDatadogSpanExporter(unittest.TestCase):
instrumentation_info = InstrumentationInfo(__name__, "0")
otel_spans = [
trace.Span(
trace._Span(
name=span_names[0],
context=span_context,
parent=parent_context,
@ -196,14 +196,14 @@ class TestDatadogSpanExporter(unittest.TestCase):
instrumentation_info=instrumentation_info,
resource=Resource({}),
),
trace.Span(
trace._Span(
name=span_names[1],
context=parent_context,
parent=None,
instrumentation_info=instrumentation_info,
resource=resource_without_service,
),
trace.Span(
trace._Span(
name=span_names[2],
context=other_context,
parent=None,
@ -289,7 +289,7 @@ class TestDatadogSpanExporter(unittest.TestCase):
is_remote=False,
)
test_span = trace.Span("test_span", context=context)
test_span = trace._Span("test_span", context=context)
test_span.start()
test_span.end()
@ -492,8 +492,8 @@ class TestDatadogSpanExporter(unittest.TestCase):
),
)
root_span = trace.Span(name="root", context=context, parent=None)
child_span = trace.Span(
root_span = trace._Span(name="root", context=context, parent=None)
child_span = trace._Span(
name="child", context=context, parent=root_span
)
root_span.start()
@ -528,7 +528,7 @@ class TestDatadogSpanExporter(unittest.TestCase):
)
sampler = sampling.TraceIdRatioBased(0.5)
span = trace.Span(
span = trace._Span(
name="sampled", context=context, parent=None, sampler=sampler
)
span.start()

View File

@ -104,7 +104,7 @@ class TestDatadogFormat(unittest.TestCase):
)
self.assertTrue(parent_context.is_remote)
child = trace.Span(
child = trace._Span(
"child",
trace_api.SpanContext(
parent_context.trace_id,
@ -149,7 +149,7 @@ class TestDatadogFormat(unittest.TestCase):
self.assertEqual(parent_context.trace_flags, constants.AUTO_REJECT)
child = trace.Span(
child = trace._Span(
"child",
trace_api.SpanContext(
parent_context.trace_id,