Skip to content

Commit c81d676

Browse files
committed
Split expect_payment_forwarded into a function called by macro
Also allowing us to pass the event manually.
1 parent 5c1642c commit c81d676

File tree

1 file changed

+30
-21
lines changed

1 file changed

+30
-21
lines changed

lightning/src/ln/functional_test_utils.rs

Lines changed: 30 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -2027,29 +2027,38 @@ macro_rules! expect_payment_path_successful {
20272027
}
20282028
}
20292029

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+
20302055
macro_rules! expect_payment_forwarded {
20312056
($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();
20332058
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);
20532062
}
20542063
}
20552064

@@ -2426,7 +2435,7 @@ pub fn pass_claimed_payment_along_route<'a, 'b, 'c>(origin_node: &Node<'a, 'b, '
24262435
}
24272436
};
24282437
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);
24302439
expected_total_fee_msat += fee as u64;
24312440
check_added_monitors!($node, 1);
24322441
let new_next_msgs = if $new_msgs {

0 commit comments

Comments
 (0)