refactor: convert MocketSSLContext.wrap_socket and wrap_bio to instance-methods

Author: betaboon

mocket/inject.py

@@ -69,7 +69,9 @@ def _inject_stdlib_ssl() -> None:
     _replace(ssl, "SSLContext", MocketSSLContext)
 
     if python_version() < Version("3.12.0"):
-        _replace(ssl, "wrap_socket", MocketSSLContext.wrap_socket)
+        from mocket.ssl.context import mock_wrap_socket
+
+        _replace(ssl, "wrap_socket", mock_wrap_socket)
 
 
 def _restore_stdlib_ssl() -> None:

mocket/ssl/context.py

@@ -58,12 +58,16 @@ def dummy_method(*args: Any, **kwargs: Any) -> Any:
         for m in self.DUMMY_METHODS:
             setattr(self, m, dummy_method)
 
-    @staticmethod
-    def wrap_socket(sock: MocketSocket, *args: Any, **kwargs: Any) -> MocketSSLSocket:
+    def wrap_socket(
+        self,
+        sock: MocketSocket,
+        *args: Any,
+        **kwargs: Any,
+    ) -> MocketSSLSocket:
         return MocketSSLSocket._create(sock, *args, **kwargs)
 
-    @staticmethod
     def wrap_bio(
+        self,
         incoming: Any,  # _ssl.MemoryBIO
         outgoing: Any,  # _ssl.MemoryBIO
         server_side: bool = False,
@@ -72,3 +76,12 @@ def wrap_bio(
         ssl_obj = MocketSSLSocket()
         ssl_obj._host = server_hostname
         return ssl_obj
+
+
+def mock_wrap_socket(
+    sock: MocketSocket,
+    *args: Any,
+    **kwargs: Any,
+) -> MocketSSLSocket:
+    context = MocketSSLContext()
+    return context.wrap_socket(sock, *args, **kwargs)