Test for add_random_cltv_offset.

Author: tnull

lightning/src/routing/router.rs

@@ -1611,7 +1611,7 @@ fn add_random_cltv_offset(route: &mut Route, payment_params: &PaymentParameters,
 mod tests {
 	use routing::scoring::{ProbabilisticScorer, ProbabilisticScoringParameters, Score};
 	use routing::network_graph::{NetworkGraph, NetGraphMsgHandler, NodeId};
-	use routing::router::{get_route, PaymentParameters, Route, RouteHint, RouteHintHop, RouteHop, RoutingFees};
+	use routing::router::{get_route, add_random_cltv_offset, PaymentParameters, Route, RouteHint, RouteHintHop, RouteHop, RoutingFees, DEFAULT_MAX_TOTAL_CLTV_EXPIRY_DELTA};
 	use chain::transaction::OutPoint;
 	use chain::keysinterface::KeysInterface;
 	use ln::features::{ChannelFeatures, InitFeatures, InvoiceFeatures, NodeFeatures};
@@ -5020,6 +5020,26 @@ mod tests {
 			Ok(_) => panic!("Expected error"),
 		}
 	}
+	#[test]
+	fn adds_cltv_offset() {
+		let (secp_ctx, network_graph, _, _, logger) = build_graph();
+		let (_, our_id, _, nodes) = get_nodes(&secp_ctx);
+
+		let scorer = test_utils::TestScorer::with_penalty(0);
+
+		let payment_params = PaymentParameters::from_node_id(nodes[6]).with_route_hints(last_hops(&nodes));
+		let keys_manager = test_utils::TestKeysInterface::new(&[0u8;32], Network::Testnet);
+		let random_seed_bytes = keys_manager.get_secure_random_bytes();
+		let mut route = get_route(&our_id, &payment_params, &network_graph, None, 100, 42, Arc::clone(&logger), &scorer, &random_seed_bytes).unwrap();
+		let cltv_expiry_deltas_before: Vec<u32> = route.paths.iter().flat_map(|p| p.iter().map(|h| h.cltv_expiry_delta).collect::<Vec<u32>>()).collect::<Vec<u32>>();
+		add_random_cltv_offset(&mut route, &payment_params, &network_graph, &random_seed_bytes);
+		let cltv_expiry_deltas_after: Vec<u32> = route.paths.iter().flat_map(|p| p.iter().map(|h| h.cltv_expiry_delta).collect::<Vec<u32>>()).collect::<Vec<u32>>();
+		assert_eq!(cltv_expiry_deltas_before.len(), cltv_expiry_deltas_after.len());
+		for i in 0..cltv_expiry_deltas_before.len() {
+			assert!(cltv_expiry_deltas_after[i] >= cltv_expiry_deltas_before[i]);
+			assert!(cltv_expiry_deltas_after[i] <= DEFAULT_MAX_TOTAL_CLTV_EXPIRY_DELTA);
+		}
+	}
 
 	#[cfg(not(feature = "no-std"))]
 	pub(super) fn random_init_seed() -> u64 {