test: added test for counter

Author: carlosvdr

tap-agent/src/agent/sender_account.rs

@@ -570,7 +570,6 @@ impl Actor for SenderAccount {
                             );
                             SenderAccount::deny_sender(&state.pgpool, state.sender).await;
                         }
-
                         state.sender_fee_tracker.add(allocation_id, value);
 
                         UNAGGREGATED_FEES
@@ -612,7 +611,6 @@ impl Actor for SenderAccount {
                     state.sender_fee_tracker.get_total_fee_outside_buffer();
                 let total_fee_greater_trigger_value =
                     total_fee_outside_buffer >= state.config.tap.rav_request_trigger_value;
-
                 let rav_result = match (
                     counter_greater_receipt_limit,
                     total_fee_greater_trigger_value,
@@ -931,13 +929,15 @@ pub mod tests {
     const TRIGGER_VALUE: u128 = 500;
     const ESCROW_VALUE: u128 = 1000;
     const BUFFER_MS: u64 = 100;
+    const RECEIPT_LIMIT: u64 = 10000;
 
     async fn create_sender_account(
         pgpool: PgPool,
         initial_allocation: HashSet<Address>,
         rav_request_trigger_value: u128,
         max_unnaggregated_fees_per_sender: u128,
         escrow_subgraph_endpoint: &str,
+        rav_request_receipt_limit: u64,
     ) -> (
         ActorRef<SenderAccountMessage>,
         tokio::task::JoinHandle<()>,
@@ -954,6 +954,7 @@ pub mod tests {
                 rav_request_timestamp_buffer_ms: BUFFER_MS,
                 rav_request_timeout_secs: 5,
                 max_unnaggregated_fees_per_sender,
+                rav_request_receipt_limit,
                 ..Default::default()
             },
             ..Default::default()
@@ -1005,6 +1006,7 @@ pub mod tests {
             TRIGGER_VALUE,
             TRIGGER_VALUE,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1045,6 +1047,7 @@ pub mod tests {
             TRIGGER_VALUE,
             TRIGGER_VALUE,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1228,6 +1231,7 @@ pub mod tests {
             TRIGGER_VALUE,
             TRIGGER_VALUE,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1264,6 +1268,7 @@ pub mod tests {
             TRIGGER_VALUE,
             TRIGGER_VALUE,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1309,6 +1314,60 @@ pub mod tests {
         handle.await.unwrap();
     }
 
+    #[sqlx::test(migrations = "../migrations")]
+    async fn test_counter_greater_limit_trigger_rav(pgpool: PgPool) {
+        let (sender_account, handle, prefix, _) = create_sender_account(
+            pgpool,
+            HashSet::new(),
+            TRIGGER_VALUE,
+            TRIGGER_VALUE,
+            DUMMY_URL,
+            1,
+        )
+        .await;
+
+        let (triggered_rav_request, _, allocation, allocation_handle) =
+            create_mock_sender_allocation(prefix, SENDER.1, *ALLOCATION_ID_0).await;
+
+        // create a fake sender allocation
+        sender_account
+            .cast(SenderAccountMessage::UpdateReceiptFees(
+                *ALLOCATION_ID_0,
+                ReceiptFees::NewReceipt(TRIGGER_VALUE - 1),
+            ))
+            .unwrap();
+
+        tokio::time::sleep(Duration::from_millis(20)).await;
+
+        assert_eq!(
+            triggered_rav_request.load(std::sync::atomic::Ordering::SeqCst),
+            0
+        );
+
+        // wait for it to be outside buffer
+        tokio::time::sleep(Duration::from_millis(BUFFER_MS)).await;
+
+        sender_account
+            .cast(SenderAccountMessage::UpdateReceiptFees(
+                *ALLOCATION_ID_0,
+                ReceiptFees::Retry,
+            ))
+            .unwrap();
+
+        tokio::time::sleep(Duration::from_millis(20)).await;
+
+        assert_eq!(
+            triggered_rav_request.load(std::sync::atomic::Ordering::SeqCst),
+            1
+        );
+
+        allocation.stop_and_wait(None, None).await.unwrap();
+        allocation_handle.await.unwrap();
+
+        sender_account.stop_and_wait(None, None).await.unwrap();
+        handle.await.unwrap();
+    }
+
     #[sqlx::test(migrations = "../migrations")]
     async fn test_remove_sender_account(pgpool: PgPool) {
         let (sender_account, handle, prefix, _) = create_sender_account(
@@ -1317,6 +1376,7 @@ pub mod tests {
             TRIGGER_VALUE,
             TRIGGER_VALUE,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1368,6 +1428,7 @@ pub mod tests {
             TRIGGER_VALUE,
             TRIGGER_VALUE,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1386,6 +1447,7 @@ pub mod tests {
             TRIGGER_VALUE,
             max_unaggregated_fees_per_sender,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1435,6 +1497,7 @@ pub mod tests {
             u128::MAX,
             max_unaggregated_fees_per_sender,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1530,6 +1593,7 @@ pub mod tests {
             TRIGGER_VALUE,
             u128::MAX,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1563,6 +1627,7 @@ pub mod tests {
             trigger_rav_request,
             u128::MAX,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1668,6 +1733,7 @@ pub mod tests {
             TRIGGER_VALUE,
             u128::MAX,
             &mock_server.uri(),
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1721,6 +1787,7 @@ pub mod tests {
             TRIGGER_VALUE,
             u128::MAX,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 
@@ -1764,6 +1831,7 @@ pub mod tests {
             TRIGGER_VALUE,
             max_unaggregated_fees_per_sender,
             DUMMY_URL,
+            RECEIPT_LIMIT,
         )
         .await;
 

tap-agent/src/agent/sender_fee_tracker.rs

@@ -80,6 +80,7 @@ impl SenderFeeTracker {
             expiring_sum.sum += value;
         }
         self.total_fee += value;
+
         let entry = self.id_to_fee.entry(id).or_default();
         entry.fee += value;
         entry.count += 1;