def test_no_span_but_in_trace(self):
with ContextHelper():
tracer = tracer_module.Tracer(sampler=AlwaysOnSampler())
wrapped_span = OpenCensusSpan()
assert wrapped_span.span_instance.name == "span"
assert wrapped_span.span_instance.context_tracer.trace_id == tracer.span_context.trace_id
wrapped_span.finish()
tracer.finish()
def test_no_span_passed_in_with_no_environ(self):
with ContextHelper() as ctx:
tracer = OpenCensusSpan.get_current_tracer()
wrapped_span = OpenCensusSpan()
assert wrapped_span.span_instance.name == "span"
assert wrapped_span.span_instance.context_tracer.span_context.trace_id == tracer.span_context.trace_id
assert ctx.orig_tracer == tracer
wrapped_span.finish()
def test_span_passed_in(self):
with ContextHelper():
tracer = tracer_module.Tracer(sampler=AlwaysOnSampler())
with tracer.start_span(name="parent") as parent:
wrapped_span = OpenCensusSpan(parent)
assert wrapped_span.span_instance.name == "parent"
assert wrapped_span.span_instance.context_tracer.trace_id == tracer.span_context.trace_id
wrapped_span.finish()
tracer.finish()
def test_start_finish(self):
with ContextHelper() as ctx:
tracer = tracer_module.Tracer(sampler=AlwaysOnSampler())
parent = OpenCensusSpan()
wrapped_class = parent.span()
assert wrapped_class.span_instance.end_time is None
wrapped_class.start()
wrapped_class.finish()
assert wrapped_class.span_instance.start_time is not None
assert wrapped_class.span_instance.end_time is not None
parent.finish()
