forked from paritytech/substrate
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
setup initial mock structure establish privelleged function test save progress first pass new_test_ext test sudo to make sure it error when non-root add set keys tests fix unused result from set_key call parent 5151bd7 author zeke <[email protected]> 1589076740 -0700 committer zeke <[email protected]> 1589350443 -0700 parent 5151bd7 author zeke <[email protected]> 1589076740 -0700 committer zeke <[email protected]> 1589350442 -0700 remove unused imports warnings fix unused result from set_key call remove unused imports warnings pre master merge Expose BlockHashCount on system metadata constants (paritytech#5960) squash fix whitespace spelling and whitespace a single pesky space add logger module to mock add logger dispatch to privlleged function sub logger in for dummy functions create first of several event tests first pass at test coverage for events comment house keeping pre master merge Expose BlockHashCount on system metadata constants (paritytech#5960) Expose BlockHashCount on system metadata constants (paritytech#5960) fix whitespace spell check implement last_seen_account storage item, event, and, logger function create vec account log and use in tests allow weight to be passed into account log refactor all log dispatchables save progress complete initial transition to refactored logger cleaning
- Loading branch information
Showing
11 changed files
with
376 additions
and
73 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,185 @@ | ||
// Copyright 2020 Parity Technologies (UK) Ltd. | ||
// This file is part of Substrate. | ||
|
||
// Substrate 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. | ||
|
||
// Substrate 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 Substrate. If not, see <http://www.gnu.org/licenses/>. | ||
|
||
//! Test utilities | ||
use super::*; | ||
use frame_support::{ | ||
impl_outer_origin, impl_outer_dispatch, impl_outer_event, parameter_types, | ||
weights::{Weight, DispatchClass} | ||
}; | ||
use sp_core::H256; | ||
// The testing primitives are very useful for avoiding having to work with signatures | ||
// or public keys. | ||
use sp_runtime::{Perbill, traits::{BlakeTwo256, IdentityLookup}, testing::Header}; | ||
use sp_io; | ||
use crate as sudo; | ||
|
||
// Logger module to track execution. | ||
pub mod logger { | ||
use super::*; | ||
use frame_system::ensure_root; | ||
|
||
pub trait Trait: system::Trait { | ||
type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>; | ||
} | ||
|
||
decl_storage! { | ||
trait Store for Module<T: Trait> as Logger { | ||
AccountLog get(fn account_log): Vec<T::AccountId>; | ||
I32Log get(fn i32_log): Vec<i32>; | ||
} | ||
} | ||
|
||
decl_event! { | ||
pub enum Event<T> where AccountId = <T as frame_system::Trait>::AccountId { | ||
AppendAccount(AccountId, i32, Weight), | ||
AppendI32(i32, Weight), | ||
} | ||
} | ||
|
||
decl_module! { | ||
pub struct Module<T: Trait> for enum Call where origin: <T as system::Trait>::Origin { | ||
fn deposit_event() = default; | ||
|
||
#[weight = FunctionOf( | ||
|args: (&i32, &Weight)| *args.1, | ||
DispatchClass::Normal, | ||
Pays::Yes, | ||
)] | ||
fn privileged_i32_log(origin, i: i32, weight: Weight){ | ||
// Ensure that the `origin` is `Root`. | ||
ensure_root(origin)?; | ||
<I32Log>::append(i); | ||
Self::deposit_event(RawEvent::AppendI32(i, weight)); | ||
} | ||
|
||
#[weight = FunctionOf( | ||
|args: (&i32, &Weight)| *args.1, | ||
DispatchClass::Normal, | ||
Pays::Yes, | ||
)] | ||
fn non_privileged_i32_log(origin, i: i32, weight: Weight){ | ||
// Ensure that the `origin` is some signed account. | ||
ensure_signed(origin)?; | ||
<I32Log>::append(i); | ||
Self::deposit_event(RawEvent::AppendI32(i, weight)); | ||
} | ||
|
||
#[weight = FunctionOf( | ||
|args: (&i32, &Weight)| *args.1, | ||
DispatchClass::Normal, | ||
Pays::Yes, | ||
)] | ||
fn non_privileged_account_log(origin, i: i32, weight: Weight) { | ||
// Ensure that the `origin` is some signed account. | ||
let sender = ensure_signed(origin)?; | ||
<AccountLog<T>>::append(sender.clone()); | ||
Self::deposit_event(RawEvent::AppendAccount(sender, i, weight)); | ||
} | ||
} | ||
} | ||
} | ||
|
||
impl_outer_origin! { | ||
pub enum Origin for Test where system = frame_system {} | ||
} | ||
|
||
mod test_events { | ||
pub use crate::Event; | ||
} | ||
|
||
impl_outer_event! { | ||
pub enum TestEvent for Test { | ||
system<T>, | ||
sudo<T>, | ||
logger<T>, | ||
} | ||
} | ||
|
||
impl_outer_dispatch! { | ||
pub enum Call for Test where origin: Origin { | ||
sudo::Sudo, | ||
logger::Logger, | ||
} | ||
} | ||
|
||
// For testing the pallet, we construct most of a mock runtime. This means | ||
// first constructing a configuration type (`Test`) which `impl`s each of the | ||
// configuration traits of pallets we want to use. | ||
#[derive(Clone, Eq, PartialEq)] | ||
pub struct Test; | ||
|
||
parameter_types! { | ||
pub const BlockHashCount: u64 = 250; | ||
pub const MaximumBlockWeight: Weight = 1024; | ||
pub const MaximumBlockLength: u32 = 2 * 1024; | ||
pub const AvailableBlockRatio: Perbill = Perbill::one(); | ||
} | ||
|
||
impl frame_system::Trait for Test { | ||
type Origin = Origin; | ||
type Call = Call; | ||
type Index = u64; | ||
type BlockNumber = u64; | ||
type Hash = H256; | ||
type Hashing = BlakeTwo256; | ||
type AccountId = u64; | ||
type Lookup = IdentityLookup<Self::AccountId>; | ||
type Header = Header; | ||
type Event = TestEvent; | ||
type BlockHashCount = BlockHashCount; | ||
type MaximumBlockWeight = MaximumBlockWeight; | ||
type DbWeight = (); | ||
type BlockExecutionWeight = (); | ||
type ExtrinsicBaseWeight = (); | ||
type MaximumBlockLength = MaximumBlockLength; | ||
type AvailableBlockRatio = AvailableBlockRatio; | ||
type Version = (); | ||
type ModuleToIndex = (); | ||
type AccountData = (); | ||
type OnNewAccount = (); | ||
type OnKilledAccount = (); | ||
} | ||
|
||
// Implement the logger module's `Trait` on the Test runtime. | ||
impl logger::Trait for Test { | ||
type Event = TestEvent; | ||
} | ||
|
||
// Implement the sudo module's `Trait` on the Test runtime. | ||
impl Trait for Test { | ||
type Event = TestEvent; | ||
type Call = Call; | ||
} | ||
|
||
// Assign back to type variables in order to make dispatched calls of these modules later. | ||
pub type Sudo = Module<Test>; | ||
pub type Logger = logger::Module<Test>; | ||
pub type System = system::Module<Test>; | ||
|
||
// New types for dispatchable functions. | ||
pub type SudoCall = sudo::Call<Test>; | ||
pub type LoggerCall = logger::Call<Test>; | ||
|
||
// Build test environment by setting the root `key` for the Genesis. | ||
pub fn new_test_ext(root_key: u64) -> sp_io::TestExternalities { | ||
let mut t = frame_system::GenesisConfig::default().build_storage::<Test>().unwrap(); | ||
GenesisConfig::<Test>{ | ||
key: root_key, | ||
}.assimilate_storage(&mut t).unwrap(); | ||
t.into() | ||
} |
Oops, something went wrong.