git.delta.rocks / unique-network / refs/commits / 4e84fdd5539f

difftreelog

refactor xcm transact forbidden test

Daniel Shiposha2023-03-27parent: #48ebb54.patch.diff
in: master

17 files changed

modifiedruntime/common/tests/mod.rsdiffbeforeafterboth
14// You should have received a copy of the GNU General Public License14// You should have received a copy of the GNU General Public License
15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
1616
17use sp_runtime::BuildStorage;17use sp_runtime::{BuildStorage, Storage};
18use sp_core::{Public, Pair};18use sp_core::{Public, Pair};
19use sp_std::vec;19use sp_std::vec;
20use up_common::types::AuraId;20use up_common::types::AuraId;
21use crate::{GenesisConfig, ParachainInfoConfig};21use crate::{Runtime, GenesisConfig, ParachainInfoConfig, RuntimeEvent, System};
22
23pub use sp_runtime::AccountId32 as AccountId;
24pub type Balance = u128;
2225
23pub mod xcm;26pub mod xcm;
27
28#[cfg(any(feature = "opal-runtime", feature = "quartz-runtime"))]
29/// PARA_ID for Opal/Sapphire/Quartz
30const PARA_ID: u32 = 2095;
31
32#[cfg(feature = "unique-runtime")]
33/// PARA_ID for Unique
34const PARA_ID: u32 = 2037;
2435
25fn get_from_seed<TPublic: Public>(seed: &str) -> <TPublic::Pair as Pair>::Public {36fn get_from_seed<TPublic: Public>(seed: &str) -> <TPublic::Pair as Pair>::Public {
26 TPublic::Pair::from_string(&format!("//{}", seed), None)37 TPublic::Pair::from_string(&format!("//{}", seed), None)
27 .expect("static values are valid; qed")38 .expect("static values are valid; qed")
28 .public()39 .public()
29}40}
41
42fn last_events(n: usize) -> Vec<RuntimeEvent> {
43 System::events().into_iter().map(|e| e.event).rev().take(n).rev().collect()
44}
45
46fn new_test_ext(balances: Vec<(AccountId, Balance)>) -> sp_io::TestExternalities {
47 let mut storage = make_basic_storage();
48
49 pallet_balances::GenesisConfig::<Runtime> { balances }
50 .assimilate_storage(&mut storage)
51 .unwrap();
52
53 let mut ext = sp_io::TestExternalities::new(storage);
54 ext.execute_with(|| System::set_block_number(1));
55 ext
56}
3057
31#[cfg(feature = "collator-selection")]58#[cfg(feature = "collator-selection")]
32fn new_test_ext(para_id: u32) -> sp_io::TestExternalities {59fn make_basic_storage() -> Storage {
33 use sp_core::{sr25519};60 use sp_core::{sr25519};
34 use sp_runtime::traits::{IdentifyAccount, Verify};61 use sp_runtime::traits::{IdentifyAccount, Verify};
35 use crate::{AccountId, Signature, SessionKeys, CollatorSelectionConfig, SessionConfig};62 use crate::{AccountId, Signature, SessionKeys, CollatorSelectionConfig, SessionConfig};
66 collator_selection: CollatorSelectionConfig { invulnerables },93 collator_selection: CollatorSelectionConfig { invulnerables },
67 session: SessionConfig { keys },94 session: SessionConfig { keys },
68 parachain_info: ParachainInfoConfig {95 parachain_info: ParachainInfoConfig {
69 parachain_id: para_id.into(),96 parachain_id: PARA_ID.into(),
70 },97 },
71 ..GenesisConfig::default()98 ..GenesisConfig::default()
72 };99 };
75}102}
76103
77#[cfg(not(feature = "collator-selection"))]104#[cfg(not(feature = "collator-selection"))]
78fn new_test_ext(para_id: u32) -> sp_io::TestExternalities {105fn make_basic_storage() -> Storage {
79 use crate::AuraConfig;106 use crate::AuraConfig;
80107
81 let cfg = GenesisConfig {108 let cfg = GenesisConfig {
86 ],113 ],
87 },114 },
88 parachain_info: ParachainInfoConfig {115 parachain_info: ParachainInfoConfig {
89 parachain_id: para_id.into(),116 parachain_id: PARA_ID.into(),
90 },117 },
91 ..GenesisConfig::default()118 ..GenesisConfig::default()
92 };119 };
modifiedruntime/common/tests/xcm.rsdiffbeforeafterboth
--- a/runtime/common/tests/xcm.rs
+++ b/runtime/common/tests/xcm.rs
@@ -14,149 +14,53 @@
 // You should have received a copy of the GNU General Public License
 // along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
 
-use xcm_executor::traits::ShouldExecute;
-use xcm::latest::prelude::*;
-use logtest::Logger;
-use crate::RuntimeCall;
-use super::new_test_ext;
-use frame_support::pallet_prelude::Weight;
-
-fn catch_xcm_barrier_log(logger: &mut Logger, expected_msg: &str) -> Result<(), String> {
-	for record in logger {
-		if record.target() == "xcm::barrier" && record.args() == expected_msg {
-			return Ok(());
-		}
-	}
-
-	Err(format!(
-		"the expected XCM barrier log `{}` is not found",
-		expected_msg
-	))
-}
-
-/// WARNING: Uses log capturing
-/// See https://docs.rs/logtest/latest/logtest/index.html#constraints
-pub fn barrier_denies_transact<B: ShouldExecute>(logger: &mut Logger) {
-	let location = MultiLocation {
-		parents: 0,
-		interior: Junctions::Here,
-	};
-
-	// We will never decode this "call",
-	// so it is irrelevant what we are passing to the `transact` cmd.
-	let fake_encoded_call = vec![0u8];
-
-	let transact_inst: Instruction<RuntimeCall> = Transact {
-		origin_kind: OriginKind::Superuser,
-		require_weight_at_most: Weight::default(),
-		call: fake_encoded_call.into(),
-	};
-
-	let mut xcm_program = vec![transact_inst];
+use xcm::{
+	VersionedXcm,
+	latest::{prelude::*, Error}
+};
+use codec::Encode;
+use crate::{Runtime, RuntimeCall, RuntimeOrigin, RuntimeEvent, PolkadotXcm};
+use super::{new_test_ext, last_events, AccountId};
+use frame_support::{
+	pallet_prelude::Weight,
+};
 
-	let max_weight = Weight::from_parts(100_000, 100_000);
-	let mut weight_credit = Weight::from_parts(100_000_000, 100_000_000);
+const ALICE: AccountId = AccountId::new([0u8; 32]);
+const BOB: AccountId = AccountId::new([1u8; 32]);
 
-	let result = B::should_execute(&location, &mut xcm_program, max_weight, &mut weight_credit);
+const INITIAL_BALANCE: u128 = 1000000000000000000_0000; // 1000 UNQ 
 
-	assert!(
-		result.is_err(),
-		"the barrier should disallow the XCM transact cmd"
-	);
+#[test]
+pub fn xcm_transact_is_forbidden() {
+	new_test_ext(vec![(ALICE, INITIAL_BALANCE)]).execute_with(|| {
+		PolkadotXcm::execute(
+			RuntimeOrigin::signed(ALICE),
+			Box::new(VersionedXcm::from(Xcm(vec![
+				Transact {
+					origin_kind: OriginKind::Native,
+					require_weight_at_most: Weight::from_parts(1000, 1000),
+					call: RuntimeCall::Balances(
+						pallet_balances::Call::<Runtime>::transfer {
+							dest: BOB.into(),
+							value: INITIAL_BALANCE / 2,
+						}
+					).encode().into(),
+				}
+			]))),
+			Weight::from_parts(1001000, 2000),
+		).expect("XCM execute must succeed, the error should be in the `PolkadotXcm::Attempted` event");
 
-	catch_xcm_barrier_log(logger, "transact XCM rejected").unwrap();
-}
-
-fn xcm_execute<B: ShouldExecute>(
-	self_para_id: u32,
-	location: &MultiLocation,
-	xcm: &mut [Instruction<RuntimeCall>],
-) -> Result<(), ()> {
-	new_test_ext(self_para_id).execute_with(|| {
-		let max_weight = Weight::from_parts(100_000, 100_000);
-		let mut weight_credit = Weight::from_parts(100_000_000, 100_000_000);
-
-		B::should_execute(&location, xcm, max_weight, &mut weight_credit)
-	})
-}
-
-fn make_multiassets(location: &MultiLocation) -> MultiAssets {
-	let id = AssetId::Concrete(location.clone());
-	let fun = Fungibility::Fungible(42);
-	let multiasset = MultiAsset { id, fun };
-
-	multiasset.into()
-}
-
-fn make_transfer_reserve_asset(location: &MultiLocation) -> Xcm<RuntimeCall> {
-	let assets = make_multiassets(location);
-	let inst = TransferReserveAsset {
-		assets,
-		dest: location.clone(),
-		xcm: Xcm(vec![]),
-	};
-
-	Xcm::<RuntimeCall>(vec![inst])
-}
-
-fn make_deposit_reserve_asset(location: &MultiLocation) -> Xcm<RuntimeCall> {
-	let assets = make_multiassets(location);
-	let inst = DepositReserveAsset {
-		assets: assets.into(),
-		dest: location.clone(),
-		xcm: Xcm(vec![]),
-	};
-
-	Xcm::<RuntimeCall>(vec![inst])
-}
-
-fn expect_transfer_location_denied<B: ShouldExecute>(
-	logger: &mut Logger,
-	self_para_id: u32,
-	location: &MultiLocation,
-	xcm: &mut [Instruction<RuntimeCall>],
-) -> Result<(), String> {
-	let result = xcm_execute::<B>(self_para_id, location, xcm);
-
-	if result.is_ok() {
-		return Err("the barrier should deny the unknown location".into());
-	}
-
-	catch_xcm_barrier_log(logger, "Unexpected deposit or transfer location")
-}
-
-/// WARNING: Uses log capturing
-/// See https://docs.rs/logtest/latest/logtest/index.html#constraints
-pub fn barrier_denies_transfer_from_unknown_location<B>(
-	logger: &mut Logger,
-	self_para_id: u32,
-) -> Result<(), String>
-where
-	B: ShouldExecute,
-{
-	const UNKNOWN_PARACHAIN_ID: u32 = 4057;
-
-	let unknown_location = MultiLocation {
-		parents: 1,
-		interior: X1(Parachain(UNKNOWN_PARACHAIN_ID)),
-	};
-
-	let mut transfer_reserve_asset = make_transfer_reserve_asset(&unknown_location);
-	let mut deposit_reserve_asset = make_deposit_reserve_asset(&unknown_location);
-
-	expect_transfer_location_denied::<B>(
-		logger,
-		self_para_id,
-		&unknown_location,
-		&mut transfer_reserve_asset.0,
-	)?;
-
-	expect_transfer_location_denied::<B>(
-		logger,
-		self_para_id,
-		&unknown_location,
-		&mut deposit_reserve_asset.0,
-	)?;
-
-	Ok(())
+		let xcm_event = &last_events(1)[0];
+		match xcm_event {
+			RuntimeEvent::PolkadotXcm(
+				pallet_xcm::Event::<Runtime>::Attempted(
+					Outcome::Incomplete(_weight, Error::NoPermission)
+				)
+			) => { /* Pass */ },
+			_ => panic!(
+				"Expected PolkadotXcm.Attempted(Incomplete(_weight, NoPermission)),\
+				found: {xcm_event:#?}"
+			)
+		}
+	});
 }
modifiedruntime/opal/Cargo.tomldiffbeforeafterboth
--- a/runtime/opal/Cargo.toml
+++ b/runtime/opal/Cargo.toml
@@ -307,8 +307,5 @@
 
 impl-trait-for-tuples = { workspace = true }
 
-[dev-dependencies]
-logtest = { workspace = true }
-
 [build-dependencies]
 substrate-wasm-builder = { workspace = true }
modifiedruntime/opal/src/lib.rsdiffbeforeafterboth
--- a/runtime/opal/src/lib.rs
+++ b/runtime/opal/src/lib.rs
@@ -41,9 +41,6 @@
 
 pub mod xcm_barrier;
 
-#[cfg(test)]
-mod tests;
-
 pub use runtime_common::*;
 
 pub const RUNTIME_NAME: &str = "opal";
deletedruntime/opal/src/tests/logcapture.rsdiffbeforeafterboth
--- a/runtime/opal/src/tests/logcapture.rs
+++ /dev/null
@@ -1,25 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-use logtest::Logger;
-use super::xcm::opal_xcm_tests;
-
-#[test]
-fn opal_log_capture_tests() {
-	let mut logger = Logger::start();
-
-	opal_xcm_tests(&mut logger);
-}
deletedruntime/opal/src/tests/mod.rsdiffbeforeafterboth
--- a/runtime/opal/src/tests/mod.rs
+++ /dev/null
@@ -1,18 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-mod logcapture;
-mod xcm;
deletedruntime/opal/src/tests/xcm.rsdiffbeforeafterboth
--- a/runtime/opal/src/tests/xcm.rs
+++ /dev/null
@@ -1,27 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-use logtest::Logger;
-use crate::{runtime_common::tests::xcm::*, xcm_barrier::Barrier};
-
-const OPAL_PARA_ID: u32 = 2095; // Same as Quartz
-
-pub fn opal_xcm_tests(logger: &mut Logger) {
-	barrier_denies_transact::<Barrier>(logger);
-
-	barrier_denies_transfer_from_unknown_location::<Barrier>(logger, OPAL_PARA_ID)
-		.expect_err("opal runtime allows any location");
-}
modifiedruntime/quartz/Cargo.tomldiffbeforeafterboth
--- a/runtime/quartz/Cargo.toml
+++ b/runtime/quartz/Cargo.toml
@@ -300,8 +300,5 @@
 
 impl-trait-for-tuples = { workspace = true }
 
-[dev-dependencies]
-logtest = { workspace = true }
-
 [build-dependencies]
 substrate-wasm-builder = { workspace = true }
modifiedruntime/quartz/src/lib.rsdiffbeforeafterboth
--- a/runtime/quartz/src/lib.rs
+++ b/runtime/quartz/src/lib.rs
@@ -41,9 +41,6 @@
 
 pub mod xcm_barrier;
 
-#[cfg(test)]
-mod tests;
-
 pub use runtime_common::*;
 
 #[cfg(feature = "become-sapphire")]
deletedruntime/quartz/src/tests/logcapture.rsdiffbeforeafterboth
--- a/runtime/quartz/src/tests/logcapture.rs
+++ /dev/null
@@ -1,25 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-use logtest::Logger;
-use super::xcm::quartz_xcm_tests;
-
-#[test]
-fn quartz_log_capture_tests() {
-	let mut logger = Logger::start();
-
-	quartz_xcm_tests(&mut logger);
-}
deletedruntime/quartz/src/tests/mod.rsdiffbeforeafterboth
--- a/runtime/quartz/src/tests/mod.rs
+++ /dev/null
@@ -1,18 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-mod logcapture;
-mod xcm;
deletedruntime/quartz/src/tests/xcm.rsdiffbeforeafterboth
--- a/runtime/quartz/src/tests/xcm.rs
+++ /dev/null
@@ -1,27 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-use logtest::Logger;
-use crate::{runtime_common::tests::xcm::*, xcm_barrier::Barrier};
-
-const QUARTZ_PARA_ID: u32 = 2095;
-
-pub fn quartz_xcm_tests(logger: &mut Logger) {
-	barrier_denies_transact::<Barrier>(logger);
-
-	barrier_denies_transfer_from_unknown_location::<Barrier>(logger, QUARTZ_PARA_ID)
-		.expect("quartz runtime denies an unknown location");
-}
modifiedruntime/unique/Cargo.tomldiffbeforeafterboth
--- a/runtime/unique/Cargo.toml
+++ b/runtime/unique/Cargo.toml
@@ -298,8 +298,5 @@
 
 impl-trait-for-tuples = { workspace = true }
 
-[dev-dependencies]
-logtest = { workspace = true }
-
 [build-dependencies]
 substrate-wasm-builder = { workspace = true }
modifiedruntime/unique/src/lib.rsdiffbeforeafterboth
--- a/runtime/unique/src/lib.rs
+++ b/runtime/unique/src/lib.rs
@@ -41,9 +41,6 @@
 
 pub mod xcm_barrier;
 
-#[cfg(test)]
-mod tests;
-
 pub use runtime_common::*;
 
 pub const RUNTIME_NAME: &str = "unique";
deletedruntime/unique/src/tests/logcapture.rsdiffbeforeafterboth
--- a/runtime/unique/src/tests/logcapture.rs
+++ /dev/null
@@ -1,25 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-use logtest::Logger;
-use super::xcm::unique_xcm_tests;
-
-#[test]
-fn unique_log_capture_tests() {
-	let mut logger = Logger::start();
-
-	unique_xcm_tests(&mut logger);
-}
deletedruntime/unique/src/tests/mod.rsdiffbeforeafterboth
--- a/runtime/unique/src/tests/mod.rs
+++ /dev/null
@@ -1,18 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-mod logcapture;
-mod xcm;
deletedruntime/unique/src/tests/xcm.rsdiffbeforeafterboth
--- a/runtime/unique/src/tests/xcm.rs
+++ /dev/null
@@ -1,27 +0,0 @@
-// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.
-// This file is part of Unique Network.
-
-// Unique Network is free software: you can redistribute it and/or modify
-// it under the terms of the GNU General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-
-// Unique Network is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU General Public License for more details.
-
-// You should have received a copy of the GNU General Public License
-// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.
-
-use logtest::Logger;
-use crate::{runtime_common::tests::xcm::*, xcm_barrier::Barrier};
-
-const UNIQUE_PARA_ID: u32 = 2037;
-
-pub fn unique_xcm_tests(logger: &mut Logger) {
-	barrier_denies_transact::<Barrier>(logger);
-
-	barrier_denies_transfer_from_unknown_location::<Barrier>(logger, UNIQUE_PARA_ID)
-		.expect("unique runtime denies an unknown location");
-}