def get_traceparent(self, headers): retval = [] for key, value in headers: if self.traceparent_name_re.match(key): retval.append((key, value)) self.assertEqual(len(retval), 1, 'expect one traceparent header, got {} {!r}'.format('more' if retval else 'zero', retval)) return Traceparent.from_string(retval[0][1])
def test_str(self): string = '00-12345678901234567890123456789012-1234567890123456-ff' traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-ff') self.assertEqual(str(traceparent), string)
def test_repr(self): string = '00-12345678901234567890123456789012-1234567890123456-ff' traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-ff') self.assertEqual(repr(traceparent), 'Traceparent({})'.format(repr(string)))
def test_from_string(self): traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-00') traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-01') traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-02') traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-03') traceparent = Traceparent.from_string( '00-12345678901234567890123456789012-1234567890123456-ff') self.assertRaises( ValueError, lambda: Traceparent.from_string( '01-12345678901234567890123456789012-1234567890123456-00')) self.assertRaises( ValueError, lambda: Traceparent.from_string( '02-12345678901234567890123456789012-1234567890123456-00')) self.assertRaises( ValueError, lambda: Traceparent.from_string( 'cc-12345678901234567890123456789012-1234567890123456-00')) self.assertRaises( ValueError, lambda: Traceparent.from_string( 'ff-12345678901234567890123456789012-1234567890123456-00')) self.assertRaises( ValueError, lambda: Traceparent.from_string( '00-00000000000000000000000000000000-1234567890123456-00')) self.assertRaises( ValueError, lambda: Traceparent.from_string( '00-12345678901234567890123456789012-0000000000000000-00'))