Add cli_exit_with_code test

bytecodealliance · Oct 6, 2024 · 4fcfe6f · 4fcfe6f
1 parent 2f48c77
commit 4fcfe6f
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/crates/test-programs/src/bin/cli_exit_with_code.rs b/crates/test-programs/src/bin/cli_exit_with_code.rs
@@ -0,0 +1,5 @@
+use test_programs::wasi::cli::exit::exit_with_code;
+
+fn main() {
+    exit_with_code(42);
+}
diff --git a/crates/test-programs/src/lib.rs b/crates/test-programs/src/lib.rs
@@ -20,6 +20,7 @@ wit_bindgen::generate!({
         "../wasi-keyvalue/wit",
     ],
     world: "wasmtime:test/test",
+    features: ["cli-exit-with-code"],
     generate_all,
 });
 

diff --git a/tests/all/cli_tests.rs b/tests/all/cli_tests.rs
@@ -1326,6 +1326,16 @@ mod test_programs {
         Ok(())
     }
 
+    #[test]
+    fn cli_exit_with_code() -> Result<()> {
+        let output = get_wasmtime_command()?
+            .args(&["run", "-Wcomponent-model", "-Scli-exit-with-code", CLI_EXIT_WITH_CODE_COMPONENT])
+            .output()?;
+        assert!(!output.status.success());
+        assert_eq!(output.status.code(), Some(42));
+        Ok(())
+    }
+
     #[test]
     fn cli_exit_panic() -> Result<()> {
         let output = get_wasmtime_command()?