class SimpleIdGeneratorTestCase(TestCase): def setUp(self): self.generator = SimpleIdGenerator() def test_trivial(self): span_id = self.generator.generate_span_id() trace_id = self.generator.generate_trace_id() self.assertIsInstance(span_id, ZipkinId) self.assertIsInstance(trace_id, ZipkinId) self.assertNotEqual(span_id.get_binary(), trace_id.get_binary())
def test_downstream_request_headers_without_parent_span_id(self): generator = SimpleIdGenerator() self.store.get.return_value = data = ZipkinData( trace_id=generator.generate_trace_id(), span_id=generator.generate_span_id(), sampled=True, flags=0 ) self.assertDictEqual(self.api.get_headers_for_downstream_request(), { 'X-B3-TraceId': data.trace_id.get_hex(), 'X-B3-SpanId': data.span_id.get_hex(), 'X-B3-Sampled': 'true', 'X-B3-Flags': '0' })
def test_downstream_request_headers_without_parent_span_id(self): generator = SimpleIdGenerator() self.store.get.return_value = data = ZipkinData( trace_id=generator.generate_trace_id(), span_id=generator.generate_span_id(), sampled=True, flags=0) self.assertDictEqual( self.api.get_headers_for_downstream_request(), { 'X-B3-TraceId': data.trace_id.get_hex(), 'X-B3-SpanId': data.span_id.get_hex(), 'X-B3-Sampled': 'true', 'X-B3-Flags': '0' })
def setUp(self): self.generator = SimpleIdGenerator()