Refactoring Checkpoint: (WIP)
This commit is contained in:
@@ -0,0 +1,360 @@
|
||||
// pezkuwi/pallets/pez-rewards/src/mock.rs (v1.0 - dev_accounts FIXED)
|
||||
|
||||
use crate as pezpallet_pez_rewards;
|
||||
use pezframe_support::{
|
||||
assert_ok, construct_runtime, parameter_types,
|
||||
traits::{
|
||||
fungibles::Mutate, AsEnsureOriginWithArg, ConstU128, ConstU32, ConstU64, OnFinalize,
|
||||
OnInitialize,
|
||||
},
|
||||
PalletId,
|
||||
};
|
||||
use pezframe_system::{EnsureRoot, EnsureSigned};
|
||||
use pezsp_core::H256;
|
||||
use pezsp_runtime::{
|
||||
traits::{BlakeTwo256, IdentityLookup},
|
||||
BuildStorage,
|
||||
};
|
||||
|
||||
// --- Dummy Trait Implementations for pezpallet-trust ---
|
||||
pub struct MockStakingScoreProvider;
|
||||
impl pezpallet_trust::StakingScoreProvider<H256, u64> for MockStakingScoreProvider {
|
||||
fn get_staking_score(_who: &H256) -> (u32, u64) {
|
||||
(0, 0)
|
||||
}
|
||||
}
|
||||
pub struct MockReferralScoreProvider;
|
||||
impl pezpallet_trust::ReferralScoreProvider<H256> for MockReferralScoreProvider {
|
||||
fn get_referral_score(_who: &H256) -> u32 {
|
||||
0
|
||||
}
|
||||
}
|
||||
pub struct MockPerwerdeScoreProvider;
|
||||
impl pezpallet_trust::PerwerdeScoreProvider<H256> for MockPerwerdeScoreProvider {
|
||||
fn get_perwerde_score(_who: &H256) -> u32 {
|
||||
0
|
||||
}
|
||||
}
|
||||
pub struct MockTikiScoreProvider;
|
||||
impl pezpallet_trust::TikiScoreProvider<H256> for MockTikiScoreProvider {
|
||||
fn get_tiki_score(_who: &H256) -> u32 {
|
||||
0
|
||||
}
|
||||
}
|
||||
pub struct MockCitizenshipStatusProvider;
|
||||
impl pezpallet_trust::CitizenshipStatusProvider<H256> for MockCitizenshipStatusProvider {
|
||||
fn is_citizen(_who: &H256) -> bool {
|
||||
false
|
||||
}
|
||||
}
|
||||
|
||||
type Block = pezframe_system::mocking::MockBlock<Test>;
|
||||
type Balance = u128;
|
||||
type BlockNumber = u64;
|
||||
type Weight = pezframe_support::weights::Weight;
|
||||
|
||||
// Configure runtime
|
||||
construct_runtime!(
|
||||
pub enum Test
|
||||
{
|
||||
System: pezframe_system,
|
||||
Balances: pezpallet_balances,
|
||||
Assets: pezpallet_assets,
|
||||
IdentityKyc: pezpallet_identity_kyc,
|
||||
Trust: pezpallet_trust,
|
||||
PezRewards: pezpallet_pez_rewards,
|
||||
}
|
||||
);
|
||||
|
||||
// --- pezframe_system::Config ---
|
||||
parameter_types! {
|
||||
pub const BlockHashCount: BlockNumber = 250;
|
||||
pub const SS58Prefix: u8 = 42;
|
||||
}
|
||||
impl pezframe_system::Config for Test {
|
||||
type BaseCallFilter = pezframe_support::traits::Everything;
|
||||
type BlockWeights = ();
|
||||
type BlockLength = ();
|
||||
type DbWeight = ();
|
||||
type RuntimeOrigin = RuntimeOrigin;
|
||||
type RuntimeCall = RuntimeCall;
|
||||
type Nonce = u64;
|
||||
type Hash = H256;
|
||||
type Hashing = BlakeTwo256;
|
||||
type AccountId = H256;
|
||||
type Lookup = IdentityLookup<Self::AccountId>;
|
||||
type Block = Block;
|
||||
type RuntimeEvent = RuntimeEvent;
|
||||
type BlockHashCount = BlockHashCount;
|
||||
type Version = ();
|
||||
type PalletInfo = PalletInfo;
|
||||
type AccountData = pezpallet_balances::AccountData<Balance>;
|
||||
type OnNewAccount = ();
|
||||
type OnKilledAccount = ();
|
||||
type SystemWeightInfo = ();
|
||||
type SS58Prefix = SS58Prefix;
|
||||
type OnSetCode = ();
|
||||
type MaxConsumers = ConstU32<16>;
|
||||
type RuntimeTask = ();
|
||||
type SingleBlockMigrations = ();
|
||||
type MultiBlockMigrator = ();
|
||||
type PreInherents = ();
|
||||
type PostInherents = ();
|
||||
type PostTransactions = ();
|
||||
type ExtensionsWeightInfo = ();
|
||||
}
|
||||
|
||||
// --- pezpallet_balances::Config ---
|
||||
parameter_types! {
|
||||
pub const ExistentialDeposit: Balance = 1;
|
||||
pub const MaxLocks: u32 = 50;
|
||||
pub const MaxReserves: u32 = 50;
|
||||
}
|
||||
impl pezpallet_balances::Config for Test {
|
||||
type MaxLocks = MaxLocks;
|
||||
type MaxReserves = MaxReserves;
|
||||
type ReserveIdentifier = [u8; 8];
|
||||
type Balance = Balance;
|
||||
type DustRemoval = ();
|
||||
type RuntimeEvent = RuntimeEvent;
|
||||
type ExistentialDeposit = ExistentialDeposit;
|
||||
type AccountStore = System;
|
||||
type WeightInfo = ();
|
||||
type RuntimeHoldReason = ();
|
||||
type RuntimeFreezeReason = ();
|
||||
type FreezeIdentifier = ();
|
||||
type MaxFreezes = ();
|
||||
type DoneSlashHandler = ();
|
||||
}
|
||||
|
||||
// --- pezpallet_assets::Config ---
|
||||
parameter_types! {
|
||||
pub const AssetDeposit: Balance = 100;
|
||||
pub const ApprovalDeposit: Balance = 1;
|
||||
pub const StringLimit: u32 = 50;
|
||||
pub const MetadataDepositBase: Balance = 10;
|
||||
pub const MetadataDepositPerByte: Balance = 1;
|
||||
pub const AssetAccountDeposit: Balance = 1;
|
||||
}
|
||||
impl pezpallet_assets::Config for Test {
|
||||
type RuntimeEvent = RuntimeEvent;
|
||||
type Balance = Balance;
|
||||
type AssetId = u32;
|
||||
type AssetIdParameter = u32;
|
||||
type Currency = Balances;
|
||||
type CreateOrigin = AsEnsureOriginWithArg<EnsureSigned<Self::AccountId>>;
|
||||
type ForceOrigin = EnsureRoot<Self::AccountId>;
|
||||
type AssetDeposit = AssetDeposit;
|
||||
type AssetAccountDeposit = AssetAccountDeposit;
|
||||
type MetadataDepositBase = MetadataDepositBase;
|
||||
type MetadataDepositPerByte = MetadataDepositPerByte;
|
||||
type ApprovalDeposit = ApprovalDeposit;
|
||||
type StringLimit = StringLimit;
|
||||
type Freezer = ();
|
||||
type Extra = ();
|
||||
type CallbackHandle = ();
|
||||
type WeightInfo = ();
|
||||
type RemoveItemsLimit = ConstU32<1000>;
|
||||
type Holder = ();
|
||||
type ReserveData = ();
|
||||
#[cfg(feature = "runtime-benchmarks")]
|
||||
type BenchmarkHelper = ();
|
||||
}
|
||||
|
||||
// --- pezpallet_identity_kyc::Config ---
|
||||
pub struct NoOpOnKycApproved;
|
||||
impl pezpallet_identity_kyc::types::OnKycApproved<H256> for NoOpOnKycApproved {
|
||||
fn on_kyc_approved(_who: &H256, _referrer: &H256) {}
|
||||
}
|
||||
|
||||
pub struct NoOpOnCitizenshipRevoked;
|
||||
impl pezpallet_identity_kyc::types::OnCitizenshipRevoked<H256> for NoOpOnCitizenshipRevoked {
|
||||
fn on_citizenship_revoked(_who: &H256) {}
|
||||
}
|
||||
|
||||
pub struct NoOpCitizenNftProvider;
|
||||
impl pezpallet_identity_kyc::types::CitizenNftProvider<H256> for NoOpCitizenNftProvider {
|
||||
fn mint_citizen_nft(_who: &H256) -> Result<(), pezsp_runtime::DispatchError> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn mint_citizen_nft_confirmed(_who: &H256) -> Result<(), pezsp_runtime::DispatchError> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn burn_citizen_nft(_who: &H256) -> Result<(), pezsp_runtime::DispatchError> {
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
parameter_types! {
|
||||
pub const KycApplicationDeposit: Balance = 100;
|
||||
pub const MaxStringLength: u32 = 128;
|
||||
pub const MaxCidLength: u32 = 128;
|
||||
}
|
||||
|
||||
impl pezpallet_identity_kyc::Config for Test {
|
||||
type RuntimeEvent = RuntimeEvent;
|
||||
type Currency = Balances;
|
||||
type GovernanceOrigin = EnsureRoot<H256>;
|
||||
type WeightInfo = ();
|
||||
type OnKycApproved = NoOpOnKycApproved;
|
||||
type OnCitizenshipRevoked = NoOpOnCitizenshipRevoked;
|
||||
type CitizenNftProvider = NoOpCitizenNftProvider;
|
||||
type KycApplicationDeposit = KycApplicationDeposit;
|
||||
type MaxStringLength = MaxStringLength;
|
||||
type MaxCidLength = MaxCidLength;
|
||||
}
|
||||
|
||||
// --- pezpallet_trust::Config ---
|
||||
pub struct MockTrustScore;
|
||||
impl pezpallet_trust::TrustScoreProvider<H256> for MockTrustScore {
|
||||
fn trust_score_of(account: &H256) -> u128 {
|
||||
if *account == alice() {
|
||||
100
|
||||
} else if *account == bob() {
|
||||
50
|
||||
} else if *account == charlie() {
|
||||
75
|
||||
} else {
|
||||
0
|
||||
}
|
||||
}
|
||||
}
|
||||
parameter_types! {
|
||||
pub const MaxBatchSize: u32 = 100;
|
||||
}
|
||||
|
||||
impl pezpallet_trust::Config for Test {
|
||||
type RuntimeEvent = RuntimeEvent;
|
||||
type WeightInfo = ();
|
||||
type Score = u128;
|
||||
type ScoreMultiplierBase = ConstU128<1>;
|
||||
type UpdateInterval = ConstU64<100>;
|
||||
type MaxBatchSize = MaxBatchSize;
|
||||
type StakingScoreSource = MockStakingScoreProvider;
|
||||
type ReferralScoreSource = MockReferralScoreProvider;
|
||||
type PerwerdeScoreSource = MockPerwerdeScoreProvider;
|
||||
type TikiScoreSource = MockTikiScoreProvider;
|
||||
type CitizenshipSource = MockCitizenshipStatusProvider;
|
||||
}
|
||||
|
||||
// --- pezpallet_pez_rewards::Config ---
|
||||
parameter_types! {
|
||||
pub const IncentivePotId: PalletId = PalletId(*b"pez/rpot");
|
||||
pub const PezAssetId: u32 = 1;
|
||||
pub ClawbackRecipient: H256 = H256::from_low_u64_be(999);
|
||||
}
|
||||
pub struct MockWeightInfo;
|
||||
impl crate::weights::WeightInfo for MockWeightInfo {
|
||||
fn initialize_rewards_system() -> Weight {
|
||||
Weight::zero()
|
||||
}
|
||||
fn record_trust_score() -> Weight {
|
||||
Weight::zero()
|
||||
}
|
||||
fn finalize_epoch() -> Weight {
|
||||
Weight::zero()
|
||||
}
|
||||
fn claim_reward() -> Weight {
|
||||
Weight::zero()
|
||||
}
|
||||
fn close_epoch() -> Weight {
|
||||
Weight::zero()
|
||||
}
|
||||
fn register_parliamentary_nft_owner() -> Weight {
|
||||
Weight::zero()
|
||||
}
|
||||
}
|
||||
impl pezpallet_pez_rewards::Config for Test {
|
||||
type RuntimeEvent = RuntimeEvent;
|
||||
type Assets = Assets;
|
||||
type TrustScoreSource = MockTrustScore;
|
||||
type IncentivePotId = IncentivePotId;
|
||||
type PezAssetId = PezAssetId;
|
||||
type ClawbackRecipient = ClawbackRecipient;
|
||||
type WeightInfo = MockWeightInfo;
|
||||
type ForceOrigin = EnsureRoot<Self::AccountId>;
|
||||
type CollectionId = u32;
|
||||
type ItemId = u32;
|
||||
}
|
||||
|
||||
// --- Helper Fonksiyonlar ---
|
||||
pub fn alice() -> H256 {
|
||||
H256::from_low_u64_be(1)
|
||||
}
|
||||
pub fn bob() -> H256 {
|
||||
H256::from_low_u64_be(2)
|
||||
}
|
||||
pub fn charlie() -> H256 {
|
||||
H256::from_low_u64_be(3)
|
||||
}
|
||||
pub fn dave() -> H256 {
|
||||
H256::from_low_u64_be(4)
|
||||
}
|
||||
|
||||
// --- new_test_ext ---
|
||||
pub fn new_test_ext() -> pezsp_io::TestExternalities {
|
||||
let mut t = pezframe_system::GenesisConfig::<Test>::default().build_storage().unwrap();
|
||||
|
||||
// BUG FIX: dev_accounts field added (Option type)
|
||||
pezpallet_balances::GenesisConfig::<Test> {
|
||||
balances: vec![
|
||||
(alice(), 1_000_000_000_000_000),
|
||||
(bob(), 1_000_000_000_000_000),
|
||||
(charlie(), 1_000_000_000_000_000),
|
||||
(dave(), 1_000_000_000_000_000),
|
||||
(ClawbackRecipient::get(), 1_000_000_000_000_000),
|
||||
],
|
||||
dev_accounts: None, // No need for dev account in test environment
|
||||
}
|
||||
.assimilate_storage(&mut t)
|
||||
.unwrap();
|
||||
|
||||
pezpallet_assets::GenesisConfig::<Test> {
|
||||
assets: vec![(PezAssetId::get(), alice(), true, 1)],
|
||||
metadata: vec![(PezAssetId::get(), b"Pez Token".to_vec(), b"PEZ".to_vec(), 12)],
|
||||
accounts: vec![(
|
||||
PezAssetId::get(),
|
||||
PezRewards::incentive_pot_account_id(),
|
||||
1_000_000_000_000_000,
|
||||
)],
|
||||
reserves: vec![],
|
||||
next_asset_id: Some(PezAssetId::get() + 1),
|
||||
}
|
||||
.assimilate_storage(&mut t)
|
||||
.unwrap();
|
||||
|
||||
let mut ext = pezsp_io::TestExternalities::new(t);
|
||||
ext.execute_with(|| {
|
||||
System::set_block_number(1);
|
||||
assert_ok!(PezRewards::initialize_rewards_system(RuntimeOrigin::root()));
|
||||
});
|
||||
ext
|
||||
}
|
||||
|
||||
// --- Block Advancement Helper ---
|
||||
pub fn advance_blocks(n: BlockNumber) {
|
||||
let target = System::block_number() + n;
|
||||
while System::block_number() < target {
|
||||
if System::block_number() > 0 {
|
||||
AllPalletsWithSystem::on_finalize(System::block_number());
|
||||
}
|
||||
System::set_block_number(System::block_number() + 1);
|
||||
AllPalletsWithSystem::on_initialize(System::block_number());
|
||||
}
|
||||
}
|
||||
|
||||
// --- Other Helper Functions ---
|
||||
pub fn pez_balance(account: &H256) -> Balance {
|
||||
Assets::balance(PezAssetId::get(), account)
|
||||
}
|
||||
|
||||
pub fn fund_incentive_pot(amount: Balance) {
|
||||
let pot = PezRewards::incentive_pot_account_id();
|
||||
assert_ok!(Assets::mint_into(PezAssetId::get(), &pot, amount));
|
||||
}
|
||||
|
||||
pub fn register_nft_owner(nft_id: u32, owner: H256) {
|
||||
PezRewards::do_register_parliamentary_nft_owner(nft_id, owner);
|
||||
}
|
||||
Reference in New Issue
Block a user