Introduce custom data test for message::ReceiveTlvs

Author: shaavan

lightning/src/ln/async_payments_tests.rs

@@ -305,6 +305,13 @@ fn ignore_unexpected_static_invoice() {
 	create_announced_chan_between_nodes_with_value(&nodes, 0, 1, 1_000_000, 0);
 	create_unannounced_chan_between_nodes_with_value(&nodes, 1, 2, 1_000_000, 0);
 
+	let recipient_tlvs = ReceiveTlvs {
+		context: Some(MessageContext::Offers(OffersContext::InvoiceRequest {
+			nonce: Nonce([42; 16]),
+		})),
+		custom_data: None,
+	};
+
 	// Initiate payment to the sender's intended offer.
 	let (offer, valid_static_invoice) =
 		create_static_invoice(&nodes[1], &nodes[2], None, &secp_ctx);

lightning/src/onion_message/functional_tests.rs

@@ -437,6 +437,33 @@ 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);