def test_translate_to_jaeger(self): self.maxDiff = None trace_id_high = '6e0c63257de34c92' trace_id_low = 'bf9efcd03927272e' trace_id = trace_id_high + trace_id_low span_id = '6e0c63257de34c92' parent_span_id = '1111111111111111' span_attributes = { 'key_bool': False, 'key_string': 'hello_world', 'key_int': 3, 'key_double': 111.22, 'key_unsupported_type': () } annotation_attributes = { 'annotation_bool': True, 'annotation_string': 'annotation_test', 'key_float': .3, 'key_unsupported_type': {} } link_attributes = {'key_bool': True} import datetime s = '2017-08-15T18:02:26.071158' time = datetime.datetime.strptime(s, '%Y-%m-%dT%H:%M:%S.%f') time_events = [ time_event.TimeEvent( timestamp=time, annotation=time_event.Annotation( description='First Annotation', attributes=attributes.Attributes(annotation_attributes))), time_event.TimeEvent(timestamp=time, message_event=time_event.MessageEvent( id='message-event-id', uncompressed_size_bytes=0, )), ] time_events2 = [ time_event.TimeEvent(timestamp=time, annotation=time_event.Annotation( description='First Annotation', attributes=None)), time_event.TimeEvent(timestamp=time, message_event=time_event.MessageEvent( id='message-event-id', uncompressed_size_bytes=0, )), ] links = [ link.Link(trace_id=trace_id, span_id=span_id, type=link.Type.CHILD_LINKED_SPAN, attributes=link_attributes), link.Link(trace_id=trace_id, span_id=span_id, type=link.Type.PARENT_LINKED_SPAN, attributes=link_attributes), link.Link(trace_id=trace_id, span_id=span_id, type=link.Type.TYPE_UNSPECIFIED, attributes=link_attributes) ] span_status = status.Status(code=200, message='success') start_time = '2017-08-15T18:02:26.071158Z' end_time = '2017-08-15T18:02:36.071158Z' span_datas = [ span_data.SpanData( name='test1', context=span_context.SpanContext(trace_id=trace_id), span_id=span_id, parent_span_id=parent_span_id, attributes=span_attributes, start_time=start_time, end_time=end_time, child_span_count=0, stack_trace=None, time_events=time_events, links=links, status=span_status, same_process_as_parent_span=None, span_kind=0, ), span_data.SpanData( name='test2', context=None, span_id=span_id, parent_span_id=None, attributes=None, start_time=start_time, end_time=end_time, child_span_count=None, stack_trace=None, time_events=time_events2, links=None, status=None, same_process_as_parent_span=None, span_kind=None, ), span_data.SpanData( name='test3', context=None, span_id=span_id, parent_span_id=None, attributes=None, start_time=start_time, end_time=end_time, child_span_count=None, stack_trace=None, time_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=None, ) ] exporter = jaeger_exporter.JaegerExporter() spans = exporter.translate_to_jaeger(span_datas) expected_spans = [ jaeger.Span( traceIdHigh=7929822056569588882, traceIdLow=-4638992594902767826, spanId=7929822056569588882, parentSpanId=1229782938247303441, operationName='test1', startTime=1502820146071158, duration=10000000, flags=1, tags=[ jaeger.Tag(key='key_bool', vType=jaeger.TagType.BOOL, vBool=False), jaeger.Tag(key='key_string', vType=jaeger.TagType.STRING, vStr='hello_world'), jaeger.Tag(key='key_int', vType=jaeger.TagType.LONG, vLong=3), jaeger.Tag(key='key_double', vType=jaeger.TagType.DOUBLE, vDouble=111.22), jaeger.Tag(key='status.code', vType=jaeger.TagType.LONG, vLong=200), jaeger.Tag(key='status.message', vType=jaeger.TagType.STRING, vStr='success') ], references=[ jaeger.SpanRef(refType=jaeger.SpanRefType.CHILD_OF, traceIdHigh=7929822056569588882, traceIdLow=-4638992594902767826, spanId=7929822056569588882), jaeger.SpanRef(refType=jaeger.SpanRefType.FOLLOWS_FROM, traceIdHigh=7929822056569588882, traceIdLow=-4638992594902767826, spanId=7929822056569588882), jaeger.SpanRef(refType=None, traceIdHigh=7929822056569588882, traceIdLow=-4638992594902767826, spanId=7929822056569588882) ], logs=[ jaeger.Log(timestamp=1502820146071158, fields=[ jaeger.Tag(key='annotation_bool', vType=jaeger.TagType.BOOL, vBool=True), jaeger.Tag(key='annotation_string', vType=jaeger.TagType.STRING, vStr='annotation_test'), jaeger.Tag(key='key_float', vType=jaeger.TagType.DOUBLE, vDouble=0.3), jaeger.Tag(key='message', vType=jaeger.TagType.STRING, vStr='First Annotation') ]) ]), jaeger.Span(operationName="test2", traceIdHigh=7929822056569588882, traceIdLow=-4638992594902767826, spanId=7929822056569588882, parentSpanId=0, startTime=1502820146071158, duration=10000000, logs=[ jaeger.Log(timestamp=1502820146071158, fields=[ jaeger.Tag( key='message', vType=jaeger.TagType.STRING, vStr='First Annotation') ]) ]), jaeger.Span(operationName="test3", traceIdHigh=7929822056569588882, traceIdLow=-4638992594902767826, spanId=7929822056569588882, parentSpanId=0, startTime=1502820146071158, duration=10000000, logs=[]) ] spans_json = [span.format_span_json() for span in spans] expected_spans_json = [ span.format_span_json() for span in expected_spans ] span = spans_json[0] expected_span = expected_spans_json[0] try: listsEqual = self.assertCountEqual except AttributeError: listsEqual = self.assertItemsEqual log = span.get('logs')[0] expected_log = expected_span.get('logs')[0] self.assertEqual(log.get('timestamp'), expected_log.get('timestamp')) listsEqual(log.get('fields'), expected_log.get('fields')) listsEqual(span.get('tags'), expected_span.get('tags')) listsEqual(span.get('references'), expected_span.get('references')) self.assertEqual(span.get('traceIdHigh'), expected_span.get('traceIdHigh')) self.assertEqual(span.get('traceIdLow'), expected_span.get('traceIdLow')) self.assertEqual(span.get('spanId'), expected_span.get('spanId')) self.assertEqual(span.get('parentSpanId'), expected_span.get('parentSpanId')) self.assertEqual(span.get('operationName'), expected_span.get('operationName')) self.assertEqual(span.get('startTime'), expected_span.get('startTime')) self.assertEqual(span.get('duration'), expected_span.get('duration')) self.assertEqual(span.get('flags'), expected_span.get('flags')) self.assertEqual(spans_json[1], expected_spans_json[1]) self.assertEqual(spans_json[2], expected_spans_json[2])
def test_translate_links(self): hex_encoder = codecs.getencoder('hex') span_data = span_data_module.SpanData( context=span_context_module.SpanContext( trace_id='6e0c63257de34c92bf9efcd03927272e'), span_id='6e0c63257de34c92', links=[ link_module.Link(trace_id='0e0c63257de34c92bf9efcd03927272e', span_id='0e0c63257de34c92', type=link_module.Type.TYPE_UNSPECIFIED, attributes=attributes_module.Attributes( attributes={ 'test_str_key': 'test_str_value', 'test_int_key': 1, 'test_bool_key': False })), link_module.Link(trace_id='1e0c63257de34c92bf9efcd03927272e', span_id='1e0c63257de34c92', type=link_module.Type.CHILD_LINKED_SPAN), link_module.Link(trace_id='2e0c63257de34c92bf9efcd03927272e', span_id='2e0c63257de34c92', type=link_module.Type.PARENT_LINKED_SPAN) ], span_kind=span_module.SpanKind.SERVER, status=None, start_time=None, end_time=None, child_span_count=None, name=None, parent_span_id=None, attributes=None, same_process_as_parent_span=False, stack_trace=None, time_events=None) pb_span = utils.translate_to_trace_proto(span_data) self.assertEqual(len(pb_span.links.link), 3) self.assertEqual( hex_encoder(pb_span.links.link[0].trace_id)[0], b'0e0c63257de34c92bf9efcd03927272e') self.assertEqual( hex_encoder(pb_span.links.link[1].trace_id)[0], b'1e0c63257de34c92bf9efcd03927272e') self.assertEqual( hex_encoder(pb_span.links.link[2].trace_id)[0], b'2e0c63257de34c92bf9efcd03927272e') self.assertEqual( hex_encoder(pb_span.links.link[0].span_id)[0], b'0e0c63257de34c92') self.assertEqual( hex_encoder(pb_span.links.link[1].span_id)[0], b'1e0c63257de34c92') self.assertEqual( hex_encoder(pb_span.links.link[2].span_id)[0], b'2e0c63257de34c92') self.assertEqual(pb_span.links.link[0].type, 0) self.assertEqual(pb_span.links.link[1].type, 1) self.assertEqual(pb_span.links.link[2].type, 2) self.assertEqual(len(pb_span.links.link[0].attributes.attribute_map), 3) self.assertEqual(len(pb_span.links.link[1].attributes.attribute_map), 0) self.assertEqual(len(pb_span.links.link[2].attributes.attribute_map), 0) self.assertEqual( pb_span.links.link[0].attributes.attribute_map['test_str_key'], trace_pb2.AttributeValue(string_value=trace_pb2.TruncatableString( value='test_str_value'))) self.assertEqual( pb_span.links.link[0].attributes.attribute_map['test_int_key'], trace_pb2.AttributeValue(int_value=1)) self.assertEqual( pb_span.links.link[0].attributes.attribute_map['test_bool_key'], trace_pb2.AttributeValue(bool_value=False))
def test_basic_span_translation(self): hex_encoder = codecs.getencoder('hex') span_data = span_data_module.SpanData( name="name", context=span_context_module.SpanContext( trace_id='6e0c63257de34c92bf9efcd03927272e'), span_id='6e0c63257de34c92', parent_span_id='6e0c63257de34c93', attributes={ 'test_str_key': 'test_str_value', 'test_int_key': 1, 'test_bool_key': False, 'test_double_key': 567.89 }, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0) pb_span = utils.translate_to_trace_proto(span_data) self.assertEqual(pb_span.name.value, "name") self.assertEqual( hex_encoder(pb_span.trace_id)[0], b'6e0c63257de34c92bf9efcd03927272e') self.assertEqual(hex_encoder(pb_span.span_id)[0], b'6e0c63257de34c92') self.assertEqual( hex_encoder(pb_span.parent_span_id)[0], b'6e0c63257de34c93') self.assertEqual(pb_span.kind, 0) self.assertEqual(len(pb_span.attributes.attribute_map), 4) self.assertEqual( pb_span.attributes.attribute_map['test_str_key'], trace_pb2.AttributeValue(string_value=trace_pb2.TruncatableString( value='test_str_value'))) self.assertEqual(pb_span.attributes.attribute_map['test_int_key'], trace_pb2.AttributeValue(int_value=1)) self.assertEqual(pb_span.attributes.attribute_map['test_bool_key'], trace_pb2.AttributeValue(bool_value=False)) self.assertEqual(pb_span.attributes.attribute_map['test_double_key'], trace_pb2.AttributeValue(double_value=567.89)) self.assertEqual(pb_span.start_time.ToJsonString(), '2017-08-15T18:02:26.071158Z') self.assertEqual(pb_span.end_time.ToJsonString(), '2017-08-15T18:02:36.071158Z') self.assertEqual(pb_span.child_span_count.value, 0) self.assertEqual(pb_span.same_process_as_parent_span.value, False) self.assertEqual(len(pb_span.time_events.time_event), 0) self.assertEqual(pb_span.status.code, 0) self.assertFalse(pb_span.status.message) self.assertEqual(len(pb_span.links.link), 0) self.assertEqual(len(pb_span.tracestate.entries), 0)
def test_emit(self): trace_id = '6e0c63257de34c92bf9efcd03927272e' span_datas = [ span_data_module.SpanData( name='span', context=span_context.SpanContext(trace_id=trace_id), span_id='1111', parent_span_id=None, attributes=None, start_time=None, end_time=None, child_span_count=None, stack_trace=None, time_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0, ) ] stackdriver_spans = { 'spans': [{ 'status': None, 'childSpanCount': None, 'links': None, 'startTime': None, 'spanId': '1111', 'attributes': { 'attributeMap': { 'g.co/agent': { 'string_value': { 'truncated_byte_count': 0, 'value': 'opencensus-python [{}]'.format( stackdriver_exporter.VERSION) } } } }, 'stackTrace': None, 'displayName': { 'truncated_byte_count': 0, 'value': 'span' }, 'name': 'projects/PROJECT/traces/{}/spans/1111'.format(trace_id), 'timeEvents': None, 'endTime': None, 'sameProcessAsParentSpan': None }] } client = mock.Mock() project_id = 'PROJECT' client.project = project_id exporter = stackdriver_exporter.StackdriverExporter( client=client, project_id=project_id) exporter.emit(span_datas) name = 'projects/{}'.format(project_id) client.batch_write_spans.assert_called_with(name, stackdriver_spans) self.assertTrue(client.batch_write_spans.called)
def test_translate_time_events(self): annotation0_ts = datetime.utcnow() + timedelta(seconds=-10) annotation1_ts = datetime.utcnow() + timedelta(seconds=-9) message0_ts = datetime.utcnow() + timedelta(seconds=-8) message1_ts = datetime.utcnow() + timedelta(seconds=-7) message2_ts = datetime.utcnow() + timedelta(seconds=-6) span_data = span_data_module.SpanData( context=span_context_module.SpanContext( trace_id='6e0c63257de34c92bf9efcd03927272e'), span_id='6e0c63257de34c92', time_events=[ time_event_module.TimeEvent( timestamp=annotation0_ts, annotation=time_event_module.Annotation( description="hi there0", attributes=attributes_module.Attributes( attributes={ 'test_str_key': 'test_str_value', 'test_int_key': 1, 'test_bool_key': False, 'test_double_key': 567.89 }))), time_event_module.TimeEvent( timestamp=annotation1_ts, annotation=time_event_module.Annotation( description="hi there1")), time_event_module.TimeEvent( timestamp=message0_ts, message_event=time_event_module.MessageEvent( id=0, type=time_event_module.Type.SENT, uncompressed_size_bytes=10, compressed_size_bytes=1)), time_event_module.TimeEvent( timestamp=message1_ts, message_event=time_event_module.MessageEvent( id=1, type=time_event_module.Type.RECEIVED, uncompressed_size_bytes=20, compressed_size_bytes=2)), time_event_module.TimeEvent( timestamp=message2_ts, message_event=time_event_module.MessageEvent( id=2, type=time_event_module.Type.TYPE_UNSPECIFIED, uncompressed_size_bytes=30, compressed_size_bytes=3)) ], span_kind=span_module.SpanKind.SERVER, status=None, start_time=None, end_time=None, child_span_count=None, name=None, parent_span_id=None, attributes=None, same_process_as_parent_span=False, stack_trace=None, links=None) pb_span = utils.translate_to_trace_proto(span_data) self.assertEqual(len(pb_span.time_events.time_event), 5) event0 = pb_span.time_events.time_event[0] event1 = pb_span.time_events.time_event[1] event2 = pb_span.time_events.time_event[2] event3 = pb_span.time_events.time_event[3] event4 = pb_span.time_events.time_event[4] self.assertEqual(event0.time.ToDatetime(), annotation0_ts) self.assertEqual(event1.time.ToDatetime(), annotation1_ts) self.assertEqual(event2.time.ToDatetime(), message0_ts) self.assertEqual(event3.time.ToDatetime(), message1_ts) self.assertEqual(event4.time.ToDatetime(), message2_ts) self.assertEqual(event0.annotation.description.value, "hi there0") self.assertEqual(event1.annotation.description.value, "hi there1") self.assertEqual(len(event0.annotation.attributes.attribute_map), 4) self.assertEqual(len(event1.annotation.attributes.attribute_map), 0) self.assertEqual( event0.annotation.attributes.attribute_map['test_str_key'], trace_pb2.AttributeValue(string_value=trace_pb2.TruncatableString( value='test_str_value'))) self.assertEqual( event0.annotation.attributes.attribute_map['test_int_key'], trace_pb2.AttributeValue(int_value=1)) self.assertEqual( event0.annotation.attributes.attribute_map['test_bool_key'], trace_pb2.AttributeValue(bool_value=False)) self.assertEqual( event0.annotation.attributes.attribute_map['test_double_key'], trace_pb2.AttributeValue(double_value=567.89)) self.assertEqual(event2.message_event.id, 0) self.assertEqual(event3.message_event.id, 1) self.assertEqual(event4.message_event.id, 2) self.assertEqual(event2.message_event.uncompressed_size, 10) self.assertEqual(event3.message_event.uncompressed_size, 20) self.assertEqual(event4.message_event.uncompressed_size, 30) self.assertEqual(event2.message_event.compressed_size, 1) self.assertEqual(event3.message_event.compressed_size, 2) self.assertEqual(event4.message_event.compressed_size, 3) self.assertEqual(event2.message_event.type, 1) self.assertEqual(event3.message_event.type, 2) self.assertEqual(event4.message_event.type, 0)
def test_translate_to_zipkin_span_kind_none(self): trace_id = '6e0c63257de34c92bf9efcd03927272e' spans_ipv4 = [ span_data_module.SpanData( name='child_span', context=span_context.SpanContext(trace_id=trace_id), span_id='6e0c63257de34c92', parent_span_id='6e0c63257de34c93', attributes={'test_key': 'test_value'}, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0, ), span_data_module.SpanData( name='child_span', context=span_context.SpanContext(trace_id=trace_id), span_id='6e0c63257de34c92', parent_span_id='6e0c63257de34c93', attributes={'test_key': 1}, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=None, ), ] trace_id = '6e0c63257de34c92bf9efcd03927272e' spans_ipv6 = [ span_data_module.SpanData( name='child_span', context=span_context.SpanContext(trace_id=trace_id), span_id='6e0c63257de34c92', parent_span_id=None, attributes={ 'test_key': False, 'test_key2': 'raw_value', # these tags are malformed and should be omitted 'test_key3': 0.1, }, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=1, ), ] ipv4 = '127.0.0.1' ipv6 = '2001:0db8:85a3:0000:0000:8a2e:0370:7334' local_endpoint_ipv4 = { 'serviceName': 'my_service', 'ipv4': ipv4, 'port': 9411, } local_endpoint_ipv6 = { 'serviceName': 'my_service', 'ipv6': ipv6, 'port': 9411, } expected_zipkin_spans_ipv4 = [ { 'traceId': '6e0c63257de34c92bf9efcd03927272e', 'id': '6e0c63257de34c92', 'parentId': '6e0c63257de34c93', 'name': 'child_span', 'timestamp': 1502820146000000, 'duration': 10000000, 'localEndpoint': local_endpoint_ipv4, 'tags': { 'test_key': 'test_value' }, }, { 'traceId': '6e0c63257de34c92bf9efcd03927272e', 'id': '6e0c63257de34c92', 'parentId': '6e0c63257de34c93', 'name': 'child_span', 'timestamp': 1502820146000000, 'duration': 10000000, 'localEndpoint': local_endpoint_ipv4, 'tags': { 'test_key': '1' }, }, ] expected_zipkin_spans_ipv6 = [ { 'traceId': '6e0c63257de34c92bf9efcd03927272e', 'id': '6e0c63257de34c92', 'name': 'child_span', 'timestamp': 1502820146000000, 'duration': 10000000, 'localEndpoint': local_endpoint_ipv6, 'tags': { 'test_key': 'False', 'test_key2': 'raw_value' }, 'kind': 'SERVER', }, ] # Test ipv4 local endpoint exporter_ipv4 = zipkin_exporter.ZipkinExporter( service_name='my_service', ipv4=ipv4) ipv4_trace = span_data_module.format_legacy_trace_json(spans_ipv4) zipkin_spans_ipv4 = exporter_ipv4.translate_to_zipkin( trace_id=trace_id, spans=ipv4_trace.get('spans')) self.assertEqual(zipkin_spans_ipv4, expected_zipkin_spans_ipv4) # Test ipv6 local endpoint exporter_ipv6 = zipkin_exporter.ZipkinExporter( service_name='my_service', ipv6=ipv6) ipv6_trace = span_data_module.format_legacy_trace_json(spans_ipv6) zipkin_spans_ipv6 = exporter_ipv6.translate_to_zipkin( trace_id=trace_id, spans=ipv6_trace.get('spans')) self.assertEqual(zipkin_spans_ipv6, expected_zipkin_spans_ipv6)
def test_basic_spans_emit(self): hex_encoder = codecs.getencoder('hex') client = mock.Mock() client.Export.return_value = iter([1]) span_data0 = span_data_module.SpanData( name="name0", context=span_context_module.SpanContext( trace_id='0e0c63257de34c92bf9efcd03927272e'), span_id='0e0c63257de34c92', parent_span_id=None, start_time=None, end_time=None, attributes=None, child_span_count=None, stack_trace=None, annotations=None, message_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0) span_data1 = span_data_module.SpanData( name="name1", context=span_context_module.SpanContext( trace_id='1e0c63257de34c92bf9efcd03927272e'), span_id='1e0c63257de34c92', parent_span_id=None, start_time=None, end_time=None, attributes=None, child_span_count=None, stack_trace=None, annotations=None, message_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0) exporter = TraceExporter(service_name=SERVICE_NAME, client=client, transport=MockTransport) exporter.emit([span_data0]) actual_request0 = list(client.Export.call_args[0][0])[0] self.assertEqual(actual_request0.node, exporter.node) pb_span0 = actual_request0.spans[0] self.assertEqual(pb_span0.name.value, "name0") self.assertEqual( hex_encoder(pb_span0.trace_id)[0], b'0e0c63257de34c92bf9efcd03927272e') self.assertEqual(hex_encoder(pb_span0.span_id)[0], b'0e0c63257de34c92') exporter.emit([span_data1]) self.assertEqual(len(client.Export.mock_calls), 2) actual_request1 = list(client.Export.call_args[0][0])[0] self.assertEqual(actual_request1.node, exporter.node) pb_span1 = actual_request1.spans[0] self.assertEqual(pb_span1.name.value, "name1") self.assertEqual( hex_encoder(pb_span1.trace_id)[0], b'1e0c63257de34c92bf9efcd03927272e') self.assertEqual(hex_encoder(pb_span1.span_id)[0], b'1e0c63257de34c92')
def test_span_generator(self): hex_encoder = codecs.getencoder('hex') span_datas = [ span_data_module.SpanData( name="name0", context=span_context_module.SpanContext( trace_id='0e0c63257de34c92bf9efcd03927272e'), span_id='0e0c63257de34c92', parent_span_id=None, attributes=None, start_time=None, end_time=None, child_span_count=None, stack_trace=None, annotations=None, message_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0), span_data_module.SpanData( name="name1", context=span_context_module.SpanContext( trace_id='1e0c63257de34c92bf9efcd03927272e'), span_id='1e0c63257de34c92', parent_span_id=None, attributes=None, start_time=None, end_time=None, child_span_count=None, stack_trace=None, annotations=None, message_events=None, links=None, status=None, same_process_as_parent_span=None, span_kind=0) ] exporter = TraceExporter(service_name=SERVICE_NAME, transport=MockTransport) export_requests = list(exporter.generate_span_requests(span_datas)) self.assertEqual(len(export_requests), 1) self.assertEqual(export_requests[0].node, exporter.node) self.assertEqual(len(export_requests[0].spans), 2) self.assertEqual(export_requests[0].spans[0].name.value, 'name0') self.assertEqual(export_requests[0].spans[1].name.value, 'name1') self.assertEqual( hex_encoder(export_requests[0].spans[0].trace_id)[0], b'0e0c63257de34c92bf9efcd03927272e') self.assertEqual( hex_encoder(export_requests[0].spans[0].span_id)[0], b'0e0c63257de34c92') self.assertEqual( hex_encoder(export_requests[0].spans[1].trace_id)[0], b'1e0c63257de34c92bf9efcd03927272e') self.assertEqual( hex_encoder(export_requests[0].spans[1].span_id)[0], b'1e0c63257de34c92')
def test_translate_to_zipkin_with_annotations(self): trace_id = '6e0c63257de34c92bf9efcd03927272e' annotation_attributes = { 'annotation_bool': True, 'annotation_string': 'annotation_test', 'key_float': .3 } s = '2017-08-15T18:02:26.071158' time = datetime.strptime(s, '%Y-%m-%dT%H:%M:%S.%f') time_events = [ time_event.TimeEvent(timestamp=time, annotation=time_event.Annotation( description='First Annotation', attributes=annotation_attributes)), time_event.TimeEvent(timestamp=time, message_event=time_event.MessageEvent( id='message-event-id', uncompressed_size_bytes=0, )), ] spans_ipv4 = [ span_data_module.SpanData( name='child_span', context=span_context.SpanContext(trace_id=trace_id), span_id='6e0c63257de34c92', parent_span_id='6e0c63257de34c93', attributes={'test_key': 'test_value'}, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=time_events, links=None, status=None, same_process_as_parent_span=None, span_kind=0, ), span_data_module.SpanData( name='child_span', context=span_context.SpanContext(trace_id=trace_id), span_id='6e0c63257de34c92', parent_span_id='6e0c63257de34c93', attributes={'test_key': 1}, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=time_events, links=None, status=None, same_process_as_parent_span=None, span_kind=None, ), ] spans_ipv6 = [ span_data_module.SpanData( name='child_span', context=span_context.SpanContext(trace_id=trace_id), span_id='6e0c63257de34c92', parent_span_id=None, attributes={ 'test_key': False, 'test_key2': 'raw_value', 'test_key3': 0.1, }, start_time='2017-08-15T18:02:26.071158Z', end_time='2017-08-15T18:02:36.071158Z', child_span_count=None, stack_trace=None, time_events=time_events, links=None, status=None, same_process_as_parent_span=None, span_kind=1, ), ] ipv4 = '127.0.0.1' ipv6 = '2001:0db8:85a3:0000:0000:8a2e:0370:7334' local_endpoint_ipv4 = { 'serviceName': 'my_service', 'ipv4': ipv4, 'port': 9411, } local_endpoint_ipv6 = { 'serviceName': 'my_service', 'ipv6': ipv6, 'port': 9411, } expected_zipkin_spans_ipv4 = [ { 'traceId': '6e0c63257de34c92bf9efcd03927272e', 'id': '6e0c63257de34c92', 'parentId': '6e0c63257de34c93', 'name': 'child_span', 'timestamp': 1502820146071158, 'duration': 10000000, 'localEndpoint': local_endpoint_ipv4, 'tags': { 'test_key': 'test_value' }, 'annotations': [{ 'timestamp': 1502820146071158, 'value': 'First Annotation' }] }, { 'traceId': '6e0c63257de34c92bf9efcd03927272e', 'id': '6e0c63257de34c92', 'parentId': '6e0c63257de34c93', 'name': 'child_span', 'timestamp': 1502820146071158, 'duration': 10000000, 'localEndpoint': local_endpoint_ipv4, 'tags': { 'test_key': '1' }, 'annotations': [{ 'timestamp': 1502820146071158, 'value': 'First Annotation' }] }, ] expected_zipkin_spans_ipv6 = [ { 'traceId': '6e0c63257de34c92bf9efcd03927272e', 'id': '6e0c63257de34c92', 'name': 'child_span', 'timestamp': 1502820146071158, 'duration': 10000000, 'localEndpoint': local_endpoint_ipv6, 'tags': { 'test_key': 'False', 'test_key2': 'raw_value', 'test_key3': '0.1' }, 'kind': 'SERVER', 'annotations': [{ 'timestamp': 1502820146071158, 'value': 'First Annotation' }] }, ] # Test ipv4 local endpoint exporter_ipv4 = zipkin_exporter.ZipkinExporter( service_name='my_service', ipv4=ipv4) zipkin_spans_ipv4 = exporter_ipv4.translate_to_zipkin( span_datas=spans_ipv4) self.assertEqual(zipkin_spans_ipv4, expected_zipkin_spans_ipv4) # Test ipv6 local endpoint exporter_ipv6 = zipkin_exporter.ZipkinExporter( service_name='my_service', ipv6=ipv6) zipkin_spans_ipv6 = exporter_ipv6.translate_to_zipkin( span_datas=spans_ipv6) self.assertEqual(zipkin_spans_ipv6, expected_zipkin_spans_ipv6)