Skip to content

Commit

Permalink
fix some sign tests
Browse files Browse the repository at this point in the history
  • Loading branch information
kmd-fl committed Jun 12, 2023
1 parent d27f94f commit e8c043d
Show file tree
Hide file tree
Showing 3 changed files with 44 additions and 35 deletions.
18 changes: 0 additions & 18 deletions crates/created-swarm/src/swarm.rs
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
*/

use std::convert::identity;
use std::path::Path;
use std::{path::PathBuf, time::Duration};

use derivative::Derivative;
Expand Down Expand Up @@ -133,23 +132,6 @@ pub async fn make_swarms_with_keypair(
.await
}

pub async fn make_swarms_with_builtins(
n: usize,
path: &Path,
keypair: Option<KeyPair>,
spell_base_dir: Option<String>,
) -> Vec<CreatedSwarm> {
make_swarms_with_cfg(n, |mut cfg| {
if let Some(keypair) = &keypair {
cfg.keypair = keypair.clone();
}
cfg.builtins_dir = Some(to_abs_path(path.into()));
cfg.spell_base_dir = spell_base_dir.clone().map(PathBuf::from);
cfg
})
.await
}

pub async fn make_swarms_with<RT: AquaRuntime, F, M, B>(
n: usize,
mut create_node: F,
Expand Down
52 changes: 37 additions & 15 deletions crates/nox-tests/tests/builtin.rs
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ extern crate fstrings;

use connected_client::ConnectedClient;
use created_swarm::{
make_swarms, make_swarms_with_builtins, make_swarms_with_keypair,
make_swarms, make_swarms_with_cfg, make_swarms_with_keypair,
make_swarms_with_transport_and_mocked_vm,
};
use eyre::{Report, WrapErr};
Expand Down Expand Up @@ -1466,15 +1466,17 @@ async fn service_stats_uninitialized() {
}
}

// Since KeyPair isn't in use for builtins anymore, we can't use this check as it is
// TODO: Need to ask Loysha to fix this test properly
/*
#[tokio::test]
async fn sign_verify() {
let kp = KeyPair::generate_ed25519();
let swarms = make_swarms_with_builtins(
1,
"tests/builtins/services".as_ref(),
Some(kp.clone()),
None,
)
let swarms = make_swarms_with_cfg(1, |mut cfg| {
cfg.disabled_system_services
.retain(|service| service != "registry");
cfg
})
.await;
let mut client = ConnectedClient::connect_to(swarms[0].multiaddr.clone())
Expand Down Expand Up @@ -1512,24 +1514,36 @@ async fn sign_verify() {
{
let data: Vec<_> = data.iter().map(|n| n.as_u64().unwrap() as u8).collect();
assert!(sig_result["success"].as_bool().unwrap());
assert!(
sig_result["success"].as_bool().unwrap(),
"result isn't success"
);
let signature = sig_result["signature"].as_array().unwrap()[0]
.as_array()
.unwrap()
.iter()
.map(|n| n.as_u64().unwrap() as u8)
.collect();
let signature = Signature::from_bytes(kp.public().get_key_format(), signature);
assert!(result);
assert!(kp.public().verify(&data, &signature).is_ok());
assert!(result, "sig.verify result is false");
assert!(
kp.public().verify(&data, &signature).is_ok(),
"kp verify failed"
);
} else {
panic!("incorrect args: expected three arguments")
}
}
*/

#[tokio::test]
async fn sign_invalid_tetraplets() {
let swarms = make_swarms_with_builtins(2, "tests/builtins/services".as_ref(), None, None).await;
let swarms = make_swarms_with_cfg(2, |mut cfg| {
cfg.disabled_system_services
.retain(|service| service != "registry");
cfg
})
.await;

let mut client = ConnectedClient::connect_to(swarms[0].multiaddr.clone())
.await
Expand Down Expand Up @@ -1594,9 +1608,12 @@ async fn sign_invalid_tetraplets() {

#[tokio::test]
async fn sig_verify_invalid_signature() {
let kp = KeyPair::generate_ed25519();
let swarms =
make_swarms_with_builtins(1, "tests/builtins/services".as_ref(), Some(kp), None).await;
let swarms = make_swarms_with_cfg(1, |mut cfg| {
cfg.disabled_system_services
.retain(|service| service != "registry");
cfg
})
.await;

let mut client = ConnectedClient::connect_to(swarms[0].multiaddr.clone())
.await
Expand Down Expand Up @@ -1783,7 +1800,12 @@ async fn json_builtins() {
#[tokio::test]
async fn insecure_sign_verify() {
let kp = KeyPair::from_secret_key(INSECURE_KEYPAIR_SEED.collect(), KeyFormat::Ed25519).unwrap();
let swarms = make_swarms_with_builtins(1, "tests/builtins/services".as_ref(), None, None).await;
let swarms = make_swarms_with_cfg(1, |mut cfg| {
cfg.disabled_system_services
.retain(|service| service != "registry");
cfg
})
.await;

let mut client = ConnectedClient::connect_to(swarms[0].multiaddr.clone())
.await
Expand Down
9 changes: 7 additions & 2 deletions crates/nox-tests/tests/spells.rs
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ use maplit::hashmap;
use serde_json::{json, Value as JValue};

use connected_client::ConnectedClient;
use created_swarm::{make_swarms, make_swarms_with_builtins};
use created_swarm::{make_swarms, make_swarms_with_cfg};
use fluence_spell_dtos::trigger_config::{ClockConfig, TriggerConfig};
use service_modules::load_module;
use spell_event_bus::api::{TriggerInfo, TriggerInfoAqua, MAX_PERIOD_SEC};
Expand Down Expand Up @@ -1355,7 +1355,12 @@ async fn resolve_global_alias() {

#[tokio::test]
async fn worker_sig_test() {
let swarms = make_swarms_with_builtins(1, "tests/builtins/services".as_ref(), None, None).await;
let swarms = make_swarms_with_cfg(1, |mut cfg| {
cfg.disabled_system_services
.retain(|service| service != "registry");
cfg
})
.await;

let mut client = ConnectedClient::connect_to(swarms[0].multiaddr.clone())
.await
Expand Down

0 comments on commit e8c043d

Please sign in to comment.