71 lines
1.8 KiB
Rust
71 lines
1.8 KiB
Rust
use super::*;
|
|
use crate::Pallet;
|
|
use frame_benchmarking::v2::*;
|
|
use frame_system::RawOrigin;
|
|
|
|
fn assert_last_event<T: Config>(generic_event: crate::Event<T>) {
|
|
let re: <T as Config>::RuntimeEvent = generic_event.into();
|
|
frame_system::Pallet::<T>::assert_last_event(re.into());
|
|
}
|
|
|
|
#[benchmarks(where <T as Config>::RuntimeCall: From<frame_system::Call<T>>)]
|
|
mod benchmarks {
|
|
use super::*;
|
|
|
|
#[benchmark]
|
|
fn set_key() {
|
|
let caller: T::AccountId = whitelisted_caller();
|
|
Key::<T>::put(&caller);
|
|
|
|
let new_sudoer: T::AccountId = account("sudoer", 0, 0);
|
|
let new_sudoer_lookup = T::Lookup::unlookup(new_sudoer.clone());
|
|
|
|
#[extrinsic_call]
|
|
_(RawOrigin::Signed(caller.clone()), new_sudoer_lookup);
|
|
|
|
assert_last_event::<T>(Event::KeyChanged { old: Some(caller), new: new_sudoer });
|
|
}
|
|
|
|
#[benchmark]
|
|
fn sudo() {
|
|
let caller: T::AccountId = whitelisted_caller();
|
|
Key::<T>::put(&caller);
|
|
|
|
let call = frame_system::Call::remark { remark: vec![] }.into();
|
|
|
|
#[extrinsic_call]
|
|
_(RawOrigin::Signed(caller), Box::new(call));
|
|
|
|
assert_last_event::<T>(Event::Sudid { sudo_result: Ok(()) })
|
|
}
|
|
|
|
#[benchmark]
|
|
fn sudo_as() {
|
|
let caller: T::AccountId = whitelisted_caller();
|
|
Key::<T>::put(caller.clone());
|
|
|
|
let call = frame_system::Call::remark { remark: vec![] }.into();
|
|
|
|
let who: T::AccountId = account("as", 0, 0);
|
|
let who_lookup = T::Lookup::unlookup(who);
|
|
|
|
#[extrinsic_call]
|
|
_(RawOrigin::Signed(caller), who_lookup, Box::new(call));
|
|
|
|
assert_last_event::<T>(Event::SudoAsDone { sudo_result: Ok(()) })
|
|
}
|
|
|
|
#[benchmark]
|
|
fn remove_key() {
|
|
let caller: T::AccountId = whitelisted_caller();
|
|
Key::<T>::put(&caller);
|
|
|
|
#[extrinsic_call]
|
|
_(RawOrigin::Signed(caller.clone()));
|
|
|
|
assert_last_event::<T>(Event::KeyRemoved {});
|
|
}
|
|
|
|
impl_benchmark_test_suite!(Pallet, crate::mock::new_bench_ext(), crate::mock::Test);
|
|
}
|