def test_extract_multi_header(self):
"""Test the extraction of B3 headers."""
child, parent, new_carrier = get_child_parent_new_carrier({
FORMAT.TRACE_ID_KEY:
self.serialized_trace_id,
FORMAT.SPAN_ID_KEY:
self.serialized_span_id,
FORMAT.PARENT_SPAN_ID_KEY:
self.serialized_parent_id,
FORMAT.SAMPLED_KEY:
"1",
})
self.assertEqual(
new_carrier[FORMAT.TRACE_ID_KEY],
b3_format.format_trace_id(child.context.trace_id),
)
self.assertEqual(
new_carrier[FORMAT.SPAN_ID_KEY],
b3_format.format_span_id(child.context.span_id),
)
self.assertEqual(
new_carrier[FORMAT.PARENT_SPAN_ID_KEY],
b3_format.format_span_id(parent.context.span_id),
)
self.assertTrue(parent.context.is_remote)
self.assertEqual(new_carrier[FORMAT.SAMPLED_KEY], "1")
def setUpClass(cls):
id_generator = ids_generator.RandomIdsGenerator()
cls.serialized_trace_id = b3_format.format_trace_id(
id_generator.generate_trace_id())
cls.serialized_span_id = b3_format.format_span_id(
id_generator.generate_span_id())
cls.serialized_parent_id = b3_format.format_span_id(
id_generator.generate_span_id())
def test_extract_single_header(self):
"""Test the extraction from a single b3 header."""
child, parent, new_carrier = get_child_parent_new_carrier(
{
FORMAT.SINGLE_HEADER_KEY: "{}-{}".format(
self.serialized_trace_id, self.serialized_span_id
)
}
)
self.assertEqual(
new_carrier[FORMAT.TRACE_ID_KEY],
b3_format.format_trace_id(child.context.trace_id),
)
self.assertEqual(
new_carrier[FORMAT.SPAN_ID_KEY],
b3_format.format_span_id(child.context.span_id),
)
self.assertEqual(new_carrier[FORMAT.SAMPLED_KEY], "1")
self.assertTrue(parent.context.is_remote)
child, parent, new_carrier = get_child_parent_new_carrier(
{
FORMAT.SINGLE_HEADER_KEY: "{}-{}-1-{}".format(
self.serialized_trace_id,
self.serialized_span_id,
self.serialized_parent_id,
)
}
)
self.assertEqual(
new_carrier[FORMAT.TRACE_ID_KEY],
b3_format.format_trace_id(child.context.trace_id),
)
self.assertEqual(
new_carrier[FORMAT.SPAN_ID_KEY],
b3_format.format_span_id(child.context.span_id),
)
self.assertEqual(
new_carrier[FORMAT.PARENT_SPAN_ID_KEY],
b3_format.format_span_id(parent.context.span_id),
)
self.assertTrue(parent.context.is_remote)
self.assertEqual(new_carrier[FORMAT.SAMPLED_KEY], "1")
