test(net): tcp client & server

Author: Berrysoft

Cargo.toml

@@ -43,9 +43,12 @@ async-io = { workspace = true, optional = true }
 futures-util = { workspace = true, optional = true }
 
 [dev-dependencies]
-compio = { workspace = true, default-features = false, features = ["fs"] }
+compio = { workspace = true, default-features = false, features = [
+    "fs",
+    "net",
+] }
 
-tokio = { workspace = true, features = ["rt", "macros", "fs", "io-util"] }
+tokio = { workspace = true, features = ["rt", "macros", "net", "io-util"] }
 futures-executor = { workspace = true }
 
 [features]
@@ -54,3 +57,7 @@ polling = ["compio/polling"]
 tokio = ["dep:tokio"]
 futures = ["dep:async-io", "dep:futures-util"]
 enable_log = ["compio-log/enable_log"]
+
+[[test]]
+name = "net"
+required-features = ["tokio"]

tests/net.rs

@@ -0,0 +1,73 @@
+use compio::runtime::Runtime;
+use compio_compat::{RuntimeCompat, TokioAdapter};
+
+#[tokio::test]
+async fn compio_client() {
+    use compio::net::TcpStream;
+    use tokio::net::TcpListener;
+
+    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
+    let addr = listener.local_addr().unwrap();
+    let server = async {
+        use tokio::io::{AsyncReadExt, AsyncWriteExt};
+
+        let (mut stream, _) = listener.accept().await.unwrap();
+        let mut buffer = [0u8; 12];
+        stream.read_exact(&mut buffer).await.unwrap();
+        stream.write_all(&buffer).await.unwrap();
+        stream.shutdown().await.unwrap();
+    };
+
+    let runtime = RuntimeCompat::<TokioAdapter>::new(Runtime::new().unwrap()).unwrap();
+    let client = runtime.execute(async {
+        use compio::io::{AsyncReadExt, AsyncWrite, AsyncWriteExt};
+
+        let mut stream = TcpStream::connect(addr).await.unwrap();
+        stream.write_all(b"hello world!").await.unwrap();
+        stream.shutdown().await.unwrap();
+        let buffer = [0u8; 12];
+        let (_, buffer) = stream.read_exact(buffer).await.unwrap();
+        assert_eq!(&buffer, b"hello world!");
+        stream.close().await.unwrap();
+    });
+
+    tokio::join!(server, client);
+}
+
+#[tokio::test]
+async fn compio_server() {
+    use compio::net::TcpListener;
+    use tokio::net::TcpStream;
+
+    let runtime = RuntimeCompat::<TokioAdapter>::new(Runtime::new().unwrap()).unwrap();
+    let listener = runtime
+        .execute(TcpListener::bind("127.0.0.1:0"))
+        .await
+        .unwrap();
+    let addr = listener.local_addr().unwrap();
+    let server = runtime.execute(async {
+        use compio::io::{AsyncRead, AsyncReadExt, AsyncWrite, AsyncWriteExt};
+
+        let (mut stream, _) = listener.accept().await.unwrap();
+        let buffer = [0u8; 12];
+        let (_, buffer) = stream.read_exact(buffer).await.unwrap();
+        stream.write_all(buffer).await.unwrap();
+        stream.shutdown().await.unwrap();
+        stream.read([0u8]).await.unwrap();
+        stream.close().await.unwrap();
+    });
+
+    let client = async {
+        use tokio::io::{AsyncReadExt, AsyncWriteExt};
+
+        let mut stream = TcpStream::connect(addr).await.unwrap();
+        stream.write_all(b"hello world!").await.unwrap();
+        stream.shutdown().await.unwrap();
+        let mut buffer = [0u8; 12];
+        stream.read_exact(&mut buffer).await.unwrap();
+        assert_eq!(&buffer, b"hello world!");
+        stream.shutdown().await.unwrap();
+    };
+
+    tokio::join!(server, client);
+}