add teleport tests

xcm/xcm-builder/src/integration_mock.rs

@@ -25,7 +25,7 @@ use sp_runtime::{testing::Header, traits::IdentityLookup, AccountId32};
 use polkadot_parachain::primitives::Id as ParaId;
 use polkadot_runtime_parachains::{configuration, origin, shared};
 use xcm::opaque::v0::MultiAsset;
-use xcm::v0::{MultiLocation::self, NetworkId};
+use xcm::v0::{Junction::*, MultiLocation::{self, *}, NetworkId};
 use xcm_executor::XcmExecutor;
 
 use crate as xcm_builder;
@@ -130,14 +130,22 @@ pub type Barrier = (
 	AllowUnpaidExecutionFrom<IsChildSystemParachain<ParaId>>,
 );
 
+parameter_types! {
+	pub const KusamaForStatemint: (MultiAsset, MultiLocation) =
+		(MultiAsset::AllConcreteFungible { id: Null }, X1(Parachain(1000)));
+}
+pub type TrustedTeleporters = (
+	xcm_builder::Case<KusamaForStatemint>,
+);
+
 pub struct XcmConfig;
 impl xcm_executor::Config for XcmConfig {
 	type Call = Call;
 	type XcmSender = mock::TestSendXcm;
 	type AssetTransactor = LocalAssetTransactor;
 	type OriginConverter = LocalOriginConverter;
 	type IsReserve = ();
-	type IsTeleporter = ();
+	type IsTeleporter = TrustedTeleporters;
 	type LocationInverter = LocationInverter<Ancestry>;
 	type Barrier = Barrier;
 	type Weigher = FixedWeightBounds<BaseXcmWeight, Call>;

xcm/xcm-builder/src/integration_tests.rs

@@ -177,12 +177,30 @@ fn teleport_to_statemine_works() {
 	let balances = vec![(ALICE, INITIAL_BALANCE), (para_acc.clone(), INITIAL_BALANCE)];
 	kusama_like_with_balances(balances).execute_with(|| {
 		let statemine_id = 1000;
+		let other_para_id = 3000;
 		let amount =  10 * ExistentialDeposit::get();
 		let teleport_effects = vec![
 			buy_execution(5), // unchecked mock value
 			Order::DepositAsset { assets: vec![All], dest: X2(Parent, Parachain(PARA_ID)) },
 		];
 		let weight = 3 * BaseXcmWeight::get();
+		let r = XcmExecutor::<XcmConfig>::execute_xcm(
+			Parachain(PARA_ID).into(),
+			Xcm::WithdrawAsset {
+				assets: vec![ConcreteFungible { id: Null, amount }],
+				effects: vec![
+					buy_execution(weight),
+					Order::InitiateTeleport {
+						assets: vec![All],
+						dest: Parachain(other_para_id).into(),
+						effects: teleport_effects.clone(),
+					},
+				],
+			},
+			weight,
+		);
+		// teleports not allowed to community chains
+		assert_eq!(r, Outcome::Error(XcmError::UntrustedTeleportLocation));
 		let r = XcmExecutor::<XcmConfig>::execute_xcm(
 			Parachain(PARA_ID).into(),
 			Xcm::WithdrawAsset {

xcm/xcm-builder/src/tests.rs

@@ -336,3 +336,43 @@ fn prepaid_result_of_query_should_get_free_execution() {
 	let r = XcmExecutor::<TestConfig>::execute_xcm(origin.clone(), message.clone(), weight_limit);
 	assert_eq!(r, Outcome::Incomplete(10, XcmError::Barrier));
 }
+
+#[test]
+fn teleport_destinations_should_be_filtered() {
+	let one = X1(AccountIndex64{index:1, network:Any});
+	let two = X1(AccountIndex64{index:2, network:Any});
+	AllowPaidFrom::set(vec![ one.clone(), two.clone() ]);
+	add_asset(1, ConcreteFungible { id: X1(Parent), amount: 100 });
+	add_asset(2, ConcreteFungible { id: X1(Parent), amount: 100 });
+	WeightPrice::set((X1(Parent), 1_000_000_000_000));
+
+	// teleport should fail to non-teleport locations
+	{
+		let origin = two.clone();
+		let message = Xcm::<TestCall>::WithdrawAsset {
+			assets: vec![ ConcreteFungible { id: X1(Parent), amount: 100 } ],	// enough for 100 units of weight.
+			effects: vec![
+				Order::<TestCall>::BuyExecution { fees: All, weight: 0, debt: 30, halt_on_error: true, xcm: vec![] },
+				Order::<TestCall>::InitiateTeleport { assets: vec![ All ], dest: two.clone(), effects: vec![] },
+			],
+		};
+		let weight_limit = 100;
+		let r = XcmExecutor::<TestConfig>::execute_xcm(origin, message, weight_limit);
+		assert_eq!(r, Outcome::Error(XcmError::UntrustedTeleportLocation));
+	}
+
+	add_teleporter(one.clone(), AllConcreteFungible { id: X1(Parent) });
+	// teleport should succed for known teleport locations
+	let origin = one.clone();
+	let message = Xcm::<TestCall>::WithdrawAsset {
+		assets: vec![ ConcreteFungible { id: X1(Parent), amount: 100 } ],	// enough for 100 units of weight.
+		effects: vec![
+			Order::<TestCall>::BuyExecution { fees: All, weight: 0, debt: 30, halt_on_error: true, xcm: vec![] },
+			Order::<TestCall>::InitiateTeleport { assets: vec![ All ], dest: one.clone(), effects: vec![] },
+		],
+	};
+	let weight_limit = 100;
+	let r = XcmExecutor::<TestConfig>::execute_xcm(origin, message, weight_limit);
+	assert_eq!(r, Outcome::Complete(50));
+	assert_eq!(assets(1), vec![ ConcreteFungible { id: X1(Parent), amount: 50 } ]);
+}