Use MaxValidatorsCount to bound ErasRewardPoints BoundedVec

paritytech · Dec 2, 2024 · ea59383 · ea59383
1 parent e986264
commit ea59383
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 19 deletions.
diff --git a/substrate/frame/staking/src/benchmarking.rs b/substrate/frame/staking/src/benchmarking.rs
@@ -130,10 +130,7 @@ pub fn create_validator_with_nominators<T: Config>(
 		assert_ok!(reward_map.try_insert(validator, reward));
 	}
 	// Give Era Points
-	let reward = EraRewardPoints::<
-		T::AccountId,
-		<T::ElectionProvider as ElectionProviderBase>::MaxWinners,
-	> {
+	let reward = EraRewardPoints::<T::AccountId, T::MaxValidatorsCount> {
 		total: points_total,
 		individual: reward_map,
 	};
@@ -880,10 +877,7 @@ mod benchmarks {
 			assert_ok!(reward_map.try_insert(validator, reward));
 		}
 		// Give Era Points
-		let reward = EraRewardPoints::<
-			T::AccountId,
-			<T::ElectionProvider as ElectionProviderBase>::MaxWinners,
-		> {
+		let reward = EraRewardPoints::<T::AccountId, T::MaxValidatorsCount> {
 			total: points_total,
 			individual: reward_map,
 		};

diff --git a/substrate/frame/staking/src/lib.rs b/substrate/frame/staking/src/lib.rs
@@ -386,15 +386,17 @@ pub struct ActiveEraInfo {
 ///
 /// This points will be used to reward validators and their respective nominators.
 #[derive(PartialEq, Encode, Decode, RuntimeDebug, TypeInfo, MaxEncodedLen)]
-#[scale_info(skip_type_params(MaxWinners))]
-pub struct EraRewardPoints<AccountId: Ord, MaxWinners: Get<u32>> {
+#[scale_info(skip_type_params(MaxValidatorsCount))]
+pub struct EraRewardPoints<AccountId: Ord, MaxValidatorsCount: Get<u32>> {
 	/// Total number of points. Equals the sum of reward points for each validator.
 	pub total: RewardPoint,
 	/// The reward points earned by a given validator.
-	pub individual: BoundedBTreeMap<AccountId, RewardPoint, MaxWinners>,
+	pub individual: BoundedBTreeMap<AccountId, RewardPoint, MaxValidatorsCount>,
 }
 
-impl<AccountId: Ord, MaxWinners: Get<u32>> Default for EraRewardPoints<AccountId, MaxWinners> {
+impl<AccountId: Ord, MaxValidatorsCount: Get<u32>> Default
+	for EraRewardPoints<AccountId, MaxValidatorsCount>
+{
 	fn default() -> Self {
 		EraRewardPoints { total: Default::default(), individual: BoundedBTreeMap::new() }
 	}

diff --git a/substrate/frame/staking/src/migrations.rs b/substrate/frame/staking/src/migrations.rs
@@ -111,7 +111,7 @@ pub mod v17 {
 					Ok(individual_rewards) => {
 						let bounded_era_rewards = EraRewardPoints::<
 							<T as frame_system::Config>::AccountId,
-							<<T as Config>::ElectionProvider as ElectionProviderBase>::MaxWinners,
+							<T as Config>::MaxValidatorsCount,
 						> {
 							individual: individual_rewards,
 							total: era_rewards.total,

diff --git a/substrate/frame/staking/src/mock.rs b/substrate/frame/staking/src/mock.rs
@@ -342,7 +342,7 @@ impl Default for ExtBuilder {
 			validator_count: 2,
 			minimum_validator_count: 0,
 			balance_factor: 1,
-			invulnerables: vec![],
+			invulnerables: BoundedVec::new(),
 			has_stakers: true,
 			initialize_first_session: true,
 			min_nominator_bond: ExistentialDeposit::get(),
@@ -376,7 +376,7 @@ impl ExtBuilder {
 		self
 	}
 	pub fn invulnerables(mut self, invulnerables: Vec<AccountId>) -> Self {
-		self.invulnerables = invulnerables;
+		self.invulnerables = BoundedVec::force_from(invulnerables).unwrap();
 		self
 	}
 	pub fn session_per_era(self, length: SessionIndex) -> Self {

diff --git a/substrate/frame/staking/src/pallet/mod.rs b/substrate/frame/staking/src/pallet/mod.rs
@@ -582,7 +582,7 @@ pub mod pallet {
 		_,
 		Twox64Concat,
 		EraIndex,
-		EraRewardPoints<T::AccountId, <T::ElectionProvider as ElectionProviderBase>::MaxWinners>,
+		EraRewardPoints<T::AccountId, T::MaxValidatorsCount>,
 		ValueQuery,
 	>;
 
@@ -1050,7 +1050,7 @@ pub mod pallet {
 		/// Get the rewards for the last [`Config::HistoryDepth`] eras.
 		pub fn eras_reward_points<EncodeLikeEraIndex>(
 			era_index: EncodeLikeEraIndex,
-		) -> EraRewardPoints<T::AccountId, <T::ElectionProvider as ElectionProviderBase>::MaxWinners>
+		) -> EraRewardPoints<T::AccountId, T::MaxValidatorsCount>
 		where
 			EncodeLikeEraIndex: codec::EncodeLike<EraIndex>,
 		{

diff --git a/substrate/frame/staking/src/tests.rs b/substrate/frame/staking/src/tests.rs
@@ -72,7 +72,6 @@ fn set_staking_configs_works() {
 			ConfigOp::Noop,
 			ConfigOp::Noop,
 			ConfigOp::Noop,
-			ConfigOp::Noop,
 			ConfigOp::Noop
 		)));
 
@@ -84,7 +83,6 @@ fn set_staking_configs_works() {
 			ConfigOp::Remove,
 			ConfigOp::Remove,
 			ConfigOp::Remove,
-			ConfigOp::Remove,
 			ConfigOp::Remove
 		));
 		assert_eq!(MinNominatorBond::<Test>::get(), 0);