From b77c2adc34486af882c9736150ea262b7eb8fd70 Mon Sep 17 00:00:00 2001 From: Yury Selivanov Date: Mon, 28 May 2018 15:21:13 -0400 Subject: [PATCH] bpo-32458: Further tune start_tls tests --- Lib/test/test_asyncio/test_sslproto.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/Lib/test/test_asyncio/test_sslproto.py b/Lib/test/test_asyncio/test_sslproto.py index 932487a9e3c639..1b2f9d2a3a2a13 100644 --- a/Lib/test/test_asyncio/test_sslproto.py +++ b/Lib/test/test_asyncio/test_sslproto.py @@ -293,7 +293,7 @@ async def client(addr): new_tr.close() - with self.tcp_server(serve) as srv: + with self.tcp_server(serve, timeout=self.TIMEOUT) as srv: self.loop.run_until_complete( asyncio.wait_for(client(srv.addr), loop=self.loop, timeout=10)) @@ -358,7 +358,7 @@ async def client(addr): new_tr.close() - with self.tcp_server(serve) as srv: + with self.tcp_server(serve, timeout=self.TIMEOUT) as srv: self.loop.run_until_complete( asyncio.wait_for(client(srv.addr), loop=self.loop, timeout=self.TIMEOUT)) @@ -412,7 +412,8 @@ async def main(proto, on_con, on_eof, on_con_lost): new_tr = await self.loop.start_tls( tr, proto, server_context, - server_side=True) + server_side=True, + ssl_handshake_timeout=self.TIMEOUT) await on_eof await on_con_lost @@ -429,7 +430,8 @@ async def run_main(): lambda: proto, '127.0.0.1', 0) addr = server.sockets[0].getsockname() - with self.tcp_client(lambda sock: client(sock, addr)): + with self.tcp_client(lambda sock: client(sock, addr), + timeout=self.TIMEOUT): await asyncio.wait_for( main(proto, on_con, on_eof, on_con_lost), loop=self.loop, timeout=self.TIMEOUT)