From e409043556c6fe54efc213daa083d679a37c2a03 Mon Sep 17 00:00:00 2001
From: Helmut Grohne <helmut@subdivi.de>
Date: Tue, 28 May 2024 09:28:23 +0200
Subject: support sendfile in async_copyfd

---
 tests/test_simple.py | 47 +++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 39 insertions(+), 8 deletions(-)

(limited to 'tests/test_simple.py')

diff --git a/tests/test_simple.py b/tests/test_simple.py
index 456e088..0c4e2b9 100644
--- a/tests/test_simple.py
+++ b/tests/test_simple.py
@@ -68,6 +68,17 @@ class IDAllocationTest(unittest.TestCase):
 
 
 class AsnycioTest(unittest.IsolatedAsyncioTestCase):
+    async def asyncSetUp(self) -> None:
+        self.loop = asyncio.get_running_loop()
+
+    def wait_readable(self, rfd: int) -> asyncio.Future[None]:
+        fut = self.loop.create_future()
+        def callback() -> None:
+            self.loop.remove_reader(rfd)
+            fut.set_result(None)
+        self.loop.add_reader(rfd, callback)
+        return fut
+
     async def test_eventfd(self) -> None:
         with linuxnamespaces.EventFD(
             1, linuxnamespaces.EventFDFlags.NONBLOCK
@@ -93,7 +104,33 @@ class AsnycioTest(unittest.IsolatedAsyncioTestCase):
             await set_ready()
             await asyncio.wait_for(fut, 10)
 
-    async def test_copyfd(self) -> None:
+    async def test_copyfd_file_sock(self) -> None:
+        sock1, sock2 = socket.socketpair()
+        with sock1, sock2, linuxnamespaces.FileDescriptor(
+            os.open("/etc/passwd", os.O_RDONLY)
+        ) as rfd:
+            fut = asyncio.ensure_future(
+                linuxnamespaces.async_copyfd(rfd, sock1.fileno(), 999)
+            )
+            await self.wait_readable(sock2.fileno())
+            self.assertGreater(len(sock2.recv(999)), 0)
+            self.assertTrue(fut.done())
+            self.assertGreater(await fut, 0)
+
+    async def test_copyfd_file_pipe(self) -> None:
+        rfdp, wfdp = linuxnamespaces.FileDescriptor.pipe(blocking=False)
+        with rfdp, wfdp, linuxnamespaces.FileDescriptor(
+            os.open("/etc/passwd", os.O_RDONLY)
+        ) as rfd:
+            fut = asyncio.ensure_future(
+                linuxnamespaces.async_copyfd(rfd, wfdp, 999)
+            )
+            await self.wait_readable(rfdp)
+            self.assertGreater(len(os.read(rfdp, 999)), 0)
+            self.assertTrue(fut.done())
+            self.assertGreater(await fut, 0)
+
+    async def test_copyfd_pipe_pipe(self) -> None:
         rfd1, wfd1 = linuxnamespaces.FileDescriptor.pipe(blocking=False)
         rfd2, wfd2 = linuxnamespaces.FileDescriptor.pipe(blocking=False)
         with wfd2, rfd2, rfd1:
@@ -104,13 +141,7 @@ class AsnycioTest(unittest.IsolatedAsyncioTestCase):
                 os.write(wfd1, b"hello")
                 await asyncio.sleep(0.000001)  # Let the loop run
                 os.write(wfd1, b"world")
-                loop = asyncio.get_running_loop()
-                fut2 = loop.create_future()
-                def callback() -> None:
-                    loop.remove_reader(rfd2)
-                    fut2.set_result(None)
-                loop.add_reader(rfd2, callback)
-                await fut2
+                await self.wait_readable(rfd2)
                 self.assertEqual(os.read(rfd2, 11), b"helloworld")
                 self.assertFalse(fut.done())
             await asyncio.sleep(0.000001)  # Let the loop run
-- 
cgit v1.2.3