permit batch test

Author: iamyxsh

integration-tests/src/evm_permit.rs

@@ -2875,6 +2875,147 @@ mod sponsored_paymaster {
 		)
 	}
 
+	fn build_permit_at_current_nonce(
+		inner_call: &RuntimeCall,
+		gas_limit: u64,
+	) -> (H160, Vec<u8>, u64, U256, u8, H256, H256) {
+		let from = alith_evm_address();
+		let secret_key = SecretKey::parse(&alith_secret_key()).unwrap();
+		let data = inner_call.encode();
+		let deadline = U256::from(1_000_000_000_000u128);
+		let nonce = pallet_evm_precompile_call_permit::NoncesStorage::get(from);
+
+		let permit = pallet_evm_precompile_call_permit::CallPermitPrecompile::<Runtime>::generate_permit(
+			CALLPERMIT,
+			from,
+			DISPATCH_ADDR,
+			U256::from(0),
+			data.clone(),
+			gas_limit,
+			nonce,
+			deadline,
+		);
+		let message = Message::parse(&permit);
+		let (rs, v) = sign(&message, &secret_key);
+		(
+			from,
+			data,
+			gas_limit,
+			deadline,
+			v.serialize(),
+			H256::from(rs.r.b32()),
+			H256::from(rs.s.b32()),
+		)
+	}
+
+	fn permit_batch_of_evm_calls(
+		paymaster: &AccountId,
+		target: H160,
+		legs: usize,
+		outer_gas_limit: u64,
+		leg_gas_limit: u64,
+	) -> frame_support::dispatch::DispatchResultWithPostInfo {
+		let leg = RuntimeCall::Dispatcher(pallet_dispatcher::Call::dispatch_evm_call {
+			call: Box::new(RuntimeCall::EVM(pallet_evm::Call::call {
+				source: alith_evm_address(),
+				target,
+				input: vec![0x06, 0xfd, 0xde, 0x03],
+				value: U256::zero(),
+				gas_limit: leg_gas_limit,
+				max_fee_per_gas: crate::evm::gas_price(),
+				max_priority_fee_per_gas: None,
+				nonce: None,
+				access_list: vec![],
+				authorization_list: vec![],
+			})),
+		});
+
+		let batch = RuntimeCall::Utility(pallet_utility::Call::batch_all { calls: vec![leg; legs] });
+
+		let (from, data, gas_limit, deadline, v, r, s) = build_permit_at_current_nonce(&batch, outer_gas_limit);
+
+		MultiTransactionPayment::dispatch_permit(
+			RuntimeOrigin::signed(paymaster.clone()),
+			from,
+			DISPATCH_ADDR,
+			U256::from(0),
+			data,
+			gas_limit,
+			deadline,
+			v,
+			r,
+			s,
+		)
+	}
+
+	#[test]
+	fn dispatch_permit_should_execute_every_leg_when_batching_evm_calls_under_one_permit() {
+		TestNet::reset();
+
+		Hydra::execute_with(|| {
+			init_omnipool_with_oracle_for_block_10();
+
+			let contract = crate::utils::contracts::deploy_contract("HydraToken", crate::contracts::deployer());
+			let paymaster = paymaster_account();
+			assert_ok!(Balances::mint_into(&paymaster, 100_000 * UNITS));
+
+			let user_acc = MockAccount::new(alith_truncated_account());
+			let initial_user_weth = user_acc.balance(WETH);
+			let initial_paymaster_hdx = Currencies::free_balance(HDX, &paymaster);
+
+			assert_ok!(permit_batch_of_evm_calls(&paymaster, contract, 14, 10_000_000, 100_000));
+
+			let batch_completed = frame_system::Pallet::<Runtime>::events().iter().any(|record| {
+				matches!(
+					&record.event,
+					RuntimeEvent::Utility(pallet_utility::Event::BatchCompleted)
+				)
+			});
+			assert!(
+				batch_completed,
+				"batch_all emits BatchCompleted only when EVERY leg succeeded"
+			);
+
+			assert_eq!(
+				user_acc.balance(WETH),
+				initial_user_weth,
+				"signer MUST NOT pay for any leg"
+			);
+			assert!(
+				Currencies::free_balance(HDX, &paymaster) < initial_paymaster_hdx,
+				"paymaster MUST bear the cost of the whole batch"
+			);
+		});
+	}
+
+	#[test]
+	fn dispatch_permit_should_fail_when_gas_limit_exceeds_what_the_paymaster_can_cover() {
+		TestNet::reset();
+
+		Hydra::execute_with(|| {
+			init_omnipool_with_oracle_for_block_10();
+
+			let contract = crate::utils::contracts::deploy_contract("HydraToken", crate::contracts::deployer());
+			let paymaster = paymaster_account();
+
+			let affordable = permit_batch_of_evm_calls(&paymaster, contract, 14, 10_000_000, 100_000);
+			assert_ok!(affordable);
+
+			let too_expensive = permit_batch_of_evm_calls(&paymaster, contract, 14, 40_000_000, 100_000)
+				.expect_err("gas budget above the paymaster's balance must be rejected");
+
+			assert_eq!(
+				too_expensive.error,
+				pallet_transaction_multi_payment::Error::<Runtime>::EvmPermitRunnerError.into(),
+				"the binding constraint is gas_limit x gas_price vs fee-payer balance, not the block gas limit"
+			);
+			assert!(
+				<Runtime as pallet_evm::Config>::BlockGasLimit::get() > U256::from(40_000_000u64),
+				"40M is well under the 60M block ceiling — this failure is affordability, not the block limit"
+			);
+		});
+	}
+
 	fn submit_signed_dispatch_permit_omni_sell(
 		paymaster: AccountId,
 		sell_amount: Balance,