Skip to content

Commit 0d19427

Browse files
authored
Make asyncio stream sendfile fail on error (was hang) (GH-14025)
1 parent ff6bb0a commit 0d19427

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

Lib/test/test_asyncio/test_streams.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1656,22 +1656,25 @@ def test_sendfile(self):
16561656

16571657
async def serve_callback(stream):
16581658
data = await stream.readline()
1659-
self.assertEqual(data, b'begin\n')
1659+
await stream.write(b'ack-' + data)
16601660
data = await stream.readline()
1661-
self.assertEqual(data, b'data\n')
1661+
await stream.write(b'ack-' + data)
16621662
data = await stream.readline()
1663-
self.assertEqual(data, b'end\n')
1664-
await stream.write(b'done\n')
1663+
await stream.write(b'ack-' + data)
16651664
await stream.close()
16661665

16671666
async def do_connect(host, port):
16681667
stream = await asyncio.connect(host, port)
16691668
await stream.write(b'begin\n')
1669+
data = await stream.readline()
1670+
self.assertEqual(b'ack-begin\n', data)
16701671
with open(support.TESTFN, 'rb') as fp:
16711672
await stream.sendfile(fp)
1673+
data = await stream.readline()
1674+
self.assertEqual(b'ack-data\n', data)
16721675
await stream.write(b'end\n')
16731676
data = await stream.readline()
1674-
self.assertEqual(data, b'done\n')
1677+
self.assertEqual(data, b'ack-end\n')
16751678
await stream.close()
16761679

16771680
async def test():

0 commit comments

Comments
 (0)