Skip to content
Snippets Groups Projects

Fix weight accounting

Merged Benjamin Gallois requested to merge 167-fix-remove-member-weight into master
Compare and Show latest version
7 files
+ 252
106
Compare changes
  • Side-by-side
  • Inline
Files
7
@@ -456,6 +456,59 @@ mod benchmarks {
}
}
#[benchmark]
fn membership_added() -> Result<(), BenchmarkError> {
let caller: T::AccountId = Identities::<T>::get(T::IdtyIndex::from(1u32))
.unwrap()
.owner_key;
let caller_origin: <T as frame_system::Config>::RuntimeOrigin =
RawOrigin::Signed(caller.clone()).into();
let owner_key: T::AccountId = account("new_identity", 2, 1);
let owner_key_origin: <T as frame_system::Config>::RuntimeOrigin =
RawOrigin::Signed(owner_key.clone()).into();
Pallet::<T>::create_identity(caller_origin.clone(), owner_key.clone())?;
let name = IdtyName("new_identity".into());
Pallet::<T>::confirm_identity(owner_key_origin.clone(), name.clone())?;
let idty_index = IdentityIndexOf::<T>::get(&owner_key).unwrap();
assert_ne!(
Identities::<T>::get(idty_index).unwrap().status,
IdtyStatus::Member
);
#[block]
{
Pallet::<T>::membership_added(idty_index);
}
assert_has_event::<T>(Event::<T>::IdtyValidated { idty_index }.into());
assert_eq!(
Identities::<T>::get(idty_index).unwrap().status,
IdtyStatus::Member
);
Ok(())
}
#[benchmark]
fn membership_removed() -> Result<(), BenchmarkError> {
let key: T::AccountId = account("new_identity", 2, 1);
let account: Account<T> = create_one_identity(key)?;
assert_eq!(
Identities::<T>::get(account.index).unwrap().status,
IdtyStatus::Member
);
#[block]
{
Pallet::<T>::membership_removed(account.index);
}
assert_eq!(
Identities::<T>::get(account.index).unwrap().status,
IdtyStatus::NotMember
);
Ok(())
}
#[benchmark]
fn prune_identities_noop() {
assert!(IdentityChangeSchedule::<T>::try_get(BlockNumberFor::<T>::zero()).is_err());
Loading