Add storage migrations for Combinatorial Tokens Upgrade (#1401)

* add todos for storage migrations * remove todo for asset storage migration * add storage migration and try-runtime tests * correct migration and tests * update todos * update migration tests * add migration to runtime, fix clippy * bump storage version * remove corrupted pools * correct clippy * correct CI
zeitgeistpm · Jan 30, 2025 · 33a736e · 33a736e
1 parent 93f0663
commit 33a736e
Show file tree

Hide file tree

Showing 4 changed files with 402 additions and 3 deletions.
diff --git a/primitives/src/asset.rs b/primitives/src/asset.rs
@@ -55,7 +55,6 @@ pub enum Asset<MarketId> {
     ForeignAsset(u32),
     ParimutuelShare(MarketId, CategoryIndex),
 }
-// TODO Needs storage migration
 
 #[cfg(feature = "runtime-benchmarks")]
 impl<MarketId: MaxEncodedLen> ZeitgeistAssetEnumerator<MarketId> for Asset<MarketId> {

diff --git a/runtime/common/src/lib.rs b/runtime/common/src/lib.rs
@@ -96,11 +96,13 @@ macro_rules! decl_common_types {
         #[cfg(feature = "runtime-benchmarks")]
         use zrml_prediction_markets::types::PredictionMarketsCombinatorialTokensBenchmarkHelper;
 
+        use zrml_neo_swaps::migration::MigratePoolStorageItems;
+
         pub type Block = generic::Block<Header, UncheckedExtrinsic>;
 
         type Address = sp_runtime::MultiAddress<AccountId, ()>;
 
-        type Migrations = ();
+        type Migrations = (MigratePoolStorageItems<Runtime, RemovableMarketIds>);
 
         pub type Executive = frame_executive::Executive<
             Runtime,

diff --git a/zrml/neo-swaps/src/lib.rs b/zrml/neo-swaps/src/lib.rs
@@ -93,7 +93,7 @@ mod pallet {
     };
     use zrml_market_commons::MarketCommonsPalletApi;
 
-    pub(crate) const STORAGE_VERSION: StorageVersion = StorageVersion::new(2);
+    pub(crate) const STORAGE_VERSION: StorageVersion = StorageVersion::new(3);
 
     // These should not be config parameters to avoid misconfigurations.
     pub(crate) const EXIT_FEE: u128 = CENT / 10;