diff options
| -rw-r--r-- | Lib/test/test_asyncio/test_unix_events.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/Lib/test/test_asyncio/test_unix_events.py b/Lib/test/test_asyncio/test_unix_events.py index 62545c0..31e7100 100644 --- a/Lib/test/test_asyncio/test_unix_events.py +++ b/Lib/test/test_asyncio/test_unix_events.py @@ -449,10 +449,12 @@ class SelectorEventLoopUnixSockSendfileTests(test_utils.TestCase): self.data = bytearray() self.fut = loop.create_future() self.transport = None + self._ready = loop.create_future() def connection_made(self, transport): self.started = True self.transport = transport + self._ready.set_result(None) def data_received(self, data): self.data.extend(data) @@ -503,13 +505,11 @@ class SelectorEventLoopUnixSockSendfileTests(test_utils.TestCase): server = self.run_loop(self.loop.create_server( lambda: proto, sock=srv_sock)) self.run_loop(self.loop.sock_connect(sock, (support.HOST, port))) + self.run_loop(proto._ready) def cleanup(): - if proto.transport is not None: - # can be None if the task was cancelled before - # connection_made callback - proto.transport.close() - self.run_loop(proto.wait_closed()) + proto.transport.close() + self.run_loop(proto.wait_closed()) server.close() self.run_loop(server.wait_closed()) |