1pub mod wasi;
 2use pollster::FutureExt as _;
 3pub use wasi::*;
 4
 5#[cfg(test)]
 6mod tests {
 7    use super::*;
 8
 9    #[test]
10    pub fn test_plugin() {
11        pub struct TestPlugin {
12            noop: WasiFn<(), ()>,
13            constant: WasiFn<(), u32>,
14            identity: WasiFn<u32, u32>,
15            add: WasiFn<(u32, u32), u32>,
16            swap: WasiFn<(u32, u32), (u32, u32)>,
17            sort: WasiFn<Vec<u32>, Vec<u32>>,
18            print: WasiFn<String, ()>,
19            and_back: WasiFn<u32, u32>,
20            imports: WasiFn<u32, u32>,
21        }
22
23        async {
24            let mut runtime = WasiPluginBuilder::new_with_default_ctx()
25                .unwrap()
26                .host_function("mystery_number", |input: u32| input + 7)
27                .unwrap()
28                .host_function("import_noop", |_: ()| ())
29                .unwrap()
30                .host_function("import_identity", |input: u32| input)
31                .unwrap()
32                .host_function("import_swap", |(a, b): (u32, u32)| (b, a))
33                .unwrap()
34                .init(include_bytes!("../../../plugins/bin/test_plugin.wasm"))
35                .await
36                .unwrap();
37
38            let plugin = TestPlugin {
39                noop: runtime.function("noop").unwrap(),
40                constant: runtime.function("constant").unwrap(),
41                identity: runtime.function("identity").unwrap(),
42                add: runtime.function("add").unwrap(),
43                swap: runtime.function("swap").unwrap(),
44                sort: runtime.function("sort").unwrap(),
45                print: runtime.function("print").unwrap(),
46                and_back: runtime.function("and_back").unwrap(),
47                imports: runtime.function("imports").unwrap(),
48            };
49
50            let unsorted = vec![1, 3, 4, 2, 5];
51            let sorted = vec![1, 2, 3, 4, 5];
52
53            assert_eq!(runtime.call(&plugin.noop, ()).await.unwrap(), ());
54            assert_eq!(runtime.call(&plugin.constant, ()).await.unwrap(), 27);
55            assert_eq!(runtime.call(&plugin.identity, 58).await.unwrap(), 58);
56            assert_eq!(runtime.call(&plugin.add, (3, 4)).await.unwrap(), 7);
57            assert_eq!(runtime.call(&plugin.swap, (1, 2)).await.unwrap(), (2, 1));
58            assert_eq!(runtime.call(&plugin.sort, unsorted).await.unwrap(), sorted);
59            assert_eq!(runtime.call(&plugin.print, "Hi!".into()).await.unwrap(), ());
60            assert_eq!(runtime.call(&plugin.and_back, 1).await.unwrap(), 8);
61            assert_eq!(runtime.call(&plugin.imports, 1).await.unwrap(), 8);
62
63            // dbg!("{}", runtime.call(&plugin.and_back, 1).await.unwrap());
64        }
65        .block_on()
66    }
67}