92 lines · 3.7 KB · Rust
RawDownload
pub mod plugin;pub use plugin::*; #[cfg(test)]mod tests {    use super::*;    use pollster::FutureExt as _;     #[test]    pub fn test_plugin() {        pub struct TestPlugin {            noop: WasiFn<(), ()>,            constant: WasiFn<(), u32>,            identity: WasiFn<u32, u32>,            add: WasiFn<(u32, u32), u32>,            swap: WasiFn<(u32, u32), (u32, u32)>,            sort: WasiFn<Vec<u32>, Vec<u32>>,            print: WasiFn<String, ()>,            and_back: WasiFn<u32, u32>,            imports: WasiFn<u32, u32>,            half_async: WasiFn<u32, u32>,            echo_async: WasiFn<String, String>,        }         async {            let mut runtime = PluginBuilder::new_default()                .unwrap()                .host_function("mystery_number", |input: u32| input + 7)                .unwrap()                .host_function("import_noop", |_: ()| ())                .unwrap()                .host_function("import_identity", |input: u32| input)                .unwrap()                .host_function("import_swap", |(a, b): (u32, u32)| (b, a))                .unwrap()                .host_function_async("import_half", |a: u32| async move { a / 2 })                .unwrap()                .host_function_async("command_async", |command: String| async move {                    let mut args = command.split(' ');                    let command = args.next().unwrap();                    smol::process::Command::new(command)                        .args(args)                        .output()                        .await                        .ok()                        .map(|output| output.stdout)                })                .unwrap()                .init(PluginBinary::Wasm(                    include_bytes!("../../../plugins/bin/test_plugin.wasm").as_ref(),                ))                .await                .unwrap();             let plugin = TestPlugin {                noop: runtime.function("noop").unwrap(),                constant: runtime.function("constant").unwrap(),                identity: runtime.function("identity").unwrap(),                add: runtime.function("add").unwrap(),                swap: runtime.function("swap").unwrap(),                sort: runtime.function("sort").unwrap(),                print: runtime.function("print").unwrap(),                and_back: runtime.function("and_back").unwrap(),                imports: runtime.function("imports").unwrap(),                half_async: runtime.function("half_async").unwrap(),                echo_async: runtime.function("echo_async").unwrap(),            };             let unsorted = vec![1, 3, 4, 2, 5];            let sorted = vec![1, 2, 3, 4, 5];             runtime.call(&plugin.noop, ()).await.unwrap();            assert_eq!(runtime.call(&plugin.constant, ()).await.unwrap(), 27);            assert_eq!(runtime.call(&plugin.identity, 58).await.unwrap(), 58);            assert_eq!(runtime.call(&plugin.add, (3, 4)).await.unwrap(), 7);            assert_eq!(runtime.call(&plugin.swap, (1, 2)).await.unwrap(), (2, 1));            assert_eq!(runtime.call(&plugin.sort, unsorted).await.unwrap(), sorted);            runtime.call(&plugin.print, "Hi!".into()).await.unwrap();            assert_eq!(runtime.call(&plugin.and_back, 1).await.unwrap(), 8);            assert_eq!(runtime.call(&plugin.imports, 1).await.unwrap(), 8);            assert_eq!(runtime.call(&plugin.half_async, 4).await.unwrap(), 2);            assert_eq!(                runtime                    .call(&plugin.echo_async, "eko".into())                    .await                    .unwrap(),                "eko\n"            );        }        .block_on()    }}