cranelift/tests/all/piped_tests.rs

#![cfg(not(miri))]

use super::cli_tests::get_wasmtime_command;
use anyhow::Result;
use std::process::Stdio;

pub fn run_wasmtime_piped(component_path: &str) -> Result<()> {
    let mut producer = get_wasmtime_command()?
        .arg("run")
        .arg("-Wcomponent-model")
        .arg("--env")
        .arg("PIPED_SIDE=PRODUCER")
        .arg(component_path)
        .stdout(Stdio::piped())
        .spawn()?;

    let mut consumer = get_wasmtime_command()?
        .arg("run")
        .arg("-Wcomponent-model")
        .arg("--env")
        .arg("PIPED_SIDE=CONSUMER")
        .arg(component_path)
        .stdin(producer.stdout.take().unwrap())
        .spawn()?;

    let producer = producer.wait()?;
    if !producer.success() {
        // make sure the consumer gets killed off
        if consumer.try_wait().is_err() {
            consumer.kill().expect("Failed to kill consumer");
        }

        panic!("Producer failed");
    }

    assert!(consumer.wait()?.success(), "Consumer failed");

    Ok(())
}

mod test_programs {
    use super::run_wasmtime_piped;
    use test_programs_artifacts::*;

    macro_rules! assert_test_exists {
        ($name:ident) => {
            #[allow(unused_imports)]
            use self::$name as _;
        };
    }
    foreach_piped!(assert_test_exists);

    // Below here is mechanical: there should be one test for every binary in
    // wasi-tests.
    #[test]
    fn piped_simple() {
        run_wasmtime_piped(PIPED_SIMPLE_COMPONENT).unwrap()
    }

    #[test]
    fn piped_multiple() {
        run_wasmtime_piped(PIPED_MULTIPLE_COMPONENT).unwrap()
    }

    #[test]
    fn piped_polling() {
        run_wasmtime_piped(PIPED_POLLING_COMPONENT).unwrap()
    }
}