From 6c3140a8c133e08082dbb4d10a5dd1e82fdd8246 Mon Sep 17 00:00:00 2001 From: coletdjnz Date: Thu, 4 Apr 2024 19:34:28 +1300 Subject: [PATCH] try this --- test/test_http_proxy.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/test/test_http_proxy.py b/test/test_http_proxy.py index f7175653b..3f6035d90 100644 --- a/test/test_http_proxy.py +++ b/test/test_http_proxy.py @@ -156,9 +156,8 @@ class HTTPConnectProxyHandler(BaseHTTPRequestHandler, HTTPProxyAuthMixin): 'path': self.path, 'proxy': ':'.join(str(y) for y in self.connection.getsockname()), } - request = self.request self.request_handler(self.request, self.client_address, self.server, proxy_info=proxy_info) - self.server.close_request(request) + self.server.close_request(self.request) class HTTPSConnectProxyHandler(HTTPConnectProxyHandler): @@ -167,8 +166,13 @@ class HTTPSConnectProxyHandler(HTTPConnectProxyHandler): sslctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER) sslctx.load_cert_chain(certfn, None) request = sslctx.wrap_socket(request, server_side=True) + self._original_request = request super().__init__(request, *args, **kwargs) + def do_CONNECT(self): + super().do_CONNECT() + self.server.close_request(self._original_request) + @contextlib.contextmanager def proxy_server(proxy_server_class, request_handler, bind_ip=None, **proxy_server_kwargs):