@@ -1287,9 +1287,9 @@ async def handle_client(reader, writer):
12871287 # Upgrade to TLS (server side)
12881288 try :
12891289 # We need the wait_for because the broken version hangs here
1290- await asyncio .wait_for (writer . start_tls ( server_context ),
1291- timeout = 2
1292- )
1290+ await asyncio .wait_for (
1291+ writer . start_tls ( server_context ),
1292+ timeout = 2 )
12931293 self .assertIsNotNone (writer .get_extra_info ('sslcontext' ))
12941294 except asyncio .TimeoutError :
12951295 self .assertIsNotNone (writer .get_extra_info ('sslcontext' ))
@@ -1311,7 +1311,7 @@ async def client(addr):
13111311 # Read buffered data before TLS
13121312 buffered = await reader .readexactly (len (BUFFERED_MSG ))
13131313 self .assertEqual (buffered , BUFFERED_MSG ,
1314- "Client didn't receive buffered data before TLS upgrade " )
1314+ "Wrong pre-TLS buffered data from server " )
13151315
13161316 # Write before TLS upgrade
13171317 writer .write (HELLO_MSG )
@@ -1320,17 +1320,17 @@ async def client(addr):
13201320 # Upgrade to TLS
13211321 try :
13221322 # We need the wait_for because the broken version hangs here
1323- await asyncio .wait_for (writer . start_tls ( client_context ),
1324- timeout = 2
1325- )
1323+ await asyncio .wait_for (
1324+ writer . start_tls ( client_context ),
1325+ timeout = 2 )
13261326 self .assertIsNotNone (writer .get_extra_info ('sslcontext' ))
13271327 except asyncio .TimeoutError :
13281328 self .assertIsNotNone (writer .get_extra_info ('sslcontext' ))
13291329
13301330 # Verify communication over TLS
13311331 tls_data = await reader .readexactly (2 )
13321332 self .assertEqual (tls_data , b'OK' ,
1333- "Client didn't receive TLS response correctly " )
1333+ "Wrong data from server after TLS upgrade " )
13341334
13351335 # Continue over TLS
13361336 writer .write (HELLO_MSG )
0 commit comments