def test_add_span(self): ctx = Context() span = Span(tracer=None, name="fake_span") ctx.add_span(span) assert ctx == span.context assert ctx.span_id == span.span_id assert ctx.get_current_span() == span assert ctx.get_current_root_span() == span span2 = Span(tracer=None, name="fake_span2") span2.parent_id = span.span_id span2._parent = span ctx.add_span(span2) assert ctx == span2.context assert ctx.span_id == span2.span_id assert ctx.get_current_span() == span2 assert ctx.get_current_root_span() == span
def test_current_root_span(self): # it should return the current active root span ctx = Context() span = Span(tracer=None, name='fake_span') ctx.add_span(span) assert span == ctx.get_current_root_span()
def test_current_root_span_none(self): # it should return none when there is no root span ctx = Context() assert ctx.get_current_root_span() is None