Skip to content

Commit

Permalink
#80 Modify add_registrat banchmarking
Browse files Browse the repository at this point in the history
  • Loading branch information
ayushmishra2005 committed Jan 28, 2021
1 parent 2eab46b commit db46e91
Showing 1 changed file with 15 additions and 3 deletions.
18 changes: 15 additions & 3 deletions pallets/polkadex-identity-pallet/src/benchmarking.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,29 @@

use super::*;
use sp_std::prelude::*;
use frame_system::RawOrigin;
use frame_system::{EventRecord, RawOrigin};
use frame_support::{ensure, traits::OnFinalize,};
use frame_benchmarking::{benchmarks, TrackedStorageKey, account};

const SEED: u32 = 0;

fn assert_last_event<T: Config>(generic_event: <T as Config>::Event) {
let events = frame_system::Module::<T>::events();
let system_event: <T as frame_system::Config>::Event = generic_event.into();
// compare to the last event record
let EventRecord { event, .. } = &events[events.len() - 1];
assert_eq!(event, &system_event);
}

benchmarks! {

add_registrar {
let account_id = account("registrar", 0, SEED);
}: _(RawOrigin::Root, account_id)
let account: T::AccountId = account("registrar", 0, SEED);
}: _(RawOrigin::Root, account.clone())
verify {
ensure!(Registrars::<T>::get(&account) == Judgement::PolkadexFoundationAccount, "RegistrarAlreadyPresent");
assert_last_event::<T>(Event::<T>::RegistrarAdded(account).into());
}

}

Expand Down

0 comments on commit db46e91

Please sign in to comment.