[AHM] Staking Account Translation

.github/workflows/clippy.yml

@@ -40,7 +40,7 @@ jobs:
       - name: Clippy
         run: |
           cargo clippy --version
-          cargo clippy --all-targets --locked --workspace --quiet
+          cargo clippy --all-targets --locked --workspace --quiet 
           cargo clippy --all-targets  --locked --workspace --features try-runtime,runtime-benchmarks,std --quiet
           cargo clippy --all-targets --locked --quiet --features try-runtime,runtime-benchmarks,std,kusama-ahm -p staging-kusama-runtime -p asset-hub-kusama-runtime
           cargo clippy --all-targets --locked --quiet --features try-runtime,runtime-benchmarks,std,polkadot-ahm -p polkadot-runtime -p asset-hub-polkadot-runtime

pallets/ah-migrator/src/account_translation.rs

@@ -40,6 +40,7 @@ impl<T: Config> Pallet<T> {
 	}
 
 	/// Translate the account if its a parachain sovereign account.
+	#[cfg(any(feature = "polkadot-ahm", feature = "kusama-ahm"))]
 	fn maybe_sovereign_translate(account: &T::AccountId) -> Option<T::AccountId> {
 		let Some(new) = crate::sovereign_account_translation::SOV_TRANSLATIONS
 			.binary_search_by_key(account, |((rc_acc, _), _)| rc_acc.clone())
@@ -64,7 +65,13 @@ impl<T: Config> Pallet<T> {
 		Some(new)
 	}
 
+	#[cfg(not(any(feature = "polkadot-ahm", feature = "kusama-ahm")))]
+	fn maybe_sovereign_translate(account: &T::AccountId) -> Option<T::AccountId> {
+		None
+	}
+
 	/// Translate the account if its derived from a parachain sovereign account.
+	#[cfg(any(feature = "polkadot-ahm", feature = "kusama-ahm"))]
 	fn maybe_derived_translate(account: &T::AccountId) -> Option<T::AccountId> {
 		let Some((new, idx)) = crate::sovereign_account_translation::DERIVED_TRANSLATIONS
 			.binary_search_by_key(account, |((rc_acc, _), _, _)| rc_acc.clone())
@@ -88,4 +95,9 @@ impl<T: Config> Pallet<T> {
 
 		Some(new.clone())
 	}
+
+	#[cfg(not(any(feature = "polkadot-ahm", feature = "kusama-ahm")))]
+	fn maybe_derived_translate(account: &T::AccountId) -> Option<T::AccountId> {
+		None
+	}
 }

pallets/ah-migrator/src/sovereign_account_translation.rs

@@ -22,7 +22,7 @@ use sp_runtime::AccountId32;
 // AUTOGENERATED BELOW
 // RC snap path: ../../polkadot.snap
 /// List of RC para to AH sibl sovereign account translation sorted by RC account.
-#[cfg(not(feature = "kusama-ahm"))]
+#[cfg(feature = "polkadot-ahm")]
 pub const SOV_TRANSLATIONS: &[((AccountId32, &'static str), (AccountId32, &'static str))] = &[
 	// para 0
 	(
@@ -822,7 +822,7 @@ pub const SOV_TRANSLATIONS: &[((AccountId32, &'static str), (AccountId32, &'stat
 ];
 
 /// List of RC para to AH sibl derived account translation sorted by RC account.
-#[cfg(not(feature = "kusama-ahm"))]
+#[cfg(feature = "polkadot-ahm")]
 pub const DERIVED_TRANSLATIONS: &[(
 	(AccountId32, &'static str),
 	u16,

pallets/ah-migrator/src/staking/checks.rs

@@ -16,9 +16,13 @@
 
 //! Checks that the staking migration succeeded.
 
-use pallet_rc_migrator::staking::{
-	message::{PortableNominations, PortableUnappliedSlash},
-	RcData,
+use codec::Encode;
+use pallet_rc_migrator::{
+	staking::{
+		message::{PortableNominations, PortableUnappliedSlash},
+		RcData,
+	},
+	types::SortByEncoded,
 };
 use sp_runtime::{AccountId32, Perbill};
 
@@ -56,84 +60,59 @@ impl<T: crate::Config> crate::types::AhMigrationCheck
 		assert_eq!(rc.chill_threshold, pallet_staking_async::ChillThreshold::<T>::get());
 
 		// Storage Maps
-		assert_eq!(rc.invulnerables, pallet_staking_async::Invulnerables::<T>::get().into_inner());
-		assert_eq!(rc.bonded, pallet_staking_async::Bonded::<T>::iter().collect::<Vec<_>>());
-		assert_eq!(
-			rc.ledger.into_iter().map(|(k, v)| (k, v.into())).collect::<Vec<_>>(),
-			pallet_staking_async::Ledger::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(rc.invulnerables, pallet_staking_async::Invulnerables::<T>::get());
+		assert_equal_items(rc.bonded, pallet_staking_async::Bonded::<T>::iter());
+		assert_equal_items(
+			rc.ledger.into_iter().map(|(k, v)| (k, v.into())),
+			pallet_staking_async::Ledger::<T>::iter(),
 		);
-		assert_eq!(
-			rc.payee
-				.into_iter()
-				.map(|(k, v)| (k, translate_reward_destination(v)))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::Payee::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.payee.into_iter().map(|(k, v)| (k, translate_reward_destination(v))),
+			pallet_staking_async::Payee::<T>::iter(),
 		);
-		assert_eq!(
-			rc.validators
-				.into_iter()
-				.map(|(k, v)| (k, translate_validator_prefs(v)))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::Validators::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.validators.into_iter().map(|(k, v)| (k, translate_validator_prefs(v))),
+			pallet_staking_async::Validators::<T>::iter(),
 		);
-		assert_eq!(
-			rc.nominators
-				.into_iter()
-				.map(|(k, v)| (k, translate_nominations(v)))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::Nominators::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.nominators.into_iter().map(|(k, v)| (k, translate_nominations(v))),
+			pallet_staking_async::Nominators::<T>::iter(),
 		);
-		assert_eq!(
+		assert_equal_items(
 			rc.virtual_stakers,
-			pallet_staking_async::VirtualStakers::<T>::iter_keys().collect::<Vec<_>>()
+			pallet_staking_async::VirtualStakers::<T>::iter_keys(),
 		);
-		assert_eq!(
-			rc.eras_stakers_overview
-				.into_iter()
-				.map(|(k1, k2, v)| (k1, k2, v.into()))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::ErasStakersOverview::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.eras_stakers_overview.into_iter().map(|(k1, k2, v)| (k1, k2, v.into())),
+			pallet_staking_async::ErasStakersOverview::<T>::iter(),
 		);
-		assert_eq!(
-			rc.eras_stakers_paged
-				.into_iter()
-				.map(|(k, v)| (k, v.into()))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::ErasStakersPaged::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.eras_stakers_paged.into_iter().map(|(k, v)| (k, v.into())),
+			pallet_staking_async::ErasStakersPaged::<T>::iter(),
 		);
-		assert_eq!(
+		assert_equal_items(
 			rc.claimed_rewards,
 			pallet_staking_async::ClaimedRewards::<T>::iter()
-				.map(|(k1, k2, v)| (k1, k2, v.into_inner()))
-				.collect::<Vec<_>>()
+				.map(|(k1, k2, v)| (k1, k2, v.into_inner())),
 		);
-		assert_eq!(
-			rc.eras_validator_prefs
-				.into_iter()
-				.map(|(k1, k2, v)| (k1, k2, v.into()))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::ErasValidatorPrefs::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.eras_validator_prefs.into_iter().map(|(k1, k2, v)| (k1, k2, v.into())),
+			pallet_staking_async::ErasValidatorPrefs::<T>::iter(),
 		);
-		assert_eq!(
+		assert_equal_items(
 			rc.eras_validator_reward,
-			pallet_staking_async::ErasValidatorReward::<T>::iter().collect::<Vec<_>>()
-		);
-		assert_eq!(
-			rc.eras_reward_points
-				.into_iter()
-				.map(|(k, v)| (k, v.into()))
-				.collect::<Vec<_>>(),
-			pallet_staking_async::ErasRewardPoints::<T>::iter().collect::<Vec<_>>()
+			pallet_staking_async::ErasValidatorReward::<T>::iter(),
 		);
-		assert_eq!(
-			rc.eras_total_stake,
-			pallet_staking_async::ErasTotalStake::<T>::iter().collect::<Vec<_>>()
+		assert_equal_items(
+			rc.eras_reward_points.into_iter().map(|(k, v)| (k, v.into())),
+			pallet_staking_async::ErasRewardPoints::<T>::iter(),
 		);
+		assert_equal_items(rc.eras_total_stake, pallet_staking_async::ErasTotalStake::<T>::iter());
 		check_unapplied_slashes::<T>(rc.unapplied_slashes);
-		assert_eq!(rc.bonded_eras, pallet_staking_async::BondedEras::<T>::get().into_inner());
-		assert_eq!(
+		assert_equal_items(rc.bonded_eras, pallet_staking_async::BondedEras::<T>::get());
+		assert_equal_items(
 			rc.validator_slash_in_era,
-			pallet_staking_async::ValidatorSlashInEra::<T>::iter().collect::<Vec<_>>()
+			pallet_staking_async::ValidatorSlashInEra::<T>::iter(),
 		);
 	}
 }
@@ -197,3 +176,13 @@ fn check_unapplied_slashes<T: crate::Config>(rc: Vec<(u32, Vec<PortableUnapplied
 
 	// TODO assert
 }
+
+/// Assert that two iterators have the same elements, regardless of their order.
+fn assert_equal_items<V: Encode, I: IntoIterator<Item = V>, J: IntoIterator<Item = V>>(
+	rc: I,
+	ah: J,
+) {
+	let rc = rc.into_iter().collect::<Vec<_>>().sort_by_encoded();
+	let ah = ah.into_iter().collect::<Vec<_>>().sort_by_encoded();
+	assert_eq!(rc, ah);
+}

pallets/ah-migrator/src/staking/staking.rs

@@ -17,7 +17,10 @@
 //! Pallet staking migration.
 
 use crate::*;
-use pallet_rc_migrator::{staking::PortableStakingMessage, types::DefensiveTruncateInto};
+use pallet_rc_migrator::{
+	staking::PortableStakingMessage,
+	types::{DefensiveTruncateInto, TranslateAccounts},
+};
 use sp_runtime::Perbill;
 
 impl<T: Config> Pallet<T> {
@@ -50,6 +53,8 @@ impl<T: Config> Pallet<T> {
 	fn do_receive_staking_message(message: PortableStakingMessage) -> Result<(), Error<T>> {
 		use PortableStakingMessage::*;
 
+		let message = message.translate_accounts(&Self::translate_account_rc_to_ah);
+
 		match message {
 			Values(values) => {
 				log::debug!(target: LOG_TARGET, "Integrating StakingValues");