add integration tests for TcpStream::connect

Author: metent

examples/tcp_stream_client.rs

@@ -0,0 +1,25 @@
+use wstd::io::{self, AsyncRead, AsyncWrite};
+use wstd::net::TcpStream;
+
+#[wstd::main]
+async fn main() -> io::Result<()> {
+    let mut args = std::env::args();
+
+    let _ = args.next();
+
+    let addr = args.next().ok_or_else(|| {
+        io::Error::new(
+            std::io::ErrorKind::InvalidInput,
+            "address argument required",
+        )
+    })?;
+
+    let mut stream = TcpStream::connect(addr).await?;
+
+    stream.write_all(b"ping\n").await?;
+
+    let mut reply = Vec::new();
+    stream.read_to_end(&mut reply).await?;
+
+    Ok(())
+}

test-programs/artifacts/tests/tcp_stream_client.rs

@@ -0,0 +1,48 @@
+use anyhow::{Context, Result};
+use std::{
+    net::TcpListener,
+    process::{Command, Stdio},
+};
+
+#[test_log::test]
+fn tcp_stream_client() -> Result<()> {
+    let server = TcpListener::bind("127.0.0.1:8081").context("binding temporary test server")?;
+    let addr = server
+        .local_addr()
+        .context("getting local listener address")?;
+
+    let child = Command::new("wasmtime")
+        .arg("run")
+        .arg("-Sinherit-network")
+        .arg(test_programs_artifacts::TCP_STREAM_CLIENT)
+        .arg(addr.to_string())
+        .stdout(Stdio::piped())
+        .spawn()
+        .context("spawning wasmtime component")?;
+
+    let (mut server_stream, _addr) = server
+        .accept()
+        .context("accepting TCP connection from component")?;
+
+    let mut buf = [0u8; 5];
+    std::io::Read::read_exact(&mut server_stream, &mut buf).context("reading ping message")?;
+    assert_eq!(&buf, b"ping\n", "expected ping from component");
+
+    std::io::Write::write_all(&mut server_stream, b"pong\n").context("writing reply")?;
+    std::io::Write::flush(&mut server_stream).context("flushing")?;
+
+    std::net::TcpStream::shutdown(&mut server_stream, std::net::Shutdown::Both)
+        .context("shutting down connection")?;
+
+    let output = child
+        .wait_with_output()
+        .context("waiting for component exit")?;
+
+    assert!(
+        output.status.success(),
+        "\nComponent exited abnormally (stderr:\n{})",
+        String::from_utf8_lossy(&output.stderr)
+    );
+
+    Ok(())
+}

test-programs/src/bin/tcp_stream_client.rs

@@ -0,0 +1 @@
+include!("../../../examples/tcp_stream_client.rs");