xcm-simulator tests

evm-template/runtime/tests/xcm_mock/mod.rs

@@ -0,0 +1,216 @@
+pub mod parachain;
+pub mod relay_chain;
+pub mod statemint_like;
+use std::{collections::BTreeMap, str::FromStr};
+
+use cumulus_primitives_core::ParaId;
+use polkadot_runtime_parachains::{
+    configuration::{GenesisConfig as ConfigurationGenesisConfig, HostConfiguration},
+    paras::{GenesisConfig as ParasGenesisConfig, ParaGenesisArgs, ParaKind},
+};
+use sp_core::{H160, U256};
+use sp_runtime::{traits::AccountIdConversion, AccountId32, BuildStorage};
+use xcm_simulator::{decl_test_network, decl_test_parachain, decl_test_relay_chain, TestExt};
+
+pub const PARAALICE: [u8; 20] = [1u8; 20];
+pub const RELAYALICE: AccountId32 = AccountId32::new([0u8; 32]);
+pub const RELAYBOB: AccountId32 = AccountId32::new([2u8; 32]);
+
+pub fn para_a_account() -> AccountId32 {
+    ParaId::from(1).into_account_truncating()
+}
+
+pub fn para_b_account() -> AccountId32 {
+    ParaId::from(2).into_account_truncating()
+}
+
+pub fn para_a_account_20() -> parachain::AccountId {
+    ParaId::from(1).into_account_truncating()
+}
+
+pub fn evm_account() -> H160 {
+    H160::from_str("1000000000000000000000000000000000000001").unwrap()
+}
+
+pub fn mock_para_genesis_info() -> ParaGenesisArgs {
+    ParaGenesisArgs {
+        genesis_head: vec![1u8].into(),
+        validation_code: vec![1u8].into(),
+        para_kind: ParaKind::Parachain,
+    }
+}
+
+pub fn mock_relay_config() -> HostConfiguration<relay_chain::BlockNumber> {
+    HostConfiguration::<relay_chain::BlockNumber> {
+        hrmp_channel_max_capacity: u32::MAX,
+        hrmp_channel_max_total_size: u32::MAX,
+        hrmp_max_parachain_inbound_channels: 10,
+        hrmp_max_parachain_outbound_channels: 10,
+        hrmp_channel_max_message_size: u32::MAX,
+        // Changed to avoid aritmetic errors within hrmp_close
+        max_downward_message_size: 100_000u32,
+        ..Default::default()
+    }
+}
+
+decl_test_parachain! {
+    pub struct ParaA {
+        Runtime = parachain::Runtime,
+        XcmpMessageHandler = parachain::MsgQueue,
+        DmpMessageHandler = parachain::MsgQueue,
+        new_ext = para_ext(1),
+    }
+}
+
+decl_test_parachain! {
+    pub struct ParaB {
+        Runtime = parachain::Runtime,
+        XcmpMessageHandler = parachain::MsgQueue,
+        DmpMessageHandler = parachain::MsgQueue,
+        new_ext = para_ext(2),
+    }
+}
+
+decl_test_parachain! {
+    pub struct ParaC {
+        Runtime = parachain::Runtime,
+        XcmpMessageHandler = parachain::MsgQueue,
+        DmpMessageHandler = parachain::MsgQueue,
+        new_ext = para_ext(3),
+    }
+}
+
+decl_test_parachain! {
+    pub struct Statemint {
+        Runtime = statemint_like::Runtime,
+        XcmpMessageHandler = statemint_like::MsgQueue,
+        DmpMessageHandler = statemint_like::MsgQueue,
+        new_ext = statemint_ext(4),
+    }
+}
+
+decl_test_relay_chain! {
+    pub struct Relay {
+        Runtime = relay_chain::Runtime,
+        RuntimeCall = relay_chain::RuntimeCall,
+        RuntimeEvent = relay_chain::RuntimeEvent,
+        XcmConfig = relay_chain::XcmConfig,
+        MessageQueue = relay_chain::MessageQueue,
+        System = relay_chain::System,
+        new_ext = relay_ext(vec![1, 2, 3, 4]),
+    }
+}
+
+decl_test_network! {
+    pub struct MockNet {
+        relay_chain = Relay,
+        parachains = vec![
+            (1, ParaA),
+            (2, ParaB),
+            (3, ParaC),
+            (4, Statemint),
+        ],
+    }
+}
+
+pub const INITIAL_BALANCE: u128 = 10_000_000_000_000_000;
+
+pub const INITIAL_EVM_BALANCE: u128 = 0;
+pub const INITIAL_EVM_NONCE: u32 = 1;
+
+pub fn para_ext(para_id: u32) -> sp_io::TestExternalities {
+    use parachain::{MsgQueue, Runtime, System};
+
+    let mut t = frame_system::GenesisConfig::<Runtime>::default().build_storage().unwrap();
+
+    pallet_balances::GenesisConfig::<Runtime> {
+        balances: vec![(PARAALICE.into(), INITIAL_BALANCE)],
+    }
+    .assimilate_storage(&mut t)
+    .unwrap();
+
+    // EVM accounts are self-sufficient.
+    let mut evm_accounts = BTreeMap::new();
+    evm_accounts.insert(
+        evm_account(),
+        fp_evm::GenesisAccount {
+            nonce: U256::from(INITIAL_EVM_NONCE),
+            balance: U256::from(INITIAL_EVM_BALANCE),
+            storage: Default::default(),
+            code: vec![
+                0x00, // STOP
+            ],
+        },
+    );
+
+    let genesis_config =
+        pallet_evm::GenesisConfig::<Runtime> { accounts: evm_accounts, ..Default::default() };
+    genesis_config.assimilate_storage(&mut t).unwrap();
+
+    let mut ext = sp_io::TestExternalities::new(t);
+    ext.execute_with(|| {
+        System::set_block_number(1);
+        MsgQueue::set_para_id(para_id.into());
+    });
+    ext
+}
+
+pub fn statemint_ext(para_id: u32) -> sp_io::TestExternalities {
+    use statemint_like::{MsgQueue, Runtime, System};
+
+    let mut t = frame_system::GenesisConfig::<Runtime>::default().build_storage().unwrap();
+
+    pallet_balances::GenesisConfig::<Runtime> {
+        balances: vec![(RELAYALICE.into(), INITIAL_BALANCE), (RELAYBOB.into(), INITIAL_BALANCE)],
+    }
+    .assimilate_storage(&mut t)
+    .unwrap();
+
+    let mut ext = sp_io::TestExternalities::new(t);
+    ext.execute_with(|| {
+        System::set_block_number(1);
+        MsgQueue::set_para_id(para_id.into());
+    });
+    ext
+}
+
+pub fn relay_ext(paras: Vec<u32>) -> sp_io::TestExternalities {
+    use relay_chain::{Runtime, System};
+
+    let mut t = frame_system::GenesisConfig::<Runtime>::default().build_storage().unwrap();
+
+    pallet_balances::GenesisConfig::<Runtime> { balances: vec![(RELAYALICE, INITIAL_BALANCE)] }
+        .assimilate_storage(&mut t)
+        .unwrap();
+
+    let para_genesis: Vec<(ParaId, ParaGenesisArgs)> =
+        paras.iter().map(|&para_id| (para_id.into(), mock_para_genesis_info())).collect();
+
+    let genesis_config = ConfigurationGenesisConfig::<Runtime> { config: mock_relay_config() };
+    genesis_config.assimilate_storage(&mut t).unwrap();
+
+    let genesis_config =
+        ParasGenesisConfig::<Runtime> { paras: para_genesis, ..Default::default() };
+    genesis_config.assimilate_storage(&mut t).unwrap();
+
+    let mut ext = sp_io::TestExternalities::new(t);
+    ext.execute_with(|| {
+        System::set_block_number(1);
+    });
+    ext
+}
+pub type RelayChainPalletXcm = pallet_xcm::Pallet<relay_chain::Runtime>;
+pub type Hrmp = polkadot_runtime_parachains::hrmp::Pallet<relay_chain::Runtime>;
+
+pub type StatemintBalances = pallet_balances::Pallet<statemint_like::Runtime>;
+pub type StatemintChainPalletXcm = pallet_xcm::Pallet<statemint_like::Runtime>;
+pub type StatemintAssets = pallet_assets::Pallet<statemint_like::Runtime>;
+
+pub type ParachainPalletXcm = pallet_xcm::Pallet<parachain::Runtime>;
+pub type Assets = pallet_assets::Pallet<parachain::Runtime, parachain::ForeignAssetInstance>;
+
+pub type Treasury = pallet_treasury::Pallet<parachain::Runtime>;
+pub type AssetManager = pallet_asset_manager::Pallet<parachain::Runtime>;
+pub type XTokens = orml_xtokens::Pallet<parachain::Runtime>;
+pub type RelayBalances = pallet_balances::Pallet<relay_chain::Runtime>;
+pub type ParaBalances = pallet_balances::Pallet<parachain::Runtime>;