@@ -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