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