def test_span_error_report(): tracer = Tracer() span = tracer.start_span('foo') error_message = 'unexpected_situation' with mock.patch.object(span, 'log_kv') as log_kv: with mock.patch.object(span, 'set_tag') as set_tag: try: with span: raise ValueError(error_message) except ValueError: pass assert set_tag.call_count == 1 assert set_tag.call_args[0] == (tags.ERROR, True) assert log_kv.call_count == 1 log_kv_args = log_kv.call_args[0][0] assert log_kv_args.get(logs.EVENT, None) is tags.ERROR assert log_kv_args.get(logs.MESSAGE, None) is error_message assert log_kv_args.get(logs.ERROR_KIND, None) is ValueError assert isinstance(log_kv_args.get(logs.ERROR_OBJECT, None), ValueError) assert isinstance(log_kv_args.get(logs.STACK, None), types.TracebackType)
def test_span(): tracer = Tracer() parent = tracer.start_span('parent') child = tracer.start_span('test', references=child_of(parent.context)) assert parent == child child.log_event('cache_hit', ['arg1', 'arg2']) with mock.patch.object(parent, 'finish') as finish: with mock.patch.object(parent, 'log_event') as log_event: try: with parent: raise ValueError() except ValueError: pass assert finish.call_count == 1 assert log_event.call_count == 1 with mock.patch.object(parent, 'finish') as finish: with mock.patch.object(parent, 'log_event') as log_event: with parent: pass assert finish.call_count == 1 assert log_event.call_count == 0 parent.set_tag('x', 'y').set_tag('z', 1) # test chaining parent.set_tag(tags.PEER_SERVICE, 'test-service') parent.set_tag(tags.PEER_HOST_IPV4, 127 << 24 + 1) parent.set_tag(tags.PEER_HOST_IPV6, '::') parent.set_tag(tags.PEER_HOSTNAME, 'uber.com') parent.set_tag(tags.PEER_PORT, 123) parent.finish()
def test_scope_manager_check_works(self): api_check = APICompatibilityCheckMixin() setattr(api_check, 'tracer', lambda: Tracer()) # these tests are expected to succeed api_check.test_start_active_span_ignore_active_span() api_check.test_start_span_propagation_ignore_active_span() # no-op tracer has a no-op ScopeManager implementation, # which means no *actual* propagation is done, # so these tests are expected to work, but asserts to fail with self.assertRaises(AssertionError): api_check.test_start_active_span() with self.assertRaises(AssertionError): api_check.test_start_active_span_parent() with self.assertRaises(AssertionError): api_check.test_start_span_propagation() with self.assertRaises(AssertionError): api_check.test_tracer_start_active_span_scope() with self.assertRaises(AssertionError): api_check.test_tracer_start_span_scope() with self.assertRaises(AssertionError): api_check.test_start_active_span_finish_on_close()
def test_extract(): tracer = Tracer() noop_span = tracer._noop_span bin_carrier = bytearray() span_ctx = tracer.extract(Format.BINARY, carrier=bin_carrier) assert noop_span.context == span_ctx text_carrier = {} span_ctx = tracer.extract(Format.TEXT_MAP, carrier=text_carrier) assert noop_span.context == span_ctx
def initialize(self): SERVICE = "🦄 Stan ❤️s Python 🦄" tracer.init(o.Options(service=SERVICE, log_level=logging.DEBUG)) print("Calling initialize " + str(self) + str(SERVICE)) today = datetime.datetime.now() print(today) self.tracer = Tracer() self.span = self.tracer.start_span(operation_name='root') self.span.set_tag('RequestHandler', 'MainHandler') self.requestmethod = "" self.clazzName = self.__class__.__name__
def test_baggage_check_works(self): api_check = APICompatibilityCheckMixin() setattr(api_check, 'tracer', lambda: Tracer()) # no-op tracer does not store baggage, so the test with default # value of `check_baggage_values()` should fail. with self.assertRaises(AssertionError): api_check.test_span_baggage() # second check that assert on empty baggage will fail too with self.assertRaises(AssertionError): api_check.test_context_baggage()
def test_inject(): tracer = Tracer() span = tracer.start_span() bin_carrier = bytearray() tracer.inject(span_context=span.context, format=Format.BINARY, carrier=bin_carrier) assert bin_carrier == bytearray() text_carrier = {} tracer.inject(span_context=span.context, format=Format.TEXT_MAP, carrier=text_carrier) assert text_carrier == {}
def test_join_trace(): tracer = Tracer() span_ctx = tracer.extract(format=Format.TEXT_MAP, carrier={}) span = tracer.start_span(operation_name='test', references=opentracing.child_of(span_ctx)) span.set_tag('x', 'y') span.set_baggage_item('a', 'b') span.log_event('z') child = tracer.start_span(operation_name='child', references=opentracing.child_of(span.context)) child.log_event('w') child.finish() span.finish()
def test_new_trace(): tracer = Tracer() span = tracer.start_span(operation_name='test') span.set_baggage_item('Fry', 'Leela') span.set_tag('x', 'y') span.log_event('z') child = tracer.start_span(operation_name='child', references=opentracing.child_of(span.context)) child.log_event('w') assert child.get_baggage_item('Fry') is None carrier = {} tracer.inject(span_context=child.context, format=Format.TEXT_MAP, carrier=carrier) assert carrier == dict() child.finish() span.finish()
def tracer(self): return Tracer()
def test_tracer(): tracer = Tracer() span = tracer.start_span(operation_name='root') child = tracer.start_span(operation_name='child', references=child_of(span)) assert span == child