55import unittest
66from instana .util .ids import header_to_long_id , header_to_id
77
8+
89class TestTraceparent (unittest .TestCase ):
910 def setUp (self ):
1011 self .tp = Traceparent ()
@@ -38,14 +39,18 @@ def test_validate_traceparent_None(self):
3839
3940 def test_get_traceparent_fields (self ):
4041 traceparent = f"00-{ self .w3cTraceId } -00f067aa0ba902b7-01"
41- version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (traceparent )
42+ version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (
43+ traceparent
44+ )
4245 self .assertEqual (trace_id , header_to_long_id (self .w3cTraceId ))
4346 self .assertEqual (parent_id , 67667974448284343 )
4447 self .assertTrue (sampled_flag )
4548
4649 def test_get_traceparent_fields_unsampled (self ):
4750 traceparent = f"00-{ self .w3cTraceId } -00f067aa0ba902b7-00"
48- version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (traceparent )
51+ version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (
52+ traceparent
53+ )
4954 self .assertEqual (trace_id , header_to_long_id (self .w3cTraceId ))
5055 self .assertEqual (parent_id , 67667974448284343 )
5156 self .assertFalse (sampled_flag )
@@ -54,28 +59,36 @@ def test_get_traceparent_fields_newer_version(self):
5459 # Although the incoming traceparent header sports a newer version number, we should still be able to parse the
5560 # parts that we understand (and consider it valid).
5661 traceparent = f"fe-{ self .w3cTraceId } -00f067aa0ba902b7-01-12345-abcd"
57- version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (traceparent )
62+ version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (
63+ traceparent
64+ )
5865 self .assertEqual (trace_id , header_to_long_id (self .w3cTraceId ))
5966 self .assertEqual (parent_id , 67667974448284343 )
6067 self .assertTrue (sampled_flag )
6168
6269 def test_get_traceparent_fields_unknown_flags (self ):
6370 traceparent = f"00-{ self .w3cTraceId } -00f067aa0ba902b7-ff"
64- version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (traceparent )
71+ version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (
72+ traceparent
73+ )
6574 self .assertEqual (trace_id , header_to_long_id (self .w3cTraceId ))
6675 self .assertEqual (parent_id , 67667974448284343 )
6776 self .assertTrue (sampled_flag )
6877
6978 def test_get_traceparent_fields_None_input (self ):
7079 traceparent = None
71- version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (traceparent )
80+ version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (
81+ traceparent
82+ )
7283 self .assertIsNone (trace_id )
7384 self .assertIsNone (parent_id )
7485 self .assertFalse (sampled_flag )
7586
7687 def test_get_traceparent_fields_string_input_no_dash (self ):
7788 traceparent = "invalid"
78- version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (traceparent )
89+ version , trace_id , parent_id , sampled_flag = self .tp .get_traceparent_fields (
90+ traceparent
91+ )
7992 self .assertIsNone (trace_id )
8093 self .assertIsNone (parent_id )
8194 self .assertFalse (sampled_flag )
@@ -86,12 +99,22 @@ def test_update_traceparent(self):
8699 in_span_id = "1234567890abcdef"
87100 level = 1
88101 expected_traceparent = "00-4bf92f3577b34da6a3ce929d0e0e4736-1234567890abcdef-01"
89- self .assertEqual (expected_traceparent , self .tp .update_traceparent (traceparent , in_trace_id , header_to_id (in_span_id ), level ))
102+ self .assertEqual (
103+ expected_traceparent ,
104+ self .tp .update_traceparent (
105+ traceparent , in_trace_id , header_to_id (in_span_id ), level
106+ ),
107+ )
90108
91109 def test_update_traceparent_None (self ):
92110 traceparent = None
93111 in_trace_id = "1234d0e0e4736234"
94112 in_span_id = "7890abcdef"
95113 level = 0
96114 expected_traceparent = "00-00000000000000001234d0e0e4736234-0000007890abcdef-00"
97- self .assertEqual (expected_traceparent , self .tp .update_traceparent (traceparent , in_trace_id , header_to_id (in_span_id ), level ))
115+ self .assertEqual (
116+ expected_traceparent ,
117+ self .tp .update_traceparent (
118+ traceparent , in_trace_id , header_to_id (in_span_id ), level
119+ ),
120+ )
0 commit comments