mirror of
https://github.com/fluencelabs/aquavm
synced 2025-03-28 02:11:03 +00:00
c332cca6b7
* chore(testing-framework)!: fix WASM test runner Native mode was used before because some package used native runner for its tests. This PR allows to explicitly select test runner for tests. Many testing-framework types are now parametrized with a runner type with almost compatible defaults. * chore(testing-framework): Add `ReleaseWasmAirRunner` * chore(testing-framework)!: Rename `AirScriptExecutor::simple` to `AirScriptExecutor::from_annotated`.
81 lines
3.1 KiB
Rust
81 lines
3.1 KiB
Rust
use air_test_utils::prelude::*;
|
|
|
|
use criterion::criterion_group;
|
|
use criterion::criterion_main;
|
|
use criterion::Criterion;
|
|
use serde_json::json;
|
|
|
|
use std::cell::RefCell;
|
|
|
|
thread_local!(static RELAY_1_VM: RefCell<TestRunner<ReleaseWasmAirRunner>> = RefCell::new(
|
|
create_custom_avm(unit_call_service(), "Relay1")
|
|
));
|
|
thread_local!(static RELAY_2_VM: RefCell<TestRunner<ReleaseWasmAirRunner>> = RefCell::new(
|
|
create_custom_avm(unit_call_service(), "Relay2")
|
|
));
|
|
thread_local!(static REMOTE_VM: RefCell<TestRunner<ReleaseWasmAirRunner>> = RefCell::new({
|
|
let members_call_service: CallServiceClosure = Box::new(|_| -> CallServiceResult {
|
|
CallServiceResult::ok(json!([["A", "Relay1"], ["B", "Relay2"]]))
|
|
});
|
|
|
|
create_custom_avm(members_call_service, "Remote")
|
|
}));
|
|
thread_local!(static CLIENT_1_VM: RefCell<TestRunner<ReleaseWasmAirRunner>> = RefCell::new(
|
|
create_custom_avm(unit_call_service(), "A")
|
|
));
|
|
thread_local!(static CLIENT_2_VM: RefCell<TestRunner<ReleaseWasmAirRunner>> = RefCell::new(
|
|
create_custom_avm(unit_call_service(), "B")
|
|
));
|
|
|
|
fn chat_sent_message_benchmark() -> Result<RawAVMOutcome, String> {
|
|
let script = r#"
|
|
(seq
|
|
(call "Relay1" ("identity" "") [] $void1)
|
|
(seq
|
|
(call "Remote" ("552196ea-b9b2-4761-98d4-8e7dba77fac4" "add") [] $void2)
|
|
(seq
|
|
(call "Remote" ("920e3ba3-cbdf-4ae3-8972-0fa2f31fffd9" "get_users") [] members)
|
|
(fold members m
|
|
(par
|
|
(seq
|
|
(call m.$.[1]! ("identity" "") [] $void)
|
|
(call m.$.[0]! ("fgemb3" "add") [] $void3)
|
|
)
|
|
(next m)
|
|
)
|
|
)
|
|
)
|
|
)
|
|
)
|
|
"#;
|
|
|
|
let run_parameters = TestRunParameters::new("A", 0, 1, "");
|
|
let result = CLIENT_1_VM
|
|
.with(|vm| vm.borrow_mut().call(script, "", "", run_parameters.clone()))
|
|
.unwrap();
|
|
let result = RELAY_1_VM
|
|
.with(|vm| vm.borrow_mut().call(script, "", result.data, run_parameters.clone()))
|
|
.unwrap();
|
|
let result = REMOTE_VM
|
|
.with(|vm| vm.borrow_mut().call(script, "", result.data, run_parameters.clone()))
|
|
.unwrap();
|
|
let res_data = result.data.clone();
|
|
let res1 = RELAY_1_VM
|
|
.with(|vm| vm.borrow_mut().call(script, "", res_data, run_parameters.clone()))
|
|
.unwrap();
|
|
CLIENT_1_VM
|
|
.with(|vm| vm.borrow_mut().call(script, "", res1.data, run_parameters.clone()))
|
|
.unwrap();
|
|
let res2 = RELAY_2_VM
|
|
.with(|vm| vm.borrow_mut().call(script, "", result.data, run_parameters.clone()))
|
|
.unwrap();
|
|
CLIENT_2_VM.with(|vm| vm.borrow_mut().call(script, "", res2.data, run_parameters.clone()))
|
|
}
|
|
|
|
fn criterion_benchmark(c: &mut Criterion) {
|
|
c.bench_function("chat_send_message", move |b| b.iter(chat_sent_message_benchmark));
|
|
}
|
|
|
|
criterion_group!(benches, criterion_benchmark);
|
|
criterion_main!(benches);
|