@@ -667,15 +667,23 @@ async def foobar_web(websocket: WebSocket) -> None:
667
667
)
668
668
669
669
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
- )
678
670
class 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
+
679
687
def test_custom_request_headers_in_span_attributes (self ):
680
688
expected = {
681
689
"http.request.header.custom_test_header_1" : (
@@ -790,15 +798,23 @@ def test_custom_response_headers_not_in_span_attributes(self):
790
798
self .assertNotIn (key , server_span .attributes )
791
799
792
800
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
- )
801
801
class 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
+
802
818
def test_custom_request_headers_in_span_attributes (self ):
803
819
expected = {
804
820
"http.request.header.custom_test_header_1" : (
@@ -913,23 +929,29 @@ def test_custom_response_headers_not_in_span_attributes(self):
913
929
self .assertNotIn (key , server_span .attributes )
914
930
915
931
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
- )
924
932
class TestNonRecordingSpanWithCustomHeaders (TestBaseWithCustomHeaders ):
925
933
def setUp (self ):
926
934
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
+
927
945
reset_trace_globals ()
928
946
set_tracer_provider (tracer_provider = NoOpTracerProvider ())
929
947
930
948
self ._app = self .create_app ()
931
949
self ._client = TestClient (self ._app )
932
950
951
+ def tearDown (self ):
952
+ self .test_env_patch .stop ()
953
+ super ().tearDown ()
954
+
933
955
def test_custom_header_not_present_in_non_recording_span (self ):
934
956
resp = self ._client .get (
935
957
"/foobar" ,
0 commit comments