X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;ds=sidebyside;f=lightning%2Fsrc%2Fonion_message%2Ffunctional_tests.rs;fp=lightning%2Fsrc%2Fonion_message%2Ffunctional_tests.rs;h=695064e467c49dc935728a117472ed1ac1a1e880;hb=39397d4e14ed87e62cf8f313de749b7b5785561e;hp=560028c5bbaaf86228b2ded9a2d4e7bec70686b1;hpb=6500c99f29d626c78c7848586a0018453f64fe45;p=rust-lightning

diff --git a/lightning/src/onion_message/functional_tests.rs b/lightning/src/onion_message/functional_tests.rs
index 560028c5..695064e4 100644
--- a/lightning/src/onion_message/functional_tests.rs
+++ b/lightning/src/onion_message/functional_tests.rs
@@ -119,3 +119,24 @@ fn too_big_packet_error() {
 	let err = nodes[0].messenger.send_onion_message(&hops, Destination::Node(hop_node_id)).unwrap_err();
 	assert_eq!(err, SendError::TooBigPacket);
 }
+
+#[test]
+fn invalid_blinded_route_error() {
+	// Make sure we error as expected if a provided blinded route has 0 or 1 hops.
+	let mut nodes = create_nodes(3);
+	let (node1, node2, node3) = (nodes.remove(0), nodes.remove(0), nodes.remove(0));
+
+	// 0 hops
+	let secp_ctx = Secp256k1::new();
+	let mut blinded_route = BlindedRoute::new::<EnforcingSigner, _, _>(&[node2.get_node_pk(), node3.get_node_pk()], &*node3.keys_manager, &secp_ctx).unwrap();
+	blinded_route.blinded_hops.clear();
+	let err = node1.messenger.send_onion_message(&[], Destination::BlindedRoute(blinded_route)).unwrap_err();
+	assert_eq!(err, SendError::TooFewBlindedHops);
+
+	// 1 hop
+	let mut blinded_route = BlindedRoute::new::<EnforcingSigner, _, _>(&[node2.get_node_pk(), node3.get_node_pk()], &*node3.keys_manager, &secp_ctx).unwrap();
+	blinded_route.blinded_hops.remove(0);
+	assert_eq!(blinded_route.blinded_hops.len(), 1);
+	let err = node1.messenger.send_onion_message(&[], Destination::BlindedRoute(blinded_route)).unwrap_err();
+	assert_eq!(err, SendError::TooFewBlindedHops);
+}