pezkuwichain
b6d35f6faf
Updated 4763 files with dual copyright: - Parity Technologies (UK) Ltd. - Dijital Kurdistan Tech Institute
86 lines
2.7 KiB
Rust
86 lines
2.7 KiB
Rust
// Copyright (C) Parity Technologies (UK) Ltd. and Dijital Kurdistan Tech Institute
|
|
// This file is part of Pezcumulus.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#![cfg(test)]
|
|
use bridge_hub_common::DenyExportMessageFrom;
|
|
use pezframe_support::{
|
|
parameter_types,
|
|
traits::{Equals, EverythingBut, ProcessMessageError::Unsupported},
|
|
};
|
|
use xcm::prelude::{
|
|
AliasOrigin, ByGenesis, ExportMessage, Here, Instruction, Location, NetworkId, Teyrchain,
|
|
Weight,
|
|
};
|
|
use xcm_executor::traits::{DenyExecution, Properties};
|
|
|
|
#[test]
|
|
fn test_deny_export_message_from() {
|
|
parameter_types! {
|
|
pub Source1: Location = Location::new(1, Teyrchain(1));
|
|
pub Source2: Location = Location::new(1, Teyrchain(2));
|
|
pub Remote1: NetworkId = ByGenesis([1;32]);
|
|
pub Remote2: NetworkId = ByGenesis([2;32]);
|
|
}
|
|
|
|
// Deny ExportMessage when both of the conditions met:
|
|
// 1: source != Source1
|
|
// 2: network == Remote1
|
|
pub type Denied = DenyExportMessageFrom<EverythingBut<Equals<Source1>>, Equals<Remote1>>;
|
|
|
|
let assert_deny_execution = |mut xcm: Vec<Instruction<()>>, origin, expected_result| {
|
|
assert_eq!(
|
|
Denied::deny_execution(
|
|
&origin,
|
|
&mut xcm,
|
|
Weight::zero(),
|
|
&mut Properties { weight_credit: Weight::zero(), message_id: None }
|
|
),
|
|
expected_result
|
|
);
|
|
};
|
|
|
|
// A message without an `ExportMessage` should pass
|
|
assert_deny_execution(vec![AliasOrigin(Here.into())], Source1::get(), Ok(()));
|
|
|
|
// `ExportMessage` from source1 and remote1 should pass
|
|
assert_deny_execution(
|
|
vec![ExportMessage { network: Remote1::get(), destination: Here, xcm: Default::default() }],
|
|
Source1::get(),
|
|
Ok(()),
|
|
);
|
|
|
|
// `ExportMessage` from source1 and remote2 should pass
|
|
assert_deny_execution(
|
|
vec![ExportMessage { network: Remote2::get(), destination: Here, xcm: Default::default() }],
|
|
Source1::get(),
|
|
Ok(()),
|
|
);
|
|
|
|
// `ExportMessage` from source2 and remote2 should pass
|
|
assert_deny_execution(
|
|
vec![ExportMessage { network: Remote2::get(), destination: Here, xcm: Default::default() }],
|
|
Source2::get(),
|
|
Ok(()),
|
|
);
|
|
|
|
// `ExportMessage` from source2 and remote1 should be banned
|
|
assert_deny_execution(
|
|
vec![ExportMessage { network: Remote1::get(), destination: Here, xcm: Default::default() }],
|
|
Source2::get(),
|
|
Err(Unsupported),
|
|
);
|
|
}
|