Skip to content

Commit 704165a

Browse files
TheBlueMattTheBlueMatt
committed
Make expect_payment_failed_conditions a function
This reduces macro generated code in tests a good bit, and moves us one step further away from using macros everywhere when we don't need to.
1 parent 8e5cf75 commit 704165a

File tree

5 files changed

+71
-64
lines changed

5 files changed

+71
-64
lines changed

‎lightning/src/ln/functional_test_utils.rs

Lines changed: 55 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ use chain::channelmonitor::ChannelMonitor;
1515
use chain::transaction::OutPoint;
1616
use ln::{PaymentPreimage, PaymentHash, PaymentSecret};
1717
use ln::channelmanager::{ChainParameters, ChannelManager, ChannelManagerReadArgs, RAACommitmentOrder, PaymentSendFailure, PaymentId, MIN_CLTV_EXPIRY_DELTA};
18-
use routing::gossip::{P2PGossipSync, NetworkGraph};
18+
use routing::gossip::{NetGraphMsgHandler, NetworkGraph, NetworkUpdate};
1919
use routing::router::{PaymentParameters, Route, get_route};
2020
use ln::features::{InitFeatures, InvoiceFeatures};
2121
use ln::msgs;
@@ -1467,7 +1467,7 @@ impl<'a> PaymentFailedConditions<'a> {
14671467
#[cfg(test)]
14681468
macro_rules! expect_payment_failed_with_update {
14691469
($node: expr, $expected_payment_hash: expr, $rejected_by_dest: expr, $scid: expr, $chan_closed: expr) => {
1470-
expect_payment_failed_conditions!($node, $expected_payment_hash, $rejected_by_dest,
1470+
$crate::ln::functional_test_utils::expect_payment_failed_conditions(&$node, $expected_payment_hash, $rejected_by_dest,
14711471
$crate::ln::functional_test_utils::PaymentFailedConditions::new().blamed_scid($scid).blamed_chan_closed($chan_closed));
14721472
}
14731473
}
@@ -1480,64 +1480,71 @@ macro_rules! expect_payment_failed {
14801480
$(
14811481
conditions = conditions.expected_htlc_error_data($expected_error_code, &$expected_error_data);
14821482
)*
1483-
expect_payment_failed_conditions!($node, $expected_payment_hash, $rejected_by_dest, conditions);
1483+
$crate::ln::functional_test_utils::expect_payment_failed_conditions(&$node, $expected_payment_hash, $rejected_by_dest, conditions);
14841484
};
14851485
}
14861486

1487-
#[cfg(test)]
1488-
macro_rules! expect_payment_failed_conditions {
1489-
($node: expr, $expected_payment_hash: expr, $rejected_by_dest: expr, $conditions: expr) => {
1490-
let events = $node.node.get_and_clear_pending_events();
1491-
assert_eq!(events.len(), 1);
1492-
let expected_payment_id = match events[0] {
1493-
Event::PaymentPathFailed { ref payment_hash, rejected_by_dest, ref error_code, ref error_data, ref path, ref retry, ref payment_id, ref network_update, .. } => {
1494-
assert_eq!(*payment_hash, $expected_payment_hash, "unexpected payment_hash");
1495-
assert_eq!(rejected_by_dest, $rejected_by_dest, "unexpected rejected_by_dest value");
1496-
assert!(retry.is_some(), "expected retry.is_some()");
1497-
assert_eq!(retry.as_ref().unwrap().final_value_msat, path.last().unwrap().fee_msat, "Retry amount should match last hop in path");
1498-
assert_eq!(retry.as_ref().unwrap().payment_params.payee_pubkey, path.last().unwrap().pubkey, "Retry payee node_id should match last hop in path");
1499-
1487+
pub fn expect_payment_failed_conditions<'a, 'b, 'c, 'd, 'e>(
1488+
node: &'a Node<'b, 'c, 'd>, expected_payment_hash: PaymentHash, rejected_by_dest: bool,
1489+
conditions: PaymentFailedConditions<'e>
1490+
) {
1491+
let mut events = node.node.get_and_clear_pending_events();
1492+
assert_eq!(events.len(), 1);
1493+
let expected_payment_id = match events.pop().unwrap() {
1494+
Event::PaymentPathFailed { payment_hash, rejected_by_dest:lulz, path, retry, payment_id, network_update,
1495+
#[cfg(test)]
1496+
error_code,
1497+
#[cfg(test)]
1498+
error_data, .. } => {
1499+
assert_eq!(payment_hash, expected_payment_hash, "unexpected payment_hash");
1500+
assert_eq!(rejected_by_dest, lulz, "unexpected rejected_by_dest value");
1501+
assert!(retry.is_some(), "expected retry.is_some()");
1502+
assert_eq!(retry.as_ref().unwrap().final_value_msat, path.last().unwrap().fee_msat, "Retry amount should match last hop in path");
1503+
assert_eq!(retry.as_ref().unwrap().payment_params.payee_pubkey, path.last().unwrap().pubkey, "Retry payee node_id should match last hop in path");
1504+
1505+
#[cfg(test)]
1506+
{
15001507
assert!(error_code.is_some(), "expected error_code.is_some() = true");
15011508
assert!(error_data.is_some(), "expected error_data.is_some() = true");
1502-
if let Some((code, data)) = $conditions.expected_htlc_error_data {
1509+
if let Some((code, data)) = conditions.expected_htlc_error_data {
15031510
assert_eq!(error_code.unwrap(), code, "unexpected error code");
15041511
assert_eq!(&error_data.as_ref().unwrap()[..], data, "unexpected error data");
15051512
}
1513+
}
15061514

1507-
if let Some(chan_closed) = $conditions.expected_blamed_chan_closed {
1508-
match network_update {
1509-
&Some($crate::routing::gossip::NetworkUpdate::ChannelUpdateMessage { ref msg }) if !chan_closed => {
1510-
if let Some(scid) = $conditions.expected_blamed_scid {
1511-
assert_eq!(msg.contents.short_channel_id, scid);
1512-
}
1513-
assert_eq!(msg.contents.flags & 2, 0);
1514-
},
1515-
&Some($crate::routing::gossip::NetworkUpdate::ChannelFailure { short_channel_id, is_permanent }) if chan_closed => {
1516-
if let Some(scid) = $conditions.expected_blamed_scid {
1517-
assert_eq!(short_channel_id, scid);
1518-
}
1519-
assert!(is_permanent);
1520-
},
1521-
Some(_) => panic!("Unexpected update type"),
1522-
None => panic!("Expected update"),
1523-
}
1515+
if let Some(chan_closed) = conditions.expected_blamed_chan_closed {
1516+
match network_update {
1517+
Some(NetworkUpdate::ChannelUpdateMessage { ref msg }) if !chan_closed => {
1518+
if let Some(scid) = conditions.expected_blamed_scid {
1519+
assert_eq!(msg.contents.short_channel_id, scid);
1520+
}
1521+
assert_eq!(msg.contents.flags & 2, 0);
1522+
},
1523+
Some(NetworkUpdate::ChannelClosed { short_channel_id, is_permanent }) if chan_closed => {
1524+
if let Some(scid) = conditions.expected_blamed_scid {
1525+
assert_eq!(short_channel_id, scid);
1526+
}
1527+
assert!(is_permanent);
1528+
},
1529+
Some(_) => panic!("Unexpected update type"),
1530+
None => panic!("Expected update"),
15241531
}
1532+
}
15251533

1526-
payment_id.unwrap()
1527-
},
1528-
_ => panic!("Unexpected event"),
1529-
};
1530-
if !$conditions.expected_mpp_parts_remain {
1531-
$node.node.abandon_payment(expected_payment_id);
1532-
let events = $node.node.get_and_clear_pending_events();
1533-
assert_eq!(events.len(), 1);
1534-
match events[0] {
1535-
Event::PaymentFailed { ref payment_hash, ref payment_id } => {
1536-
assert_eq!(*payment_hash, $expected_payment_hash, "unexpected second payment_hash");
1537-
assert_eq!(*payment_id, expected_payment_id);
1538-
}
1539-
_ => panic!("Unexpected second event"),
1534+
payment_id.unwrap()
1535+
},
1536+
_ => panic!("Unexpected event"),
1537+
};
1538+
if !conditions.expected_mpp_parts_remain {
1539+
node.node.abandon_payment(expected_payment_id);
1540+
let events = node.node.get_and_clear_pending_events();
1541+
assert_eq!(events.len(), 1);
1542+
match events[0] {
1543+
Event::PaymentFailed { ref payment_hash, ref payment_id } => {
1544+
assert_eq!(*payment_hash, expected_payment_hash, "unexpected second payment_hash");
1545+
assert_eq!(*payment_id, expected_payment_id);
15401546
}
1547+
_ => panic!("Unexpected second event"),
15411548
}
15421549
}
15431550
}

‎lightning/src/ln/functional_tests.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9725,7 +9725,7 @@ fn do_test_dup_htlc_second_rejected(test_for_second_fail_panic: bool) {
97259725
nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &fail_updates_1.update_fail_htlcs[0]);
97269726
commitment_signed_dance!(nodes[0], nodes[1], fail_updates_1.commitment_signed, false);
97279727

9728-
expect_payment_failed_conditions!(nodes[0], our_payment_hash, true, PaymentFailedConditions::new().mpp_parts_remain());
9728+
expect_payment_failed_conditions(&nodes[0], our_payment_hash, true, PaymentFailedConditions::new().mpp_parts_remain());
97299729

97309730
claim_payment(&nodes[0], &[&nodes[1]], our_payment_preimage);
97319731
}
@@ -9830,7 +9830,7 @@ fn test_inconsistent_mpp_params() {
98309830
nodes[0].node.handle_update_fail_htlc(&nodes[2].node.get_our_node_id(), &fail_updates_2.update_fail_htlcs[0]);
98319831
commitment_signed_dance!(nodes[0], nodes[2], fail_updates_2.commitment_signed, false);
98329832

9833-
expect_payment_failed_conditions!(nodes[0], our_payment_hash, true, PaymentFailedConditions::new().mpp_parts_remain());
9833+
expect_payment_failed_conditions(&nodes[0], our_payment_hash, true, PaymentFailedConditions::new().mpp_parts_remain());
98349834

98359835
nodes[0].node.send_payment_along_path(&route.paths[1], &payment_params_opt, &our_payment_hash, &Some(our_payment_secret), 15_000_000, cur_height, payment_id, &None).unwrap();
98369836
check_added_monitors!(nodes[0], 1);

‎lightning/src/ln/onion_route_tests.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -854,7 +854,7 @@ fn test_phantom_onion_hmac_failure() {
854854
.blamed_scid(phantom_scid)
855855
.blamed_chan_closed(true)
856856
.expected_htlc_error_data(0x8000 | 0x4000 | 5, &sha256_of_onion);
857-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, fail_conditions);
857+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, fail_conditions);
858858
}
859859

860860
#[test]
@@ -927,7 +927,7 @@ fn test_phantom_invalid_onion_payload() {
927927
.blamed_scid(phantom_scid)
928928
.blamed_chan_closed(true)
929929
.expected_htlc_error_data(0x4000 | 22, &error_data);
930-
expect_payment_failed_conditions!(nodes[0], payment_hash, true, fail_conditions);
930+
expect_payment_failed_conditions(&nodes[0], payment_hash, true, fail_conditions);
931931
}
932932

933933
#[test]
@@ -983,7 +983,7 @@ fn test_phantom_final_incorrect_cltv_expiry() {
983983
let mut fail_conditions = PaymentFailedConditions::new()
984984
.blamed_scid(phantom_scid)
985985
.expected_htlc_error_data(18, &error_data);
986-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, fail_conditions);
986+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, fail_conditions);
987987
}
988988

989989
#[test]
@@ -1028,7 +1028,7 @@ fn test_phantom_failure_too_low_cltv() {
10281028
let mut fail_conditions = PaymentFailedConditions::new()
10291029
.blamed_scid(phantom_scid)
10301030
.expected_htlc_error_data(17, &error_data);
1031-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, fail_conditions);
1031+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, fail_conditions);
10321032
}
10331033

10341034
#[test]
@@ -1076,7 +1076,7 @@ fn test_phantom_failure_too_low_recv_amt() {
10761076
let mut fail_conditions = PaymentFailedConditions::new()
10771077
.blamed_scid(phantom_scid)
10781078
.expected_htlc_error_data(0x4000 | 15, &error_data);
1079-
expect_payment_failed_conditions!(nodes[0], payment_hash, true, fail_conditions);
1079+
expect_payment_failed_conditions(&nodes[0], payment_hash, true, fail_conditions);
10801080
}
10811081

10821082
#[test]
@@ -1123,7 +1123,7 @@ fn test_phantom_dust_exposure_failure() {
11231123
.blamed_scid(channel.0.contents.short_channel_id)
11241124
.blamed_chan_closed(false)
11251125
.expected_htlc_error_data(0x1000 | 7, &err_data);
1126-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, fail_conditions);
1126+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, fail_conditions);
11271127
}
11281128

11291129
#[test]
@@ -1174,5 +1174,5 @@ fn test_phantom_failure_reject_payment() {
11741174
let mut fail_conditions = PaymentFailedConditions::new()
11751175
.blamed_scid(phantom_scid)
11761176
.expected_htlc_error_data(0x4000 | 15, &error_data);
1177-
expect_payment_failed_conditions!(nodes[0], payment_hash, true, fail_conditions);
1177+
expect_payment_failed_conditions(&nodes[0], payment_hash, true, fail_conditions);
11781178
}

‎lightning/src/ln/payment_tests.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ fn retry_single_path_payment() {
7070
check_added_monitors!(nodes[1], 1);
7171
nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &htlc_updates.update_fail_htlcs[0]);
7272
commitment_signed_dance!(nodes[0], nodes[1], htlc_updates.commitment_signed, false);
73-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain());
73+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain());
7474

7575
// Rebalance the channel so the retry succeeds.
7676
send_payment(&nodes[2], &vec!(&nodes[1])[..], 3_000_000);
@@ -173,7 +173,7 @@ fn mpp_retry() {
173173
check_added_monitors!(nodes[2], 1);
174174
nodes[0].node.handle_update_fail_htlc(&nodes[2].node.get_our_node_id(), &htlc_updates.update_fail_htlcs[0]);
175175
commitment_signed_dance!(nodes[0], nodes[2], htlc_updates.commitment_signed, false);
176-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain());
176+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain());
177177

178178
// Rebalance the channel so the second half of the payment can succeed.
179179
send_payment(&nodes[3], &vec!(&nodes[2])[..], 1_500_000);
@@ -251,7 +251,7 @@ fn do_mpp_receive_timeout(send_partial_mpp: bool) {
251251
check_added_monitors!(nodes[1], 1);
252252
commitment_signed_dance!(nodes[0], nodes[1], htlc_fail_updates_1_0.commitment_signed, false);
253253

254-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain().expected_htlc_error_data(23, &[][..]));
254+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain().expected_htlc_error_data(23, &[][..]));
255255
} else {
256256
// Pass half of the payment along the second path.
257257
pass_along_path(&nodes[0], &[&nodes[2], &nodes[3]], 200_000, payment_hash, Some(payment_secret), events.remove(0), true, None);
@@ -521,7 +521,7 @@ fn do_retry_with_no_persist(confirm_before_reload: bool) {
521521
confirm_transaction(&nodes[0], &first_htlc_timeout_tx);
522522
}
523523
nodes[0].tx_broadcaster.txn_broadcasted.lock().unwrap().clear();
524-
expect_payment_failed_conditions!(nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain());
524+
expect_payment_failed_conditions(&nodes[0], payment_hash, false, PaymentFailedConditions::new().mpp_parts_remain());
525525

526526
// Finally, retry the payment (which was reloaded from the ChannelMonitor when nodes[0] was
527527
// reloaded) via a route over the new channel, which work without issue and eventually be

‎lightning/src/ln/priv_short_conf_tests.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,7 @@ fn test_inbound_scid_privacy() {
460460
nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &updates.update_fail_htlcs[0]);
461461
commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, false);
462462

463-
expect_payment_failed_conditions!(nodes[0], payment_hash_2, false,
463+
expect_payment_failed_conditions(&nodes[0], payment_hash_2, false,
464464
PaymentFailedConditions::new().blamed_scid(last_hop[0].short_channel_id.unwrap())
465465
.blamed_chan_closed(true).expected_htlc_error_data(0x4000|10, &[0; 0]));
466466
}
@@ -537,7 +537,7 @@ fn test_scid_alias_returned() {
537537
err_data.extend_from_slice(&ChannelUpdate::TYPE.to_be_bytes());
538538
err_data.extend_from_slice(&msg.encode());
539539

540-
expect_payment_failed_conditions!(nodes[0], payment_hash, false,
540+
expect_payment_failed_conditions(&nodes[0], payment_hash, false,
541541
PaymentFailedConditions::new().blamed_scid(last_hop[0].inbound_scid_alias.unwrap())
542542
.blamed_chan_closed(false).expected_htlc_error_data(0x1000|7, &err_data));
543543

@@ -560,7 +560,7 @@ fn test_scid_alias_returned() {
560560
err_data.extend_from_slice(&(msg.serialized_length() as u16 + 2).to_be_bytes());
561561
err_data.extend_from_slice(&ChannelUpdate::TYPE.to_be_bytes());
562562
err_data.extend_from_slice(&msg.encode());
563-
expect_payment_failed_conditions!(nodes[0], payment_hash, false,
563+
expect_payment_failed_conditions(&nodes[0], payment_hash, false,
564564
PaymentFailedConditions::new().blamed_scid(last_hop[0].inbound_scid_alias.unwrap())
565565
.blamed_chan_closed(false).expected_htlc_error_data(0x1000|12, &err_data));
566566
}

0 commit comments

Comments
 (0)