Skip to content

Commit

Permalink
chore: fix typo
Browse files Browse the repository at this point in the history
Signed-off-by: Gustavo Inacio <[email protected]>
  • Loading branch information
gusinacio committed Oct 1, 2024
1 parent 0599fee commit 758c4dc
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 13 deletions.
2 changes: 1 addition & 1 deletion tap-agent/src/agent/sender_account.rs
Original file line number Diff line number Diff line change
Expand Up @@ -606,7 +606,7 @@ impl Actor for SenderAccount {
state.add_to_denylist().await;
}

if state.sender_fee_tracker.get_total_fee_outsite_buffer()
if state.sender_fee_tracker.get_total_fee_outside_buffer()
>= state.config.tap.rav_request_trigger_value
{
tracing::debug!(
Expand Down
24 changes: 12 additions & 12 deletions tap-agent/src/agent/sender_fee_tracker.rs
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,7 @@ impl SenderFeeTracker {
self.total_fee
}

pub fn get_total_fee_outsite_buffer(&mut self) -> u128 {
pub fn get_total_fee_outside_buffer(&mut self) -> u128 {
self.total_fee - self.get_buffer_fee().min(self.total_fee)
}

Expand Down Expand Up @@ -222,70 +222,70 @@ mod tests {
const BUFFER_WINDOW: Duration = Duration::from_millis(20);
let mut tracker = SenderFeeTracker::new(BUFFER_WINDOW);
assert_eq!(tracker.get_heaviest_allocation_id(), None);
assert_eq!(tracker.get_total_fee_outsite_buffer(), 0);
assert_eq!(tracker.get_total_fee_outside_buffer(), 0);
assert_eq!(tracker.get_total_fee(), 0);

tracker.add(allocation_id_0, 10);
assert_eq!(tracker.get_heaviest_allocation_id(), None);
assert_eq!(tracker.get_total_fee_outsite_buffer(), 0);
assert_eq!(tracker.get_total_fee_outside_buffer(), 0);
assert_eq!(tracker.get_total_fee(), 10);

sleep(BUFFER_WINDOW);

assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_0));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 10);
assert_eq!(tracker.get_total_fee_outside_buffer(), 10);
assert_eq!(tracker.get_total_fee(), 10);

tracker.add(allocation_id_2, 20);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_0));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 10);
assert_eq!(tracker.get_total_fee_outside_buffer(), 10);
assert_eq!(tracker.get_total_fee(), 30);

sleep(BUFFER_WINDOW);

tracker.block_allocation_id(allocation_id_2);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_0));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 30);
assert_eq!(tracker.get_total_fee_outside_buffer(), 30);
assert_eq!(tracker.get_total_fee(), 30);

tracker.unblock_allocation_id(allocation_id_2);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_2));

tracker.add(allocation_id_1, 30);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_2));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 30);
assert_eq!(tracker.get_total_fee_outside_buffer(), 30);
assert_eq!(tracker.get_total_fee(), 60);

sleep(BUFFER_WINDOW);

assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_1));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 60);
assert_eq!(tracker.get_total_fee_outside_buffer(), 60);
assert_eq!(tracker.get_total_fee(), 60);

tracker.add(allocation_id_2, 20);
tracker.update(allocation_id_2, 0);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_1));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 20);
assert_eq!(tracker.get_total_fee_outside_buffer(), 20);
assert_eq!(tracker.get_total_fee(), 40);

sleep(BUFFER_WINDOW);

tracker.add(allocation_id_2, 100);
tracker.update(allocation_id_2, 0);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_1));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 0);
assert_eq!(tracker.get_total_fee_outside_buffer(), 0);
assert_eq!(tracker.get_total_fee(), 40);

sleep(BUFFER_WINDOW);

tracker.update(allocation_id_1, 0);
assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_0));
assert_eq!(tracker.get_total_fee_outsite_buffer(), 10);
assert_eq!(tracker.get_total_fee_outside_buffer(), 10);
assert_eq!(tracker.get_total_fee(), 10);

tracker.update(allocation_id_0, 0);
assert_eq!(tracker.get_heaviest_allocation_id(), None);
assert_eq!(tracker.get_total_fee_outsite_buffer(), 0);
assert_eq!(tracker.get_total_fee_outside_buffer(), 0);
assert_eq!(tracker.get_total_fee(), 0);
}
}

0 comments on commit 758c4dc

Please sign in to comment.