diff --git a/pallets/collator-selection/src/benchmarking.rs b/pallets/collator-selection/src/benchmarking.rs index cc91fd88ec..ce26c020ff 100644 --- a/pallets/collator-selection/src/benchmarking.rs +++ b/pallets/collator-selection/src/benchmarking.rs @@ -84,12 +84,14 @@ fn validator(c: u32) -> (T::AccountId, ("candidate", c, 1000), keys::(c)) } -fn register_validators(count: u32) { +fn register_validators(count: u32) -> Vec { let validators = (0..count).map(|c| validator::(c)).collect::>(); - for (who, keys) in validators { + for (who, keys) in validators.clone() { >::set_keys(RawOrigin::Signed(who).into(), keys, Vec::new()).unwrap(); } + + validators.into_iter().map(|(who, _)| who).collect() } fn register_candidates(count: u32) { @@ -107,7 +109,7 @@ benchmarks! { set_invulnerables { let b in 1 .. T::MaxInvulnerables::get(); - let new_invulnerables = (0..b).map(|c| account("candidate", c, SEED)).collect::>(); + let new_invulnerables = register_validators::(b); let origin = T::UpdateOrigin::successful_origin(); }: { assert_ok!(