test: add tracker test for ongoing ravs

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

tap-agent/src/agent/sender_fee_tracker.rs

@@ -161,11 +161,11 @@ impl SenderFeeTracker {
     }
 
     pub fn get_total_fee(&self) -> u128 {
-        self.total_fee
+        self.total_fee - self.fees_requesting
     }
 
     pub fn get_total_fee_outside_buffer(&mut self) -> u128 {
-        self.total_fee - self.get_buffer_fee().min(self.total_fee)
+        self.get_total_fee() - self.get_buffer_fee().min(self.total_fee)
     }
 
     pub fn get_buffer_fee(&mut self) -> u128 {
@@ -371,4 +371,35 @@ mod tests {
         assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_1));
         assert_eq!(tracker.get_total_fee(), 30);
     }
+
+    #[test]
+    fn test_ongoing_rav_requests() {
+        let allocation_id_0 = address!("abababababababababababababababababababab");
+        let allocation_id_1 = address!("bcbcbcbcbcbcbcbcbcbcbcbcbcbcbcbcbcbcbcbc");
+        let allocation_id_2 = address!("cdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcd");
+
+        let mut tracker = SenderFeeTracker::default();
+        assert_eq!(tracker.get_heaviest_allocation_id(), None);
+        assert_eq!(tracker.get_total_fee_outside_buffer(), 0);
+        assert_eq!(tracker.get_total_fee(), 0);
+
+        tracker.add(allocation_id_0, 10);
+        tracker.add(allocation_id_1, 20);
+        tracker.add(allocation_id_2, 30);
+        assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_2));
+        assert_eq!(tracker.get_total_fee(), 60);
+        assert_eq!(tracker.get_total_fee_outside_buffer(), 60);
+
+        tracker.start_rav_request(allocation_id_2);
+
+        assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_1));
+        assert_eq!(tracker.get_total_fee(), 30);
+        assert_eq!(tracker.get_total_fee_outside_buffer(), 30);
+
+        tracker.finish_rav_request(allocation_id_2);
+
+        assert_eq!(tracker.get_heaviest_allocation_id(), Some(allocation_id_2));
+        assert_eq!(tracker.get_total_fee(), 60);
+        assert_eq!(tracker.get_total_fee_outside_buffer(), 60);
+    }
 }