Skip to content

Commit 6fe1a55

Browse files
committed
fix: adjusting dao benchmarks for new threshold validation
1 parent 8e8ca97 commit 6fe1a55

File tree

1 file changed

+18
-14
lines changed

1 file changed

+18
-14
lines changed

substrate-node/pallets/pallet-dao/src/benchmarking.rs

Lines changed: 18 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ benchmarks! {
2626
propose {
2727
let caller: T::AccountId = whitelisted_caller();
2828
assert_ok!(_add_council_member::<T>(caller.clone()));
29-
let threshold = 1;
29+
let threshold = 2;
3030
let proposal: T::Proposal = SystemCall::<T>::remark { remark: b"remark".to_vec() }.into();
3131
let description = b"some_description".to_vec();
3232
let link = b"some_link".to_vec();
@@ -53,7 +53,7 @@ benchmarks! {
5353
// vote()
5454
vote {
5555
let farmer: T::AccountId = account("Alice", 0, 0);
56-
_prepare_farm_with_node::<T>(farmer.clone());
56+
_prepare_farm_with_node::<T>(farmer.clone(), b"testfarm", 1);
5757
let farm_id = 1;
5858

5959
let caller: T::AccountId = whitelisted_caller();
@@ -80,7 +80,7 @@ benchmarks! {
8080
// veto()
8181
veto {
8282
let farmer: T::AccountId = account("Alice", 0, 0);
83-
_prepare_farm_with_node::<T>(farmer.clone());
83+
_prepare_farm_with_node::<T>(farmer.clone(), b"testfarm", 1);
8484
let farm_id = 1;
8585

8686
let caller: T::AccountId = whitelisted_caller();
@@ -98,15 +98,19 @@ benchmarks! {
9898
// close()
9999
close {
100100
let farmer: T::AccountId = account("Alice", 0, 0);
101-
_prepare_farm_with_node::<T>(farmer.clone());
102-
let farm_id = 1;
101+
let farmer2: T::AccountId = account("Bob", 0, 0);
102+
103+
_prepare_farm_with_node::<T>(farmer.clone(), b"testfarm", 1);
104+
_prepare_farm_with_node::<T>(farmer2.clone(), b"testfarm2", 2);
103105

104106
let caller: T::AccountId = whitelisted_caller();
105107
let proposal_hash = _create_proposal::<T>(caller.clone());
106108
let proposal_index = 0;
107109

108110
let approve = false;
109-
DaoModule::<T>::vote(RawOrigin::Signed(farmer.clone()).into(), farm_id, proposal_hash, approve).unwrap();
111+
DaoModule::<T>::vote(RawOrigin::Signed(farmer.clone()).into(), 1, proposal_hash, approve).unwrap();
112+
DaoModule::<T>::vote(RawOrigin::Signed(farmer2.clone()).into(), 2, proposal_hash, approve).unwrap();
113+
110114
}: _(RawOrigin::Signed(caller.clone()), proposal_hash, proposal_index)
111115
verify {
112116
assert!(DaoModule::<T>::proposal_list(proposal_hash).is_none());
@@ -127,10 +131,10 @@ fn assert_last_event<T: Config>(generic_event: <T as Config>::RuntimeEvent) {
127131
assert_eq!(event, &system_event);
128132
}
129133

130-
pub fn _prepare_farm_with_node<T: Config>(source: T::AccountId) {
134+
pub fn _prepare_farm_with_node<T: Config>(source: T::AccountId, farm_name: &[u8], farm_id : u32) {
131135
_create_twin::<T>(source.clone());
132-
_create_farm::<T>(source.clone());
133-
_create_node::<T>(source.clone());
136+
_create_farm::<T>(source.clone(), farm_name);
137+
_create_node::<T>(source.clone(), farm_id);
134138
}
135139

136140
fn _create_twin<T: Config>(source: T::AccountId) {
@@ -147,7 +151,7 @@ fn _create_twin<T: Config>(source: T::AccountId) {
147151
));
148152
}
149153

150-
fn _create_farm<T: Config>(source: T::AccountId) {
154+
fn _create_farm<T: Config>(source: T::AccountId, farm_name: &[u8]) {
151155
let mut pub_ips = Vec::new();
152156
pub_ips.push(IP4 {
153157
ip: get_public_ip_ip_input(b"185.206.122.33/24"),
@@ -160,12 +164,12 @@ fn _create_farm<T: Config>(source: T::AccountId) {
160164

161165
assert_ok!(TfgridModule::<T>::create_farm(
162166
RawOrigin::Signed(source).into(),
163-
b"testfarm".to_vec().try_into().unwrap(),
167+
farm_name.to_vec().try_into().unwrap(),
164168
pub_ips.clone().try_into().unwrap(),
165169
));
166170
}
167171

168-
fn _create_node<T: Config>(source: T::AccountId) {
172+
fn _create_node<T: Config>(source: T::AccountId, farm_id: u32) {
169173
let resources = ResourcesInput {
170174
hru: 1024 * GIGABYTE,
171175
sru: 512 * GIGABYTE,
@@ -183,7 +187,7 @@ fn _create_node<T: Config>(source: T::AccountId) {
183187

184188
assert_ok!(TfgridModule::<T>::create_node(
185189
RawOrigin::Signed(source.clone()).into(),
186-
1,
190+
farm_id,
187191
resources,
188192
location,
189193
Vec::new().try_into().unwrap(),
@@ -196,7 +200,7 @@ fn _create_node<T: Config>(source: T::AccountId) {
196200
pub fn _create_proposal<T: Config>(source: T::AccountId) -> T::Hash {
197201
assert_ok!(_add_council_member::<T>(source.clone()));
198202

199-
let threshold = 1;
203+
let threshold = 2;
200204
let proposal: T::Proposal = SystemCall::<T>::remark {
201205
remark: b"remark".to_vec(),
202206
}

0 commit comments

Comments
 (0)