From ea5ad794760e9b562f788be90954a49f02916d42 Mon Sep 17 00:00:00 2001 From: Giorgio Salluzzo Date: Sun, 20 Oct 2024 11:09:03 +0200 Subject: [PATCH] Refactoring FakeSSLContext. (#257) --- mocket/mocket.py | 20 ++------------------ 1 file changed, 2 insertions(+), 18 deletions(-) diff --git a/mocket/mocket.py b/mocket/mocket.py index cbd42ca9..570150a8 100644 --- a/mocket/mocket.py +++ b/mocket/mocket.py @@ -25,7 +25,6 @@ from .compat import basestring, byte_type, decode_from_bytes, encode_to_bytes, text_type from .utils import ( - SSL_PROTOCOL, MocketMode, MocketSocketCore, get_mocketize, @@ -98,20 +97,9 @@ def check_hostname(self): def check_hostname(self, _): self._check_hostname = False - def __init__(self, sock=None, server_hostname=None, _context=None, *args, **kwargs): + def __init__(self, *args, **kwargs): self._set_dummy_methods() - if isinstance(sock, MocketSocket): - self.sock = sock - self.sock._host = server_hostname - self.sock.true_socket = true_ssl_socket( - sock=self.sock.true_socket, - server_hostname=server_hostname, - _context=true_ssl_context(protocol=SSL_PROTOCOL), - ) - elif isinstance(sock, int) and true_ssl_context: - self.context = true_ssl_context(sock) - def _set_dummy_methods(self): def dummy_method(*args, **kwargs): pass @@ -120,7 +108,7 @@ def dummy_method(*args, **kwargs): setattr(self, m, dummy_method) @staticmethod - def wrap_socket(sock=sock, *args, **kwargs): + def wrap_socket(sock, *args, **kwargs): sock.kwargs = kwargs sock._secure_socket = True return sock @@ -131,10 +119,6 @@ def wrap_bio(incoming, outcoming, *args, **kwargs): ssl_obj._host = kwargs["server_hostname"] return ssl_obj - def __getattr__(self, name): - if self.sock is not None: - return getattr(self.sock, name) - def create_connection(address, timeout=None, source_address=None): s = socket.socket(socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_TCP)