File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
sdk/core/azure-core-tracing-opentelemetry/tests Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 33# Licensed under the MIT License.
44# ------------------------------------
55import os
6+ from typing import Generator
67
8+ from azure .core .settings import settings
79from opentelemetry import trace
810from opentelemetry .sdk .trace import TracerProvider
911from opentelemetry .sdk .trace .export .in_memory_span_exporter import InMemorySpanExporter
@@ -25,11 +27,13 @@ def enable_tracing():
2527
2628
2729@pytest .fixture (scope = "function" )
28- def tracing_helper () -> TracingTestHelper :
30+ def tracing_helper () -> Generator [TracingTestHelper , None , None ]:
31+ settings .tracing_implementation = "opentelemetry"
2932 span_exporter = InMemorySpanExporter ()
3033 processor = SimpleSpanProcessor (span_exporter )
3134 trace .get_tracer_provider ().add_span_processor (processor )
32- return TracingTestHelper (trace .get_tracer (__name__ ), span_exporter )
35+ yield TracingTestHelper (trace .get_tracer (__name__ ), span_exporter )
36+ settings .tracing_implementation = None
3337
3438
3539@pytest .fixture (scope = "session" )
You can’t perform that action at this time.
0 commit comments