1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
use anyhow::{anyhow, Result};
use test_programs_artifacts::{foreach_keyvalue, KEYVALUE_MAIN_COMPONENT};
use wasmtime::{
component::{Component, Linker, ResourceTable},
Store,
};
use wasmtime_wasi::{bindings::Command, WasiCtx, WasiCtxBuilder, WasiView};
use wasmtime_wasi_keyvalue::{WasiKeyValue, WasiKeyValueCtx, WasiKeyValueCtxBuilder};
struct Ctx {
table: ResourceTable,
wasi_ctx: WasiCtx,
wasi_keyvalue_ctx: WasiKeyValueCtx,
}
impl WasiView for Ctx {
fn table(&mut self) -> &mut ResourceTable {
&mut self.table
}
fn ctx(&mut self) -> &mut WasiCtx {
&mut self.wasi_ctx
}
}
async fn run_wasi(path: &str, ctx: Ctx) -> Result<()> {
let engine = test_programs_artifacts::engine(|config| {
config.async_support(true);
});
let mut store = Store::new(&engine, ctx);
let component = Component::from_file(&engine, path)?;
let mut linker = Linker::new(&engine);
wasmtime_wasi::add_to_linker_async(&mut linker)?;
wasmtime_wasi_keyvalue::add_to_linker(&mut linker, |h: &mut Ctx| {
WasiKeyValue::new(&h.wasi_keyvalue_ctx, &mut h.table)
})?;
let command = Command::instantiate_async(&mut store, &component, &linker).await?;
command
.wasi_cli_run()
.call_run(&mut store)
.await?
.map_err(|()| anyhow!("command returned with failing exit status"))
}
macro_rules! assert_test_exists {
($name:ident) => {
#[allow(unused_imports)]
use self::$name as _;
};
}
foreach_keyvalue!(assert_test_exists);
#[tokio::test(flavor = "multi_thread")]
async fn keyvalue_main() -> Result<()> {
run_wasi(
KEYVALUE_MAIN_COMPONENT,
Ctx {
table: ResourceTable::new(),
wasi_ctx: WasiCtxBuilder::new().inherit_stderr().build(),
wasi_keyvalue_ctx: WasiKeyValueCtxBuilder::new()
.in_memory_data([("atomics_key", "5")])
.build(),
},
)
.await
}
|
