@@ -466,9 +466,9 @@ def test_err(self):
466466
467467
468468class CustomHeaderSimpleHTTPRequestHandler (SimpleHTTPRequestHandler ):
469- custom_headers = None
470- def __init__ (self , * args , directory = None , response_headers = None , ** kwargs ):
471- super ().__init__ (* args , response_headers = self .custom_headers , ** kwargs )
469+ extra_response_headers = None
470+ def __init__ (self , * args , directory = None , extra_response_headers = None , ** kwargs ):
471+ super ().__init__ (* args , extra_response_headers = self .extra_response_headers , ** kwargs )
472472
473473
474474class SimpleHTTPServerTestCase (BaseTestCase ):
@@ -829,17 +829,17 @@ def test_path_without_leading_slash(self):
829829 self .assertEqual (response .getheader ("Location" ),
830830 self .tempdir_name + "/?hi=1" )
831831
832- def test_custom_headers_list_dir (self ):
833- with mock .patch .object (self .request_handler , 'custom_headers ' , new = [
832+ def test_extra_headers_list_dir (self ):
833+ with mock .patch .object (self .request_handler , 'extra_response_headers ' , new = [
834834 ('X-Test1' , 'test1' ),
835835 ('X-Test2' , 'test2' ),
836836 ]):
837837 response = self .request (self .base_url + '/' )
838838 self .assertEqual (response .getheader ("X-Test1" ), 'test1' )
839839 self .assertEqual (response .getheader ("X-Test2" ), 'test2' )
840840
841- def test_custom_headers_get_file (self ):
842- with mock .patch .object (self .request_handler , 'custom_headers ' , new = [
841+ def test_extra_response_headers_get_file (self ):
842+ with mock .patch .object (self .request_handler , 'extra_response_headers ' , new = [
843843 ('Set-Cookie' , 'test1=value1' ),
844844 ('Set-Cookie' , 'test2=value2' ),
845845 ('X-Test1' , 'value3' ),
@@ -1491,7 +1491,7 @@ def test_unknown_flag(self, _):
14911491
14921492 @mock .patch ('http.server._make_server' , wraps = server ._make_server )
14931493 @mock .patch .object (HTTPServer , 'serve_forever' )
1494- def test_response_headers_arg (self , _ , mock_make_server ):
1494+ def test_extra_response_headers_arg (self , _ , mock_make_server ):
14951495 server ._main (
14961496 ['-H' , 'Set-Cookie' , 'k=v' , '-H' , 'Set-Cookie' , 'k2=v2' , '8080' ]
14971497 )
@@ -1508,11 +1508,11 @@ def test_response_headers_arg(self, _, mock_make_server):
15081508 ) as mock_handler_init :
15091509 mock_handler_init .return_value = None
15101510 # finish_request instantiates a request handler class,
1511- # ensure response_headers are passed to it
1511+ # ensure extra_response_headers are passed to it
15121512 httpd .finish_request (mock .Mock (), '127.0.0.1' )
15131513 mock_handler_init .assert_called_once_with (
15141514 mock .ANY , mock .ANY , mock .ANY , directory = mock .ANY ,
1515- response_headers = [
1515+ extra_response_headers = [
15161516 ['Set-Cookie' , 'k=v' ], ['Set-Cookie' , 'k2=v2' ]
15171517 ]
15181518 )
0 commit comments