diff --git a/ext/opentelemetry-ext-aiohttp-client/src/opentelemetry/ext/aiohttp_client/__init__.py b/ext/opentelemetry-ext-aiohttp-client/src/opentelemetry/ext/aiohttp_client/__init__.py index 3c4e7f4eda8..77dadbd6453 100644 --- a/ext/opentelemetry-ext-aiohttp-client/src/opentelemetry/ext/aiohttp_client/__init__.py +++ b/ext/opentelemetry-ext-aiohttp-client/src/opentelemetry/ext/aiohttp_client/__init__.py @@ -175,9 +175,7 @@ async def on_request_start( trace.propagation.set_span_in_context(trace_config_ctx.span) ) - propagators.inject( - tracer, type(params.headers).__setitem__, params.headers - ) + propagators.inject(type(params.headers).__setitem__, params.headers) async def on_request_end( unused_session: aiohttp.ClientSession, diff --git a/ext/opentelemetry-ext-aiohttp-client/tests/test_aiohttp_client_integration.py b/ext/opentelemetry-ext-aiohttp-client/tests/test_aiohttp_client_integration.py index ae29801797c..8cf5048e588 100644 --- a/ext/opentelemetry-ext-aiohttp-client/tests/test_aiohttp_client_integration.py +++ b/ext/opentelemetry-ext-aiohttp-client/tests/test_aiohttp_client_integration.py @@ -71,7 +71,8 @@ def _http_request( """Helper to start an aiohttp test server and send an actual HTTP request to it.""" async def do_request(): - async def default_handler(unused_request): + async def default_handler(request): + assert "traceparent" in request.headers return aiohttp.web.Response(status=int(status_code)) handler = request_handler or default_handler @@ -281,8 +282,9 @@ async def do_request(url): self.memory_exporter.clear() def test_timeout(self): - async def request_handler(unused_request): + async def request_handler(request): await asyncio.sleep(1) + assert "traceparent" in request.headers return aiohttp.web.Response() host, port = self._http_request( @@ -312,6 +314,7 @@ def test_too_many_redirects(self): async def request_handler(request): # Create a redirect loop. location = request.url + assert "traceparent" in request.headers raise aiohttp.web.HTTPFound(location=location) host, port = self._http_request(