referrerpolicy=no-referrer-when-downgrade

pallet_bridge_parachains/
benchmarking.rs

1// Copyright (C) Parity Technologies (UK) Ltd.
2// This file is part of Parity Bridges Common.
3
4// Parity Bridges Common is free software: you can redistribute it and/or modify
5// it under the terms of the GNU General Public License as published by
6// the Free Software Foundation, either version 3 of the License, or
7// (at your option) any later version.
8
9// Parity Bridges Common is distributed in the hope that it will be useful,
10// but WITHOUT ANY WARRANTY; without even the implied warranty of
11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12// GNU General Public License for more details.
13
14// You should have received a copy of the GNU General Public License
15// along with Parity Bridges Common.  If not, see <http://www.gnu.org/licenses/>.
16
17//! Parachains finality pallet benchmarking.
18
19use crate::{
20	weights_ext::DEFAULT_PARACHAIN_HEAD_SIZE, Call, RelayBlockHash, RelayBlockHasher,
21	RelayBlockNumber,
22};
23
24use bp_polkadot_core::parachains::{ParaHash, ParaHeadsProof, ParaId};
25use bp_runtime::UnverifiedStorageProofParams;
26use frame_benchmarking::{account, benchmarks_instance_pallet};
27use frame_system::RawOrigin;
28use sp_std::prelude::*;
29
30/// Pallet we're benchmarking here.
31pub struct Pallet<T: Config<I>, I: 'static = ()>(crate::Pallet<T, I>);
32
33/// Trait that must be implemented by runtime to benchmark the parachains finality pallet.
34pub trait Config<I: 'static>: crate::Config<I> {
35	/// Returns vector of supported parachains.
36	fn parachains() -> Vec<ParaId>;
37	/// Generate parachain heads proof and prepare environment for verifying this proof.
38	fn prepare_parachain_heads_proof(
39		parachains: &[ParaId],
40		parachain_head_size: u32,
41		proof_params: UnverifiedStorageProofParams,
42	) -> (RelayBlockNumber, RelayBlockHash, ParaHeadsProof, Vec<(ParaId, ParaHash)>);
43}
44
45benchmarks_instance_pallet! {
46	where_clause {
47		where
48			<T as pallet_bridge_grandpa::Config<T::BridgesGrandpaPalletInstance>>::BridgedChain:
49				bp_runtime::Chain<
50					BlockNumber = RelayBlockNumber,
51					Hash = RelayBlockHash,
52					Hasher = RelayBlockHasher,
53				>,
54	}
55
56	// Benchmark `submit_parachain_heads` extrinsic with different number of parachains.
57	submit_parachain_heads_with_n_parachains {
58		let p in 1..(T::parachains().len() + 1) as u32;
59
60		let sender = account("sender", 0, 0);
61		let mut parachains = T::parachains();
62		let _ = if p <= parachains.len() as u32 {
63			parachains.split_off(p as usize)
64		} else {
65			Default::default()
66		};
67		tracing::trace!(target: crate::LOG_TARGET, "=== {:?}", parachains.len());
68		let (relay_block_number, relay_block_hash, parachain_heads_proof, parachains_heads) = T::prepare_parachain_heads_proof(
69			&parachains,
70			DEFAULT_PARACHAIN_HEAD_SIZE,
71			UnverifiedStorageProofParams::default(),
72		);
73		let at_relay_block = (relay_block_number, relay_block_hash);
74	}: submit_parachain_heads(RawOrigin::Signed(sender), at_relay_block, parachains_heads, parachain_heads_proof)
75	verify {
76		for parachain in parachains {
77			assert!(crate::Pallet::<T, I>::best_parachain_head(parachain).is_some());
78		}
79	}
80
81	// Benchmark `submit_parachain_heads` extrinsic with 1kb proof size.
82	submit_parachain_heads_with_1kb_proof {
83		let sender = account("sender", 0, 0);
84		let parachains = vec![T::parachains()[0]];
85		let (relay_block_number, relay_block_hash, parachain_heads_proof, parachains_heads) = T::prepare_parachain_heads_proof(
86			&parachains,
87			DEFAULT_PARACHAIN_HEAD_SIZE,
88			UnverifiedStorageProofParams::from_db_size(1024),
89		);
90		let at_relay_block = (relay_block_number, relay_block_hash);
91	}: submit_parachain_heads(RawOrigin::Signed(sender), at_relay_block, parachains_heads, parachain_heads_proof)
92	verify {
93		for parachain in parachains {
94			assert!(crate::Pallet::<T, I>::best_parachain_head(parachain).is_some());
95		}
96	}
97
98	// Benchmark `submit_parachain_heads` extrinsic with 16kb proof size.
99	submit_parachain_heads_with_16kb_proof {
100		let sender = account("sender", 0, 0);
101		let parachains = vec![T::parachains()[0]];
102		let (relay_block_number, relay_block_hash, parachain_heads_proof, parachains_heads) = T::prepare_parachain_heads_proof(
103			&parachains,
104			DEFAULT_PARACHAIN_HEAD_SIZE,
105			UnverifiedStorageProofParams::from_db_size(16 * 1024),
106		);
107		let at_relay_block = (relay_block_number, relay_block_hash);
108	}: submit_parachain_heads(RawOrigin::Signed(sender), at_relay_block, parachains_heads, parachain_heads_proof)
109	verify {
110		for parachain in parachains {
111			assert!(crate::Pallet::<T, I>::best_parachain_head(parachain).is_some());
112		}
113	}
114
115	impl_benchmark_test_suite!(Pallet, crate::mock::new_test_ext(), crate::mock::TestRuntime)
116}