def test_no_span_passed_in_with_no_environ(self, tracer):
with tracer.start_as_current_span("Root") as parent:
with OpenTelemetrySpan() as wrapped_span:
assert wrapped_span.span_instance.name == "span"
assert wrapped_span.span_instance is trace.get_current_span()
assert parent is trace.get_current_span()
def test_span_passed_in(self, tracer):
with tracer.start_as_current_span(name="parent") as parent:
wrapped_span = OpenTelemetrySpan(parent)
assert wrapped_span.span_instance.name == "parent"
assert parent is tracer.get_current_span()
assert wrapped_span.span_instance is tracer.get_current_span()
assert parent is tracer.get_current_span()
def test_span(self, tracer):
with tracer.start_as_current_span("Root") as parent:
with OpenTelemetrySpan() as wrapped_span:
assert wrapped_span.span_instance is trace.get_current_span()
with wrapped_span.span() as child:
assert child.span_instance.name == "span"
assert child.span_instance is trace.get_current_span()
assert child.span_instance.parent is wrapped_span.span_instance.context
def test_links(self, tracer):
with tracer.start_as_current_span("Root") as parent:
og_header = {
"traceparent":
"00-2578531519ed94423ceae67588eff2c9-231ebdc614cb9ddd-01"
}
with OpenTelemetrySpan() as wrapped_class:
OpenTelemetrySpan.link_from_headers(og_header)
assert len(wrapped_class.span_instance.links) == 1
link = wrapped_class.span_instance.links[0]
assert link.context.trace_id == int(
"2578531519ed94423ceae67588eff2c9", 16)
assert link.context.span_id == int("231ebdc614cb9ddd", 16)
with OpenTelemetrySpan() as wrapped_class:
OpenTelemetrySpan.link(
"00-2578531519ed94423ceae67588eff2c9-231ebdc614cb9ddd-01")
assert len(wrapped_class.span_instance.links) == 1
link = wrapped_class.span_instance.links[0]
assert link.context.trace_id == int(
"2578531519ed94423ceae67588eff2c9", 16)
assert link.context.span_id == int("231ebdc614cb9ddd", 16)
def test_set_http_attributes(self, tracer):
with tracer.start_as_current_span("Root") as parent:
wrapped_class = OpenTelemetrySpan(span=parent)
request = mock.Mock()
setattr(request, "method", "GET")
setattr(request, "url", "some url")
response = mock.Mock()
setattr(request, "headers", {})
setattr(response, "status_code", 200)
wrapped_class.set_http_attributes(request)
assert wrapped_class.span_instance.attributes.get(
"http.method") == request.method
assert wrapped_class.span_instance.attributes.get(
"component") == "http"
assert wrapped_class.span_instance.attributes.get(
"http.url") == request.url
assert wrapped_class.span_instance.attributes.get(
"http.status_code") == 504
assert wrapped_class.span_instance.attributes.get(
"http.user_agent") is None
request.headers["User-Agent"] = "some user agent"
wrapped_class.set_http_attributes(request, response)
assert wrapped_class.span_instance.attributes.get(
"http.status_code") == response.status_code
assert wrapped_class.span_instance.attributes.get(
"http.user_agent") == request.headers.get("User-Agent")
def test_links_with_attribute(self, tracer):
with tracer.start_as_current_span("Root") as parent:
attributes = {"attribute1": 1, "attribute2": 2}
og_header = {
"traceparent":
"00-2578531519ed94423ceae67588eff2c9-231ebdc614cb9ddd-02"
}
with OpenTelemetrySpan() as wrapped_class:
OpenTelemetrySpan.link_from_headers(og_header, attributes)
assert len(wrapped_class.span_instance.links) == 1
link = wrapped_class.span_instance.links[0]
assert link.context.trace_id == int(
"2578531519ed94423ceae67588eff2c9", 16)
assert link.context.span_id == int("231ebdc614cb9ddd", 16)
assert "attribute1" in link.attributes
assert "attribute2" in link.attributes
assert link.attributes == attributes
with OpenTelemetrySpan() as wrapped_class:
OpenTelemetrySpan.link(
"00-2578531519ed94423ceae67588eff2c9-231ebdc614cb9ddd-02",
attributes)
assert len(wrapped_class.span_instance.links) == 1
link = wrapped_class.span_instance.links[0]
assert link.context.trace_id == int(
"2578531519ed94423ceae67588eff2c9", 16)
assert link.context.span_id == int("231ebdc614cb9ddd", 16)
assert "attribute1" in link.attributes
assert "attribute2" in link.attributes
assert link.attributes == attributes
def test_start_finish(self, tracer):
with tracer.start_as_current_span("Root") as parent:
wrapped_class = OpenTelemetrySpan()
assert wrapped_class.span_instance.start_time is not None
assert wrapped_class.span_instance.end_time is None
wrapped_class.start()
assert wrapped_class.span_instance.start_time is not None
assert wrapped_class.span_instance.end_time is None
wrapped_class.finish()
assert wrapped_class.span_instance.start_time is not None
assert wrapped_class.span_instance.end_time is not None
def test_get_span_from_thread(tracer):
result = []
def get_span_from_thread(output):
current_span = OpenTelemetrySpan.get_current_span()
output.append(current_span)
with tracer.start_as_current_span(name="TestSpan") as span:
thread = threading.Thread(
target=OpenTelemetrySpan.with_current_context(get_span_from_thread),
args=(result,)
)
thread.start()
thread.join()
assert span is result[0]
def test_change_context(self, tracer):
with tracer.start_as_current_span("Root") as parent:
with OpenTelemetrySpan() as wrapped_class:
with OpenTelemetrySpan.change_context(parent):
assert tracer.get_current_span() is parent
def test_add_attribute(self, tracer):
with tracer.start_as_current_span("Root") as parent:
wrapped_class = OpenTelemetrySpan(span=parent)
wrapped_class.add_attribute("test", "test2")
assert wrapped_class.span_instance.attributes["test"] == "test2"
assert parent.attributes["test"] == "test2"
def get_span_from_thread(output):
current_span = OpenTelemetrySpan.get_current_span()
output.append(current_span)