mirror of
https://github.com/fluencelabs/aquavm
synced 2024-12-04 15:20:16 +00:00
chore(tests): add tests for topology hops (#616)
This commit is contained in:
Mike Voronov
GitHub
parent
dadbba37e9
commit
be8c97c35c
@ -24,3 +24,4 @@ mod scopes;
|
||||
mod signatures;
|
||||
mod streams;
|
||||
mod tetraplets;
|
||||
mod topology_hops;
|
||||
|
||||
@ -0,0 +1,171 @@
|
||||
/*
|
||||
* Copyright 2023 Fluence Labs Limited
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
use air_test_utils::prelude::*;
|
||||
use fluence_keypair::KeyFormat;
|
||||
use fluence_keypair::KeyPair;
|
||||
|
||||
#[test]
|
||||
fn call_generates_hop() {
|
||||
let client_peer_id = "client";
|
||||
let mut client_vm = create_avm(unit_call_service(), client_peer_id);
|
||||
|
||||
let test_peer_id = "test_peer_id";
|
||||
|
||||
let script = f!(r#"
|
||||
(call "{test_peer_id}" ("" "") [])
|
||||
"#);
|
||||
|
||||
let client_result = checked_call_vm!(client_vm, <_>::default(), script, "", "");
|
||||
assert_next_pks!(&client_result.next_peer_pks, &[test_peer_id]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn call_with_join_behaviour() {
|
||||
let relay_peer_id = "relay";
|
||||
let mut relay_vm = create_avm(unit_call_service(), relay_peer_id);
|
||||
let client_peer_id = "client";
|
||||
let mut client_vm = create_avm(unit_call_service(), client_peer_id);
|
||||
|
||||
let friend_peer_id = "friend";
|
||||
let mut friend_vm = create_avm(unit_call_service(), friend_peer_id);
|
||||
let friend_relay_peer_id = "friend_relay";
|
||||
let mut friend_relay_vm = create_avm(unit_call_service(), friend_relay_peer_id);
|
||||
|
||||
let script = f!(r#"
|
||||
(xor
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(ap "{relay_peer_id}" -relay- )
|
||||
(ap "{client_peer_id}" me)
|
||||
)
|
||||
(ap "{relay_peer_id}" myRelay)
|
||||
)
|
||||
(ap "{friend_peer_id}" friend)
|
||||
)
|
||||
(ap "{friend_relay_peer_id}" friendRelay)
|
||||
)
|
||||
(par
|
||||
(seq
|
||||
(seq
|
||||
(call -relay- ("op" "noop") [])
|
||||
(call friendRelay ("op" "noop") [])
|
||||
)
|
||||
(xor
|
||||
(seq
|
||||
(seq
|
||||
(call friend ("testo" "getString") ["testo string via friend "] str)
|
||||
(call friendRelay ("op" "noop") [])
|
||||
)
|
||||
(call -relay- ("op" "noop") [])
|
||||
)
|
||||
(seq
|
||||
(seq
|
||||
(call friendRelay ("op" "noop") [])
|
||||
(call -relay- ("op" "noop") [])
|
||||
)
|
||||
(call %init_peer_id% ("errorHandlingSrv" "error") [%last_error% 1])
|
||||
)
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("lp" "print") ["string in par"])
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("lp" "print") [str])
|
||||
)
|
||||
(xor
|
||||
(call %init_peer_id% ("callbackSrv" "response") ["finish"])
|
||||
(call %init_peer_id% ("errorHandlingSrv" "error") [%last_error% 2])
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("errorHandlingSrv" "error") [%last_error% 3])
|
||||
)
|
||||
"#);
|
||||
|
||||
let keypair = KeyPair::generate(KeyFormat::Ed25519);
|
||||
let run_parameters = TestRunParameters::new(client_peer_id.to_string(), 0, 0, "".to_string());
|
||||
|
||||
let client_result = client_vm
|
||||
.runner
|
||||
.call(
|
||||
&script,
|
||||
"",
|
||||
"",
|
||||
client_peer_id,
|
||||
0,
|
||||
0,
|
||||
None,
|
||||
<_>::default(),
|
||||
&keypair,
|
||||
"".to_string(),
|
||||
)
|
||||
.unwrap();
|
||||
assert_next_pks!(&client_result.next_peer_pks, &[relay_peer_id]);
|
||||
|
||||
let relay_result = checked_call_vm!(relay_vm, run_parameters.clone(), &script, "", client_result.data);
|
||||
assert_next_pks!(&relay_result.next_peer_pks, &[friend_relay_peer_id]);
|
||||
|
||||
let friend_relay_result = checked_call_vm!(
|
||||
friend_relay_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
"",
|
||||
relay_result.data.clone()
|
||||
);
|
||||
assert_next_pks!(&friend_relay_result.next_peer_pks, &[friend_peer_id]);
|
||||
|
||||
let friend_result = checked_call_vm!(
|
||||
friend_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
"",
|
||||
friend_relay_result.data.clone()
|
||||
);
|
||||
assert_next_pks!(&friend_result.next_peer_pks, &[friend_relay_peer_id]);
|
||||
|
||||
let friend_relay_result = checked_call_vm!(
|
||||
friend_relay_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
friend_relay_result.data,
|
||||
friend_result.data
|
||||
);
|
||||
assert_next_pks!(&friend_relay_result.next_peer_pks, &[relay_peer_id]);
|
||||
|
||||
let relay_result = checked_call_vm!(
|
||||
relay_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
relay_result.data,
|
||||
friend_relay_result.data
|
||||
);
|
||||
assert_next_pks!(&relay_result.next_peer_pks, &[client_peer_id]);
|
||||
|
||||
let client_result = checked_call_vm!(client_vm, run_parameters.clone(), &script, "", "");
|
||||
let client_result = checked_call_vm!(
|
||||
client_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
client_result.data,
|
||||
relay_result.data
|
||||
);
|
||||
assert!(client_result.next_peer_pks.is_empty());
|
||||
}
|
||||
@ -0,0 +1,195 @@
|
||||
/*
|
||||
* Copyright 2023 Fluence Labs Limited
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
use air_test_utils::prelude::*;
|
||||
use fluence_keypair::KeyFormat;
|
||||
use fluence_keypair::KeyPair;
|
||||
|
||||
#[test]
|
||||
fn canon_generates_hop() {
|
||||
let client_peer_id = "client";
|
||||
let mut client_vm = create_avm(unit_call_service(), client_peer_id);
|
||||
|
||||
let test_peer_id = "test_peer_id";
|
||||
|
||||
let script = f!(r#"
|
||||
(canon "{test_peer_id}" $empty_stream #canon_stream)
|
||||
"#);
|
||||
|
||||
let client_result = checked_call_vm!(client_vm, <_>::default(), script, "", "");
|
||||
assert_next_pks!(&client_result.next_peer_pks, &[test_peer_id]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn canon_with_join_behaviour() {
|
||||
let relay_peer_id = "relay";
|
||||
let mut relay_vm = create_avm(unit_call_service(), relay_peer_id);
|
||||
let client_peer_id = "client";
|
||||
let mut client_vm = create_avm(unit_call_service(), client_peer_id);
|
||||
|
||||
let friend_peer_id = "friend";
|
||||
let mut friend_vm = create_avm(unit_call_service(), friend_peer_id);
|
||||
let friend_relay_peer_id = "friend_relay";
|
||||
let mut friend_relay_vm = create_avm(unit_call_service(), friend_relay_peer_id);
|
||||
|
||||
let script = f!(r#"
|
||||
(xor
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(seq
|
||||
(ap "{relay_peer_id}" -relay- )
|
||||
(ap "{client_peer_id}" me)
|
||||
)
|
||||
(ap "{relay_peer_id}" myRelay)
|
||||
)
|
||||
(ap "{friend_peer_id}" friend)
|
||||
)
|
||||
(ap "{friend_relay_peer_id}" friendRelay)
|
||||
)
|
||||
(par
|
||||
(seq
|
||||
(seq
|
||||
(new $hop-stream-drop
|
||||
(new #hop-canon-drop
|
||||
(canon -relay- $hop-stream-drop #hop-canon-drop)
|
||||
)
|
||||
)
|
||||
(new $hop-stream-drop
|
||||
(new #hop-canon-drop
|
||||
(canon friendRelay $hop-stream-drop #hop-canon-drop)
|
||||
)
|
||||
)
|
||||
)
|
||||
(xor
|
||||
(seq
|
||||
(seq
|
||||
(call friend ("testo" "getString") ["testo string via friend "] str)
|
||||
(new $hop-stream-drop
|
||||
(new #hop-canon-drop
|
||||
(canon friendRelay $hop-stream-drop #hop-canon-drop)
|
||||
)
|
||||
)
|
||||
)
|
||||
(new $hop-stream-drop
|
||||
(new #hop-canon-drop
|
||||
(canon -relay- $hop-stream-drop #hop-canon-drop)
|
||||
)
|
||||
)
|
||||
)
|
||||
(seq
|
||||
(seq
|
||||
(new $hop-stream-drop
|
||||
(new #hop-canon-drop
|
||||
(canon friendRelay $hop-stream-drop #hop-canon-drop)
|
||||
)
|
||||
)
|
||||
(new $hop-stream-drop
|
||||
(new #hop-canon-drop
|
||||
(canon -relay- $hop-stream-drop #hop-canon-drop)
|
||||
)
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("errorHandlingSrv" "error") [%last_error% 1])
|
||||
)
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("lp" "print") ["string in par"])
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("lp" "print") [str])
|
||||
)
|
||||
(xor
|
||||
(call %init_peer_id% ("callbackSrv" "response") ["finish"])
|
||||
(call %init_peer_id% ("errorHandlingSrv" "error") [%last_error% 2])
|
||||
)
|
||||
)
|
||||
(call %init_peer_id% ("errorHandlingSrv" "error") [%last_error% 3])
|
||||
)
|
||||
"#);
|
||||
|
||||
let keypair = KeyPair::generate(KeyFormat::Ed25519);
|
||||
let run_parameters = TestRunParameters::new(client_peer_id.to_string(), 0, 0, "".to_string());
|
||||
|
||||
let client_result = client_vm
|
||||
.runner
|
||||
.call(
|
||||
&script,
|
||||
"",
|
||||
"",
|
||||
client_peer_id,
|
||||
0,
|
||||
0,
|
||||
None,
|
||||
<_>::default(),
|
||||
&keypair,
|
||||
"".to_string(),
|
||||
)
|
||||
.unwrap();
|
||||
assert_next_pks!(&client_result.next_peer_pks, &[relay_peer_id]);
|
||||
|
||||
let relay_result = checked_call_vm!(relay_vm, run_parameters.clone(), &script, "", client_result.data);
|
||||
assert_next_pks!(&relay_result.next_peer_pks, &[friend_relay_peer_id]);
|
||||
|
||||
let friend_relay_result = checked_call_vm!(
|
||||
friend_relay_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
"",
|
||||
relay_result.data.clone()
|
||||
);
|
||||
assert_next_pks!(&friend_relay_result.next_peer_pks, &[friend_peer_id]);
|
||||
|
||||
let friend_result = checked_call_vm!(
|
||||
friend_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
"",
|
||||
friend_relay_result.data.clone()
|
||||
);
|
||||
assert_next_pks!(&friend_result.next_peer_pks, &[friend_relay_peer_id]);
|
||||
|
||||
let friend_relay_result = checked_call_vm!(
|
||||
friend_relay_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
friend_relay_result.data,
|
||||
friend_result.data
|
||||
);
|
||||
assert_next_pks!(&friend_relay_result.next_peer_pks, &[relay_peer_id]);
|
||||
|
||||
let relay_result = checked_call_vm!(
|
||||
relay_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
relay_result.data,
|
||||
friend_relay_result.data
|
||||
);
|
||||
assert_next_pks!(&relay_result.next_peer_pks, &[client_peer_id]);
|
||||
|
||||
let client_result = checked_call_vm!(client_vm, run_parameters.clone(), &script, "", "");
|
||||
let client_result = checked_call_vm!(
|
||||
client_vm,
|
||||
run_parameters.clone(),
|
||||
&script,
|
||||
client_result.data,
|
||||
relay_result.data
|
||||
);
|
||||
assert!(client_result.next_peer_pks.is_empty());
|
||||
}
|
||||
18
air/tests/test_module/features/topology_hops/mod.rs
Normal file
18
air/tests/test_module/features/topology_hops/mod.rs
Normal file
@ -0,0 +1,18 @@
|
||||
/*
|
||||
* Copyright 2023 Fluence Labs Limited
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
mod call_guided_topology;
|
||||
mod canon_guided_topology;
|
||||
Loading…
Reference in New Issue
Block a user