This commit is contained in:
coletdjnz 2024-07-14 11:50:17 +12:00
parent 86b32ee0da
commit 6c585b9de8
No known key found for this signature in database
GPG Key ID: 91984263BB39894A

View File

@ -160,7 +160,7 @@ def handle(self):
import websockets.sync.server
self.request.settimeout(None)
protocol = websockets.ServerProtocol()
connection = websockets.sync.server.ServerConnection(socket=self.request, protocol=protocol, close_timeout=2)
connection = websockets.sync.server.ServerConnection(socket=self.request, protocol=protocol, close_timeout=0)
connection.handshake()
for message in connection:
if message == 'proxy_info':
@ -438,11 +438,11 @@ def test_http_connect_source_address(self, handler, ctx):
assert proxy_info['client_address'][0] == source_address
@pytest.mark.skipif(urllib3 is None, reason='requires urllib3 to test')
@pytest.mark.skip_handler_if(
'Websockets', lambda request:
(platform.python_implementation() == 'PyPy'
and request.getfixturevalue('ctx').REQUEST_PROTO == 'wss'),
'PyPy sometimes fails with WSS over TLS tests, unknown reason')
# @pytest.mark.skip_handler_if(
# 'Websockets', lambda request:
# (platform.python_implementation() == 'PyPy'
# and request.getfixturevalue('ctx').REQUEST_PROTO == 'wss'),
# 'PyPy sometimes fails with WSS over TLS tests, unknown reason')
def test_https_connect_proxy(self, handler, ctx):
with ctx.http_server(HTTPSConnectProxyHandler) as server_address:
with handler(verify=False, proxies={ctx.REQUEST_PROTO: f'https://{server_address}'}) as rh:
@ -452,11 +452,11 @@ def test_https_connect_proxy(self, handler, ctx):
assert 'Proxy-Authorization' not in proxy_info['headers']
@pytest.mark.skipif(urllib3 is None, reason='requires urllib3 to test')
@pytest.mark.skip_handler_if(
'Websockets', lambda request:
(platform.python_implementation() == 'PyPy'
and request.getfixturevalue('ctx').REQUEST_PROTO == 'wss'),
'PyPy sometimes fails with WSS over TLS tests, unknown reason')
# @pytest.mark.skip_handler_if(
# 'Websockets', lambda request:
# (platform.python_implementation() == 'PyPy'
# and request.getfixturevalue('ctx').REQUEST_PROTO == 'wss'),
# 'PyPy sometimes fails with WSS over TLS tests, unknown reason')
def test_https_connect_verify_failed(self, handler, ctx):
with ctx.http_server(HTTPSConnectProxyHandler) as server_address:
with handler(verify=True, proxies={ctx.REQUEST_PROTO: f'https://{server_address}'}) as rh:
@ -467,11 +467,11 @@ def test_https_connect_verify_failed(self, handler, ctx):
ctx.proxy_info_request(rh)
@pytest.mark.skipif(urllib3 is None, reason='requires urllib3 to test')
@pytest.mark.skip_handler_if(
'Websockets', lambda request:
(platform.python_implementation() == 'PyPy'
and request.getfixturevalue('ctx').REQUEST_PROTO == 'wss'),
'PyPy sometimes fails with WSS over TLS tests, unknown reason')
# @pytest.mark.skip_handler_if(
# 'Websockets', lambda request:
# (platform.python_implementation() == 'PyPy'
# and request.getfixturevalue('ctx').REQUEST_PROTO == 'wss'),
# 'PyPy sometimes fails with WSS over TLS tests, unknown reason')
def test_https_connect_proxy_auth(self, handler, ctx):
with ctx.http_server(HTTPSConnectProxyHandler, username='test', password='test') as server_address:
with handler(verify=False, proxies={ctx.REQUEST_PROTO: f'https://test:test@{server_address}'}) as rh: