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