Implement set_block_number for runtime-bencmarks in RelayBlockNumberP…

…roiver (#1187)
paritytech · Apr 21, 2022 · 08bc38d · 08bc38d
1 parent 8b31bce
commit 08bc38d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pallets/parachain-system/Cargo.toml b/pallets/parachain-system/Cargo.toml
@@ -73,3 +73,7 @@ std = [
 	"sp-trie/std",
 	"xcm/std"
 ]
+
+runtime-benchmarks = [
+	"sp-runtime/runtime-benchmarks"
+]
diff --git a/pallets/parachain-system/src/lib.rs b/pallets/parachain-system/src/lib.rs
@@ -1047,4 +1047,11 @@ impl<T: Config> BlockNumberProvider for RelaychainBlockNumberProvider<T> {
 			.map(|d| d.relay_parent_number)
 			.unwrap_or_default()
 	}
+	#[cfg(feature = "runtime-benchmarks")]
+	fn set_block_number(block: Self::BlockNumber) {
+		if let Some(mut validation_data) = Pallet::<T>::validation_data() {
+			validation_data.relay_parent_number = block;
+			ValidationData::<T>::put(validation_data)
+		}
+	}
 }