add try-runtime testing

Signed-off-by: Yi <flame0743@gmail.com>
Manta-Network · Aug 5, 2022 · 394a0ca · 394a0ca
1 parent 0edb2cd
commit 394a0ca
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 2 deletions.
diff --git a/runtime/calamari/Cargo.toml b/runtime/calamari/Cargo.toml
@@ -155,6 +155,7 @@ try-runtime = [
   'pallet-assets/try-runtime',
   'pallet-tx-pause/try-runtime',
   'pallet-asset-manager/try-runtime',
+  'runtime-common/try-runtime',
 ]
 # Set timing constants (e.g. session period) to faster versions to speed up testing.
 fast-runtime = []

diff --git a/runtime/common/Cargo.toml b/runtime/common/Cargo.toml
@@ -14,7 +14,6 @@ pallet-transaction-payment = { git = 'https://github.com/paritytech/substrate.gi
 sp-runtime = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.26", default-features = false }
 sp-std = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.26", default-features = false }
 
-
 # Self dependencies
 manta-primitives = { path = '../../primitives/manta', default-features = false }
 
@@ -66,3 +65,6 @@ std = [
   "pallet-transaction-payment/std",
   "sp-runtime/std",
 ]
+try-runtime = [
+  "frame-support/try-runtime",
+]
diff --git a/runtime/common/src/migration.rs b/runtime/common/src/migration.rs
@@ -20,8 +20,9 @@
 
 use frame_support::{
     dispatch::Weight,
+    ensure,
     migrations::migrate_from_pallet_version_to_storage_version,
-    traits::{GetStorageVersion, OnRuntimeUpgrade, PalletInfoAccess},
+    traits::{GetStorageVersion, OnRuntimeUpgrade, PalletInfoAccess, StorageVersion},
     weights::constants::RocksDbWeight,
 };
 
@@ -43,6 +44,27 @@ where
         let db_weight = RocksDbWeight::get();
         migrate_from_pallet_version_to_storage_version::<T>(&db_weight)
     }
+
+    #[cfg(feature = "try-runtime")]
+    fn pre_upgrade() -> Result<(), &'static str> {
+        let storage_version = StorageVersion::get::<T>();
+        frame_support::debug(&"----PreUpgrade----");
+        frame_support::debug(&T::module_name());
+        frame_support::debug(&T::name());
+        frame_support::debug(&storage_version);
+        Ok(())
+    }
+
+    #[cfg(feature = "try-runtime")]
+    fn post_upgrade() -> Result<(), &'static str> {
+        let storage_version = StorageVersion::get::<T>();
+        frame_support::debug(&"----PostUpgrade----");
+        frame_support::debug(&T::module_name());
+        frame_support::debug(&T::name());
+        frame_support::debug(&storage_version);
+        ensure!(storage_version == StorageVersion::new(1), T::module_name());
+        Ok(())
+    }
 }
 
 #[cfg(test)]