ghost-node/pallets/networks/src/tests.rs

1307 lines
46 KiB
Rust

use super::*;
use frame_support::{assert_err, assert_ok};
use mock::{
ExtBuilder, GhostNetworks, MaxNetworks, RandomAccount, RegistererAccount, RemoverAccount,
RewardCurve, RuntimeEvent, RuntimeOrigin, System, Test, UpdaterAccount,
};
use pallet_staking::EraPayout;
use sp_runtime::DispatchError;
const CURRENT_CURVE: NetworkCurve = NetworkCurve::Secp256k1;
fn prepare_network_data(
maybe_in_share: Option<u32>,
maybe_out_share: Option<u32>,
) -> (u32, NetworkData) {
(
1u32,
ghost_helpers::networks::NetworkDataBuilder::default()
.with_gatekeeper(
hex::decode("02f9308a019258c31049344f85f89d5229b531c845836f99b08601f113bce036f9")
.expect("Invalid public key"),
)
.with_default_endpoints(vec![
b"https://some-mainnet.infura.io/v3/demo-key".to_vec(),
b"https://rpc4ever.info/api".to_vec(),
])
.with_selector(vec![1u8; 4])
.with_network_type(NetworkType::Evm)
.with_network_curve(CURRENT_CURVE)
.with_rate_limit_delay(6)
.with_finality_delay(69)
.with_block_deviation(420)
.with_incoming_share(maybe_in_share.unwrap_or_default())
.with_outgoing_share(maybe_out_share.unwrap_or_default())
.build(),
)
}
fn register_and_check_network(chain_id: u32, network: NetworkData) {
let network_curve = network.curve;
assert_ok!(GhostNetworks::register_network(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
network.clone()
));
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
assert_eq!(NetworkIndexes::<Test>::get(), vec![chain_id]);
assert!(NetworkCurves::<Test>::contains_key(&network_curve));
assert!(NetworkCurvesVec::<Test>::get().contains(&network_curve));
}
#[test]
fn could_add_network_from_authority() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_eq!(NetworkIndexes::<Test>::get(), vec![]);
assert!(!NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(!NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
assert_ok!(GhostNetworks::register_network(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
network.clone(),
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkRegistered { chain_id },
));
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
assert_eq!(NetworkIndexes::<Test>::get(), vec![chain_id]);
assert!(NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
});
}
#[test]
fn could_not_add_network_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_eq!(NetworkIndexes::<Test>::get(), vec![]);
assert!(!NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(!NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
assert_err!(
GhostNetworks::register_network(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
network.clone(),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::register_network(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
network.clone(),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::register_network(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
network,
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_eq!(NetworkIndexes::<Test>::get(), vec![]);
assert!(!NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(!NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
});
}
#[test]
fn could_update_network_selector_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_selector =
BoundedVec::<u8, ConstU32<MAX_SELECTOR_LEN>>::try_from(vec![69, 69, 69, 69])
.unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_selector(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_selector.clone()
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkSelectorUpdated { chain_id },
));
let mut final_network = network.clone();
final_network.selector = new_selector;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_add_network_endpoint_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let raw_endpoint =
BoundedVec::<u8, ConstU32<MAX_ENDPOINT_LEN>>::try_from(
b"https:://new-endpoint.my-server.com/v1/my-super-secret-key".to_vec(),
).unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
None,
Some(raw_endpoint.clone()),
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkEndpointAdded { chain_id },
));
let current_network = Networks::<Test>::get(chain_id).unwrap();
assert_eq!(
current_network.default_endpoints.len(),
network.default_endpoints.len() + 1
);
assert_eq!(
current_network
.default_endpoints
.last()
.cloned()
.unwrap_or_default(),
raw_endpoint
);
assert_ne!(&network, &current_network);
});
}
#[test]
fn could_remove_network_endpoint_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let index_to_remove = 0u32;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
Some(index_to_remove),
None,
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkEndpointRemoved {
chain_id,
index: index_to_remove,
},
));
let current_network = Networks::<Test>::get(chain_id).unwrap();
assert_eq!(
current_network.default_endpoints.len(),
network.default_endpoints.len() - 1
);
assert_ne!(
current_network
.default_endpoints
.get(index_to_remove as usize),
network.default_endpoints.get(index_to_remove as usize)
);
assert_ne!(&network, &current_network);
});
}
#[test]
fn could_update_network_endpoint_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let index_to_update = 0u32;
let raw_endpoint =
BoundedVec::<u8, ConstU32<MAX_ENDPOINT_LEN>>::try_from(
b"https:://new-endpoint.my-server.com/v1/my-super-secret-key".to_vec(),
).unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
Some(index_to_update),
Some(raw_endpoint.clone()),
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkEndpointUpdated { chain_id, index: index_to_update },
));
let previous_endpoints_len = network.default_endpoints.len();
let mut final_network = network.clone();
if let Some(endpoint_by_index) = final_network
.default_endpoints
.get_mut(index_to_update as usize)
{
*endpoint_by_index = raw_endpoint.clone();
}
let current_network = Networks::<Test>::get(chain_id).unwrap();
assert_eq!(
current_network.default_endpoints.len(),
previous_endpoints_len
);
assert_eq!(
current_network
.default_endpoints
.get(index_to_update as usize)
.cloned()
.unwrap_or_default(),
raw_endpoint
);
assert_ne!(&network, &final_network);
});
}
#[test]
fn could_update_network_finality_delay_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_finality_delay = 1337;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_finality_delay(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_finality_delay
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkFinalityDelayUpdated {
chain_id,
finality_delay: new_finality_delay,
},
));
let mut final_network = network.clone();
final_network.finality_delay = new_finality_delay;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_network_rate_limit_delay_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_rate_limit_delay = 1337;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_rate_limit_delay(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_rate_limit_delay
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkRateLimitDelayUpdated {
chain_id,
rate_limit_delay: new_rate_limit_delay,
},
));
let mut final_network = network.clone();
final_network.rate_limit_delay = new_rate_limit_delay;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_network_block_deviation_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_block_deviation = 1337;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_block_deviation(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_block_deviation
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkBlockDeviationUpdated {
chain_id,
block_deviation: new_block_deviation,
},
));
let mut final_network = network.clone();
final_network.block_deviation = new_block_deviation;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_network_type_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_type = NetworkType::Utxo;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_type(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_type.clone()
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkTypeUpdated {
chain_id,
network_type: new_type.clone(),
},
));
let mut final_network = network.clone();
final_network.r#type = new_type;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_network_curve_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_curve = NetworkCurve::Ed25519;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert!(!NetworkCurves::<Test>::contains_key(&new_curve));
assert!(NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(!NetworkCurvesVec::<Test>::get().contains(&new_curve));
assert!(NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
assert_ok!(GhostNetworks::update_network_curve(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_curve.clone()
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkCurveUpdated {
chain_id,
network_curve: new_curve.clone(),
},
));
let mut final_network = network.clone();
final_network.r#curve = new_curve;
assert!(NetworkCurves::<Test>::contains_key(&new_curve));
assert!(!NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(NetworkCurvesVec::<Test>::get().contains(&new_curve));
assert!(!NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_network_gatekeeper_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_gatekeeper = BoundedVec::<u8, ConstU32<MAX_GATEKEEPER_LEN>>::try_from(
hex::decode("04b2c7e3d9f9f8a5b6c7d8e9f0a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0c1d2e3f4a5b6c7d8e9f0a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0")
.expect("Invalid public key"),
).unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_network_gatekeeper(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_gatekeeper.clone()
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkGatekeeperUpdated { chain_id },
));
let mut final_network = network.clone();
final_network.gatekeeper = new_gatekeeper;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_incoming_network_share_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_incoming_share = 69;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_incoming_network_share(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_incoming_share
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkIncomingShareUpdated {
chain_id,
incoming_share: new_incoming_share,
},
));
let mut final_network = network.clone();
final_network.incoming_share = new_incoming_share;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_update_outgoing_network_share_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let new_outgoing_share = 69;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::update_outgoing_network_share(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_outgoing_share
));
System::assert_last_event(RuntimeEvent::GhostNetworks(
crate::Event::NetworkOutgoingShareUpdated {
chain_id,
outgoing_share: new_outgoing_share,
},
));
let mut final_network = network.clone();
final_network.outgoing_share = new_outgoing_share;
assert_eq!(Networks::<Test>::get(chain_id), Some(final_network.clone()));
assert_ne!(network, final_network);
});
}
#[test]
fn could_not_update_network_endpoint_from_random_account() {
ExtBuilder::build().execute_with(|| {
let index_to_update = 0u32;
let raw_endpoint = BoundedVec::<u8, ConstU32<MAX_ENDPOINT_LEN>>::try_from(
b"https:://new-endpoint.my-server.com/v1/my-super-secret-key".to_vec(),
).unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
Some(index_to_update),
Some(raw_endpoint.clone()),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
Some(index_to_update),
Some(raw_endpoint.clone()),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
Some(index_to_update),
Some(raw_endpoint.clone()),
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_add_network_endpoint_from_random_account() {
ExtBuilder::build().execute_with(|| {
let raw_endpoint = BoundedVec::<u8, ConstU32<MAX_ENDPOINT_LEN>>::try_from(
b"https:://new-endpoint.my-server.com/v1/my-super-secret-key".to_vec(),
).unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
None,
Some(raw_endpoint.clone()),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
None,
Some(raw_endpoint.clone()),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
None,
Some(raw_endpoint.clone()),
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_remove_network_endpoint_from_random_account() {
ExtBuilder::build().execute_with(|| {
let index_to_remove = 0u32;
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
Some(index_to_remove),
None,
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
Some(index_to_remove),
None,
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
Some(index_to_remove),
None,
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_finality_delay_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
let finality_delay = 1337;
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_finality_delay(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
finality_delay
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_finality_delay(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
finality_delay
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_finality_delay(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
finality_delay
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_rate_limit_delay_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
let rate_limit_delay = 1337;
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_rate_limit_delay(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
rate_limit_delay
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_rate_limit_delay(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
rate_limit_delay
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_rate_limit_delay(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
rate_limit_delay
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_release_delay_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
let block_deviation = 1337;
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_block_deviation(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
block_deviation
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_block_deviation(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
block_deviation
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_block_deviation(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
block_deviation
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_type_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_type(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
NetworkType::Utxo
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_type(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
NetworkType::Utxo
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_type(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
NetworkType::Utxo
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_curve_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_curve(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
NetworkCurve::Ed25519,
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_curve(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
NetworkCurve::Ed25519,
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_curve(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
NetworkCurve::Ed25519,
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_gatekeeper_from_random_account() {
ExtBuilder::build().execute_with(|| {
let new_gatekeeper = BoundedVec::<u8, ConstU32<MAX_GATEKEEPER_LEN>>::try_from(
hex::decode("04b2c7e3d9f9f8a5b6c7d8e9f0a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0c1d2e3f4a5b6c7d8e9f0a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0")
.expect("Invalid public key"),
).unwrap();
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_network_gatekeeper(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
new_gatekeeper.clone(),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_gatekeeper(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
new_gatekeeper.clone(),
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_network_gatekeeper(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
new_gatekeeper,
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_incoming_share_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_incoming_network_share(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
69
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_incoming_network_share(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
69
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_incoming_network_share(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
69
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_network_outgoing_share_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::update_outgoing_network_share(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id,
69
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_outgoing_network_share(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
69
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::update_outgoing_network_share(
RuntimeOrigin::signed(RandomAccount::get()),
chain_id,
69
),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
});
}
#[test]
fn could_not_update_endpoint_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let raw_endpoint = BoundedVec::<u8, ConstU32<MAX_ENDPOINT_LEN>>::try_from(
b"https:://new-endpoint.my-server.com/v1/my-super-secret-key".to_vec(),
).unwrap();
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
Some(0u32),
Some(raw_endpoint),
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_add_endpoint_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let raw_endpoint = BoundedVec::<u8, ConstU32<MAX_ENDPOINT_LEN>>::try_from(
b"https:://new-endpoint.my-server.com/v1/my-super-secret-key".to_vec(),
).unwrap();
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
None,
Some(raw_endpoint),
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_remove_endpoint_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_endpoint(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
Some(0u32),
None,
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_finality_delay_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_finality_delay(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
1337
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_rate_limit_delay_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_rate_limit_delay(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
1337
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_release_delay_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_block_deviation(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
1337
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_type_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_type(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
NetworkType::Utxo
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_curve_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_curve(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
NetworkCurve::Ed25519,
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_gatekeeper_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let new_gatekeeper = BoundedVec::<u8, ConstU32<MAX_GATEKEEPER_LEN>>::try_from(
hex::decode("04b2c7e3d9f9f8a5b6c7d8e9f0a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0c1d2e3f4a5b6c7d8e9f0a1b2c3d4e5f6a7b8c9d0e1f2a3b4c5d6e7f8a9b0")
.expect("Invalid public key"),
).unwrap();
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_network_gatekeeper(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
new_gatekeeper,
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_incoming_share_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_incoming_network_share(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
1337
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_not_update_outgoing_share_for_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::update_outgoing_network_share(
RuntimeOrigin::signed(UpdaterAccount::get()),
chain_id,
1337
),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
});
}
#[test]
fn could_remove_network_from_authority_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_ok!(GhostNetworks::remove_network(
RuntimeOrigin::signed(RemoverAccount::get()),
chain_id,
));
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_eq!(NetworkIndexes::<Test>::get(), vec![]);
assert!(!NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(!NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
});
}
#[test]
fn could_not_remove_network_from_random_account() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
register_and_check_network(chain_id, network.clone());
assert_err!(
GhostNetworks::remove_network(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id
),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::remove_network(RuntimeOrigin::signed(UpdaterAccount::get()), chain_id),
DispatchError::BadOrigin
);
assert_err!(
GhostNetworks::remove_network(RuntimeOrigin::signed(RandomAccount::get()), chain_id),
DispatchError::BadOrigin
);
assert_eq!(Networks::<Test>::get(chain_id), Some(network));
assert_eq!(NetworkIndexes::<Test>::get(), vec![chain_id]);
assert!(NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
});
}
#[test]
fn could_not_remove_non_existent_network() {
ExtBuilder::build().execute_with(|| {
let chain_id: u32 = 1;
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_err!(
GhostNetworks::remove_network(RuntimeOrigin::signed(RemoverAccount::get()), chain_id),
crate::Error::<Test>::NetworkDoesNotExist
);
assert_eq!(Networks::<Test>::get(chain_id), None);
assert_eq!(NetworkIndexes::<Test>::get(), vec![]);
assert!(!NetworkCurves::<Test>::contains_key(&CURRENT_CURVE));
assert!(!NetworkCurvesVec::<Test>::get().contains(&CURRENT_CURVE));
});
}
#[test]
fn bridge_storage_is_empty_by_default() {
ExtBuilder::build().execute_with(|| {
let state = NetworkImbalance::<Test>::get();
assert_eq!(state.incoming, Default::default());
assert_eq!(state.outgoing, Default::default());
assert_eq!(state.curve_share, Default::default());
});
}
#[test]
fn check_substrate_guarantees_not_to_overflow_u128() {
ExtBuilder::build().execute_with(|| {
let reward_curve = RewardCurve::get();
let mut n: u128 = 69;
let mut d: u128 = 100;
loop {
n = match n.checked_mul(1_000) {
Some(value) => value,
None => break,
};
d = match d.checked_mul(1_000) {
Some(value) => value,
None => break,
};
assert_eq!(
reward_curve.calculate_for_fraction_times_denominator(n, d),
d
);
}
});
}
#[test]
fn check_substrate_guarantees_not_to_overflow_u64() {
ExtBuilder::build().execute_with(|| {
let reward_curve = RewardCurve::get();
let mut n: u64 = 69;
let mut d: u64 = 100;
loop {
n = match n.checked_mul(1_000) {
Some(value) => value,
None => break,
};
d = match d.checked_mul(1_000) {
Some(value) => value,
None => break,
};
assert_eq!(
reward_curve.calculate_for_fraction_times_denominator(n, d),
d
);
}
});
}
#[test]
fn check_substrate_guarantees_not_to_overflow_u32() {
ExtBuilder::build().execute_with(|| {
let reward_curve = RewardCurve::get();
let mut n: u32 = 69;
let mut d: u32 = 100;
loop {
n = match n.checked_mul(1_000) {
Some(value) => value,
None => break,
};
d = match d.checked_mul(1_000) {
Some(value) => value,
None => break,
};
assert_eq!(
reward_curve.calculate_for_fraction_times_denominator(n, d),
d
);
}
});
}
#[test]
fn error_on_max_networks_overflow() {
ExtBuilder::build().execute_with(|| {
let (chain_id, network) = prepare_network_data(None, None);
let max_networks = MaxNetworks::get();
for index in 0..max_networks {
let other_chain_id = chain_id.saturating_add(index);
assert_ok!(GhostNetworks::register_network(
RuntimeOrigin::signed(RegistererAccount::get()),
other_chain_id,
network.clone(),
));
}
assert_err!(
GhostNetworks::register_network(
RuntimeOrigin::signed(RegistererAccount::get()),
chain_id.saturating_add(max_networks),
network.clone(),
),
crate::Error::<Test>::TooManyNetworks,
);
});
}
#[test]
fn bridge_inbound_and_outbound_flows() {
ExtBuilder::build().execute_with(|| {
let in_share = Some(50_000_000); // 5%
let out_share = Some(100_000_000); // 10%
let (chain_id, network) = prepare_network_data(in_share, out_share);
register_and_check_network(chain_id, network);
let inbound_amount: u128 = 10_000;
let expected_in_curve: u128 = 500;
let expected_in_final: u128 = 9500;
let final_amount = Pallet::<Test>::register_incoming(
&chain_id,
inbound_amount,
);
assert_ok!(final_amount, expected_in_final);
let imbalance = NetworkImbalance::<Test>::get();
assert_eq!(imbalance.incoming, expected_in_final);
assert_eq!(imbalance.curve_share, expected_in_curve);
assert_eq!(imbalance.outgoing, 0);
let gatekeeper_balance = GatekeeperAmounts::<Test>::get(&chain_id);
assert_eq!(gatekeeper_balance, inbound_amount);
let outbound_amount: u128 = 5_000;
let expected_out_curve: u128 = 500;
let expected_out_final: u128 = 4500;
let final_amount = Pallet::<Test>::register_outgoing(&chain_id, outbound_amount);
assert_ok!(final_amount, expected_out_final);
let current_imbalance = NetworkImbalance::<Test>::get();
assert_eq!(current_imbalance.incoming, expected_in_final);
assert_eq!(current_imbalance.outgoing, expected_out_final);
assert_eq!(current_imbalance.curve_share, 1000);
assert_eq!(current_imbalance.curve_share, expected_in_curve + expected_out_curve);
let final_gatekeeper_balance = GatekeeperAmounts::<Test>::get(&chain_id);
assert_eq!(final_gatekeeper_balance, 5_000);
let evil_outbound_amount: u128 = 6_000;
let final_amount_result = Pallet::<Test>::register_outgoing(&chain_id, evil_outbound_amount);
assert!(final_amount_result.is_err());
let post_fail_imbalance = NetworkImbalance::<Test>::get();
assert_eq!(post_fail_imbalance.curve_share, 1000);
assert_eq!(post_fail_imbalance.outgoing, expected_out_final);
let total_accumulated_shares = 1000;
assert_eq!(current_imbalance.curve_share, total_accumulated_shares);
let total_staked_ideal: u128 = 69_000_000;
let total_issuance: u128 = 100_000_000;
let (payout, rest_payout) = BridgedInflationCurve::<RewardCurve, Test>::era_payout(
total_staked_ideal,
total_issuance,
0,
);
assert_eq!(payout + rest_payout, total_accumulated_shares);
let post_era_imbalance = NetworkImbalance::<Test>::get();
assert_eq!(post_era_imbalance.incoming, 0);
assert_eq!(post_era_imbalance.outgoing, 0);
assert_eq!(post_era_imbalance.curve_share, 0);
});
}
#[test]
fn test_extreme_overflow_protection() {
ExtBuilder::build().execute_with(|| {
let in_share = Some(50_000_000); // 5%
let out_share = Some(100_000_000); // 10%
let (chain_id, network) = prepare_network_data(in_share, out_share);
register_and_check_network(chain_id, network);
let near_max = u128::MAX - 1000;
NetworkImbalance::<Test>::put(NetworkImbalanceState {
outgoing: 0,
incoming: near_max,
curve_share: 0,
});
let massive_amount = 100_000_000;
let result = Pallet::<Test>::register_incoming(&chain_id, massive_amount);
assert!(result.is_err());
GatekeeperAmounts::<Test>::insert(&chain_id, 420);
let result = Pallet::<Test>::register_outgoing(&chain_id, massive_amount);
assert!(result.is_err());
});
}