Add socket test using fibers

symmetryinvestments · Aug 21, 2024 · 517b863 · 517b863
1 parent 3467493
commit 517b863
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/tests/ut/concurrency/io.d b/tests/ut/concurrency/io.d
@@ -49,7 +49,7 @@ unittest {
 }
 
 @safe
-@("acceptAsync.connectAsync")
+@("acceptAsync.connectAsync.basic")
 unittest {
     import concurrency.io.socket;
     auto fd = listenTcp("127.0.0.1", 0);
@@ -70,3 +70,28 @@ unittest {
 	closeSocket(socket);
 	closeSocket(fd);
 }
+
+@safe
+@("acceptAsync.connectAsync.fiber")
+unittest {
+    import concurrency.io.socket;
+    import concurrency.fiber;
+
+    auto io = IOContext.construct(12);
+    io.run(fiber({
+        auto fd = listenTcp("127.0.0.1", 0);
+        auto socket = tcpSocket();
+        auto port = fd.getPort();
+
+        auto result = whenAll(
+            acceptAsync(fd),
+            connectAsync(socket, "127.0.0.1", port),
+        ).yield().value;
+
+        auto client = result[0];
+
+        closeSocket(client.fd);
+        closeSocket(socket);
+        closeSocket(fd);
+    })).syncWait.assumeOk;
+}