Skip to content

Commit 9b571f2

Browse files
committed
Update extract_invoice to also return the reply_path.
- This will be utilised in the following commit for a test.
1 parent 5c27061 commit 9b571f2

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

lightning/src/ln/offers_tests.rs

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -200,12 +200,12 @@ fn extract_invoice_request<'a, 'b, 'c>(
200200
}
201201
}
202202

203-
fn extract_invoice<'a, 'b, 'c>(node: &Node<'a, 'b, 'c>, message: &OnionMessage) -> Bolt12Invoice {
203+
fn extract_invoice<'a, 'b, 'c>(node: &Node<'a, 'b, 'c>, message: &OnionMessage) -> (Bolt12Invoice, BlindedPath) {
204204
match node.onion_messenger.peel_onion_message(message) {
205-
Ok(PeeledOnion::Receive(message, _, _)) => match message {
205+
Ok(PeeledOnion::Receive(message, _, reply_path)) => match message {
206206
ParsedOnionMessageContents::Offers(offers_message) => match offers_message {
207207
OffersMessage::InvoiceRequest(invoice_request) => panic!("Unexpected invoice_request: {:?}", invoice_request),
208-
OffersMessage::Invoice(invoice) => invoice,
208+
OffersMessage::Invoice(invoice) => (invoice, reply_path.unwrap()),
209209
OffersMessage::InvoiceError(error) => panic!("Unexpected invoice_error: {:?}", error),
210210
},
211211
ParsedOnionMessageContents::Custom(message) => panic!("Unexpected custom message: {:?}", message),
@@ -553,7 +553,7 @@ fn creates_and_pays_for_offer_using_two_hop_blinded_path() {
553553
let onion_message = charlie.onion_messenger.next_onion_message_for_peer(david_id).unwrap();
554554
david.onion_messenger.handle_onion_message(&charlie_id, &onion_message);
555555

556-
let invoice = extract_invoice(david, &onion_message);
556+
let (invoice, _) = extract_invoice(david, &onion_message);
557557
assert_eq!(invoice.amount_msats(), 10_000_000);
558558
assert_ne!(invoice.signing_pubkey(), alice_id);
559559
assert!(!invoice.payment_paths().is_empty());
@@ -632,7 +632,7 @@ fn creates_and_pays_for_refund_using_two_hop_blinded_path() {
632632
let onion_message = charlie.onion_messenger.next_onion_message_for_peer(david_id).unwrap();
633633
david.onion_messenger.handle_onion_message(&charlie_id, &onion_message);
634634

635-
let invoice = extract_invoice(david, &onion_message);
635+
let (invoice, _) = extract_invoice(david, &onion_message);
636636
assert_eq!(invoice, expected_invoice);
637637

638638
assert_eq!(invoice.amount_msats(), 10_000_000);
@@ -699,7 +699,7 @@ fn creates_and_pays_for_offer_using_one_hop_blinded_path() {
699699
let onion_message = alice.onion_messenger.next_onion_message_for_peer(bob_id).unwrap();
700700
bob.onion_messenger.handle_onion_message(&alice_id, &onion_message);
701701

702-
let invoice = extract_invoice(bob, &onion_message);
702+
let (invoice, _) = extract_invoice(bob, &onion_message);
703703
assert_eq!(invoice.amount_msats(), 10_000_000);
704704
assert_ne!(invoice.signing_pubkey(), alice_id);
705705
assert!(!invoice.payment_paths().is_empty());
@@ -752,7 +752,7 @@ fn creates_and_pays_for_refund_using_one_hop_blinded_path() {
752752
let onion_message = alice.onion_messenger.next_onion_message_for_peer(bob_id).unwrap();
753753
bob.onion_messenger.handle_onion_message(&alice_id, &onion_message);
754754

755-
let invoice = extract_invoice(bob, &onion_message);
755+
let (invoice, _) = extract_invoice(bob, &onion_message);
756756
assert_eq!(invoice, expected_invoice);
757757

758758
assert_eq!(invoice.amount_msats(), 10_000_000);
@@ -814,7 +814,7 @@ fn pays_for_offer_without_blinded_paths() {
814814
let onion_message = alice.onion_messenger.next_onion_message_for_peer(bob_id).unwrap();
815815
bob.onion_messenger.handle_onion_message(&alice_id, &onion_message);
816816

817-
let invoice = extract_invoice(bob, &onion_message);
817+
let (invoice, _) = extract_invoice(bob, &onion_message);
818818
route_bolt12_payment(bob, &[alice], &invoice);
819819
expect_recent_payment!(bob, RecentPaymentDetails::Pending, payment_id);
820820

@@ -855,7 +855,7 @@ fn pays_for_refund_without_blinded_paths() {
855855
let onion_message = alice.onion_messenger.next_onion_message_for_peer(bob_id).unwrap();
856856
bob.onion_messenger.handle_onion_message(&alice_id, &onion_message);
857857

858-
let invoice = extract_invoice(bob, &onion_message);
858+
let (invoice, _) = extract_invoice(bob, &onion_message);
859859
assert_eq!(invoice, expected_invoice);
860860

861861
route_bolt12_payment(bob, &[alice], &invoice);
@@ -1352,7 +1352,7 @@ fn fails_paying_invoice_more_than_once() {
13521352

13531353
// David pays the first invoice
13541354
let payment_context = PaymentContext::Bolt12Refund(Bolt12RefundContext {});
1355-
let invoice1 = extract_invoice(david, &onion_message);
1355+
let (invoice1, _) = extract_invoice(david, &onion_message);
13561356

13571357
route_bolt12_payment(david, &[charlie, bob, alice], &invoice1);
13581358
expect_recent_payment!(david, RecentPaymentDetails::Pending, payment_id);
@@ -1374,7 +1374,7 @@ fn fails_paying_invoice_more_than_once() {
13741374
let onion_message = charlie.onion_messenger.next_onion_message_for_peer(david_id).unwrap();
13751375
david.onion_messenger.handle_onion_message(&charlie_id, &onion_message);
13761376

1377-
let invoice2 = extract_invoice(david, &onion_message);
1377+
let (invoice2, _) = extract_invoice(david, &onion_message);
13781378
assert_eq!(invoice1.payer_metadata(), invoice2.payer_metadata());
13791379

13801380
// David sends an error instead of paying the second invoice

0 commit comments

Comments
 (0)