mirror of
https://github.com/zed-industries/zed.git
synced 2024-10-28 01:07:09 +00:00
8ba2f77148
Co-authored-by: Mikayla <mikayla@zed.dev>
93 lines
3.7 KiB
Rust
93 lines
3.7 KiB
Rust
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()
|
|
}
|
|
}
|