artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-04-07 17:33:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: always use SSLTransport if available

Browse Source 
(so it is used for both ends of tls-in-tls)
This commit is contained in:
coletdjnz 2024-05-18 14:02:26 +12:00
parent c01179b581
commit f5cfe9e00a
No known key found for this signature in database
GPG Key ID: 91984263BB39894A
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/test_http_proxy.py
View File

@ -139,7 +139,7 @@ class HTTPSProxyHandler(HTTPProxyHandler):
certfn = os.path.join(TEST_DIR, 'testcert.pem')
sslctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
sslctx.load_cert_chain(certfn, None)
if isinstance(request, ssl.SSLSocket):
if SSLTransport:
request = SSLTransport(request, ssl_context=sslctx, server_side=True)
else:
request = sslctx.wrap_socket(request, server_side=True)
@ -208,7 +208,10 @@ class HTTPSConnectProxyHandler(HTTPConnectProxyHandler):
certfn = os.path.join(TEST_DIR, 'testcert.pem')
sslctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
sslctx.load_cert_chain(certfn, None)
request = sslctx.wrap_socket(request, server_side=True)
if SSLTransport:
request = SSLTransport(request, ssl_context=sslctx, server_side=True)
else:
request = sslctx.wrap_socket(request, server_side=True)
self._original_request = request
super().__init__(request, *args, **kwargs)