add test

Author: bisgaard-itis

instrumentation/opentelemetry-instrumentation-aiohttp-server/tests/test_aiohttp_server_integration.py

@@ -32,7 +32,7 @@
 from opentelemetry.test.globals_test import reset_trace_globals
 from opentelemetry.test.test_base import TestBase
 from opentelemetry.util._importlib_metadata import entry_points
-
+from opentelemetry.sdk.trace.sampling import ParentBased, TraceIdRatioBased
 
 class HTTPMethod(Enum):
     """HTTP methods and descriptions"""
@@ -76,9 +76,9 @@ def fixture_suppress():
 
 @pytest_asyncio.fixture(name="server_fixture")
 async def fixture_server_fixture(tracer, aiohttp_server, suppress):
-    _, memory_exporter = tracer
+    tracer_provider, memory_exporter = tracer
 
-    AioHttpServerInstrumentor().instrument()
+    AioHttpServerInstrumentor().instrument(tracer_provider=tracer_provider)
 
     app = aiohttp.web.Application()
     app.add_routes([aiohttp.web.get("/test-path", default_handler)])
@@ -195,3 +195,34 @@ async def handler(request):
     # Clean up
     AioHttpServerInstrumentor().uninstrument()
     memory_exporter.clear()
+
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "tracer", [TestBase().create_tracer_provider(sampler=ParentBased(TraceIdRatioBased(0.05)))]
+)
+async def test_non_global_tracer_provider(
+    tracer,
+    server_fixture,
+    aiohttp_client,
+):
+    n_requests = 1000
+    collection_ratio = 0.05
+    n_expected_trace_ids = n_requests * collection_ratio
+
+    _, memory_exporter = tracer
+    server, _ = server_fixture
+
+    assert len(memory_exporter.get_finished_spans()) == 0
+
+    client = await aiohttp_client(server)
+    for _ in range(n_requests):
+        await client.get("/test-path")
+
+    trace_ids = {
+        span.context.trace_id
+        for span in memory_exporter.get_finished_spans()
+        if span.context is not None
+    }
+    assert 0.5 * n_expected_trace_ids <= len(trace_ids) <= 1.5 * n_expected_trace_ids