Update tests for UnpaidExecution fix

polkadot-fellows · Aug 16, 2024 · e718c64 · e718c64
1 parent 63cb34d
commit e718c64
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 4 deletions.
diff --git a/integration-tests/emulated/tests/coretime/coretime-kusama/src/tests/coretime_interface.rs b/integration-tests/emulated/tests/coretime/coretime-kusama/src/tests/coretime_interface.rs
@@ -137,8 +137,6 @@ fn transact_hardcoded_weights_are_sane() {
 		Kusama::ext_wrapper(|| {
 			block_number_cursor = <Kusama as Chain>::System::block_number();
 		});
-
-		dbg!(&block_number_cursor);
 	}
 
 	// In this block we trigger assign core.
@@ -201,4 +199,25 @@ fn transact_hardcoded_weights_are_sane() {
 			]
 		);
 	});
+
+	// Here we receive and process the notify_revenue XCM with zero revenue.
+	CoretimeKusama::execute_with(|| {
+		// Hooks don't run in emulated tests - workaround.
+		<CoretimeKusama as CoretimeKusamaPallet>::Broker::on_initialize(
+			<CoretimeKusama as Chain>::System::block_number(),
+		);
+
+		assert_expected_events!(
+			CoretimeKusama,
+			vec![
+				CoretimeEvent::MessageQueue(
+					pallet_message_queue::Event::Processed { success: true, .. }
+				) => {},
+				// Zero revenue in first timeslice so history is immediately dropped.
+				CoretimeEvent::Broker(
+					pallet_broker::Event::HistoryDropped { when: 0, revenue: 0 }
+				) => {},
+			]
+		);
+	});
 }
diff --git a/integration-tests/emulated/tests/coretime/coretime-polkadot/src/tests/coretime_interface.rs b/integration-tests/emulated/tests/coretime/coretime-polkadot/src/tests/coretime_interface.rs
@@ -137,8 +137,6 @@ fn transact_hardcoded_weights_are_sane() {
 		Polkadot::ext_wrapper(|| {
 			block_number_cursor = <Polkadot as Chain>::System::block_number();
 		});
-
-		dbg!(&block_number_cursor);
 	}
 
 	// In this block we trigger assign core.
@@ -201,4 +199,25 @@ fn transact_hardcoded_weights_are_sane() {
 			]
 		);
 	});
+
+	// Here we receive and process the notify_revenue XCM with zero revenue.
+	CoretimePolkadot::execute_with(|| {
+		// Hooks don't run in emulated tests - workaround.
+		<CoretimePolkadot as CoretimePolkadotPallet>::Broker::on_initialize(
+			<CoretimePolkadot as Chain>::System::block_number(),
+		);
+
+		assert_expected_events!(
+			CoretimePolkadot,
+			vec![
+				CoretimeEvent::MessageQueue(
+					pallet_message_queue::Event::Processed { success: true, .. }
+				) => {},
+				// Zero revenue in first timeslice so history is immediately dropped.
+				CoretimeEvent::Broker(
+					pallet_broker::Event::HistoryDropped { when: 0, revenue: 0 }
+				) => {},
+			]
+		);
+	});
 }