fix: Complete snowbridge pezpallet rebrand and critical bug fixes

- snowbridge-pezpallet-* → pezsnowbridge-pezpallet-* (201 refs) - pallet/ directories → pezpallet/ (4 locations) - Fixed pezpallet.rs self-include recursion bug - Fixed sc-chain-spec hardcoded crate name in derive macro - Reverted .pezpallet_by_name() to .pallet_by_name() (subxt API) - Added BizinikiwiConfig type alias for zombienet tests - Deleted obsolete session state files Verified: pezsnowbridge-pezpallet-*, pezpallet-staking, pezpallet-staking-async, pezframe-benchmarking-cli all pass cargo check
2025-12-16 09:57:23 +03:00
parent eea003e14d
commit 3139ffa25e
3022 changed files with 42157 additions and 23579 deletions
@@ -0,0 +1,198 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Parity Bridges Common.
+
+// Parity Bridges Common is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Parity Bridges Common is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Parity Bridges Common.  If not, see <http://www.gnu.org/licenses/>.
+
+//! Benchmarks for the relayers Pezpallet.
+
+#![cfg(feature = "runtime-benchmarks")]
+
+use crate::*;
+
+use pezframe_benchmarking::v2::*;
+use pezframe_support::{assert_ok, weights::Weight};
+use pezframe_system::RawOrigin;
+use pezsp_runtime::traits::One;
+
+/// Reward amount that is (hopefully) is larger than existential deposit across all chains.
+const REWARD_AMOUNT: u32 = u32::MAX;
+
+/// Pezpallet we're benchmarking here.
+pub struct Pezpallet<T: Config<I>, I: 'static = ()>(crate::Pezpallet<T, I>);
+
+/// Trait that must be implemented by runtime.
+pub trait Config<I: 'static = ()>: crate::Config<I> {
+	/// `T::Reward` to use in benchmarks.
+	fn bench_reward() -> Self::Reward;
+	/// Prepare environment for paying given reward for serving given lane.
+	fn prepare_rewards_account(
+		reward_kind: Self::Reward,
+		reward: Self::RewardBalance,
+	) -> Option<BeneficiaryOf<Self, I>>;
+	/// Give enough balance to given account.
+	fn deposit_account(account: Self::AccountId, balance: Self::Balance);
+}
+
+fn assert_last_event<T: Config<I>, I: 'static>(
+	generic_event: <T as pezpallet::Config<I>>::RuntimeEvent,
+) {
+	pezframe_system::Pezpallet::<T>::assert_last_event(generic_event.into());
+}
+
+#[instance_benchmarks(
+	where
+		BeneficiaryOf<T, I>: From<<T as pezframe_system::Config>::AccountId>,
+)]
+mod benchmarks {
+	use super::*;
+
+	#[benchmark]
+	fn claim_rewards() {
+		let relayer: T::AccountId = whitelisted_caller();
+		let reward_kind = T::bench_reward();
+		let reward_balance = T::RewardBalance::from(REWARD_AMOUNT);
+		let _ = T::prepare_rewards_account(reward_kind, reward_balance);
+		RelayerRewards::<T, I>::insert(&relayer, reward_kind, reward_balance);
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(relayer.clone()), reward_kind);
+
+		// we can't check anything here, because `PaymentProcedure` is responsible for
+		// payment logic, so we assume that if call has succeeded, the procedure has
+		// also completed successfully
+		assert_last_event::<T, I>(
+			Event::RewardPaid {
+				relayer: relayer.clone(),
+				reward_kind,
+				reward_balance,
+				beneficiary: relayer.into(),
+			}
+			.into(),
+		);
+	}
+
+	#[benchmark]
+	fn claim_rewards_to() -> Result<(), BenchmarkError> {
+		let relayer: T::AccountId = whitelisted_caller();
+		let reward_kind = T::bench_reward();
+		let reward_balance = T::RewardBalance::from(REWARD_AMOUNT);
+
+		let Some(alternative_beneficiary) = T::prepare_rewards_account(reward_kind, reward_balance)
+		else {
+			return Err(BenchmarkError::Override(BenchmarkResult::from_weight(Weight::MAX)));
+		};
+		RelayerRewards::<T, I>::insert(&relayer, reward_kind, reward_balance);
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(relayer.clone()), reward_kind, alternative_beneficiary.clone());
+
+		// we can't check anything here, because `PaymentProcedure` is responsible for
+		// payment logic, so we assume that if call has succeeded, the procedure has
+		// also completed successfully
+		assert_last_event::<T, I>(
+			Event::RewardPaid {
+				relayer: relayer.clone(),
+				reward_kind,
+				reward_balance,
+				beneficiary: alternative_beneficiary,
+			}
+			.into(),
+		);
+
+		Ok(())
+	}
+
+	#[benchmark]
+	fn register() {
+		let relayer: T::AccountId = whitelisted_caller();
+		let valid_till = pezframe_system::Pezpallet::<T>::block_number()
+			.saturating_add(crate::Pezpallet::<T, I>::required_registration_lease())
+			.saturating_add(One::one())
+			.saturating_add(One::one());
+		T::deposit_account(relayer.clone(), crate::Pezpallet::<T, I>::required_stake());
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(relayer.clone()), valid_till);
+
+		assert!(crate::Pezpallet::<T, I>::is_registration_active(&relayer));
+	}
+
+	#[benchmark]
+	fn deregister() {
+		let relayer: T::AccountId = whitelisted_caller();
+		let valid_till = pezframe_system::Pezpallet::<T>::block_number()
+			.saturating_add(crate::Pezpallet::<T, I>::required_registration_lease())
+			.saturating_add(One::one())
+			.saturating_add(One::one());
+		T::deposit_account(relayer.clone(), crate::Pezpallet::<T, I>::required_stake());
+		crate::Pezpallet::<T, I>::register(RawOrigin::Signed(relayer.clone()).into(), valid_till)
+			.unwrap();
+		pezframe_system::Pezpallet::<T>::set_block_number(valid_till.saturating_add(One::one()));
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(relayer.clone()));
+
+		assert!(!crate::Pezpallet::<T, I>::is_registration_active(&relayer));
+	}
+
+	// Benchmark `slash_and_deregister` method of the pezpallet. We are adding this weight to
+	// the weight of message delivery call if `BridgeRelayersTransactionExtension` signed extension
+	// is deployed at runtime level.
+	#[benchmark]
+	fn slash_and_deregister() {
+		// prepare and register relayer account
+		let relayer: T::AccountId = whitelisted_caller();
+		let valid_till = pezframe_system::Pezpallet::<T>::block_number()
+			.saturating_add(crate::Pezpallet::<T, I>::required_registration_lease())
+			.saturating_add(One::one())
+			.saturating_add(One::one());
+		T::deposit_account(relayer.clone(), crate::Pezpallet::<T, I>::required_stake());
+		assert_ok!(crate::Pezpallet::<T, I>::register(
+			RawOrigin::Signed(relayer.clone()).into(),
+			valid_till
+		));
+
+		// create slash destination account
+		let slash_destination: T::AccountId = whitelisted_caller();
+		T::deposit_account(slash_destination.clone(), Zero::zero());
+
+		#[block]
+		{
+			crate::Pezpallet::<T, I>::slash_and_deregister(
+				&relayer,
+				bp_relayers::ExplicitOrAccountParams::Explicit::<_, ()>(slash_destination),
+			);
+		}
+
+		assert!(!crate::Pezpallet::<T, I>::is_registration_active(&relayer));
+	}
+
+	// Benchmark `register_relayer_reward` method of the pezpallet. We are adding this weight to
+	// the weight of message delivery call if `BridgeRelayersTransactionExtension` signed extension
+	// is deployed at runtime level.
+	#[benchmark]
+	fn register_relayer_reward() {
+		let reward_kind = T::bench_reward();
+		let relayer: T::AccountId = whitelisted_caller();
+
+		#[block]
+		{
+			crate::Pezpallet::<T, I>::register_relayer_reward(reward_kind, &relayer, One::one());
+		}
+
+		assert_eq!(RelayerRewards::<T, I>::get(relayer, &reward_kind), Some(One::one()));
+	}
+
+	impl_benchmark_test_suite!(Pezpallet, crate::mock::new_test_ext(), crate::mock::TestRuntime);
+}