From a4d0c99e6b086e0bc6af929b4c8f242552a7f025 Mon Sep 17 00:00:00 2001
From: Ankan <10196091+Ank4n@users.noreply.github.com>
Date: Sat, 24 Sep 2022 21:32:35 +0200
Subject: [PATCH] re add the migration checks for staking (#12330)

Co-authored-by: parity-processbot <>
---
 substrate/frame/staking/src/migrations.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/substrate/frame/staking/src/migrations.rs b/substrate/frame/staking/src/migrations.rs
index f47545af69..8f37ae30dd 100644
--- a/substrate/frame/staking/src/migrations.rs
+++ b/substrate/frame/staking/src/migrations.rs
@@ -35,6 +35,11 @@ pub mod v12 {
 	impl<T: Config> OnRuntimeUpgrade for MigrateToV12<T> {
 		#[cfg(feature = "try-runtime")]
 		fn pre_upgrade() -> Result<Vec<u8>, &'static str> {
+			frame_support::ensure!(
+				StorageVersion::<T>::get() == Releases::V11_0_0,
+				"Expected v11 before upgrading to v12"
+			);
+
 			frame_support::ensure!(
 				T::HistoryDepth::get() == HistoryDepth::<T>::get(),
 				"Provided value of HistoryDepth should be same as the existing storage value"
@@ -129,6 +134,11 @@ pub mod v11 {
 
 		#[cfg(feature = "try-runtime")]
 		fn post_upgrade(_state: Vec<u8>) -> Result<(), &'static str> {
+			frame_support::ensure!(
+				StorageVersion::<T>::get() == crate::Releases::V11_0_0,
+				"wrong version after the upgrade"
+			);
+
 			let old_pallet_name = N::get();
 			let new_pallet_name = <P as PalletInfoAccess>::name();