fix SIM117 multiple-with-statements

Signed-off-by: Gaëtan Lehmann <[email protected]>

Author: glehmann

tests/security/test_tls.py

@@ -26,11 +26,11 @@ def test_tls_disabled(host: str, protocol_name: str):
 
     with pytest.raises(ssl.SSLError):
         context = ssl.SSLContext(protocol)
-        with socket.create_connection((str(host), PORT), timeout=10) as sock:
-            with context.wrap_socket(sock, server_hostname=str(host)) as ssock:
-                ssock.do_handshake()
-                # If we reach this point, the protocol is enabled (test should fail)
-                pytest.fail(f"Protocol {protocol} should be disabled but connection succeeded")
+        with socket.create_connection((str(host), PORT), timeout=10) as sock, \
+             context.wrap_socket(sock, server_hostname=str(host)) as ssock:
+            ssock.do_handshake()
+            # If we reach this point, the protocol is enabled (test should fail)
+            pytest.fail(f"Protocol {protocol} should be disabled but connection succeeded")
 
 @pytest.mark.parametrize("protocol_name", ["TLSv1.2"])
 def test_enabled(host: str, protocol_name: str):
@@ -48,9 +48,9 @@ def test_enabled(host: str, protocol_name: str):
 
     try:
         context = ssl.SSLContext(protocol)
-        with socket.create_connection((str(host), PORT), timeout=10) as sock:
-            with context.wrap_socket(sock, server_hostname=str(host)) as ssock:
-                ssock.do_handshake()
-                assert ssock.version()
+        with socket.create_connection((str(host), PORT), timeout=10) as sock, \
+             context.wrap_socket(sock, server_hostname=str(host)) as ssock:
+            ssock.do_handshake()
+            assert ssock.version()
     except ssl.SSLError as e:
         pytest.fail(f"{protocol_name} should be enabled, but got SSLError: {e}")