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