From 16083b382c6cd11a11dba8a5f6a77ecbb535a00f Mon Sep 17 00:00:00 2001
From: Anton <anton.kalyaev@gmail.com>
Date: Fri, 16 Jun 2023 16:17:01 +0400
Subject: [PATCH] [frame/im-online] remove ensure from pre_upgrade (#14401)

* [frame/im-online] remove ensure from pre_upgrade

modify ensure in post_upgrade to allow for future upgrades
refs https://github.com/paritytech/substrate/pull/14251#discussion_r1230973672

* cargo fmt
---
 frame/im-online/src/migration.rs | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/frame/im-online/src/migration.rs b/frame/im-online/src/migration.rs
index 2f567ce778414..84652965972e3 100644
--- a/frame/im-online/src/migration.rs
+++ b/frame/im-online/src/migration.rs
@@ -61,8 +61,6 @@ pub mod v1 {
 	impl<T: Config> OnRuntimeUpgrade for Migration<T> {
 		#[cfg(feature = "try-runtime")]
 		fn pre_upgrade() -> Result<Vec<u8>, TryRuntimeError> {
-			ensure!(StorageVersion::get::<Pallet<T>>() == 0, "can only upgrade from version 0");
-
 			let count = v0::ReceivedHeartbeats::<T>::iter().count();
 			log::info!(target: TARGET, "Migrating {} received heartbeats", count);
 
@@ -111,7 +109,7 @@ pub mod v1 {
 					old_received_heartbeats
 				);
 			}
-			ensure!(StorageVersion::get::<Pallet<T>>() == 1, "must upgrade");
+			ensure!(StorageVersion::get::<Pallet<T>>() >= 1, "must upgrade");
 
 			Ok(())
 		}