Commit 0d194277 authored by Andrew Svetlov's avatar Andrew Svetlov Committed by GitHub

Make asyncio stream sendfile fail on error (was hang) (GH-14025)

parent ff6bb0aa
......@@ -1656,22 +1656,25 @@ os.close(fd)
async def serve_callback(stream):
data = await stream.readline()
self.assertEqual(data, b'begin\n')
await stream.write(b'ack-' + data)
data = await stream.readline()
self.assertEqual(data, b'data\n')
await stream.write(b'ack-' + data)
data = await stream.readline()
self.assertEqual(data, b'end\n')
await stream.write(b'done\n')
await stream.write(b'ack-' + data)
await stream.close()
async def do_connect(host, port):
stream = await asyncio.connect(host, port)
await stream.write(b'begin\n')
data = await stream.readline()
self.assertEqual(b'ack-begin\n', data)
with open(support.TESTFN, 'rb') as fp:
await stream.sendfile(fp)
data = await stream.readline()
self.assertEqual(b'ack-data\n', data)
await stream.write(b'end\n')
data = await stream.readline()
self.assertEqual(data, b'done\n')
self.assertEqual(data, b'ack-end\n')
await stream.close()
async def test():
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment