wip(pallet-communities): initial parmeters for benchmarks

virto-network · Mar 8, 2024 · 898e729 · 898e729
1 parent 8edfc00
commit 898e729
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 10 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/pallets/communities/src/benchmarking.rs b/pallets/communities/src/benchmarking.rs
@@ -1,5 +1,7 @@
 //! Benchmarking setup for pallet-communities
 #![cfg(feature = "runtime-benchmarks")]
+use self::types::{CommunityIdOf, PalletsOriginOf};
+
 use super::*;
 
 use self::{

diff --git a/pallets/communities/src/lib.rs b/pallets/communities/src/lib.rs
@@ -196,6 +196,11 @@ pub mod pallet {
 	use sp_runtime::traits::StaticLookup;
 	use types::{PollIndexOf, *};
 
+	#[cfg(feature = "runtime-benchmarks")]
+	pub trait BenchmarkHelper<T: Config> {
+		fn get_community_id() -> CommunityIdOf<T>;
+	}
+
 	#[pallet::pallet]
 	pub struct Pallet<T>(_);
 

diff --git a/pallets/communities/src/tests/mock.rs b/pallets/communities/src/tests/mock.rs
@@ -9,6 +9,7 @@ use frame_support::{
 	PalletId,
 };
 use frame_system::{EnsureRoot, EnsureRootWithSuccess, EnsureSigned};
+use pallet_communities::{types::CommunityIdOf, BenchmarkHelper};
 use pallet_referenda::{TrackIdOf, TrackInfoOf, TracksInfo};
 use parity_scale_codec::Compact;
 use sp_core::H256;