2020 normalise_request_header_name ,
2121 normalise_response_header_name ,
2222 OTEL_PYTHON_CAPTURE_REQUEST_HEADERS ,
23- OTEL_PYTHON_CAPTURE_RESPONSE_HEADERS
23+ OTEL_PYTHON_CAPTURE_RESPONSE_HEADERS ,
2424)
2525
2626
2727class TestCaptureCustomHeaders (TestBase ):
2828 @patch .dict (
2929 "os.environ" ,
30- {
31- OTEL_PYTHON_CAPTURE_REQUEST_HEADERS : "User-Agent,Test-Header"
32- },
30+ {OTEL_PYTHON_CAPTURE_REQUEST_HEADERS : "User-Agent,Test-Header" },
3331 )
3432 def test_get_custom_request_header (self ):
35- custom_headers_to_capture = get_custom_headers (OTEL_PYTHON_CAPTURE_REQUEST_HEADERS )
36- assert custom_headers_to_capture == ['User-Agent' , 'Test-Header' ]
33+ custom_headers_to_capture = get_custom_headers (
34+ OTEL_PYTHON_CAPTURE_REQUEST_HEADERS
35+ )
36+ assert custom_headers_to_capture == ["User-Agent" , "Test-Header" ]
3737
3838 @patch .dict (
3939 "os.environ" ,
@@ -42,13 +42,19 @@ def test_get_custom_request_header(self):
4242 },
4343 )
4444 def test_get_custom_response_header (self ):
45- custom_headers_to_capture = get_custom_headers (OTEL_PYTHON_CAPTURE_RESPONSE_HEADERS )
46- assert custom_headers_to_capture == ['content-type' , 'content-length' , 'test-header' ]
45+ custom_headers_to_capture = get_custom_headers (
46+ OTEL_PYTHON_CAPTURE_RESPONSE_HEADERS
47+ )
48+ assert custom_headers_to_capture == [
49+ "content-type" ,
50+ "content-length" ,
51+ "test-header" ,
52+ ]
4753
4854 def test_normalise_request_header_name (self ):
49- key = normalise_request_header_name (' Test-Header' )
50- assert key == ' http.request.header.test_header'
55+ key = normalise_request_header_name (" Test-Header" )
56+ assert key == " http.request.header.test_header"
5157
5258 def test_normalise_response_header_name (self ):
53- key = normalise_response_header_name (' Test-Header' )
54- assert key == ' http.response.header.test_header'
59+ key = normalise_response_header_name (" Test-Header" )
60+ assert key == " http.response.header.test_header"
0 commit comments