@@ -26,7 +26,7 @@ benchmarks! {
26
26
propose {
27
27
let caller: T :: AccountId = whitelisted_caller( ) ;
28
28
assert_ok!( _add_council_member:: <T >( caller. clone( ) ) ) ;
29
- let threshold = 1 ;
29
+ let threshold = 2 ;
30
30
let proposal: T :: Proposal = SystemCall :: <T >:: remark { remark: b"remark" . to_vec( ) } . into( ) ;
31
31
let description = b"some_description" . to_vec( ) ;
32
32
let link = b"some_link" . to_vec( ) ;
@@ -53,7 +53,7 @@ benchmarks! {
53
53
// vote()
54
54
vote {
55
55
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 ) ;
57
57
let farm_id = 1 ;
58
58
59
59
let caller: T :: AccountId = whitelisted_caller( ) ;
@@ -80,7 +80,7 @@ benchmarks! {
80
80
// veto()
81
81
veto {
82
82
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 ) ;
84
84
let farm_id = 1 ;
85
85
86
86
let caller: T :: AccountId = whitelisted_caller( ) ;
@@ -98,15 +98,19 @@ benchmarks! {
98
98
// close()
99
99
close {
100
100
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 ) ;
103
105
104
106
let caller: T :: AccountId = whitelisted_caller( ) ;
105
107
let proposal_hash = _create_proposal:: <T >( caller. clone( ) ) ;
106
108
let proposal_index = 0 ;
107
109
108
110
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
+
110
114
} : _( RawOrigin :: Signed ( caller. clone( ) ) , proposal_hash, proposal_index)
111
115
verify {
112
116
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) {
127
131
assert_eq ! ( event, & system_event) ;
128
132
}
129
133
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 ) {
131
135
_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 ) ;
134
138
}
135
139
136
140
fn _create_twin < T : Config > ( source : T :: AccountId ) {
@@ -147,7 +151,7 @@ fn _create_twin<T: Config>(source: T::AccountId) {
147
151
) ) ;
148
152
}
149
153
150
- fn _create_farm < T : Config > ( source : T :: AccountId ) {
154
+ fn _create_farm < T : Config > ( source : T :: AccountId , farm_name : & [ u8 ] ) {
151
155
let mut pub_ips = Vec :: new ( ) ;
152
156
pub_ips. push ( IP4 {
153
157
ip : get_public_ip_ip_input ( b"185.206.122.33/24" ) ,
@@ -160,12 +164,12 @@ fn _create_farm<T: Config>(source: T::AccountId) {
160
164
161
165
assert_ok ! ( TfgridModule :: <T >:: create_farm(
162
166
RawOrigin :: Signed ( source) . into( ) ,
163
- b"testfarm" . to_vec( ) . try_into( ) . unwrap( ) ,
167
+ farm_name . to_vec( ) . try_into( ) . unwrap( ) ,
164
168
pub_ips. clone( ) . try_into( ) . unwrap( ) ,
165
169
) ) ;
166
170
}
167
171
168
- fn _create_node < T : Config > ( source : T :: AccountId ) {
172
+ fn _create_node < T : Config > ( source : T :: AccountId , farm_id : u32 ) {
169
173
let resources = ResourcesInput {
170
174
hru : 1024 * GIGABYTE ,
171
175
sru : 512 * GIGABYTE ,
@@ -183,7 +187,7 @@ fn _create_node<T: Config>(source: T::AccountId) {
183
187
184
188
assert_ok ! ( TfgridModule :: <T >:: create_node(
185
189
RawOrigin :: Signed ( source. clone( ) ) . into( ) ,
186
- 1 ,
190
+ farm_id ,
187
191
resources,
188
192
location,
189
193
Vec :: new( ) . try_into( ) . unwrap( ) ,
@@ -196,7 +200,7 @@ fn _create_node<T: Config>(source: T::AccountId) {
196
200
pub fn _create_proposal < T : Config > ( source : T :: AccountId ) -> T :: Hash {
197
201
assert_ok ! ( _add_council_member:: <T >( source. clone( ) ) ) ;
198
202
199
- let threshold = 1 ;
203
+ let threshold = 2 ;
200
204
let proposal: T :: Proposal = SystemCall :: < T > :: remark {
201
205
remark : b"remark" . to_vec ( ) ,
202
206
}
0 commit comments