Thea: Fisherman and Batch Signing

pallets/thea-executor/src/benchmarking.rs

@@ -18,21 +18,37 @@
 
 #![cfg(feature = "runtime-benchmarks")]
 use super::*;
-
-use sp_runtime::traits::AccountIdConversion;
-use sp_std::{boxed::Box, vec, vec::Vec};
-
+use crate::Pallet as TheaExecutor;
 use frame_benchmarking::v1::{account, benchmarks};
 use frame_support::traits::{
-	fungible::Mutate as NativeMutate,
+	fungible::{Inspect as NativeInspect, Mutate as NativeMutate},
 	fungibles::{Create, Inspect, Mutate},
-	Get,
+	Get, OnInitialize,
 };
-use frame_system::RawOrigin;
-use sp_runtime::SaturatedConversion;
-use thea_primitives::types::{AssetMetadata, Deposit};
+use frame_system::{pallet_prelude::BlockNumberFor, RawOrigin};
+use parity_scale_codec::Decode;
+use polkadex_primitives::UNIT_BALANCE;
+use sp_runtime::{traits::AccountIdConversion, SaturatedConversion};
+use sp_std::{boxed::Box, collections::btree_set::BTreeSet, vec, vec::Vec};
+use thea_primitives::types::{AssetMetadata, Deposit, Withdraw};
 use xcm::VersionedMultiLocation;
 
+fn create_deposit<T: Config>(recipient: T::AccountId) -> Vec<Deposit<T::AccountId>> {
+	let mut pending_deposits = vec![];
+	let asset_id = 100;
+	for _i in 1..20 {
+		let deposit: Deposit<T::AccountId> = Deposit {
+			id: vec![],
+			recipient: recipient.clone(),
+			asset_id,
+			amount: 1_000_000_000_000,
+			extra: vec![],
+		};
+		pending_deposits.push(deposit);
+	}
+	pending_deposits
+}
+
 benchmarks! {
 	set_withdrawal_fee {
 		let r in 1 .. 1000;
@@ -53,21 +69,6 @@ benchmarks! {
 		assert_eq!(<Metadata<T>>::get(asset_id), Some(metadata));
 	}
 
-	claim_deposit {
-		let r in 1 .. 1000;
-		let account = account::<T::AccountId>("alice", 1, r);
-		let asset_id: <T as pallet::Config>::AssetId = 100u128.into();
-		let deposits = create_deposit::<T>(account.clone());
-		let metadata = AssetMetadata::new(10).unwrap();
-		<Metadata<T>>::insert(100, metadata);
-		<T as pallet::Config>::Currency::mint_into(&account, 100_000_000_000_000u128.saturated_into()).unwrap();
-		<ApprovedDeposits<T>>::insert(account.clone(), deposits);
-	}: _(RawOrigin::Signed(account.clone()), 10,account.clone())
-	verify {
-		let current_balance = <T as pallet::Config>::Assets::balance(asset_id.into(), &account);
-		assert_eq!(current_balance, 1_000_000_000_000_000u128.saturated_into());
-	}
-
 	withdraw {
 		let r in 1 .. 1000;
 		let asset_id: <T as pallet::Config>::AssetId = 100u128.into();
@@ -111,22 +112,80 @@ benchmarks! {
 		let ready_withdrawal = <ReadyWithdrawals<T>>::get(<frame_system::Pallet<T>>::block_number(), network_id);
 		assert_eq!(ready_withdrawal.len(), 1);
 	}
-}
 
-fn create_deposit<T: Config>(recipient: T::AccountId) -> Vec<Deposit<T::AccountId>> {
-	let mut pending_deposits = vec![];
-	let asset_id = 100;
-	for _i in 1..20 {
-		let deposit: Deposit<T::AccountId> = Deposit {
+	ethereum_withdraw {
+		let r in 1 .. 1000;
+		let asset_id: <T as pallet::Config>::AssetId = 100u128.into();
+		let admin = account::<T::AccountId>("admin", 1, r);
+		let network_id = 2;
+		<T as pallet::Config>::Assets::create(asset_id.into(), admin, true, 1u128.saturated_into()).unwrap();
+		let pallet_acc = T::TheaPalletId::get().into_account_truncating();
+		<T as pallet::Config>::Currency::mint_into(&pallet_acc, 100_000_000_000_000_000_000u128.saturated_into()).unwrap();
+		let account = account::<T::AccountId>("alice", 1, r);
+		<T as pallet::Config>::Assets::mint_into(asset_id.into(), &account, 100_000_000_000_000_000_000u128.saturated_into()).unwrap();
+		<T as pallet::Config>::Currency::mint_into(&account, 100_000_000_000_000u128.saturated_into()).unwrap();
+		let metadata = AssetMetadata::new(10).unwrap();
+		<Metadata<T>>::insert(100, metadata);
+		<WithdrawalFees<T>>::insert(network_id, 1_000);
+		let beneficiary: sp_core::H160 = sp_core::H160::default();
+	}: _(RawOrigin::Signed(account.clone()), 100, 1_000_000_000_000, beneficiary, true, false)
+	verify {
+		let ready_withdrawal = <ReadyWithdrawals<T>>::get(<frame_system::Pallet<T>>::block_number(), network_id);
+		assert_eq!(ready_withdrawal.len(), 1);
+	}
+
+	on_initialize {
+		let x in 1 .. 1_000;
+		let y in 1 .. 1_000;
+		let network_len: usize = x as usize;
+		let network_len: u8 = network_len as u8;
+		let withdrawal = Withdraw {
 			id: vec![],
-			recipient: recipient.clone(),
-			asset_id,
+			asset_id: 100,
 			amount: 1_000_000_000_000,
+			destination: vec![],
+			is_blocked: false,
 			extra: vec![],
 		};
-		pending_deposits.push(deposit);
+		let mut withdrawal_vec = Vec::new();
+		for _ in 0..y {
+			withdrawal_vec.push(withdrawal.clone());
+		}
+		let block_no: u32 = 10;
+		let mut networks: BTreeSet<u8> = BTreeSet::new();
+		for i in 0..network_len {
+			networks.insert(i);
+		}
+		let block_no: BlockNumberFor<T> = block_no.into();
+		for network_id in networks {
+			<ReadyWithdrawals<T>>::insert(block_no, network_id, withdrawal_vec.clone());
+		}
+	}: {
+		TheaExecutor::<T>::on_initialize(block_no);
+	}
+
+	burn_native_tokens{
+		let account: T::AccountId = T::AccountId::decode(&mut &[0u8; 32][..]).unwrap();
+		<T as pallet::Config>::Currency::mint_into(&account, (100000*UNIT_BALANCE).saturated_into()).unwrap();
+	}: _(RawOrigin::Root, account.clone(), UNIT_BALANCE)
+	verify {
+		assert_eq!(<T as pallet::Config>::Currency::balance(&account), (99999 * UNIT_BALANCE).saturated_into());
+	}
+
+	claim_deposit {
+		let r in 1 .. 1000;
+		let account = account::<T::AccountId>("alice", 1, r);
+		let asset_id: <T as pallet::Config>::AssetId = 100u128.into();
+		let deposits = create_deposit::<T>(account.clone());
+		let metadata = AssetMetadata::new(10).unwrap();
+		<Metadata<T>>::insert(100, metadata);
+		<T as pallet::Config>::Currency::mint_into(&account, 100_000_000_000_000u128.saturated_into()).unwrap();
+		<ApprovedDeposits<T>>::insert(account.clone(), deposits);
+	}: _(RawOrigin::Signed(account.clone()), 10,account.clone())
+	verify {
+		let current_balance = <T as pallet::Config>::Assets::balance(asset_id.into(), &account);
+		assert_eq!(current_balance, 1_000_000_000_000_000u128.saturated_into());
 	}
-	pending_deposits
 }
 
 #[cfg(test)]