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 3
3
# Licensed under the MIT License.
4
4
# ------------------------------------
5
5
import os
6
+ from typing import Generator
6
7
8
+ from azure .core .settings import settings
7
9
from opentelemetry import trace
8
10
from opentelemetry .sdk .trace import TracerProvider
9
11
from opentelemetry .sdk .trace .export .in_memory_span_exporter import InMemorySpanExporter
@@ -25,11 +27,13 @@ def enable_tracing():
25
27
26
28
27
29
@pytest .fixture (scope = "function" )
28
- def tracing_helper () -> TracingTestHelper :
30
+ def tracing_helper () -> Generator [TracingTestHelper , None , None ]:
31
+ settings .tracing_implementation = "opentelemetry"
29
32
span_exporter = InMemorySpanExporter ()
30
33
processor = SimpleSpanProcessor (span_exporter )
31
34
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
33
37
34
38
35
39
@pytest .fixture (scope = "session" )
You can’t perform that action at this time.
0 commit comments