mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-20 00:01:03 +00:00
Participation Lottery Pallet (#7221)
* Basic design * start adding tests * finish tests * clean up crates * use call index for match * finish benchmarks * add to runtime * fix * cargo run --release --features=runtime-benchmarks --manifest-path=bin/node/cli/Cargo.toml -- benchmark --chain=dev --steps=50 --repeat=20 --pallet=pallet_lottery --extrinsic=* --execution=wasm --wasm-execution=compiled --heap-pages=4096 --output=./frame/lottery/src/weights.rs --template=./.maintain/frame-weight-template.hbs * more efficient storage * cargo run --release --features=runtime-benchmarks --manifest-path=bin/node/cli/Cargo.toml -- benchmark --chain=dev --steps=50 --repeat=20 --pallet=pallet_lottery --extrinsic=* --execution=wasm --wasm-execution=compiled --heap-pages=4096 --output=./frame/lottery/src/weights.rs --template=./.maintain/frame-weight-template.hbs * Update lib.rs * Update bin/node/runtime/src/lib.rs * trait -> config * add repeating lottery * new benchmarks * fix build * move trait for warning * feedback from @xlc * add stop_repeat * fix * cargo run --release --features=runtime-benchmarks --manifest-path=bin/node/cli/Cargo.toml -- benchmark --chain=dev --steps=50 --repeat=20 --pallet=pallet_lottery --extrinsic=* --execution=wasm --wasm-execution=compiled --heap-pages=4096 --output=./frame/lottery/src/weights.rs --template=./.maintain/frame-weight-template.hbs * Support static calls * cargo run --release --features=runtime-benchmarks --manifest-path=bin/node/cli/Cargo.toml -- benchmark --chain=dev --steps=50 --repeat=20 --pallet=pallet_lottery --extrinsic=* --execution=wasm --wasm-execution=compiled --heap-pages=4096 --output=./frame/lottery/src/weights.rs --template=./.maintain/frame-weight-template.hbs * fix test * add loop to mitigate modulo bias * Update weights for worst case scenario loop * Initialize pot with ED * cargo run --release --features=runtime-benchmarks --manifest-path=bin/node/cli/Cargo.toml -- benchmark --chain=dev --steps=50 --repeat=20 --pallet=pallet_lottery --extrinsic=* --execution=wasm --wasm-execution=compiled --heap-pages=4096 --output=./frame/lottery/src/weights.rs --template=./.maintain/frame-weight-template.hbs Co-authored-by: Parity Benchmarking Bot <admin@parity.io>
This commit is contained in:
Shawn Tabrizi
@@ -0,0 +1,192 @@
|
||||
// This file is part of Substrate.
|
||||
|
||||
// Copyright (C) 2020 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.
|
||||
|
||||
//! Lottery pallet benchmarking.
|
||||
|
||||
#![cfg(feature = "runtime-benchmarks")]
|
||||
|
||||
use super::*;
|
||||
|
||||
use frame_system::RawOrigin;
|
||||
use frame_support::traits::{OnInitialize, UnfilteredDispatchable};
|
||||
use frame_benchmarking::{benchmarks, account, whitelisted_caller};
|
||||
use sp_runtime::traits::{Bounded, Zero};
|
||||
|
||||
use crate::Module as Lottery;
|
||||
|
||||
// Set up and start a lottery
|
||||
fn setup_lottery<T: Config>(repeat: bool) -> Result<(), &'static str> {
|
||||
let price = T::Currency::minimum_balance();
|
||||
let length = 10u32.into();
|
||||
let delay = 5u32.into();
|
||||
// Calls will be maximum length...
|
||||
let mut calls = vec![
|
||||
frame_system::Call::<T>::set_code(vec![]).into();
|
||||
T::MaxCalls::get().saturating_sub(1)
|
||||
];
|
||||
// Last call will be the match for worst case scenario.
|
||||
calls.push(frame_system::Call::<T>::remark(vec![]).into());
|
||||
let origin = T::ManagerOrigin::successful_origin();
|
||||
Lottery::<T>::set_calls(origin.clone(), calls)?;
|
||||
Lottery::<T>::start_lottery(origin, price, length, delay, repeat)?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
benchmarks! {
|
||||
_ { }
|
||||
|
||||
buy_ticket {
|
||||
let caller = whitelisted_caller();
|
||||
T::Currency::make_free_balance_be(&caller, BalanceOf::<T>::max_value());
|
||||
setup_lottery::<T>(false)?;
|
||||
// force user to have a long vec of calls participating
|
||||
let set_code_index: CallIndex = Lottery::<T>::call_to_index(
|
||||
&frame_system::Call::<T>::set_code(vec![]).into()
|
||||
)?;
|
||||
let already_called: (u32, Vec<CallIndex>) = (
|
||||
LotteryIndex::get(),
|
||||
vec![
|
||||
set_code_index;
|
||||
T::MaxCalls::get().saturating_sub(1)
|
||||
],
|
||||
);
|
||||
Participants::<T>::insert(&caller, already_called);
|
||||
|
||||
let call = frame_system::Call::<T>::remark(vec![]);
|
||||
}: _(RawOrigin::Signed(caller), Box::new(call.into()))
|
||||
verify {
|
||||
assert_eq!(TicketsCount::get(), 1);
|
||||
}
|
||||
|
||||
set_calls {
|
||||
let n in 0 .. T::MaxCalls::get() as u32;
|
||||
let calls = vec![frame_system::Call::<T>::remark(vec![]).into(); n as usize];
|
||||
|
||||
let call = Call::<T>::set_calls(calls);
|
||||
let origin = T::ManagerOrigin::successful_origin();
|
||||
assert!(CallIndices::get().is_empty());
|
||||
}: { call.dispatch_bypass_filter(origin)? }
|
||||
verify {
|
||||
if !n.is_zero() {
|
||||
assert!(!CallIndices::get().is_empty());
|
||||
}
|
||||
}
|
||||
|
||||
start_lottery {
|
||||
let price = BalanceOf::<T>::max_value();
|
||||
let end = 10u32.into();
|
||||
let payout = 5u32.into();
|
||||
|
||||
let call = Call::<T>::start_lottery(price, end, payout, true);
|
||||
let origin = T::ManagerOrigin::successful_origin();
|
||||
}: { call.dispatch_bypass_filter(origin)? }
|
||||
verify {
|
||||
assert!(crate::Lottery::<T>::get().is_some());
|
||||
}
|
||||
|
||||
stop_repeat {
|
||||
setup_lottery::<T>(true)?;
|
||||
assert_eq!(crate::Lottery::<T>::get().unwrap().repeat, true);
|
||||
let call = Call::<T>::stop_repeat();
|
||||
let origin = T::ManagerOrigin::successful_origin();
|
||||
}: { call.dispatch_bypass_filter(origin)? }
|
||||
verify {
|
||||
assert_eq!(crate::Lottery::<T>::get().unwrap().repeat, false);
|
||||
}
|
||||
|
||||
on_initialize_end {
|
||||
setup_lottery::<T>(false)?;
|
||||
let winner = account("winner", 0, 0);
|
||||
// User needs more than min balance to get ticket
|
||||
T::Currency::make_free_balance_be(&winner, T::Currency::minimum_balance() * 10u32.into());
|
||||
// Make sure lottery account has at least min balance too
|
||||
let lottery_account = Lottery::<T>::account_id();
|
||||
T::Currency::make_free_balance_be(&lottery_account, T::Currency::minimum_balance() * 10u32.into());
|
||||
// Buy a ticket
|
||||
let call = frame_system::Call::<T>::remark(vec![]);
|
||||
Lottery::<T>::buy_ticket(RawOrigin::Signed(winner.clone()).into(), Box::new(call.into()))?;
|
||||
// Kill user account for worst case
|
||||
T::Currency::make_free_balance_be(&winner, 0u32.into());
|
||||
// Assert that lotto is set up for winner
|
||||
assert_eq!(TicketsCount::get(), 1);
|
||||
assert!(!Lottery::<T>::pot().1.is_zero());
|
||||
}: {
|
||||
// Generate `MaxGenerateRandom` numbers for worst case scenario
|
||||
for i in 0 .. T::MaxGenerateRandom::get() {
|
||||
Lottery::<T>::generate_random_number(i);
|
||||
}
|
||||
// Start lottery has block 15 configured for payout
|
||||
Lottery::<T>::on_initialize(15u32.into());
|
||||
}
|
||||
verify {
|
||||
assert!(crate::Lottery::<T>::get().is_none());
|
||||
assert_eq!(TicketsCount::get(), 0);
|
||||
assert_eq!(Lottery::<T>::pot().1, 0u32.into());
|
||||
assert!(!T::Currency::free_balance(&winner).is_zero())
|
||||
}
|
||||
|
||||
on_initialize_repeat {
|
||||
setup_lottery::<T>(true)?;
|
||||
let winner = account("winner", 0, 0);
|
||||
// User needs more than min balance to get ticket
|
||||
T::Currency::make_free_balance_be(&winner, T::Currency::minimum_balance() * 10u32.into());
|
||||
// Make sure lottery account has at least min balance too
|
||||
let lottery_account = Lottery::<T>::account_id();
|
||||
T::Currency::make_free_balance_be(&lottery_account, T::Currency::minimum_balance() * 10u32.into());
|
||||
// Buy a ticket
|
||||
let call = frame_system::Call::<T>::remark(vec![]);
|
||||
Lottery::<T>::buy_ticket(RawOrigin::Signed(winner.clone()).into(), Box::new(call.into()))?;
|
||||
// Kill user account for worst case
|
||||
T::Currency::make_free_balance_be(&winner, 0u32.into());
|
||||
// Assert that lotto is set up for winner
|
||||
assert_eq!(TicketsCount::get(), 1);
|
||||
assert!(!Lottery::<T>::pot().1.is_zero());
|
||||
}: {
|
||||
// Generate `MaxGenerateRandom` numbers for worst case scenario
|
||||
for i in 0 .. T::MaxGenerateRandom::get() {
|
||||
Lottery::<T>::generate_random_number(i);
|
||||
}
|
||||
// Start lottery has block 15 configured for payout
|
||||
Lottery::<T>::on_initialize(15u32.into());
|
||||
}
|
||||
verify {
|
||||
assert!(crate::Lottery::<T>::get().is_some());
|
||||
assert_eq!(LotteryIndex::get(), 2);
|
||||
assert_eq!(TicketsCount::get(), 0);
|
||||
assert_eq!(Lottery::<T>::pot().1, 0u32.into());
|
||||
assert!(!T::Currency::free_balance(&winner).is_zero())
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::mock::{new_test_ext, Test};
|
||||
use frame_support::assert_ok;
|
||||
|
||||
#[test]
|
||||
fn test_benchmarks() {
|
||||
new_test_ext().execute_with(|| {
|
||||
assert_ok!(test_benchmark_buy_ticket::<Test>());
|
||||
assert_ok!(test_benchmark_set_calls::<Test>());
|
||||
assert_ok!(test_benchmark_start_lottery::<Test>());
|
||||
assert_ok!(test_benchmark_stop_repeat::<Test>());
|
||||
assert_ok!(test_benchmark_on_initialize_end::<Test>());
|
||||
assert_ok!(test_benchmark_on_initialize_repeat::<Test>());
|
||||
});
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user