fix pallet claims benchmark

galacticcouncil · Nov 9, 2023 · 628ea58 · 628ea58
1 parent 7042d35
commit 628ea58
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/pallets/claims/src/benchmarking.rs b/pallets/claims/src/benchmarking.rs
@@ -38,16 +38,12 @@ benchmarks! {
 		let eth_address = EthereumAddress(hex!["8202c0af5962b750123ce1a9b12e1c30a4973557"]);
 
 		let initial_balance = T::Currency::free_balance(&caller);
+		let amount_to_claim = 1_000_000_000_000_000_000_u128;
 
-		Claims::<T>::insert(eth_address, T::CurrencyBalance::from(1_000_000_000_000_000_000_u128).into());
+		Claims::<T>::insert(eth_address, T::CurrencyBalance::from(amount_to_claim).into());
 	}: _(RawOrigin::Signed(caller.clone()), EcdsaSignature(signature))
 	verify {
-		let expected_balance = T::CurrencyBalance::from(1_010_000_000_000_000_000_u128);
-
-		#[cfg(test)]
-		let expected_balance = T::CurrencyBalance::from(1_000_000_000_000_000_000_u128);
-
-		assert_eq!(T::Currency::free_balance(&caller), initial_balance + expected_balance.into());
+		assert_eq!(T::Currency::free_balance(&caller), initial_balance + T::CurrencyBalance::from(amount_to_claim).into());
 		assert_eq!(Claims::<T>::get(eth_address), T::CurrencyBalance::from(0u128).into());
 	}
 }