diff --git a/substrate/frame/ranked-collective/src/lib.rs b/substrate/frame/ranked-collective/src/lib.rs index 84be1243c8afc810f98860ba8403bca727c1140f..a65d184d6bf0635ae62d0b65a71073f32a1a520a 100644 --- a/substrate/frame/ranked-collective/src/lib.rs +++ b/substrate/frame/ranked-collective/src/lib.rs @@ -256,8 +256,7 @@ impl<T: Config<I>, I: 'static, const MIN_RANK: u16> EnsureOrigin<T::RuntimeOrigi #[cfg(feature = "runtime-benchmarks")] fn try_successful_origin() -> Result<T::RuntimeOrigin, ()> { - let who = IndexToId::<T, I>::get(MIN_RANK, 0).ok_or(())?; - Ok(frame_system::RawOrigin::Signed(who).into()) + EnsureRankedMember::<T, I, MIN_RANK>::try_successful_origin() } } @@ -279,8 +278,7 @@ impl<T: Config<I>, I: 'static, const MIN_RANK: u16> EnsureOrigin<T::RuntimeOrigi #[cfg(feature = "runtime-benchmarks")] fn try_successful_origin() -> Result<T::RuntimeOrigin, ()> { - let who = IndexToId::<T, I>::get(MIN_RANK, 0).ok_or(())?; - Ok(frame_system::RawOrigin::Signed(who).into()) + EnsureRankedMember::<T, I, MIN_RANK>::try_successful_origin() } } @@ -302,7 +300,9 @@ impl<T: Config<I>, I: 'static, const MIN_RANK: u16> EnsureOrigin<T::RuntimeOrigi #[cfg(feature = "runtime-benchmarks")] fn try_successful_origin() -> Result<T::RuntimeOrigin, ()> { - let who = IndexToId::<T, I>::get(MIN_RANK, 0).ok_or(())?; + let who = frame_benchmarking::account::<T::AccountId>("successful_origin", 0, 0); + crate::Pallet::<T, I>::do_add_member_to_rank(who.clone(), MIN_RANK) + .expect("Could not add members for benchmarks"); Ok(frame_system::RawOrigin::Signed(who).into()) } }