try to fix

Author: adriangb

instrumentation/opentelemetry-instrumentation-asgi/tests/test_asgi_custom_headers.py

@@ -352,7 +352,7 @@ def setUp(self):
             }
         )
         super().setUp()
-    
+
     def tearDown(self):
         os.environ.pop(OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS, None)
         os.environ.pop(OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST, None)

instrumentation/opentelemetry-instrumentation-django/src/opentelemetry/instrumentation/django/middleware/otel_middleware.py

@@ -45,6 +45,7 @@
 from opentelemetry.util.http import (
     OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS,
     OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST,
+    OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE,
     SanitizeValue,
     _parse_active_request_count_attrs,
     _parse_duration_attrs,
@@ -351,14 +352,14 @@ def process_response(self, request, response):
 
                     custom_res_attributes = (
                         asgi_collect_custom_headers_attributes(
-                            custom_headers,
+                            {'headers': custom_headers},
                             SanitizeValue(
                                 get_custom_headers(
                                     OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS
                                 )
                             ),
                             get_custom_headers(
-                                OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST
+                                OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE
                             ),
                             normalise_request_header_name,
                         )

instrumentation/opentelemetry-instrumentation-django/tests/test_middleware_asgi.py

@@ -537,7 +537,8 @@ async def test_http_custom_response_headers_in_span_attributes(self):
             ),
             "http.response.header.my_secret_header": ("[REDACTED]",),
         }
-        await self.async_client.get("/traced_custom_header/")
+        resp = await self.async_client.get("/traced_custom_header/")
+        assert resp.status_code == 200
         spans = self.exporter.get_finished_spans()
         self.assertEqual(len(spans), 1)