Skip to content

Commit f8fa0cf

Browse files
committed
Resolve flake8 warnings in test utilities
1 parent 20189ed commit f8fa0cf

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

tests/test_tcp.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)