fix clippy

gear-tech · Nov 12, 2024 · ec46dfb · ec46dfb
1 parent af369bc
commit ec46dfb
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/pallets/gear-messenger/src/migrations/context_store.rs b/pallets/gear-messenger/src/migrations/context_store.rs
@@ -24,7 +24,14 @@ use common::{
 };
 
 #[cfg(feature = "try-runtime")]
-use {frame_support::ensure, sp_runtime::TryRuntimeError, sp_std::vec::Vec};
+use {
+    frame_support::ensure,
+    sp_runtime::{
+        codec::{Decode, Encode},
+        TryRuntimeError,
+    },
+    sp_std::vec::Vec,
+};
 
 use frame_support::{
     traits::{GetStorageVersion, OnRuntimeUpgrade, StorageVersion},
@@ -113,15 +120,17 @@ impl<T: Config> OnRuntimeUpgrade for RemoveCommitStorage<T> {
                 "Current storage version is not allowed for migration, check migration code in order to allow it."
             );
 
-            Ok(vec![1])
+            Ok(Some(1u64).encode())
         } else {
-            Ok(vec![0])
+            Ok(Option::<u64>::None.encode())
         }
     }
 
     #[cfg(feature = "try-runtime")]
     fn post_upgrade(state: Vec<u8>) -> Result<(), TryRuntimeError> {
-        if state[0] == 1 {
+        if let Some(1) = Option::<u64>::decode(&mut state.as_ref())
+            .map_err(|_| "`pre_upgrade` provided an invalid state")?
+        {
             ensure!(
                 Pallet::<T>::on_chain_storage_version() == MIGRATE_TO_VERSION,
                 "incorrect storage version after migration"

diff --git a/pallets/gear-program/src/migrations/add_section_sizes.rs b/pallets/gear-program/src/migrations/add_section_sizes.rs
@@ -28,7 +28,6 @@ use {
         codec::{Decode, Encode},
         TryRuntimeError,
     },
-    sp_std::vec,
     sp_std::vec::Vec,
 };