1414
1515
1616import pytest
17+ from opentelemetry import trace as trace_api
18+ from opentelemetry .exporter .otlp .proto .grpc .trace_exporter import (
19+ OTLPSpanExporter ,
20+ )
21+ from opentelemetry .sdk import trace as trace_sdk
22+ from opentelemetry .sdk .trace .export import BatchSpanProcessor
1723
1824from veadk .tracing .telemetry .exporters .apmplus_exporter import (
1925 APMPlusExporter ,
2935)
3036from veadk .tracing .telemetry .opentelemetry_tracer import OpentelemetryTracer
3137
32- from opentelemetry import trace as trace_api
33- from opentelemetry .sdk import trace as trace_sdk
34- from opentelemetry .sdk .trace .export import BatchSpanProcessor
35- from opentelemetry .exporter .otlp .proto .grpc .trace_exporter import (
36- OTLPSpanExporter ,
37- )
38-
3938APP_NAME = "app"
4039USER_ID = "testuser"
4140SESSION_ID = "testsession"
4241
42+
4343def init_exporters ():
4444 cozeloop_exporter = CozeloopExporter (
4545 config = CozeloopExporterConfig (
@@ -68,13 +68,15 @@ def init_exporters():
6868 )
6969 return [cozeloop_exporter , apmplus_exporter , tls_exporter ]
7070
71+
7172def gen_span_processor (endpoint : str ):
7273 otlp_exporter = OTLPSpanExporter (
7374 endpoint = endpoint ,
7475 )
7576 span_processor = BatchSpanProcessor (otlp_exporter )
7677 return span_processor
7778
79+
7880@pytest .mark .asyncio
7981async def test_tracing ():
8082 exporters = init_exporters ()
@@ -86,6 +88,7 @@ async def test_tracing():
8688 # TODO: Ensure the tracing provider is set correctly after loading SDK
8789 # TODO: Ensure the tracing provider is set correctly after loading SDK
8890
91+
8992@pytest .mark .asyncio
9093async def test_tracing_with_global_provider ():
9194 exporters = init_exporters ()
@@ -113,5 +116,3 @@ async def test_tracing_with_apmplus_global_provider():
113116
114117 # apmplus exporter won't init again
115118 assert len (tracer .exporters ) == 4 # with extra 2 built-in exporters
116-
117-
0 commit comments