Refactoring Checkpoint: (WIP)

pezcumulus/teyrchains/pezpallets/pez-treasury/src/benchmarking.rs

@@ -0,0 +1,92 @@
+// pezkuwi/pallets/pez-treasury/src/benchmarking.rs
+
+#![cfg(feature = "runtime-benchmarks")]
+
+use super::*;
+use crate::Pallet as PezTreasury;
+use pezframe_benchmarking::v2::*;
+use pezframe_support::traits::{
+	fungibles::{Inspect, Mutate},
+	Get, // HATA GİDERİLDİ: .get() fonksiyonu için bu trait eklendi
+};
+use pezframe_system::RawOrigin;
+use pezsp_runtime::traits::{Saturating, Zero};
+
+#[benchmarks]
+mod benchmarks {
+	use super::*;
+
+	#[benchmark]
+	fn initialize_treasury() {
+		crate::TreasuryStartBlock::<T>::kill();
+		crate::HalvingInfo::<T>::kill();
+		crate::NextReleaseMonth::<T>::kill();
+
+		#[extrinsic_call]
+		initialize_treasury(RawOrigin::Root);
+
+		assert!(crate::TreasuryStartBlock::<T>::get().is_some());
+		let halving_info = crate::HalvingInfo::<T>::get();
+		assert_eq!(halving_info.current_period, 0);
+		assert!(!halving_info.monthly_amount.is_zero());
+	}
+
+	#[benchmark]
+	fn force_genesis_distribution() {
+		// Clear the flag to allow benchmark run (tests the new storage operation)
+		crate::GenesisDistributionDone::<T>::kill();
+
+		#[block]
+		{
+			PezTreasury::<T>::do_genesis_distribution().unwrap();
+		}
+
+		let treasury_account = PezTreasury::<T>::treasury_account_id();
+		let presale_account = T::PresaleAccount::get();
+		let founder_account = T::FounderAccount::get();
+
+		assert!(!T::Assets::balance(T::PezAssetId::get(), &treasury_account).is_zero());
+		assert!(!T::Assets::balance(T::PezAssetId::get(), &presale_account).is_zero());
+		assert!(!T::Assets::balance(T::PezAssetId::get(), &founder_account).is_zero());
+	}
+
+	#[benchmark]
+	fn release_monthly_funds() {
+		// Setup
+		crate::TreasuryStartBlock::<T>::kill();
+		crate::HalvingInfo::<T>::kill();
+		crate::NextReleaseMonth::<T>::kill();
+		crate::GenesisDistributionDone::<T>::kill();
+		// Deprecated `remove_all` yerine `clear` kullanılıyor.
+		crate::MonthlyReleases::<T>::clear(u32::MAX, None);
+
+		// First do genesis distribution to properly fund the treasury
+		PezTreasury::<T>::do_genesis_distribution().unwrap();
+		PezTreasury::<T>::do_initialize_treasury().unwrap();
+
+		let treasury_account = PezTreasury::<T>::treasury_account_id();
+		let initial_monthly_amount = PezTreasury::<T>::halving_info().monthly_amount;
+		let incentive_amount = initial_monthly_amount * 75u32.into() / 100u32.into();
+		let government_amount = initial_monthly_amount.saturating_sub(incentive_amount);
+
+		// Ensure treasury has MORE than enough balance for the release
+		// Mint additional 10x the monthly amount to ensure sufficient balance
+		let _ = T::Assets::mint_into(
+			T::PezAssetId::get(),
+			&treasury_account,
+			initial_monthly_amount * 10u32.into(),
+		);
+
+		let current_block = pezframe_system::Pallet::<T>::block_number();
+		let target_block = current_block + crate::BLOCKS_PER_MONTH.into() + 1u32.into();
+		pezframe_system::Pallet::<T>::set_block_number(target_block);
+
+		#[extrinsic_call]
+		release_monthly_funds(RawOrigin::Root);
+
+		assert_eq!(PezTreasury::<T>::get_incentive_pot_balance(), incentive_amount);
+		assert_eq!(PezTreasury::<T>::get_government_pot_balance(), government_amount);
+	}
+
+	impl_benchmark_test_suite!(PezTreasury, crate::mock::new_test_ext(), crate::mock::Test);
+}