Introduce custom data test for message::ReceiveTlvs

Author: shaavan

lightning/src/ln/async_payments_tests.rs

@@ -286,6 +286,7 @@ fn static_invoice_unknown_required_features() {
 		context: Some(MessageContext::Offers(OffersContext::InvoiceRequest {
 			nonce: Nonce([42; 16]),
 		})),
+		custom_data: None,
 	};
 
 	let blinded_paths_to_always_online_node = nodes[1]
@@ -371,6 +372,7 @@ fn ignore_unexpected_static_invoice() {
 		context: Some(MessageContext::Offers(OffersContext::InvoiceRequest {
 			nonce: Nonce([42; 16]),
 		})),
+		custom_data: None,
 	};
 
 	// Initiate payment to the sender's intended offer.
@@ -515,6 +517,7 @@ fn pays_static_invoice() {
 		context: Some(MessageContext::Offers(OffersContext::InvoiceRequest {
 			nonce: Nonce([42; 16]),
 		})),
+		custom_data: None,
 	};
 
 	let blinded_paths_to_always_online_node = nodes[1]

lightning/src/onion_message/functional_tests.rs

@@ -416,6 +416,24 @@ fn one_blinded_hop() {
 	pass_along_path(&nodes);
 }
 
+#[test]
+fn one_blinded_hop_with_custom_data() {
+	let nodes = create_nodes(2);
+	let test_msg = TestCustomMessage::Pong;
+
+	let secp_ctx = Secp256k1::new();
+	let recipient_tlvs = ReceiveTlvs {
+		context: Some(MessageContext::Custom(Vec::new())),
+		custom_data: Some(vec![42; 42]),
+	};
+	let blinded_path = BlindedMessagePath::new(&[], nodes[1].node_id, recipient_tlvs, &*nodes[1].entropy_source, &secp_ctx).unwrap();
+	let destination = Destination::BlindedPath(blinded_path);
+	let instructions = MessageSendInstructions::WithoutReplyPath { destination };
+	nodes[0].messenger.send_onion_message(test_msg, instructions).unwrap();
+	nodes[1].custom_message_handler.expect_message_with_custom_data(TestCustomMessage::Pong, vec![42; 42]);
+	pass_along_path(&nodes);
+}
+
 #[test]
 fn two_unblinded_two_blinded() {
 	let nodes = create_nodes(5);