opensource/opentelemetry-python-contrib
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git synced 2025-08-02 11:31:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor common test code to use utility method (#700)

Browse Source
This commit is contained in:
alrex
2021-10-14 15:19:02 -07:00
committed by GitHub
parent 3ebfdb1a54
commit face2a3272
6 changed files with 15 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
instrumentation/opentelemetry-instrumentation-django/tests/test_middleware.py
View File

@ -34,12 +34,7 @@ from opentelemetry.sdk.trace import Span
from opentelemetry.semconv.trace import SpanAttributes
from opentelemetry.test.test_base import TestBase
from opentelemetry.test.wsgitestutil import WsgiTestBase
from opentelemetry.trace import (
SpanKind,
StatusCode,
format_span_id,
format_trace_id,
)
from opentelemetry.trace import SpanKind, StatusCode
from opentelemetry.util.http import get_excluded_urls, get_traced_request_attrs
# pylint: disable=import-error
@ -345,15 +340,8 @@ class TestMiddleware(TestBase, WsgiTestBase):
set_global_response_propagator(TraceResponsePropagator())
response = Client().get("/span_name/1234/")
span = self.memory_exporter.get_finished_spans()[0]
self.assertIn("traceresponse", response.headers)
self.assertEqual(
response.headers["Access-Control-Expose-Headers"], "traceresponse",
)
self.assertEqual(
response.headers["traceresponse"],
f"00-{format_trace_id(span.get_span_context().trace_id)}-{format_span_id(span.get_span_context().span_id)}-01",
self.assertTraceResponseHeaderMatchesSpan(
response.headers, self.memory_exporter.get_finished_spans()[0]
)
self.memory_exporter.clear()