pezsnowbridge-test-utils 0.1.0

Snowbridge test utilities
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2023 Snowfork <hello@snowfork.com>
use pezbp_relayers::{PaymentProcedure, RewardLedger, RewardsAccountOwner, RewardsAccountParams};
use codec::{Decode, DecodeWithMemTracking, Encode, MaxEncodedLen};
use pezframe_support::{parameter_types, pezpallet_prelude::DispatchResult, pezsp_runtime};
use scale_info::TypeInfo;
use xcm::opaque::latest::Location;

/// Showcasing that we can handle multiple different rewards with the same pezpallet.
#[derive(
	Clone,
	Copy,
	Debug,
	Decode,
	DecodeWithMemTracking,
	Encode,
	Eq,
	MaxEncodedLen,
	PartialEq,
	TypeInfo,
)]
pub enum BridgeReward {
	/// Rewards for Snowbridge.
	Snowbridge,
}

pub struct MockPaymentProcedure;

// Provide a no-op or mock implementation for the required trait
impl PaymentProcedure<pezsp_runtime::AccountId32, RewardsAccountParams<u64>, u128>
	for MockPaymentProcedure
{
	type Error = DispatchResult;
	type Beneficiary = Location;
	fn pay_reward(
		_who: &pezsp_runtime::AccountId32,
		_reward_params: pezbp_relayers::RewardsAccountParams<u64>,
		_reward_balance: u128,
		_beneficiary: Self::Beneficiary,
	) -> Result<(), Self::Error> {
		Ok(())
	}
}

impl From<BridgeReward> for RewardsAccountParams<u64> {
	fn from(_bridge_reward: BridgeReward) -> Self {
		RewardsAccountParams::new(1, [0; 4], RewardsAccountOwner::ThisChain)
	}
}

parameter_types! {
	pub static RegisteredRewardsCount: u128 = 0;
	pub static RegisteredRewardAmount: u128 = 0;
}

pub struct MockRewardLedger;

impl RewardLedger<pezsp_runtime::AccountId32, BridgeReward, u128> for MockRewardLedger {
	fn register_reward(
		_relayer: &pezsp_runtime::AccountId32,
		_reward: BridgeReward,
		reward_balance: u128,
	) {
		RegisteredRewardsCount::set(RegisteredRewardsCount::get().saturating_add(1));
		RegisteredRewardAmount::set(reward_balance);
	}
}