tests: add schedule and cancel upgrade ITs

centrifuge · Aug 14, 2024 · 67672b0 · 67672b0
1 parent c812198
commit 67672b0
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/runtime/integration-tests/src/cases/lp/pool_management.rs b/runtime/integration-tests/src/cases/lp/pool_management.rs
@@ -879,3 +879,40 @@ fn recover_assets<T: Runtime>() {
 		);
 	});
 }
+
+#[test_runtimes([development])]
+fn schedule_upgrade<T: Runtime>() {
+	let mut env = super::setup_full::<T>();
+	env.state_mut(|evm| {
+		assert_ok!(pallet_liquidity_pools::Pallet::<T>::schedule_upgrade(
+			<T as frame_system::Config>::RuntimeOrigin::root(),
+			EVM_DOMAIN_CHAIN_ID,
+			evm.deployed(names::POOL_MANAGER).address().into()
+		));
+
+		utils::process_gateway_message::<T>(utils::verify_gateway_message_success::<T>);
+	});
+}
+
+#[test_runtimes([development])]
+fn cancel_upgrade<T: Runtime>() {
+	let mut env = super::setup_full::<T>();
+	env.state_mut(|evm| {
+		assert_ok!(pallet_liquidity_pools::Pallet::<T>::schedule_upgrade(
+			<T as frame_system::Config>::RuntimeOrigin::root(),
+			EVM_DOMAIN_CHAIN_ID,
+			evm.deployed(names::POOL_MANAGER).address().into()
+		));
+
+		utils::process_gateway_message::<T>(utils::verify_gateway_message_success::<T>);
+	});
+	env.state_mut(|evm| {
+		assert_ok!(pallet_liquidity_pools::Pallet::<T>::cancel_upgrade(
+			<T as frame_system::Config>::RuntimeOrigin::root(),
+			EVM_DOMAIN_CHAIN_ID,
+			evm.deployed(names::POOL_MANAGER).address().into()
+		));
+
+		utils::process_gateway_message::<T>(utils::verify_gateway_message_success::<T>);
+	});
+}