@@ -535,8 +535,8 @@ def release_conn(self):
535535
536536
537537# Tests for SSL Context Reuse (fix for OpenSSL 3.0+ performance issues)
538- @patch (' ssl.create_default_context' )
539- @patch (' urllib3.PoolManager' )
538+ @patch (" ssl.create_default_context" )
539+ @patch (" urllib3.PoolManager" )
540540def test_ssl_context_created_with_ca_cert (mock_pool_manager , mock_create_context ):
541541 """Test that SSL context is created with CA certificate file."""
542542 mock_ssl_context = MagicMock ()
@@ -559,11 +559,11 @@ def test_ssl_context_created_with_ca_cert(mock_pool_manager, mock_create_context
559559 # Verify SSL context was passed to PoolManager
560560 mock_pool_manager .assert_called_once ()
561561 call_kwargs = mock_pool_manager .call_args [1 ]
562- assert call_kwargs [' ssl_context' ] == mock_ssl_context
562+ assert call_kwargs [" ssl_context" ] == mock_ssl_context
563563
564564
565- @patch (' ssl.create_default_context' )
566- @patch (' urllib3.PoolManager' )
565+ @patch (" ssl.create_default_context" )
566+ @patch (" urllib3.PoolManager" )
567567def test_ssl_context_loads_client_certificate (mock_pool_manager , mock_create_context ):
568568 """Test that SSL context loads client certificate and key when provided."""
569569 mock_ssl_context = MagicMock ()
@@ -591,12 +591,14 @@ def test_ssl_context_loads_client_certificate(mock_pool_manager, mock_create_con
591591 # Verify SSL context was passed to PoolManager
592592 mock_pool_manager .assert_called_once ()
593593 call_kwargs = mock_pool_manager .call_args [1 ]
594- assert call_kwargs [' ssl_context' ] == mock_ssl_context
594+ assert call_kwargs [" ssl_context" ] == mock_ssl_context
595595
596596
597- @patch ('ssl.create_default_context' )
598- @patch ('urllib3.PoolManager' )
599- def test_ssl_context_disables_verification_when_verify_ssl_false (mock_pool_manager , mock_create_context ):
597+ @patch ("ssl.create_default_context" )
598+ @patch ("urllib3.PoolManager" )
599+ def test_ssl_context_disables_verification_when_verify_ssl_false (
600+ mock_pool_manager , mock_create_context
601+ ):
600602 """Test that SSL context disables verification when verify_ssl=False."""
601603 mock_ssl_context = MagicMock ()
602604 mock_create_context .return_value = mock_ssl_context
@@ -622,11 +624,11 @@ def test_ssl_context_disables_verification_when_verify_ssl_false(mock_pool_manag
622624 # Verify SSL context was passed to PoolManager
623625 mock_pool_manager .assert_called_once ()
624626 call_kwargs = mock_pool_manager .call_args [1 ]
625- assert call_kwargs [' ssl_context' ] == mock_ssl_context
627+ assert call_kwargs [" ssl_context" ] == mock_ssl_context
626628
627629
628- @patch (' ssl.create_default_context' )
629- @patch (' urllib3.ProxyManager' )
630+ @patch (" ssl.create_default_context" )
631+ @patch (" urllib3.ProxyManager" )
630632def test_ssl_context_used_with_proxy_manager (mock_proxy_manager , mock_create_context ):
631633 """Test that SSL context is passed to ProxyManager when proxy is configured."""
632634 mock_ssl_context = MagicMock ()
@@ -655,14 +657,16 @@ def test_ssl_context_used_with_proxy_manager(mock_proxy_manager, mock_create_con
655657 # Verify SSL context was passed to ProxyManager
656658 mock_proxy_manager .assert_called_once ()
657659 call_kwargs = mock_proxy_manager .call_args [1 ]
658- assert call_kwargs [' ssl_context' ] == mock_ssl_context
659- assert call_kwargs [' proxy_url' ] == "http://proxy:8080"
660- assert call_kwargs [' proxy_headers' ] == {"Proxy-Auth" : "token" }
660+ assert call_kwargs [" ssl_context" ] == mock_ssl_context
661+ assert call_kwargs [" proxy_url" ] == "http://proxy:8080"
662+ assert call_kwargs [" proxy_headers" ] == {"Proxy-Auth" : "token" }
661663
662664
663- @patch ('ssl.create_default_context' )
664- @patch ('urllib3.PoolManager' )
665- def test_ssl_context_reuse_performance_optimization (mock_pool_manager , mock_create_context ):
665+ @patch ("ssl.create_default_context" )
666+ @patch ("urllib3.PoolManager" )
667+ def test_ssl_context_reuse_performance_optimization (
668+ mock_pool_manager , mock_create_context
669+ ):
666670 """Test that SSL context creation is called only once per client instance."""
667671 mock_ssl_context = MagicMock ()
668672 mock_create_context .return_value = mock_ssl_context
@@ -685,7 +689,7 @@ def test_ssl_context_reuse_performance_optimization(mock_pool_manager, mock_crea
685689 # Verify the same SSL context instance is reused
686690 mock_pool_manager .assert_called_once ()
687691 call_kwargs = mock_pool_manager .call_args [1 ]
688- assert call_kwargs [' ssl_context' ] is mock_ssl_context
692+ assert call_kwargs [" ssl_context" ] is mock_ssl_context
689693
690694 # Verify context was not created again during subsequent operations
691695 mock_create_context .reset_mock ()
@@ -697,8 +701,8 @@ def test_ssl_context_reuse_performance_optimization(mock_pool_manager, mock_crea
697701 mock_create_context .assert_not_called ()
698702
699703
700- @patch (' ssl.create_default_context' )
701- @patch (' urllib3.PoolManager' )
704+ @patch (" ssl.create_default_context" )
705+ @patch (" urllib3.PoolManager" )
702706def test_ssl_context_with_all_ssl_options (mock_pool_manager , mock_create_context ):
703707 """Test SSL context creation with all SSL configuration options set."""
704708 mock_ssl_context = MagicMock ()
@@ -725,11 +729,17 @@ def test_ssl_context_with_all_ssl_options(mock_pool_manager, mock_create_context
725729
726730 # Verify SSL verification settings were NOT modified (verify_ssl=True)
727731 # check_hostname and verify_mode should remain at their default secure values
728- assert not hasattr (mock_ssl_context , 'check_hostname' ) or mock_ssl_context .check_hostname
729- assert not hasattr (mock_ssl_context , 'verify_mode' ) or mock_ssl_context .verify_mode != ssl .CERT_NONE
732+ assert (
733+ not hasattr (mock_ssl_context , "check_hostname" )
734+ or mock_ssl_context .check_hostname
735+ )
736+ assert (
737+ not hasattr (mock_ssl_context , "verify_mode" )
738+ or mock_ssl_context .verify_mode != ssl .CERT_NONE
739+ )
730740
731741 # Verify SSL context was passed to PoolManager
732742 mock_pool_manager .assert_called_once ()
733743 call_kwargs = mock_pool_manager .call_args [1 ]
734- assert call_kwargs [' ssl_context' ] == mock_ssl_context
735- assert call_kwargs [' maxsize' ] == 8
744+ assert call_kwargs [" ssl_context" ] == mock_ssl_context
745+ assert call_kwargs [" maxsize" ] == 8
0 commit comments