Skip to content

Commit 57f0b95

Browse files
Krayt78seemantaggarwalbkchr
authored
Migrating cumulus-pallet-session-benchmarking to Benchmarking V2 (#6564)
# Description Migrating cumulus-pallet-session-benchmarking to the new benchmarking syntax v2. This is a part of #6202 --------- Co-authored-by: seemantaggarwal <32275622+seemantaggarwal@users.noreply.github.com> Co-authored-by: Bastian Köcher <git@kchr.de>
1 parent 2d53238 commit 57f0b95

1 file changed

Lines changed: 29 additions & 9 deletions

File tree

  • cumulus/pallets/session-benchmarking/src

cumulus/pallets/session-benchmarking/src/inner.rs

Lines changed: 29 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -14,29 +14,49 @@
1414
// limitations under the License.
1515

1616
//! Benchmarking setup for pallet-session.
17+
#![cfg(feature = "runtime-benchmarks")]
1718

1819
use alloc::{vec, vec::Vec};
1920

2021
use codec::Decode;
21-
use frame_benchmarking::{benchmarks, whitelisted_caller};
22+
use frame_benchmarking::v2::*;
2223
use frame_system::RawOrigin;
2324
use pallet_session::*;
2425
pub struct Pallet<T: Config>(pallet_session::Pallet<T>);
2526
pub trait Config: pallet_session::Config {}
2627

27-
benchmarks! {
28-
set_keys {
28+
#[benchmarks]
29+
mod benchmarks {
30+
use super::*;
31+
32+
#[benchmark]
33+
fn set_keys() -> Result<(), BenchmarkError> {
2934
let caller: T::AccountId = whitelisted_caller();
3035
frame_system::Pallet::<T>::inc_providers(&caller);
3136
let keys = T::Keys::decode(&mut sp_runtime::traits::TrailingZeroInput::zeroes()).unwrap();
32-
let proof: Vec<u8> = vec![0,1,2,3];
33-
}: _(RawOrigin::Signed(caller), keys, proof)
37+
let proof: Vec<u8> = vec![0, 1, 2, 3];
38+
39+
#[extrinsic_call]
40+
_(RawOrigin::Signed(caller), keys, proof);
41+
42+
Ok(())
43+
}
3444

35-
purge_keys {
45+
#[benchmark]
46+
fn purge_keys() -> Result<(), BenchmarkError> {
3647
let caller: T::AccountId = whitelisted_caller();
3748
frame_system::Pallet::<T>::inc_providers(&caller);
3849
let keys = T::Keys::decode(&mut sp_runtime::traits::TrailingZeroInput::zeroes()).unwrap();
39-
let proof: Vec<u8> = vec![0,1,2,3];
40-
let _t = pallet_session::Pallet::<T>::set_keys(RawOrigin::Signed(caller.clone()).into(), keys, proof);
41-
}: _(RawOrigin::Signed(caller))
50+
let proof: Vec<u8> = vec![0, 1, 2, 3];
51+
let _t = pallet_session::Pallet::<T>::set_keys(
52+
RawOrigin::Signed(caller.clone()).into(),
53+
keys,
54+
proof,
55+
);
56+
57+
#[extrinsic_call]
58+
_(RawOrigin::Signed(caller));
59+
60+
Ok(())
61+
}
4262
}

0 commit comments

Comments
 (0)