Fix post_update version check in try-runtime tests

gear-tech · May 28, 2024 · 989e2fe · 989e2fe
1 parent adae978
commit 989e2fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pallets/gear-program/src/migrations/allocations.rs b/pallets/gear-program/src/migrations/allocations.rs
@@ -118,7 +118,7 @@ impl<T: Config> OnRuntimeUpgrade for MigrateAllocations<T> {
                 "incorrect count of elements old {} != new {}",
             );
             ensure!(
-                Pallet::<T>::current_storage_version() == MIGRATE_TO_VERSION,
+                Pallet::<T>::on_chain_storage_version() == MIGRATE_TO_VERSION,
                 "incorrect storage version after migration"
             );
         }

diff --git a/pallets/gear-program/src/migrations/v7.rs b/pallets/gear-program/src/migrations/v7.rs
@@ -183,7 +183,7 @@ where
                 "Active programs total balance is not as expected: {} != {}",
             );
             ensure!(
-                Pallet::<T>::current_storage_version() == MIGRATE_TO_VERSION,
+                Pallet::<T>::on_chain_storage_version() == MIGRATE_TO_VERSION,
                 "incorrect storage version after migration"
             );
         }