Skip to content

Commit 2acbdb9

Browse files
committed
test: Add result form to TriggerRavRequest test
1 parent d5da7cc commit 2acbdb9

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

tap-agent/src/agent/sender_account.rs

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1099,18 +1099,20 @@ pub mod tests {
10991099
}
11001100

11011101
impl MockSenderAllocation {
1102-
pub fn new_with_triggered_rav_request() -> (Self, Arc<AtomicU32>, Arc<Mutex<u128>>) {
1102+
pub fn new_with_triggered_rav_request(
1103+
) -> (Self, Result<(Arc<AtomicU32>, Arc<Mutex<u128>>), ()>) {
11031104
let triggered_rav_request = Arc::new(AtomicU32::new(0));
11041105
let unaggregated_fees = Arc::new(Mutex::new(0));
1106+
let mock_rav_request_response =
1107+
(triggered_rav_request.clone(), unaggregated_fees.clone());
11051108
(
11061109
Self {
1107-
triggered_rav_request: triggered_rav_request.clone(),
1110+
triggered_rav_request: triggered_rav_request,
11081111
receipts: Arc::new(Mutex::new(Vec::new())),
11091112
next_rav_value: Arc::new(Mutex::new(0)),
1110-
next_unaggregated_fees_value: unaggregated_fees.clone(),
1113+
next_unaggregated_fees_value: unaggregated_fees,
11111114
},
1112-
triggered_rav_request,
1113-
unaggregated_fees,
1115+
Ok(mock_rav_request_response),
11141116
)
11151117
}
11161118

@@ -1211,8 +1213,9 @@ pub mod tests {
12111213
ActorRef<SenderAllocationMessage>,
12121214
JoinHandle<()>,
12131215
) {
1214-
let (mock_sender_allocation, triggered_rav_request, next_unaggregated_fees) =
1216+
let (mock_sender_allocation, rav_request_result) =
12151217
MockSenderAllocation::new_with_triggered_rav_request();
1218+
let (triggered_rav_request, next_unaggregated_fees) = rav_request_result.unwrap();
12161219

12171220
let name = format!("{}:{}:{}", prefix, sender, allocation);
12181221
let (sender_account, join_handle) =

0 commit comments

Comments
 (0)