Test anti-probing penalty.

Author: tnull

lightning/src/routing/gossip.rs

@@ -128,6 +128,9 @@ pub struct NetworkGraph<L: Deref> where L::Target: Logger {
 	genesis_hash: BlockHash,
 	logger: L,
 	// Lock order: channels -> nodes
+	#[cfg(test)]
+	pub(crate) channels: RwLock<BTreeMap<u64, ChannelInfo>>,
+	#[cfg(not(test))]
 	channels: RwLock<BTreeMap<u64, ChannelInfo>>,
 	nodes: RwLock<BTreeMap<NodeId, NodeInfo>>,
 }

lightning/src/routing/scoring.rs

@@ -2126,4 +2126,49 @@ mod tests {
 		let usage = ChannelUsage { amount_msat: 1_001, ..usage };
 		assert_eq!(scorer.channel_penalty_msat(42, &source, &target, usage), u64::max_value());
 	}
+
+	#[test]
+	fn adds_anti_probing_penalty() {
+		let logger = TestLogger::new();
+		let network_graph = network_graph(&logger);
+		let source = source_node_id();
+		let target = target_node_id();
+		let params = ProbabilisticScoringParameters {
+			anti_probing_penalty_msat: 10_000,
+			..ProbabilisticScoringParameters::zero_penalty()
+		};
+		let scorer = ProbabilisticScorer::new(params, &network_graph, &logger);
+
+		// Check we receive no penalty for a low htlc_maximum_msat.
+		let usage = ChannelUsage {
+			amount_msat: 512_000,
+			inflight_htlc_msat: 0,
+			effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: Some(1_000) },
+		};
+		assert_eq!(scorer.channel_penalty_msat(42, &source, &target, usage), 0);
+
+		// Check we receive anti-probing penalty for htlc_maximum_msat == channel_capacity.
+		let usage = ChannelUsage {
+			amount_msat: 512_000,
+			inflight_htlc_msat: 0,
+			effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: Some(1_024_000) },
+		};
+		assert_eq!(scorer.channel_penalty_msat(42, &source, &target, usage), 10_000);
+
+		// Check we receive anti-probing penalty for htlc_maximum_msat == channel_capacity/2.
+		let usage = ChannelUsage {
+			amount_msat: 512_000,
+			inflight_htlc_msat: 0,
+			effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: Some(512_000) },
+		};
+		assert_eq!(scorer.channel_penalty_msat(42, &source, &target, usage), 10_000);
+
+		// Check we receive no anti-probing penalty for htlc_maximum_msat == channel_capacity/2 - 1.
+		let usage = ChannelUsage {
+			amount_msat: 512_000,
+			inflight_htlc_msat: 0,
+			effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: Some(511_999) },
+		};
+		assert_eq!(scorer.channel_penalty_msat(42, &source, &target, usage), 0);
+	}
 }