Skip to content

Commit 7815e54

Browse files
authored
Merge pull request #4214 from elnosh/commitment-signed-dance-macro
replace `commitment_signed_dance` macro in test utils
2 parents fea54b4 + 1bbc692 commit 7815e54

21 files changed

+362
-347
lines changed

lightning-dns-resolver/src/lib.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ mod test {
180180
use lightning::types::payment::PaymentHash;
181181
use lightning::util::logger::Logger;
182182

183-
use lightning::{commitment_signed_dance, expect_payment_claimed, get_htlc_update_msgs};
183+
use lightning::{expect_payment_claimed, get_htlc_update_msgs};
184184
use lightning_types::string::UntrustedString;
185185

186186
use std::ops::Deref;
@@ -418,7 +418,7 @@ mod test {
418418
check_added_monitors(&nodes[0], 1);
419419
let updates = get_htlc_update_msgs!(nodes[0], payee_id);
420420
nodes[1].node.handle_update_add_htlc(payer_id, &updates.update_add_htlcs[0]);
421-
commitment_signed_dance!(nodes[1], nodes[0], updates.commitment_signed, false);
421+
do_commitment_signed_dance(&nodes[1], &nodes[0], &updates.commitment_signed, false, false);
422422
expect_and_process_pending_htlcs(&nodes[1], false);
423423

424424
let claimable_events = nodes[1].node.get_and_clear_pending_events();
@@ -452,7 +452,7 @@ mod test {
452452
check_added_monitors(&nodes[1], 1);
453453
let mut updates = get_htlc_update_msgs!(nodes[1], payer_id);
454454
nodes[0].node.handle_update_fulfill_htlc(payee_id, updates.update_fulfill_htlcs.remove(0));
455-
commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, false);
455+
do_commitment_signed_dance(&nodes[0], &nodes[1], &updates.commitment_signed, false, false);
456456

457457
expect_payment_sent(&nodes[0], our_payment_preimage, None, true, true);
458458
}

lightning-tests/src/upgrade_downgrade_tests.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -477,14 +477,15 @@ fn do_test_0_1_htlc_forward_after_splice(fail_htlc: bool) {
477477

478478
let updates = get_htlc_update_msgs(&nodes[1], &node_a_id);
479479
nodes[0].node.handle_update_fail_htlc(node_b_id, &updates.update_fail_htlcs[0]);
480-
commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, false);
480+
do_commitment_signed_dance(&nodes[0], &nodes[1], &updates.commitment_signed, false, false);
481481
let conditions = PaymentFailedConditions::new();
482482
expect_payment_failed_conditions(&nodes[0], pay_hash, false, conditions);
483483
} else {
484484
check_added_monitors(&nodes[1], 1);
485485
let forward_event = SendEvent::from_node(&nodes[1]);
486486
nodes[2].node.handle_update_add_htlc(node_b_id, &forward_event.msgs[0]);
487-
commitment_signed_dance!(nodes[2], nodes[1], forward_event.commitment_msg, false);
487+
let commitment = &forward_event.commitment_msg;
488+
do_commitment_signed_dance(&nodes[2], &nodes[1], commitment, false, false);
488489

489490
expect_and_process_pending_htlcs(&nodes[2], false);
490491
expect_payment_claimable!(nodes[2], pay_hash, pay_secret, 1_000_000);

lightning/src/ln/async_payments_tests.rs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -587,7 +587,8 @@ fn lock_in_htlc_for_static_invoice(
587587
let payment_hash = update_add.payment_hash;
588588
assert!(update_add.hold_htlc.is_some());
589589
sender_lsp.node.handle_update_add_htlc(sender.node.get_our_node_id(), &update_add);
590-
commitment_signed_dance!(sender_lsp, sender, &commitment_update.commitment_signed, false, true);
590+
let commitment = &commitment_update.commitment_signed;
591+
do_commitment_signed_dance(sender_lsp, sender, commitment, false, true);
591592
payment_hash
592593
}
593594

@@ -3240,7 +3241,7 @@ fn async_payment_mpp() {
32403241
let payment_hash = update_add.payment_hash;
32413242
assert!(update_add.hold_htlc.is_some());
32423243
lsp_a.node.handle_update_add_htlc(sender.node.get_our_node_id(), &update_add);
3243-
commitment_signed_dance!(lsp_a, sender, &commitment_update.commitment_signed, false, true);
3244+
do_commitment_signed_dance(lsp_a, sender, &commitment_update.commitment_signed, false, true);
32443245
lsp_a.node.process_pending_htlc_forwards();
32453246

32463247
// HTLC 2
@@ -3252,7 +3253,7 @@ fn async_payment_mpp() {
32523253
let update_add = commitment_update.update_add_htlcs[0].clone();
32533254
assert!(update_add.hold_htlc.is_some());
32543255
lsp_b.node.handle_update_add_htlc(sender.node.get_our_node_id(), &update_add);
3255-
commitment_signed_dance!(lsp_b, sender, &commitment_update.commitment_signed, false, true);
3256+
do_commitment_signed_dance(lsp_b, sender, &commitment_update.commitment_signed, false, true);
32563257
lsp_b.node.process_pending_htlc_forwards();
32573258

32583259
// held htlc <> release_htlc dance

lightning/src/ln/blinded_payment_tests.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -721,7 +721,7 @@ fn do_blinded_intercept_payment(intercept_node_fails: bool) {
721721
SendEvent::from_event(events.remove(0))
722722
};
723723
nodes[1].node.handle_update_add_htlc(nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
724-
commitment_signed_dance!(nodes[1], nodes[0], &payment_event.commitment_msg, false, true);
724+
do_commitment_signed_dance(&nodes[1], &nodes[0], &payment_event.commitment_msg, false, true);
725725
expect_and_process_pending_htlcs(&nodes[1], false);
726726

727727
let events = nodes[1].node.get_and_clear_pending_events();
@@ -761,7 +761,7 @@ fn do_blinded_intercept_payment(intercept_node_fails: bool) {
761761
SendEvent::from_event(events.remove(0))
762762
};
763763
nodes[2].node.handle_update_add_htlc(nodes[1].node.get_our_node_id(), &payment_event.msgs[0]);
764-
commitment_signed_dance!(nodes[2], nodes[1], &payment_event.commitment_msg, false, true);
764+
do_commitment_signed_dance(&nodes[2], &nodes[1], &payment_event.commitment_msg, false, true);
765765
expect_and_process_pending_htlcs(&nodes[2], false);
766766

767767
expect_payment_claimable!(&nodes[2], payment_hash, payment_secret, amt_msat, None, nodes[2].node.get_our_node_id());
@@ -1027,7 +1027,7 @@ fn do_multi_hop_receiver_fail(check: ReceiveCheckFail) {
10271027
check_added_monitors!(nodes[2], 1);
10281028

10291029
nodes[2].node.handle_shutdown(nodes[1].node.get_our_node_id(), &node_1_shutdown);
1030-
commitment_signed_dance!(nodes[2], nodes[1], (), false, true, false, false);
1030+
assert!(commitment_signed_dance_through_cp_raa(&nodes[2], &nodes[1], false, false).is_none());
10311031
expect_and_process_pending_htlcs(&nodes[2], false);
10321032
expect_htlc_handling_failed_destinations!(nodes[2].node.get_and_clear_pending_events(), &[HTLCHandlingFailureType::Receive { payment_hash }]);
10331033
check_added_monitors(&nodes[2], 1);
@@ -1452,7 +1452,7 @@ fn fails_receive_tlvs_authentication() {
14521452
assert!(update_fail.update_fail_htlcs.len() == 1);
14531453
let fail_msg = &update_fail.update_fail_htlcs[0];
14541454
nodes[0].node.handle_update_fail_htlc(nodes[1].node.get_our_node_id(), fail_msg);
1455-
commitment_signed_dance!(nodes[0], nodes[1], update_fail.commitment_signed, false);
1455+
do_commitment_signed_dance(&nodes[0], &nodes[1], &update_fail.commitment_signed, false, false);
14561456
expect_payment_failed_conditions(
14571457
&nodes[0], payment_hash, true,
14581458
PaymentFailedConditions::new().expected_htlc_error_data(LocalHTLCFailureReason::InvalidOnionPayload, &[]),

lightning/src/ln/bolt11_payment_tests.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ fn payment_metadata_end_to_end_for_invoice_with_amount() {
7676
check_added_monitors(&nodes[0], 1);
7777
let send_event = SendEvent::from_node(&nodes[0]);
7878
nodes[1].node.handle_update_add_htlc(nodes[0].node.get_our_node_id(), &send_event.msgs[0]);
79-
commitment_signed_dance!(nodes[1], nodes[0], &send_event.commitment_msg, false);
79+
do_commitment_signed_dance(&nodes[1], &nodes[0], &send_event.commitment_msg, false, false);
8080

8181
expect_and_process_pending_htlcs(&nodes[1], false);
8282

@@ -144,7 +144,7 @@ fn payment_metadata_end_to_end_for_invoice_with_no_amount() {
144144
check_added_monitors(&nodes[0], 1);
145145
let send_event = SendEvent::from_node(&nodes[0]);
146146
nodes[1].node.handle_update_add_htlc(nodes[0].node.get_our_node_id(), &send_event.msgs[0]);
147-
commitment_signed_dance!(nodes[1], nodes[0], &send_event.commitment_msg, false);
147+
do_commitment_signed_dance(&nodes[1], &nodes[0], &send_event.commitment_msg, false, false);
148148

149149
expect_and_process_pending_htlcs(&nodes[1], false);
150150

0 commit comments

Comments
 (0)