Refactoring Checkpoint: (WIP)

pezcumulus/teyrchains/pezpallets/identity-kyc/src/mock.rs

@@ -0,0 +1,140 @@
+use crate as pezpallet_identity_kyc;
+use pezframe_support::{
+	construct_runtime, derive_impl, parameter_types,
+	traits::{ConstU128, ConstU32},
+};
+use pezframe_system::EnsureRoot;
+use pezsp_core::H256;
+use pezsp_runtime::BuildStorage;
+
+type Block = pezframe_system::mocking::MockBlock<Test>;
+pub type AccountId = u64;
+pub type Balance = u128;
+
+// Founding citizen for genesis tests
+pub const FOUNDER: AccountId = 100;
+pub const CITIZEN_1: AccountId = 1;
+pub const CITIZEN_2: AccountId = 2;
+pub const APPLICANT: AccountId = 3;
+
+construct_runtime!(
+	pub enum Test
+	{
+		System: pezframe_system,
+		Balances: pezpallet_balances,
+		IdentityKyc: pezpallet_identity_kyc,
+	}
+);
+
+#[derive_impl(pezframe_system::config_preludes::TestDefaultConfig)]
+impl pezframe_system::Config for Test {
+	type Block = Block;
+	type AccountData = pezpallet_balances::AccountData<Balance>;
+}
+
+#[derive_impl(pezpallet_balances::config_preludes::TestDefaultConfig)]
+impl pezpallet_balances::Config for Test {
+	type Balance = Balance;
+	type ExistentialDeposit = ConstU128<1>;
+	type AccountStore = System;
+}
+
+parameter_types! {
+	pub const KycApplicationDepositAmount: Balance = 100;
+	pub const MaxStringLen: u32 = 50;
+	pub const MaxCidLen: u32 = 128;
+}
+
+// Mock implementation for OnKycApproved hook
+// UPDATED: Now includes referrer parameter
+pub struct MockOnKycApproved;
+impl crate::types::OnKycApproved<AccountId> for MockOnKycApproved {
+	fn on_kyc_approved(_who: &AccountId, _referrer: &AccountId) {
+		// No-op for tests - in real runtime this triggers referral pallet
+	}
+}
+
+// Mock implementation for OnCitizenshipRevoked hook
+pub struct MockOnCitizenshipRevoked;
+impl crate::types::OnCitizenshipRevoked<AccountId> for MockOnCitizenshipRevoked {
+	fn on_citizenship_revoked(_who: &AccountId) {
+		// No-op for tests - in real runtime this triggers penalty system
+	}
+}
+
+// Mock implementation for CitizenNftProvider
+pub struct MockCitizenNftProvider;
+impl crate::types::CitizenNftProvider<AccountId> for MockCitizenNftProvider {
+	fn mint_citizen_nft(_who: &AccountId) -> pezsp_runtime::DispatchResult {
+		Ok(())
+	}
+
+	fn mint_citizen_nft_confirmed(_who: &AccountId) -> pezsp_runtime::DispatchResult {
+		Ok(())
+	}
+
+	fn burn_citizen_nft(_who: &AccountId) -> pezsp_runtime::DispatchResult {
+		Ok(())
+	}
+}
+
+impl crate::Config for Test {
+	type RuntimeEvent = RuntimeEvent;
+	type Currency = Balances;
+	type GovernanceOrigin = EnsureRoot<Self::AccountId>;
+	type WeightInfo = ();
+	type OnKycApproved = MockOnKycApproved;
+	type OnCitizenshipRevoked = MockOnCitizenshipRevoked;
+	type CitizenNftProvider = MockCitizenNftProvider;
+	type KycApplicationDeposit = KycApplicationDepositAmount;
+	type MaxStringLength = MaxStringLen;
+	type MaxCidLength = MaxCidLen;
+}
+
+/// Build test externalities with founding citizens
+pub fn new_test_ext() -> pezsp_io::TestExternalities {
+	let mut t = pezframe_system::GenesisConfig::<Test>::default().build_storage().unwrap();
+
+	pezpallet_balances::GenesisConfig::<Test> {
+		balances: vec![
+			(FOUNDER, 1_000_000),
+			(CITIZEN_1, 10_000),
+			(CITIZEN_2, 10_000),
+			(APPLICANT, 10_000),
+		],
+		..Default::default()
+	}
+	.assimilate_storage(&mut t)
+	.unwrap();
+
+	// Add founding citizen via genesis config
+	pezpallet_identity_kyc::GenesisConfig::<Test> {
+		founding_citizens: vec![
+			(FOUNDER, H256::from_low_u64_be(1)),   // Founder is pre-approved
+			(CITIZEN_1, H256::from_low_u64_be(2)), // Citizen 1 is pre-approved
+		],
+		_phantom: Default::default(),
+	}
+	.assimilate_storage(&mut t)
+	.unwrap();
+
+	let mut ext = pezsp_io::TestExternalities::new(t);
+	ext.execute_with(|| System::set_block_number(1));
+	ext
+}
+
+/// Build test externalities without founding citizens (for edge case tests)
+pub fn new_test_ext_empty() -> pezsp_io::TestExternalities {
+	let mut t = pezframe_system::GenesisConfig::<Test>::default().build_storage().unwrap();
+
+	pezpallet_balances::GenesisConfig::<Test> {
+		balances: vec![(FOUNDER, 1_000_000), (CITIZEN_1, 10_000), (APPLICANT, 10_000)],
+		..Default::default()
+	}
+	.assimilate_storage(&mut t)
+	.unwrap();
+
+	let mut ext = pezsp_io::TestExternalities::new(t);
+	ext.execute_with(|| System::set_block_number(1));
+	ext
+}