More checks

seadanda · Sep 29, 2024 · 41b53cb · 41b53cb
1 parent ac10dc3
commit 41b53cb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/system-parachains/coretime/coretime-polkadot/src/migrations.rs b/system-parachains/coretime/coretime-polkadot/src/migrations.rs
@@ -149,6 +149,10 @@ impl FixMigration {
 			(2101, 298800),
 		];
 
+		if PotentialRenewals::<Runtime>::iter().count() != premature_renewals.len() {
+			log::error!(target: TARGET, "Unexpected renewal state!");
+		}
+
 		if PotentialRenewals::<Runtime>::clear(premature_renewals.len() as u32, None)
 			.maybe_cursor
 			.is_some()
@@ -253,6 +257,9 @@ impl OnRuntimeUpgrade for FixMigration {
 		let sale_info = SaleInfo::<Runtime>::get().unwrap();
 		let leases = Leases::<Runtime>::get();
 		let pre_upgrade_state = (sale_info, leases);
+
+		assert_eq!(PotentialRenewals::<Runtime>::iter().count(), INCORRECT_RENEWAL_IDS.len());
+
 		Ok(pre_upgrade_state.encode())
 	}