Skip to content

Commit a68b325

Browse files
foriequal0mergify[bot]
authored andcommitted
Fix reversed delegator-delegatee relationship
1 parent 9651de1 commit a68b325

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

core/src/consensus/tendermint/stake/action_data.rs

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)