def test_from_carrier_keys_not_exist(self):
carrier = {}
propagator = text_format.TextFormatPropagator()
span_context = propagator.from_carrier(carrier)
self.assertIsNotNone(span_context.trace_id)
# Span_id should be None here which indicates no parent span_id for
# the child spans
self.assertIsNone(span_context.span_id)
self.assertTrue(span_context.enabled)
def test_to_carrier_no_span_id(self):
test_trace_id = '6e0c63257de34c92bf9efcd03927272e'
test_options = '1'
span_context = mock.Mock()
span_context.trace_id = test_trace_id
span_context.span_id = None
span_context.trace_options.trace_options_byte = test_options
carrier = {}
propagator = text_format.TextFormatPropagator()
carrier = propagator.to_carrier(span_context, carrier)
self.assertEqual(carrier[text_format._TRACE_ID_KEY], test_trace_id)
self.assertIsNone(carrier.get(text_format._SPAN_ID_KEY))
self.assertEqual(carrier[text_format._TRACE_OPTIONS_KEY], test_options)
def test_to_carrier_no_span_id(self):
test_trace_id = '6e0c63257de34c92bf9efcd03927272e'
test_options = 1
span_context = mock.Mock()
span_context.trace_id = test_trace_id
span_context.span_id = None
span_context.enabled = bool(test_options)
carrier = {}
propagator = text_format.TextFormatPropagator()
carrier = propagator.to_carrier(span_context, carrier)
self.assertEqual(carrier[text_format._TRACE_ID_KEY], test_trace_id)
self.assertIsNone(carrier.get(text_format._SPAN_ID_KEY))
self.assertEqual(carrier[text_format._ENABLED_TRACE_KEY],
str(bool(test_options)))
def test_to_carrier_has_span_id(self):
test_trace_id = '6e0c63257de34c92bf9efcd03927272e'
test_span_id = '00f067aa0ba902b7'
test_options = '2'
span_context = mock.Mock()
span_context.trace_id = test_trace_id
span_context.span_id = test_span_id
span_context.trace_options.trace_options_byte = test_options
carrier = {}
propagator = text_format.TextFormatPropagator()
carrier = propagator.to_carrier(span_context, carrier)
self.assertEqual(carrier[text_format._TRACE_ID_KEY], test_trace_id)
self.assertEqual(carrier[text_format._SPAN_ID_KEY], str(test_span_id))
self.assertEqual(carrier[text_format._TRACE_OPTIONS_KEY], test_options)
def test_from_carrier_keys_exist(self):
test_trace_id = '6e0c63257de34c92bf9efcd03927272e'
test_span_id = 1234
test_options = 1
carrier = {
text_format._TRACE_ID_KEY: test_trace_id,
text_format._SPAN_ID_KEY: test_span_id,
text_format._ENABLED_TRACE_KEY: test_options,
}
propagator = text_format.TextFormatPropagator()
span_context = propagator.from_carrier(carrier)
self.assertEqual(span_context.trace_id, test_trace_id)
self.assertEqual(span_context.span_id, test_span_id)
self.assertEqual(span_context.enabled, bool(test_options))
