mirror of
https://github.com/open-telemetry/opentelemetry-python-contrib.git
synced 2025-07-30 05:32:30 +08:00
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
import mock
|
|
|
|
from ddtrace import helpers
|
|
|
|
from .base import BaseTracerTestCase
|
|
from .util import override_global_tracer
|
|
|
|
|
|
class HelpersTestCase(BaseTracerTestCase):
|
|
"""Test suite for ``ddtrace`` helpers"""
|
|
def test_correlation_identifiers(self):
|
|
# ensures the right correlation identifiers are
|
|
# returned when a Trace is active
|
|
with override_global_tracer(self.tracer):
|
|
span = self.tracer.trace('MockSpan')
|
|
active_trace_id, active_span_id = span.trace_id, span.span_id
|
|
trace_id, span_id = helpers.get_correlation_ids()
|
|
|
|
self.assertEqual(trace_id, active_trace_id)
|
|
self.assertEqual(span_id, active_span_id)
|
|
|
|
def test_correlation_identifiers_without_trace(self):
|
|
# ensures `None` is returned if no Traces are active
|
|
with override_global_tracer(self.tracer):
|
|
trace_id, span_id = helpers.get_correlation_ids()
|
|
|
|
self.assertIsNone(trace_id)
|
|
self.assertIsNone(span_id)
|
|
|
|
def test_correlation_identifiers_with_disabled_trace(self):
|
|
# ensures `None` is returned if tracer is disabled
|
|
with override_global_tracer(self.tracer):
|
|
self.tracer.enabled = False
|
|
self.tracer.trace('MockSpan')
|
|
trace_id, span_id = helpers.get_correlation_ids()
|
|
|
|
self.assertIsNone(trace_id)
|
|
self.assertIsNone(span_id)
|
|
|
|
def test_correlation_identifiers_missing_context(self):
|
|
# ensures we return `None` if there is no current context
|
|
self.tracer.get_call_context = mock.MagicMock(return_value=None)
|
|
|
|
with override_global_tracer(self.tracer):
|
|
trace_id, span_id = helpers.get_correlation_ids()
|
|
|
|
self.assertIsNone(trace_id)
|
|
self.assertIsNone(span_id)
|