@@ -167,8 +167,8 @@ impl<'a> Delegation<'a> {
167167 }
168168
169169 #[ cfg( test) ]
170- pub fn get_quantity ( & self , delegator : & Address ) -> StakeQuantity {
171- self . delegatees . get ( delegator ) . cloned ( ) . unwrap_or ( 0 )
170+ pub fn get_quantity ( & self , delegatee : & Address ) -> StakeQuantity {
171+ self . delegatees . get ( delegatee ) . cloned ( ) . unwrap_or ( 0 )
172172 }
173173
174174 #[ cfg( test) ]
@@ -442,23 +442,23 @@ mod tests {
442442 let mut state = helpers:: get_temp_state ( ) ;
443443
444444 // Prepare
445- let delegatee = Address :: random ( ) ;
446- let delegators : Vec < _ > = ( 0 ..10 ) . map ( |_| Address :: random ( ) ) . collect ( ) ;
445+ let delegator = Address :: random ( ) ;
446+ let delegatees : Vec < _ > = ( 0 ..10 ) . map ( |_| Address :: random ( ) ) . collect ( ) ;
447447 let delegation_amount: HashMap < & Address , StakeQuantity > =
448- delegators . iter ( ) . map ( |address| ( address, rng. gen_range ( 0 , 100 ) ) ) . collect ( ) ;
448+ delegatees . iter ( ) . map ( |address| ( address, rng. gen_range ( 0 , 100 ) ) ) . collect ( ) ;
449449
450450 // Do delegate
451- let mut delegation = Delegation :: load_from_state ( & state, & delegatee ) . unwrap ( ) ;
452- for delegator in delegators . iter ( ) {
453- delegation. add_quantity ( * delegator , delegation_amount[ delegator ] ) . unwrap ( )
451+ let mut delegation = Delegation :: load_from_state ( & state, & delegator ) . unwrap ( ) ;
452+ for delegatee in delegatees . iter ( ) {
453+ delegation. add_quantity ( * delegatee , delegation_amount[ delegatee ] ) . unwrap ( )
454454 }
455455 delegation. save_to_state ( & mut state) . unwrap ( ) ;
456456
457457 // assert
458- let delegation = Delegation :: load_from_state ( & state, & delegatee ) . unwrap ( ) ;
459- assert_eq ! ( delegation. iter( ) . count( ) , delegators . len( ) ) ;
460- for delegator in delegators . iter ( ) {
461- assert_eq ! ( delegation. get_quantity( delegator ) , delegation_amount[ delegator ] ) ;
458+ let delegation = Delegation :: load_from_state ( & state, & delegator ) . unwrap ( ) ;
459+ assert_eq ! ( delegation. iter( ) . count( ) , delegatees . len( ) ) ;
460+ for delegatee in delegatees . iter ( ) {
461+ assert_eq ! ( delegation. get_quantity( delegatee ) , delegation_amount[ delegatee ] ) ;
462462 }
463463 }
464464}
0 commit comments