Skip to content

Commit f9701cb

Browse files
committed
Add a test for mixed sock_send and sock_sendfile
1 parent 099dc56 commit f9701cb

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

Lib/test/test_asyncio/test_unix_events.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,20 @@ def test_sendfile_zero_size(self):
561561
self.assertEqual(ret, 0)
562562
self.assertEqual(self.file.tell(), 0)
563563

564+
def test_mix_sendfile_and_regular_send(self):
565+
buf = b'1234567890' * 1024 * 1024 # 10 MB
566+
sock, proto = self.prepare()
567+
self.run_loop(self.loop.sock_sendall(sock, buf))
568+
ret = self.run_loop(self.loop.sock_sendfile(sock, self.file))
569+
self.run_loop(self.loop.sock_sendall(sock, buf))
570+
sock.close()
571+
self.run_loop(proto.wait_closed())
572+
573+
self.assertEqual(ret, len(self.DATA))
574+
expected = buf + self.DATA + buf
575+
self.assertEqual(proto.data, expected)
576+
self.assertEqual(self.file.tell(), len(self.DATA))
577+
564578

565579
class UnixReadPipeTransportTests(test_utils.TestCase):
566580

0 commit comments

Comments
 (0)