test-programs

Author: pchickey

Cargo.lock

@@ -1,21 +1,26 @@
 [workspace]
-members = ["wasmtime", "cli"]
+members = ["cli", "wasmtime", "test-programs", "test-programs/artifacts"]
 
 [workspace.package]
 version = "0.1.0"
 edition = "2024"
 
 [workspace.dependencies]
+clap = { version = "4", features = ["derive"] }
 component-init = { path = "." }
 component-init-wasmtime = { path = "./wasmtime" }
 anyhow = "1.0.89"
 async-trait = "0.1.83"
 futures = "0.3.31"
+tokio = { version = "1", features = ["full"] }
+test-programs = { path = "./test-programs" }
+test-programs-artifacts = { path = "./test-programs/artifacts" }
 wasm-encoder = { version = "0.235.0", features = ["wasmparser"] }
 wasmparser = "0.235.0"
 wasmtime = "34.0.1"
-wasmtime-wasi = "34.0.1"
-wasmtime-wasi-http = "34.0.1"
+# wasmtime-wasi = "34.0.1"
+# wasmtime-wasi-http = "34.0.1"
+wit-bindgen = "0.43.0"
 
 [package]
 name = "component-init"

Cargo.toml

@@ -9,6 +9,6 @@ path = "src/main.rs"
 
 [dependencies]
 anyhow.workspace = true
-clap = { version = "4", features = ["derive"] }
+clap.workspace = true
 component-init-wasmtime.workspace = true
-tokio = { version = "1", features = ["full"] }
+tokio.workspace = true

cli/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "test-programs"
+version.workspace = true
+edition.workspace = true
+publish = false
+
+[dependencies]
+wit-bindgen.workspace = true

test-programs/Cargo.toml

@@ -0,0 +1,11 @@
+[package]
+name = "test-programs-artifacts"
+version = "0.1.0"
+edition.workspace = true
+publish = false
+
+[dependencies]
+
+[build-dependencies]
+cargo_metadata = "0.18"
+heck = "0.5"

test-programs/artifacts/Cargo.toml

@@ -0,0 +1,62 @@
+use heck::ToShoutySnakeCase;
+use std::env::var_os;
+use std::path::PathBuf;
+use std::process::Command;
+
+fn main() {
+    let out_dir = PathBuf::from(var_os("OUT_DIR").expect("OUT_DIR env var exists"));
+
+    let meta = cargo_metadata::MetadataCommand::new()
+        .exec()
+        .expect("cargo metadata");
+    let test_programs_meta = meta
+        .packages
+        .iter()
+        .find(|p| p.name == "test-programs")
+        .expect("test-programs is in cargo metadata");
+    let test_programs_root = test_programs_meta.manifest_path.parent().unwrap();
+    println!(
+        "cargo:rerun-if-changed={}",
+        test_programs_root.as_os_str().to_str().unwrap()
+    );
+
+    let status = Command::new("cargo")
+        .arg("build")
+        .arg("--target=wasm32-wasip2")
+        .arg("--package=test-programs")
+        .env("CARGO_TARGET_DIR", &out_dir)
+        .env("CARGO_PROFILE_DEV_DEBUG", "2")
+        .env("RUSTFLAGS", rustflags())
+        .env_remove("CARGO_ENCODED_RUSTFLAGS")
+        .status()
+        .expect("cargo build test programs");
+    assert!(status.success());
+
+    let mut generated_code = "// THIS FILE IS GENERATED CODE\n".to_string();
+
+    for binary in test_programs_meta
+        .targets
+        .iter()
+        .filter(|t| t.kind == ["bin"])
+    {
+        let component_path = out_dir
+            .join("wasm32-wasip2")
+            .join("debug")
+            .join(format!("{}.wasm", binary.name));
+
+        let const_name = binary.name.to_shouty_snake_case();
+        generated_code += &format!(
+            "pub const {const_name}: &str = {:?};\n",
+            component_path.as_os_str().to_str().expect("path is str")
+        );
+    }
+
+    std::fs::write(out_dir.join("gen.rs"), generated_code).unwrap();
+}
+
+fn rustflags() -> &'static str {
+    match option_env!("RUSTFLAGS") {
+        Some(s) if s.contains("-D warnings") => "-D warnings",
+        _ => "",
+    }
+}

test-programs/artifacts/build.rs

@@ -0,0 +1 @@
+include!(concat!(env!("OUT_DIR"), "/gen.rs"));

test-programs/artifacts/src/lib.rs

@@ -0,0 +1,26 @@
+use std::cell::Cell;
+
+wit_bindgen::generate!({
+    inline: r"
+    package this:wit;
+    world w {
+        export component-init: func();
+    }"
+});
+
+const IS_INITIALIZED: Cell<bool> = Cell::new(false);
+
+struct S;
+impl Guest for S {
+    fn component_init() {
+        let before = IS_INITIALIZED.replace(true);
+        assert!(!before, "component should only be initialized once");
+    }
+}
+
+export!(S);
+
+fn main() {
+    let initialized = IS_INITIALIZED.get();
+    assert!(initialized, "component was not initialized")
+}

test-programs/src/bin/test.rs

@@ -8,5 +8,9 @@ anyhow.workspace = true
 async-trait.workspace = true
 component-init.workspace = true
 wasmtime.workspace = true
-wasmtime-wasi.workspace = true
-wasmtime-wasi-http.workspace = true
+# wasmtime-wasi.workspace = true
+# wasmtime-wasi-http.workspace = true
+
+[dev-dependencies]
+tokio.workspace = true
+test-programs-artifacts.workspace = true

wasmtime/Cargo.toml

@@ -0,0 +1,15 @@
+use anyhow::Result;
+use test_programs_artifacts::TEST;
+
+#[tokio::test]
+async fn initialize_test_component() -> Result<()> {
+    let component = std::fs::read(TEST)?;
+
+    // TODO: execute `component`. it should trap.
+
+    let out = component_init_wasmtime::initialize(&component).await?;
+
+    // TODO: execute `out`. it should not trap.
+
+    Ok(())
+}