Add Consideration::ensure_successful calls to benchmarks

Author: DavidK

substrate/frame/proxy/src/benchmarking.rs

@@ -43,6 +43,10 @@ fn add_proxies<T: Config>(n: u32, maybe_who: Option<T::AccountId>) -> Result<(),
 			BlockNumberFor::<T>::zero(),
 		)?;
 	}
+	T::ProxyConsideration::ensure_successful(
+		&caller,
+		Footprint::from_mel::<(ProxiesVec<T>, ProxyTicketOf<T>)>(),
+	);
 	Ok(())
 }
 
@@ -63,6 +67,10 @@ fn add_announcements<T: Config>(
 			T::ProxyType::default(),
 			BlockNumberFor::<T>::zero(),
 		)?;
+		T::ProxyConsideration::ensure_successful(
+			&real,
+			Footprint::from_mel::<(ProxiesVec<T>, ProxyTicketOf<T>)>(),
+		);
 		real
 	};
 	let real_lookup = T::Lookup::unlookup(real);
@@ -73,6 +81,10 @@ fn add_announcements<T: Config>(
 			T::CallHasher::hash_of(&("add_announcement", n)),
 		)?;
 	}
+	T::AnnouncementConsideration::ensure_successful(
+		&caller,
+		Footprint::from_mel::<(AnnouncementsVec<T>, AnnouncementTicketOf<T>)>(),
+	);
 	Ok(())
 }
 
@@ -108,6 +120,7 @@ benchmarks! {
 			real_lookup.clone(),
 			T::CallHasher::hash_of(&call),
 		)?;
+		T::AnnouncementConsideration::ensure_successful(&delegate, Footprint::from_mel::<(AnnouncementsVec<T>, AnnouncementTicketOf<T>)>());
 		add_announcements::<T>(a, Some(delegate.clone()), None)?;
 	}: _(RawOrigin::Signed(caller), delegate_lookup, real_lookup, Some(T::ProxyType::default()), Box::new(call))
 	verify {
@@ -128,6 +141,7 @@ benchmarks! {
 			real_lookup.clone(),
 			T::CallHasher::hash_of(&call),
 		)?;
+		T::AnnouncementConsideration::ensure_successful(&caller, Footprint::from_mel::<(AnnouncementsVec<T>, AnnouncementTicketOf<T>)>());
 		add_announcements::<T>(a, Some(caller.clone()), None)?;
 	}: _(RawOrigin::Signed(caller.clone()), real_lookup, T::CallHasher::hash_of(&call))
 	verify {
@@ -150,6 +164,7 @@ benchmarks! {
 			real_lookup,
 			T::CallHasher::hash_of(&call),
 		)?;
+		T::AnnouncementConsideration::ensure_successful(&caller, Footprint::from_mel::<(AnnouncementsVec<T>, AnnouncementTicketOf<T>)>());
 		add_announcements::<T>(a, Some(caller.clone()), None)?;
 	}: _(RawOrigin::Signed(real), caller_lookup, T::CallHasher::hash_of(&call))
 	verify {
@@ -177,6 +192,7 @@ benchmarks! {
 		let p in 1 .. (T::MaxProxies::get() - 1) => add_proxies::<T>(p, None)?;
 		let caller: T::AccountId = whitelisted_caller();
 		let real = T::Lookup::unlookup(account("target", T::MaxProxies::get(), SEED));
+		T::ProxyConsideration::ensure_successful(&caller, Footprint::from_mel::<(ProxiesVec<T>, ProxyTicketOf<T>)>());
 	}: _(
 		RawOrigin::Signed(caller.clone()),
 		real,
@@ -215,6 +231,7 @@ benchmarks! {
 	create_pure {
 		let p in 1 .. (T::MaxProxies::get() - 1) => add_proxies::<T>(p, None)?;
 		let caller: T::AccountId = whitelisted_caller();
+		T::ProxyConsideration::ensure_successful(&caller, Footprint::from_mel::<(ProxiesVec<T>, ProxyTicketOf<T>)>());
 	}: _(
 		RawOrigin::Signed(caller.clone()),
 		T::ProxyType::default(),
@@ -237,11 +254,12 @@ benchmarks! {
 		let caller: T::AccountId = whitelisted_caller();
 		let caller_lookup = T::Lookup::unlookup(caller.clone());
 		Pallet::<T>::create_pure(
-			RawOrigin::Signed(whitelisted_caller()).into(),
+			RawOrigin::Signed(caller.clone()).into(),
 			T::ProxyType::default(),
 			BlockNumberFor::<T>::zero(),
 			0
 		)?;
+		T::ProxyConsideration::ensure_successful(&caller, Footprint::from_mel::<(ProxiesVec<T>, ProxyTicketOf<T>)>());
 		let height = system::Pallet::<T>::block_number();
 		let ext_index = system::Pallet::<T>::extrinsic_index().unwrap_or(0);
 		let pure_account = Pallet::<T>::pure_account(&caller, &T::ProxyType::default(), 0, None);