@@ -667,15 +667,23 @@ async def foobar_web(websocket: WebSocket) -> None:
667667 )
668668
669669
670- @patch .dict (
671- "os.environ" ,
672- {
673- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS : ".*my-secret.*" ,
674- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,Regex-Test-Header-.*,Regex-Invalid-Test-Header-.*,.*my-secret.*" ,
675- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,my-custom-regex-header-.*,invalid-regex-header-.*,.*my-secret.*" ,
676- },
677- )
678670class TestHTTPAppWithCustomHeaders (TestBaseWithCustomHeaders ):
671+ def setUp (self ):
672+ super ().setUp ()
673+ self .test_env_patch = patch .dict (
674+ "os.environ" ,
675+ {
676+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS : ".*my-secret.*" ,
677+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,Regex-Test-Header-.*,Regex-Invalid-Test-Header-.*,.*my-secret.*" ,
678+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,my-custom-regex-header-.*,invalid-regex-header-.*,.*my-secret.*" ,
679+ },
680+ )
681+ self .test_env_patch .start ()
682+
683+ def tearDown (self ):
684+ self .test_env_patch .stop ()
685+ super ().tearDown ()
686+
679687 def test_custom_request_headers_in_span_attributes (self ):
680688 expected = {
681689 "http.request.header.custom_test_header_1" : (
@@ -790,15 +798,23 @@ def test_custom_response_headers_not_in_span_attributes(self):
790798 self .assertNotIn (key , server_span .attributes )
791799
792800
793- @patch .dict (
794- "os.environ" ,
795- {
796- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS : ".*my-secret.*" ,
797- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,Regex-Test-Header-.*,Regex-Invalid-Test-Header-.*,.*my-secret.*" ,
798- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,my-custom-regex-header-.*,invalid-regex-header-.*,.*my-secret.*" ,
799- },
800- )
801801class TestWebSocketAppWithCustomHeaders (TestBaseWithCustomHeaders ):
802+ def setUp (self ):
803+ super ().setUp ()
804+ self .test_env_patch = patch .dict (
805+ "os.environ" ,
806+ {
807+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS : ".*my-secret.*" ,
808+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,Regex-Test-Header-.*,Regex-Invalid-Test-Header-.*,.*my-secret.*" ,
809+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,my-custom-regex-header-.*,invalid-regex-header-.*,.*my-secret.*" ,
810+ },
811+ )
812+ self .test_env_patch .start ()
813+
814+ def tearDown (self ):
815+ self .test_env_patch .stop ()
816+ super ().tearDown ()
817+
802818 def test_custom_request_headers_in_span_attributes (self ):
803819 expected = {
804820 "http.request.header.custom_test_header_1" : (
@@ -913,23 +929,29 @@ def test_custom_response_headers_not_in_span_attributes(self):
913929 self .assertNotIn (key , server_span .attributes )
914930
915931
916- @patch .dict (
917- "os.environ" ,
918- {
919- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS : ".*my-secret.*" ,
920- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,Regex-Test-Header-.*,Regex-Invalid-Test-Header-.*,.*my-secret.*" ,
921- OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,my-custom-regex-header-.*,invalid-regex-header-.*,.*my-secret.*" ,
922- },
923- )
924932class TestNonRecordingSpanWithCustomHeaders (TestBaseWithCustomHeaders ):
925933 def setUp (self ):
926934 super ().setUp ()
935+ self .test_env_patch = patch .dict (
936+ "os.environ" ,
937+ {
938+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SANITIZE_FIELDS : ".*my-secret.*" ,
939+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_REQUEST : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,Regex-Test-Header-.*,Regex-Invalid-Test-Header-.*,.*my-secret.*" ,
940+ OTEL_INSTRUMENTATION_HTTP_CAPTURE_HEADERS_SERVER_RESPONSE : "Custom-Test-Header-1,Custom-Test-Header-2,Custom-Test-Header-3,my-custom-regex-header-.*,invalid-regex-header-.*,.*my-secret.*" ,
941+ },
942+ )
943+ self .test_env_patch .start ()
944+
927945 reset_trace_globals ()
928946 set_tracer_provider (tracer_provider = NoOpTracerProvider ())
929947
930948 self ._app = self .create_app ()
931949 self ._client = TestClient (self ._app )
932950
951+ def tearDown (self ):
952+ self .test_env_patch .stop ()
953+ super ().tearDown ()
954+
933955 def test_custom_header_not_present_in_non_recording_span (self ):
934956 resp = self ._client .get (
935957 "/foobar" ,
0 commit comments