Ecosystem/serai
Ecosystem/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-02-02 03:06:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add subsequent_vote test

Browse source 
This is the contracts/extension that triggered a Rust ICE, as noted in 
my issue there.
This commit is contained in:
Luke Parker 2022-07-18 02:53:45 -04:00
parent aa0d364fc2
commit 21e555192c
2 changed files with 44 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
contracts/extension/src/lib.rs
View file

@ -34,6 +34,16 @@ impl Environment for SeraiEnvironment {
type ChainExtension = SeraiExtension;
}
pub fn test_validators() -> Vec<AccountId> {
vec![
AccountId::from([1; 32]),
AccountId::from([2; 32]),
AccountId::from([3; 32]),
AccountId::from([4; 32]),
AccountId::from([5; 32]),
]
}
pub fn test_register() {
struct ExtensionLen;
impl ink_env::test::ChainExtension for ExtensionLen {
@ -42,7 +52,7 @@ pub fn test_register() {
}
fn call(&mut self, _: &[u8], output: &mut Vec<u8>) -> u32 {
scale::Encode::encode_to(&5u16, output);
scale::Encode::encode_to(&u16::try_from(test_validators().len()).unwrap(), output);
0
}
}
@ -72,13 +82,7 @@ pub fn test_register() {
let potential = AccountId::decode(&mut &input[1 ..]).unwrap(); // TODO: Why is this 1 ..?
let mut presence = false;
for validator in [
AccountId::from([1; 32]),
AccountId::from([2; 32]),
AccountId::from([3; 32]),
AccountId::from([4; 32]),
AccountId::from([5; 32])
].clone() {
for validator in test_validators() {
if potential == validator {
presence = true;
}

35
contracts/multisig/lib.rs
View file

@ -220,20 +220,22 @@ mod multisig {
assert_eq!(multisig.validator_set(), [0; 32]);
}
/// Non-existent curves error accordingly.
#[ink::test]
fn non_existent_curve() {
assert_eq!(Multisig::new().key(0), Err(Error::NonExistentCurve));
}
/// Validators can vote on keys.
#[ink::test]
fn vote() {
serai_extension::test_register();
ink_env::test::set_caller::<ink_env::DefaultEnvironment>(AccountId::from([1; 32]));
let keys = vec![vec![0, 1], vec![2, 3]];
let mut multisig = Multisig::new();
let keys = vec![vec![0, 1], vec![2, 3]];
ink_env::test::set_caller::<ink_env::DefaultEnvironment>(AccountId::from([1; 32]));
multisig.vote(keys.clone()).unwrap();
let emitted_events = ink_env::test::recorded_events().collect::<Vec<_>>();
assert_eq!(emitted_events.len(), 1);
assert_vote(
@ -248,5 +250,32 @@ mod multisig {
Some(keys),
);
}
/// Subsequent votes don't re-emit the full keys.
#[ink::test]
fn subsequent_vote() {
serai_extension::test_register();
let keys = vec![vec![0, 1], vec![2, 3]];
let mut multisig = Multisig::new();
ink_env::test::set_caller::<ink_env::DefaultEnvironment>(AccountId::from([1; 32]));
multisig.vote(keys.clone()).unwrap();
ink_env::test::set_caller::<ink_env::DefaultEnvironment>(AccountId::from([2; 32]));
multisig.vote(keys.clone()).unwrap();
let emitted_events = ink_env::test::recorded_events().collect::<Vec<_>>();
assert_eq!(emitted_events.len(), 2);
assert_vote(
&emitted_events[1],
AccountId::from([2; 32]),
[0xff; 32],
{
let mut hash = [0; 32];
ink_env::hash_encoded::<Blake2x256, _>(&keys, &mut hash);
hash
},
None,
);
}
}
}