mirror of
https://github.com/python/cpython.git
synced 2025-08-27 20:25:18 +00:00
Make asyncio stream sendfile fail on error (was hang) (GH-14025)
This commit is contained in:
parent
ff6bb0aa95
commit
0d1942774a
1 changed files with 8 additions and 5 deletions
|
@ -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():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue