diff --git a/test/test_http_proxy.py b/test/test_http_proxy.py index 1658c7949..102fc4d4f 100644 --- a/test/test_http_proxy.py +++ b/test/test_http_proxy.py @@ -163,7 +163,7 @@ def handle(self): protocol = websockets.ServerProtocol() connection = websockets.sync.server.ServerConnection(socket=self.request, protocol=protocol, close_timeout=10) try: - connection.handshake() + connection.handshake(timeout=5.0) for message in connection: if message == 'proxy_info': connection.send(json.dumps(self.proxy_info)) @@ -256,7 +256,7 @@ class HTTPProxyTestContext(abc.ABC): def http_server(self, server_class, *args, **kwargs): server = proxy_server(server_class, self.REQUEST_HANDLER_CLASS, *args, **kwargs) - time.sleep(1) # ensure server is up + time.sleep(1) return server @abc.abstractmethod diff --git a/yt_dlp/networking/_websockets.py b/yt_dlp/networking/_websockets.py index 46b2cbd9f..b2918af77 100644 --- a/yt_dlp/networking/_websockets.py +++ b/yt_dlp/networking/_websockets.py @@ -227,7 +227,7 @@ def _send(self, request): open_timeout=timeout, user_agent_header=None, ssl_context=ssl_context, - close_timeout=0, # not ideal, but prevents yt-dlp hanging + close_timeout=0.1, # not ideal, but prevents yt-dlp hanging ) return WebsocketsResponseAdapter(conn, url=request.url)