File tree Expand file tree Collapse file tree 1 file changed +14
-7
lines changed
Expand file tree Collapse file tree 1 file changed +14
-7
lines changed Original file line number Diff line number Diff line change @@ -62,16 +62,23 @@ def unwrap(self) -> MocketSocket:
6262 return self ._original_socket
6363
6464 @classmethod
65- def _create (cls , sock : MocketSocket , * args : Any , ** kwargs : Any ) -> MocketSSLSocket :
65+ def _create (
66+ cls ,
67+ sock : MocketSocket ,
68+ ssl_context : ssl .SSLContext | None = None ,
69+ server_hostname : str | None = None ,
70+ * args : Any ,
71+ ** kwargs : Any ,
72+ ) -> MocketSSLSocket :
6673 ssl_socket = MocketSSLSocket ()
6774 ssl_socket ._original_socket = sock
75+ ssl_socket ._true_socket = sock ._true_socket
6876
69- from mocket .urllib3 import true_ssl_wrap_socket
70-
71- ssl_socket ._true_socket = true_ssl_wrap_socket (
72- sock ._true_socket ,
73- ** kwargs ,
74- )
77+ if ssl_context :
78+ ssl_socket ._true_socket = ssl_context .wrap_socket (
79+ sock = ssl_socket ._true_socket ,
80+ server_hostname = server_hostname ,
81+ )
7582
7683 ssl_socket ._kwargs = kwargs
7784
You can’t perform that action at this time.
0 commit comments