refactor: send just the value

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

Author: gusinacio

tap-agent/src/agent/sender_account.rs

@@ -23,7 +23,6 @@ use tap_core::rav::SignedRAV;
 use thegraph_core::Address;
 use tracing::{error, Level};
 
-use super::sender_accounts_manager::NewReceiptNotification;
 use super::sender_allocation::{SenderAllocation, SenderAllocationArgs};
 use crate::agent::sender_allocation::SenderAllocationMessage;
 use crate::agent::sender_fee_tracker::SenderFeeTracker;
@@ -80,7 +79,7 @@ type Balance = U256;
 
 #[derive(Debug, Eq, PartialEq)]
 pub enum ReceiptFees {
-    NewReceipt(NewReceiptNotification),
+    NewReceipt(u128),
     UpdateValue(UnaggregatedReceipts),
     Retry,
 }
@@ -569,15 +568,15 @@ impl Actor for SenderAccount {
                 }
 
                 match receipt_fees {
-                    ReceiptFees::NewReceipt(receipt) => {
-                        state.sender_fee_tracker.add(allocation_id, receipt.value);
+                    ReceiptFees::NewReceipt(value) => {
+                        state.sender_fee_tracker.add(allocation_id, value);
 
                         UNAGGREGATED_FEES
                             .with_label_values(&[
                                 &state.sender.to_string(),
                                 &allocation_id.to_string(),
                             ])
-                            .add(receipt.value as f64);
+                            .add(value as f64);
                     }
                     ReceiptFees::UpdateValue(unaggregated_fees) => {
                         state

tap-agent/src/agent/sender_allocation.rs

@@ -249,7 +249,7 @@ impl Actor for SenderAllocation {
                         .sender_account_ref
                         .cast(SenderAccountMessage::UpdateReceiptFees(
                             state.allocation_id,
-                            ReceiptFees::NewReceipt(notification),
+                            ReceiptFees::NewReceipt(fees),
                         ))?;
                 }
             }
@@ -1074,10 +1074,7 @@ pub mod tests {
         // should emit update aggregate fees message to sender account
         let expected_message = SenderAccountMessage::UpdateReceiptFees(
             *ALLOCATION_ID_0,
-            ReceiptFees::UpdateValue(UnaggregatedReceipts {
-                last_id: 1,
-                value: 20,
-            }),
+            ReceiptFees::NewReceipt(20u128),
         );
         let last_message_emitted = last_message_emitted.lock().unwrap();
         assert_eq!(last_message_emitted.len(), 2);