mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-07-09 04:47:24 +00:00
Stanly Johnson
d2ac8bd941
* commented use of common * hack to pass tests * another hack * remove all commented code * fix the easy tests * temp hack * follow through comma hack until better solution * patch macro * missed one * update benchmarks * update docs * fix docs * removed too much * fix changes Co-authored-by: Shawn Tabrizi <shawntabrizi@gmail.com>
122 lines
3.8 KiB
Rust
122 lines
3.8 KiB
Rust
// This file is part of Substrate.
|
|
|
|
// Copyright (C) 2020-2021 Parity Technologies (UK) Ltd.
|
|
// 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.
|
|
|
|
//! Treasury pallet benchmarking.
|
|
|
|
#![cfg(feature = "runtime-benchmarks")]
|
|
|
|
use super::*;
|
|
|
|
use frame_system::RawOrigin;
|
|
use frame_benchmarking::{benchmarks_instance, account};
|
|
use frame_support::traits::OnInitialize;
|
|
|
|
use crate::Module as Treasury;
|
|
|
|
const SEED: u32 = 0;
|
|
|
|
// Create the pre-requisite information needed to create a treasury `propose_spend`.
|
|
fn setup_proposal<T: Config<I>, I: Instance>(u: u32) -> (
|
|
T::AccountId,
|
|
BalanceOf<T, I>,
|
|
<T::Lookup as StaticLookup>::Source,
|
|
) {
|
|
let caller = account("caller", u, SEED);
|
|
let value: BalanceOf<T, I> = T::ProposalBondMinimum::get().saturating_mul(100u32.into());
|
|
let _ = T::Currency::make_free_balance_be(&caller, value);
|
|
let beneficiary = account("beneficiary", u, SEED);
|
|
let beneficiary_lookup = T::Lookup::unlookup(beneficiary);
|
|
(caller, value, beneficiary_lookup)
|
|
}
|
|
|
|
// Create proposals that are approved for use in `on_initialize`.
|
|
fn create_approved_proposals<T: Config<I>, I: Instance>(n: u32) -> Result<(), &'static str> {
|
|
for i in 0 .. n {
|
|
let (caller, value, lookup) = setup_proposal::<T, I>(i);
|
|
Treasury::<T, I>::propose_spend(
|
|
RawOrigin::Signed(caller).into(),
|
|
value,
|
|
lookup
|
|
)?;
|
|
let proposal_id = <ProposalCount<I>>::get() - 1;
|
|
Treasury::<T, I>::approve_proposal(RawOrigin::Root.into(), proposal_id)?;
|
|
}
|
|
ensure!(<Approvals<I>>::get().len() == n as usize, "Not all approved");
|
|
Ok(())
|
|
}
|
|
|
|
fn setup_pot_account<T: Config<I>, I: Instance>() {
|
|
let pot_account = Treasury::<T, I>::account_id();
|
|
let value = T::Currency::minimum_balance().saturating_mul(1_000_000_000u32.into());
|
|
let _ = T::Currency::make_free_balance_be(&pot_account, value);
|
|
}
|
|
|
|
benchmarks_instance! {
|
|
|
|
propose_spend {
|
|
let (caller, value, beneficiary_lookup) = setup_proposal::<T, _>(SEED);
|
|
// Whitelist caller account from further DB operations.
|
|
let caller_key = frame_system::Account::<T>::hashed_key_for(&caller);
|
|
frame_benchmarking::benchmarking::add_to_whitelist(caller_key.into());
|
|
}: _(RawOrigin::Signed(caller), value, beneficiary_lookup)
|
|
|
|
reject_proposal {
|
|
let (caller, value, beneficiary_lookup) = setup_proposal::<T, _>(SEED);
|
|
Treasury::<T, _>::propose_spend(
|
|
RawOrigin::Signed(caller).into(),
|
|
value,
|
|
beneficiary_lookup
|
|
)?;
|
|
let proposal_id = Treasury::<T, _>::proposal_count() - 1;
|
|
}: _(RawOrigin::Root, proposal_id)
|
|
|
|
approve_proposal {
|
|
let (caller, value, beneficiary_lookup) = setup_proposal::<T, _>(SEED);
|
|
Treasury::<T, _>::propose_spend(
|
|
RawOrigin::Signed(caller).into(),
|
|
value,
|
|
beneficiary_lookup
|
|
)?;
|
|
let proposal_id = Treasury::<T, _>::proposal_count() - 1;
|
|
}: _(RawOrigin::Root, proposal_id)
|
|
|
|
on_initialize_proposals {
|
|
let p in 0 .. 100;
|
|
setup_pot_account::<T, _>();
|
|
create_approved_proposals::<T, _>(p)?;
|
|
}: {
|
|
Treasury::<T, _>::on_initialize(T::BlockNumber::zero());
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::tests::{new_test_ext, Test};
|
|
use frame_support::assert_ok;
|
|
|
|
#[test]
|
|
fn test_benchmarks() {
|
|
new_test_ext().execute_with(|| {
|
|
assert_ok!(test_benchmark_propose_spend::<Test>());
|
|
assert_ok!(test_benchmark_reject_proposal::<Test>());
|
|
assert_ok!(test_benchmark_approve_proposal::<Test>());
|
|
assert_ok!(test_benchmark_on_initialize_proposals::<Test>());
|
|
});
|
|
}
|
|
}
|