precompile: Update precompile test

centrifuge · Sep 8, 2023 · 255c0ca · 255c0ca
1 parent 97b8fa4
commit 255c0ca
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 48 deletions.
diff --git a/pallets/liquidity-pools-gateway/axelar-gateway-precompile/src/lib.rs b/pallets/liquidity-pools-gateway/axelar-gateway-precompile/src/lib.rs
@@ -12,16 +12,13 @@
 #![cfg_attr(not(feature = "std"), no_std)]
 
 use cfg_types::domain_address::{Domain, DomainAddress};
-use codec::alloc::string::ToString;
-use ethabi::Token;
 use fp_evm::PrecompileHandle;
-use frame_support::StorageHasher;
 use pallet_evm::{ExitError, PrecompileFailure};
 use precompile_utils::prelude::*;
-use sp_core::{bounded::BoundedVec, ConstU32, H160, H256, U256};
+use sp_core::{bounded::BoundedVec, ConstU32, H256, U256};
 use sp_runtime::{
 	traits::{BlakeTwo256, Hash},
-	DispatchError, DispatchResult,
+	DispatchError,
 };
 use sp_std::vec::Vec;
 

diff --git a/runtime/common/src/evm/precompile.rs b/runtime/common/src/evm/precompile.rs
@@ -46,7 +46,7 @@ const ECRECOVERPUBLICKEY_ADDR: Addr = addr(1026);
 /// Liquidity-Pool logic on centrifuge.
 ///
 /// The precompile implements
-const LP_AXELAR_GATEWAY: Addr = addr(2048);
+pub const LP_AXELAR_GATEWAY: Addr = addr(2048);
 
 pub struct CentrifugePrecompiles<R>(PhantomData<R>);
 

diff --git a/runtime/integration-tests/src/evm/precompile.rs b/runtime/integration-tests/src/evm/precompile.rs
@@ -18,19 +18,22 @@ use cfg_traits::{ethereum::EthereumTransactor, liquidity_pools::Codec};
 use cfg_types::{
 	domain_address::{Domain, DomainAddress},
 	fixed_point::Rate,
+	tokens::{CurrencyId, CustomMetadata, GeneralCurrencyIndex},
 };
 use codec::Encode;
 use ethabi::{Contract, Function, Param, ParamType, Token};
 use ethereum::{LegacyTransaction, TransactionAction, TransactionSignature, TransactionV2};
 use frame_support::{assert_err, assert_ok, dispatch::RawOrigin};
 use fudge::primitives::Chain;
 use hex::ToHex;
+use orml_traits::{asset_registry::AssetMetadata, MultiCurrency};
 use pallet_evm::{AddressMapping, FeeCalculator};
 use pallet_liquidity_pools::Message;
-use runtime_common::account_conversion::AccountConverter;
+use runtime_common::{account_conversion::AccountConverter, evm::precompile::LP_AXELAR_GATEWAY};
 use sp_core::{Get, H160, H256, U256};
 use sp_runtime::traits::{BlakeTwo256, Hash};
 use tokio::runtime::Handle;
+use xcm::{v3::MultiLocation, VersionedMultiLocation};
 
 use crate::{
 	chain::centrifuge::{
@@ -45,48 +48,82 @@ use crate::{
 	},
 };
 
-// Axelar gateway contract code deployed on Algol as of 08.09.2023.
-const AXELAR_GATEWAY_CONTRACT_CODE: &str = "6080604052600436106100745760003560e01c80639ded06df1161004e5780639ded06df1461020c578063bd02d0f51461022d578063c031a18014610268578063dc97d96214610288576100ab565b806321f8a721146101325780637ae1cfca1461019f578063986e791a146101df576100ab565b366100ab576040517f858d70bd00000000000000000000000000000000000000000000000000000000815260040160405180910390fd5b7f360894a13ba1a3210667c828492db98dca3e2076cc3735a920a3ca505d382bbc600090815260026020527f11141f466c69fd409e1990e063b49cd6d61ed2ecff27a2e402e259ca6b9a01a35473ffffffffffffffffffffffffffffffffffffffff169036908037600080366000845af43d6000803e80801561012d573d6000f35b3d6000fd5b34801561013e57600080fd5b5061017561014d366004610374565b60009081526002602052604090205473ffffffffffffffffffffffffffffffffffffffff1690565b60405173ffffffffffffffffffffffffffffffffffffffff90911681526020015b60405180910390f35b3480156101ab57600080fd5b506101cf6101ba366004610374565b60009081526004602052604090205460ff1690565b6040519015158152602001610196565b3480156101eb57600080fd5b506101ff6101fa366004610374565b6102b5565b60405161019691906103da565b34801561021857600080fd5b5061022b6102273660046103f4565b5050565b005b34801561023957600080fd5b5061025a610248366004610374565b60009081526020819052604090205490565b604051908152602001610196565b34801561027457600080fd5b506101ff610283366004610374565b610357565b34801561029457600080fd5b5061025a6102a3366004610374565b60009081526005602052604090205490565b60008181526001602052604090208054606091906102d290610466565b80601f01602080910402602001604051908101604052809291908181526020018280546102fe90610466565b801561034b5780601f106103205761010080835404028352916020019161034b565b820191906000526020600020905b81548152906001019060200180831161032e57829003601f168201915b50505050509050919050565b60008181526003602052604090208054606091906102d290610466565b60006020828403121561038657600080fd5b5035919050565b6000815180845260005b818110156103b357602081850181015186830182015201610397565b818111156103c5576000602083870101525b50601f01601f19169290920160200192915050565b6020815260006103ed602083018461038d565b9392505050565b6000806020838503121561040757600080fd5b823567ffffffffffffffff8082111561041f57600080fd5b818501915085601f83011261043357600080fd5b81358181111561044257600080fd5b86602082850101111561045457600080fd5b60209290920196919550909350505050565b600181811c9082168061047a57607f821691505b602082108114156104b4577f4e487b7100000000000000000000000000000000000000000000000000000000600052602260045260246000fd5b5091905056fea264697066735822122069dff699a34cf8bf80b00e2030730390655a53519c00ba0e4306c3269ae7a2fa64736f6c63430008090033";
-
 #[tokio::test]
 async fn axelar_precompile_execute() {
 	let mut env = env::test_env_default(Handle::current());
 
 	env.evolve().unwrap();
 
-	// Contract address must be high enough to not map to the precompile space.
-	let contract_address = H160::from_low_u64_be(1_000_001);
+	let sender_address = H160::from_low_u64_be(1_000_002);
 
-	mint_balance_into_derived_account(&mut env, contract_address, 1_000_000 * CFG);
-	deploy_contract(
-		&mut env,
-		contract_address,
-		hex::decode(TEST_CONTRACT_CODE).unwrap(),
-	);
+	mint_balance_into_derived_account(&mut env, sender_address, 1_000_000 * CFG);
 
-	let contract_address = env
+	let derived_sender_account = env
 		.with_state(Chain::Para(PARA_ID), || {
-			pallet_evm::AccountCodes::<Runtime>::iter()
-				.find(|(address, code)| code.len() > 0)
-				.unwrap()
-				.0
+			<Runtime as pallet_evm::Config>::AddressMapping::into_account_id(sender_address)
 		})
 		.unwrap();
 
-	let sender_address = H160::from_low_u64_be(1_000_002);
-	mint_balance_into_derived_account(&mut env, sender_address, 1_000_000 * CFG);
-	let sender_derived_account = env
+	let receiver_address = H160::from_low_u64_be(1_000_003);
+
+	// mint_balance_into_derived_account(&mut env, receiver_address, 1_000_000 *
+	// CFG);
+
+	let derived_receiver_account = env
 		.with_state(Chain::Para(PARA_ID), || {
-			<Runtime as pallet_evm::Config>::AddressMapping::into_account_id(sender_address)
+			<Runtime as pallet_evm::Config>::AddressMapping::into_account_id(receiver_address)
 		})
 		.unwrap();
 
-	let test_msg = Message::<Domain, PoolId, TrancheId, Balance, Rate>::AddCurrency{
-		currency: 0,
-		evm_address: sender_address.0,
+	let source_address = H160::from_low_u64_be(1111);
+	let evm_chain_name = String::from("Ethereum");
+	let evm_chain_id = 0;
+
+	let currency_id = CurrencyId::ForeignAsset(123456);
+
+	let currency_metadata = AssetMetadata {
+		decimals: 18,
+		name: "Test".into(),
+		symbol: "TST".into(),
+		existential_deposit: 1_000_000,
+		location: Some(VersionedMultiLocation::V3(MultiLocation::here())),
+		additional: CustomMetadata {
+			transferability: Default::default(),
+			mintable: true,
+			permissioned: false,
+			pool_currency: false,
+		},
 	};
 
-	let source_address = H160::from_low_u64_be(5678);
+	env.with_mut_state(Chain::Para(PARA_ID), || {
+		orml_asset_registry::Pallet::<Runtime>::register_asset(
+			RuntimeOrigin::root(),
+			currency_metadata,
+			Some(currency_id),
+		)
+		.unwrap();
+
+		orml_tokens::Pallet::<Runtime>::deposit(
+			currency_id,
+			&derived_sender_account,
+			1_000_000_000_000 * 10u128.saturating_pow(18),
+		)
+		.unwrap();
+	})
+	.unwrap();
+
+	let general_currency_id = env
+		.with_state(Chain::Para(PARA_ID), || {
+			pallet_liquidity_pools::Pallet::<Runtime>::try_get_general_index(currency_id).unwrap()
+		})
+		.unwrap();
+
+	let msg = Message::<Domain, PoolId, TrancheId, Balance, Rate>::Transfer {
+		currency: general_currency_id,
+		sender: derived_sender_account.clone().into(),
+		receiver: derived_receiver_account.into(),
+		amount: 100,
+	};
 
 	env.with_mut_state(Chain::Para(PARA_ID), || {
 		axelar_gateway_precompile::Pallet::<Runtime>::set_gateway(
@@ -97,20 +134,22 @@ async fn axelar_precompile_execute() {
 
 		axelar_gateway_precompile::Pallet::<Runtime>::set_converter(
 			RuntimeOrigin::root(),
-			BlakeTwo256::hash("Ethereum".as_bytes()),
+			BlakeTwo256::hash(evm_chain_name.as_bytes()),
 			SourceConverter {
-				domain: Domain::EVM(0),
+				domain: Domain::EVM(evm_chain_id),
 			},
 		)
 		.unwrap();
 
 		pallet_liquidity_pools_gateway::Pallet::<Runtime>::add_instance(
 			RuntimeOrigin::root(),
-			DomainAddress::EVM(0, source_address.0),
+			DomainAddress::EVM(evm_chain_id, source_address.0),
 		)
 		.unwrap();
 	});
 
+	let command_id = H256::from_low_u64_be(5678);
+
 	let test_input = Contract {
 		constructor: None,
 		functions: BTreeMap::<String, Vec<Function>>::from([(
@@ -153,21 +192,19 @@ async fn axelar_precompile_execute() {
 	.map_err(|_| "cannot retrieve test contract function")
 	.unwrap()
 	.encode_input(&[
-		Token::FixedBytes(H256::from_low_u64_be(1234).0.to_vec()),
-		Token::String(String::from("Ethereum")),
+		Token::FixedBytes(command_id.0.to_vec()),
+		Token::String(evm_chain_name),
 		Token::String(String::from_utf8(source_address.as_fixed_bytes().to_vec()).unwrap()),
-		Token::Bytes(test_msg.serialize()),
+		Token::Bytes(msg.serialize()),
 	])
 	.map_err(|_| "cannot encode input for test contract function")
 	.unwrap();
 
-	let lp_axelar_gateway_precompile_address: H160 = addr(2048).into();
-
 	env.with_state(Chain::Para(PARA_ID), || {
 		assert_ok!(pallet_evm::Pallet::<Runtime>::call(
-			RawOrigin::Signed(sender_derived_account).into(),
+			RawOrigin::Signed(derived_sender_account).into(),
 			sender_address,
-			lp_axelar_gateway_precompile_address,
+			LP_AXELAR_GATEWAY.into(),
 			test_input.to_vec(),
 			U256::from(0),
 			0x100000,
@@ -179,10 +216,3 @@ async fn axelar_precompile_execute() {
 	})
 	.unwrap();
 }
-
-fn addr(a: u64) -> [u8; 20] {
-	let b = a.to_be_bytes();
-	[
-		0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, b[0], b[1], b[2], b[3], b[4], b[5], b[6], b[7],
-	]
-}