add unit test for io redirection

Signed-off-by: David Justice <[email protected]>

Cargo.toml

@@ -39,7 +39,9 @@ flate2 = "1"
 tar = "0.4"
 
 [dev-dependencies]
-tempdir = { workspace = true}
+tempdir = { workspace = true }
+tempfile = { workspace = true }
+rand = { worspace = true }
 
 [features]
 default = ["keyvalue", "distributed-locking", "messaging", "runtime-configs", "sql", "http-server", "http-client"]
@@ -83,6 +85,7 @@ tokio = { version = "1", features = ["full"] }
 tracing = { version = "0.1", features = ["log"] }
 toml = "0.5"
 tempdir = "0.3"
+tempfile = "3.4"
 log = { version = "0.4", default-features = false }
 env_logger = "0.9"
 as-any = "0.3"

src/commands/run.rs

@@ -353,3 +353,53 @@ fn maybe_add_named_capability_to_store(
 
     Ok(())
 }
+
+#[cfg(test)]
+mod unittest {
+    use std::path::{Path, PathBuf};
+    use std::fs::File;
+    use tempfile::tempdir;
+    use tokio::fs;
+    use rand::Rng;
+    use slight_runtime::IORedirects;
+    use crate::commands::run::{handle_run, RunArgs};
+
+    #[tokio::test]
+    async fn test_handle_run_with_io() -> anyhow::Result<()> {
+        let module = "./src/commands/test/io-test.wasm";
+        assert!(Path::new(module).exists());
+        let slightfile = "./src/commands/test/slightfile.toml";
+        assert!(Path::new(slightfile).exists());
+
+        let tmp_dir = tempdir()?;
+        let stdin_path = tmp_dir.path().join("stdin");
+        let stdout_path = tmp_dir.path().join("stdout");
+        let stderr_path = tmp_dir.path().join("stderr");
+
+        let canary: String = rand::thread_rng()
+            .gen_ascii_chars()
+            .take(7)
+            .map(char::from)
+            .collect();
+        fs::write(&stdin_path, &canary).await?;
+        let _ = File::create(&stdout_path)?;
+        let _ = File::create(&stderr_path)?;
+
+        let args = RunArgs{
+            module: PathBuf::from(module),
+            slightfile: PathBuf::from(slightfile),
+            io_redirects: Some(IORedirects{
+                stdin_path: PathBuf::from(&stdin_path),
+                stdout_path: PathBuf::from(&stdout_path),
+                stderr_path: PathBuf::from(&stderr_path),
+            })
+        };
+
+        let _ = handle_run(args).await?;
+        let stdout_output = fs::read_to_string(&stdout_path).await?;
+        assert_eq!(stdout_output, canary);
+        let stderr_output = fs::read_to_string(&stderr_path).await?;
+        assert_eq!(stderr_output, format!("error: {canary}"));
+        Ok(())
+    }
+}

src/commands/test/slightfile.toml

@@ -0,0 +1,5 @@
+specversion = "0.2"
+
+[[capability]]
+resource = "configs.envvars"
+name = "env-configs"

tests/Cargo.toml

@@ -17,6 +17,7 @@ tokio = { workspace = true }
 anyhow = { workspace = true }
 mosquitto-rs = { version = "0.4.0", features = ["vendored-openssl", "vendored-mosquitto"] }
 tempdir = { workspace = true }
+rand = { workspace = true }
 
 [target.'cfg(unix)'.dev-dependencies]
 signal-child = "1"

tests/build.rs

@@ -7,6 +7,7 @@ const WIT_DIRECTORY: &str = "wit/*";
 const KEYVALUE_TEST_PATH: &str = "./keyvalue-test";
 const HTTP_TEST_PATH: &str = "./http-test";
 const CONFIGS_TEST_PATH: &str = "./configs-test";
+const IO_TEST_PATH: &str = "./io-test";
 
 fn main() {
     println!("cargo:rerun-if-changed=build.rs");
@@ -28,6 +29,7 @@ fn main() {
             std::process::exit(1);
         }
         // Build test wasm modules
+        cargo_wasi_build(IO_TEST_PATH);
         cargo_wasi_build(KEYVALUE_TEST_PATH);
         cargo_wasi_build(HTTP_TEST_PATH);
         cargo_wasi_build(CONFIGS_TEST_PATH);

tests/io-test/Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+name = "io-test"
+version = "0.1.0"
+edition = "2021"
+authors = [ "DeisLabs Engineering Team" ]
+
+[[bin]]
+name = "io-test"
+test = false
+
+[dependencies]
+anyhow = "1"
+
+[workspace]

tests/io-test/slightfile.toml

@@ -0,0 +1 @@
+specversion = "0.2"

tests/io-test/src/main.rs

@@ -0,0 +1,11 @@
+use std::io;
+use std::io::Write;
+use anyhow::Result;
+
+fn main() -> Result<()> {
+    let mut buffer = String::new();
+    io::stdin().read_line(&mut buffer)?;
+    io::stdout().write_all(buffer.as_bytes())?;
+    io::stderr().write_all(format!("error: {buffer}").as_bytes())?;
+    Ok(())
+}