test_helpers.py

from ddtrace import helpers

from nose.tools import eq_, ok_

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()

        eq_(trace_id, active_trace_id)
        eq_(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()

        ok_(trace_id is None)
        ok_(span_id is None)

    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()

        ok_(trace_id is None)
        ok_(span_id is None)