Refactoring Checkpoint: (WIP)

pezcumulus/teyrchains/pezpallets/presale/src/mock.rs

@@ -0,0 +1,195 @@
+use crate as pezpallet_presale;
+use pezframe_support::{
+	parameter_types,
+	traits::{ConstU128, ConstU16, ConstU32, ConstU64},
+	PalletId,
+};
+use pezsp_core::H256;
+use pezsp_runtime::{
+	traits::{BlakeTwo256, IdentityLookup},
+	BuildStorage,
+};
+
+type Block = pezframe_system::mocking::MockBlock<Test>;
+
+// Configure a mock runtime to test the pallet.
+pezframe_support::construct_runtime!(
+	pub enum Test
+	{
+		System: pezframe_system,
+		Balances: pezpallet_balances,
+		Assets: pezpallet_assets,
+		Presale: pezpallet_presale,
+	}
+);
+
+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 = u64;
+	type Lookup = IdentityLookup<Self::AccountId>;
+	type Block = Block;
+	type RuntimeEvent = RuntimeEvent;
+	type BlockHashCount = ConstU64<250>;
+	type Version = ();
+	type PalletInfo = PalletInfo;
+	type AccountData = pezpallet_balances::AccountData<u128>;
+	type OnNewAccount = ();
+	type OnKilledAccount = ();
+	type SystemWeightInfo = ();
+	type SS58Prefix = ConstU16<42>;
+	type OnSetCode = ();
+	type MaxConsumers = ConstU32<16>;
+	type RuntimeTask = ();
+	type ExtensionsWeightInfo = ();
+	type SingleBlockMigrations = ();
+	type MultiBlockMigrator = ();
+	type PreInherents = ();
+	type PostInherents = ();
+	type PostTransactions = ();
+}
+
+impl pezpallet_balances::Config for Test {
+	type MaxLocks = ();
+	type MaxReserves = ();
+	type ReserveIdentifier = [u8; 8];
+	type Balance = u128;
+	type RuntimeEvent = RuntimeEvent;
+	type DustRemoval = ();
+	type ExistentialDeposit = ConstU128<1>;
+	type AccountStore = System;
+	type WeightInfo = ();
+	type FreezeIdentifier = ();
+	type MaxFreezes = ();
+	type RuntimeHoldReason = ();
+	type RuntimeFreezeReason = ();
+	type DoneSlashHandler = ();
+}
+
+impl pezpallet_assets::Config for Test {
+	type RuntimeEvent = RuntimeEvent;
+	type Balance = u128;
+	type AssetId = u32;
+	type AssetIdParameter = u32;
+	type Currency = Balances;
+	type CreateOrigin =
+		pezframe_support::traits::AsEnsureOriginWithArg<pezframe_system::EnsureSigned<u64>>;
+	type ForceOrigin = pezframe_system::EnsureRoot<u64>;
+	type AssetDeposit = ConstU128<1>;
+	type AssetAccountDeposit = ConstU128<0>; // No deposit required for test environment
+	type MetadataDepositBase = ConstU128<1>;
+	type MetadataDepositPerByte = ConstU128<1>;
+	type ApprovalDeposit = ConstU128<1>;
+	type StringLimit = ConstU32<50>;
+	type Freezer = ();
+	type Extra = ();
+	type WeightInfo = ();
+	type RemoveItemsLimit = ConstU32<1000>;
+	type CallbackHandle = ();
+	type Holder = ();
+	#[cfg(feature = "runtime-benchmarks")]
+	type BenchmarkHelper = ();
+}
+
+parameter_types! {
+	pub const PresalePalletId: PalletId = PalletId(*b"py/prsal");
+	pub const PlatformFeePercent: u8 = 2;
+	pub const MaxContributors: u32 = 10000;
+	pub const MaxBonusTiers: u32 = 5;
+	pub const MaxWhitelistedAccounts: u32 = 10000;
+	pub PlatformTreasuryAccount: u64 = 999;
+	pub StakingRewardPoolAccount: u64 = 998;
+}
+
+impl pezpallet_presale::Config for Test {
+	type RuntimeEvent = RuntimeEvent;
+	type AssetId = u32;
+	type Balance = u128;
+	type Assets = Assets;
+	type PalletId = PresalePalletId;
+	type PlatformTreasury = PlatformTreasuryAccount;
+	type StakingRewardPool = StakingRewardPoolAccount;
+	type PlatformFeePercent = PlatformFeePercent;
+	type MaxContributors = MaxContributors;
+	type MaxBonusTiers = MaxBonusTiers;
+	type MaxWhitelistedAccounts = MaxWhitelistedAccounts;
+	type CreatePresaleOrigin = pezframe_system::EnsureSigned<u64>;
+	type EmergencyOrigin = pezframe_system::EnsureRoot<u64>;
+	type PresaleWeightInfo = crate::weights::BizinikiwiWeight<Test>;
+}
+
+// Build genesis storage according to the mock runtime.
+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![
+			(1, 1_000_000_000_000_000),   // Alice
+			(2, 1_000_000_000_000_000),   // Bob
+			(3, 1_000_000_000_000_000),   // Charlie
+			(999, 1_000_000_000_000_000), // Platform Treasury
+			(998, 1_000_000_000_000_000), // Staking Pool
+		],
+		dev_accounts: None,
+	}
+	.assimilate_storage(&mut t)
+	.unwrap();
+
+	let mut ext = pezsp_io::TestExternalities::new(t);
+	ext.execute_with(|| System::set_block_number(1));
+	ext
+}
+
+// Helper to create assets
+pub fn create_assets() {
+	use pezframe_support::assert_ok;
+
+	// Create PEZ asset (ID: 1)
+	assert_ok!(Assets::force_create(
+		RuntimeOrigin::root(),
+		1u32,
+		1, // Alice as admin
+		true,
+		1
+	));
+
+	// Create wUSDT asset (ID: 2)
+	assert_ok!(Assets::force_create(
+		RuntimeOrigin::root(),
+		2u32,
+		1, // Alice as admin
+		true,
+		1
+	));
+}
+
+// Helper to mint assets to accounts
+pub fn mint_assets(asset_id: u32, account: u64, amount: u128) {
+	use pezframe_support::assert_ok;
+
+	assert_ok!(Assets::mint(RuntimeOrigin::signed(1), asset_id.into(), account, amount));
+}
+
+// Helper to get presale sub-account treasury for a specific presale ID
+pub fn presale_treasury(presale_id: u32) -> u64 {
+	use pezsp_io::hashing::blake2_256;
+
+	// Create a unique account ID for each presale by hashing pezpallet_id + presale_id
+	// This matches the logic in pezpallet_presale::Pallet::presale_account_id
+	let pezpallet_id = PresalePalletId::get();
+	let mut buf = Vec::new();
+	buf.extend_from_slice(&pezpallet_id.0[..]);
+	buf.extend_from_slice(&presale_id.to_le_bytes());
+	let hash = blake2_256(&buf);
+
+	// Convert hash to u64 (since Test uses u64 as AccountId)
+	// Take first 8 bytes and convert to u64
+	u64::from_le_bytes([hash[0], hash[1], hash[2], hash[3], hash[4], hash[5], hash[6], hash[7]])
+}