@@ -2027,29 +2027,38 @@ macro_rules! expect_payment_path_successful {
2027
2027
}
2028
2028
}
2029
2029
2030
+ pub fn expect_payment_forwarded < CM : AChannelManager , H : NodeHolder < CM =CM > > (
2031
+ event : Event , node : & H , prev_node : & H , next_node : & H , expected_fee : Option < u64 > ,
2032
+ upstream_force_closed : bool , downstream_force_closed : bool
2033
+ ) {
2034
+ match event {
2035
+ Event :: PaymentForwarded {
2036
+ fee_earned_msat, prev_channel_id, claim_from_onchain_tx, next_channel_id,
2037
+ outbound_amount_forwarded_msat : _
2038
+ } => {
2039
+ assert_eq ! ( fee_earned_msat, expected_fee) ;
2040
+ if !upstream_force_closed {
2041
+ // Is the event prev_channel_id in one of the channels between the two nodes?
2042
+ assert ! ( node. node( ) . list_channels( ) . iter( ) . any( |x| x. counterparty. node_id == prev_node. node( ) . get_our_node_id( ) && x. channel_id == prev_channel_id. unwrap( ) ) ) ;
2043
+ }
2044
+ // We check for force closures since a force closed channel is removed from the
2045
+ // node's channel list
2046
+ if !downstream_force_closed {
2047
+ assert ! ( node. node( ) . list_channels( ) . iter( ) . any( |x| x. counterparty. node_id == next_node. node( ) . get_our_node_id( ) && x. channel_id == next_channel_id. unwrap( ) ) ) ;
2048
+ }
2049
+ assert_eq ! ( claim_from_onchain_tx, downstream_force_closed) ;
2050
+ } ,
2051
+ _ => panic ! ( "Unexpected event" ) ,
2052
+ }
2053
+ }
2054
+
2030
2055
macro_rules! expect_payment_forwarded {
2031
2056
( $node: expr, $prev_node: expr, $next_node: expr, $expected_fee: expr, $upstream_force_closed: expr, $downstream_force_closed: expr) => {
2032
- let events = $node. node. get_and_clear_pending_events( ) ;
2057
+ let mut events = $node. node. get_and_clear_pending_events( ) ;
2033
2058
assert_eq!( events. len( ) , 1 ) ;
2034
- match events[ 0 ] {
2035
- Event :: PaymentForwarded {
2036
- fee_earned_msat, prev_channel_id, claim_from_onchain_tx, next_channel_id,
2037
- outbound_amount_forwarded_msat: _
2038
- } => {
2039
- assert_eq!( fee_earned_msat, $expected_fee) ;
2040
- if !$upstream_force_closed {
2041
- // Is the event prev_channel_id in one of the channels between the two nodes?
2042
- assert!( $node. node. list_channels( ) . iter( ) . any( |x| x. counterparty. node_id == $prev_node. node. get_our_node_id( ) && x. channel_id == prev_channel_id. unwrap( ) ) ) ;
2043
- }
2044
- // We check for force closures since a force closed channel is removed from the
2045
- // node's channel list
2046
- if !$downstream_force_closed {
2047
- assert!( $node. node. list_channels( ) . iter( ) . any( |x| x. counterparty. node_id == $next_node. node. get_our_node_id( ) && x. channel_id == next_channel_id. unwrap( ) ) ) ;
2048
- }
2049
- assert_eq!( claim_from_onchain_tx, $downstream_force_closed) ;
2050
- } ,
2051
- _ => panic!( "Unexpected event" ) ,
2052
- }
2059
+ $crate:: ln:: functional_test_utils:: expect_payment_forwarded(
2060
+ events. pop( ) . unwrap( ) , & $node, & $prev_node, & $next_node, $expected_fee,
2061
+ $upstream_force_closed, $downstream_force_closed) ;
2053
2062
}
2054
2063
}
2055
2064
@@ -2426,7 +2435,7 @@ pub fn pass_claimed_payment_along_route<'a, 'b, 'c>(origin_node: &Node<'a, 'b, '
2426
2435
}
2427
2436
} ;
2428
2437
if $idx == 1 { fee += expected_extra_fees[ i] ; }
2429
- expect_payment_forwarded!( $node, $next_node, $prev_node, Some ( fee as u64 ) , false , false ) ;
2438
+ expect_payment_forwarded!( * $node, $next_node, $prev_node, Some ( fee as u64 ) , false , false ) ;
2430
2439
expected_total_fee_msat += fee as u64 ;
2431
2440
check_added_monitors!( $node, 1 ) ;
2432
2441
let new_next_msgs = if $new_msgs {
0 commit comments