gossip: Test NodeFailure events

Author: dunxen

lightning/src/routing/gossip.rs

@@ -2265,6 +2265,7 @@ mod tests {
 
 		let node_1_privkey = &SecretKey::from_slice(&[42; 32]).unwrap();
 		let node_2_privkey = &SecretKey::from_slice(&[41; 32]).unwrap();
+		let node_2_id = PublicKey::from_secret_key(&secp_ctx, node_2_privkey);
 
 		{
 			// There is no nodes in the table at the beginning.
@@ -2354,7 +2355,54 @@ mod tests {
 		assert_eq!(network_graph.read_only().channels().len(), 0);
 		// Nodes are also deleted because there are no associated channels anymore
 		assert_eq!(network_graph.read_only().nodes().len(), 0);
-		// TODO: Test NetworkUpdate::NodeFailure, which is not implemented yet.
+
+		// Announce a channel to test permanent node failure
+		let valid_channel_announcement = get_signed_channel_announcement(|_| {}, node_1_privkey, node_2_privkey, &secp_ctx);
+		let short_channel_id = valid_channel_announcement.contents.short_channel_id;
+		let chain_source: Option<&test_utils::TestChainSource> = None;
+		assert!(network_graph.update_channel_from_announcement(&valid_channel_announcement, &chain_source).is_ok());
+		assert!(network_graph.read_only().channels().get(&short_channel_id).is_some());
+
+		// Non-permanent node failure does not delete any nodes or channels
+		network_graph.handle_event(&Event::PaymentPathFailed {
+			payment_id: None,
+			payment_hash: PaymentHash([0; 32]),
+			rejected_by_dest: false,
+			all_paths_failed: true,
+			path: vec![],
+			network_update: Some(NetworkUpdate::NodeFailure {
+				node_id: node_2_id,
+				is_permanent: false,
+			}),
+			short_channel_id: None,
+			retry: None,
+			error_code: None,
+			error_data: None,
+		});
+
+		assert!(network_graph.read_only().channels().get(&short_channel_id).is_some());
+		assert!(network_graph.read_only().nodes().get(&NodeId::from_pubkey(&node_2_id)).is_some());
+
+		// Permanent node failure deletes node and its channels
+		network_graph.handle_event(&Event::PaymentPathFailed {
+			payment_id: None,
+			payment_hash: PaymentHash([0; 32]),
+			rejected_by_dest: false,
+			all_paths_failed: true,
+			path: vec![],
+			network_update: Some(NetworkUpdate::NodeFailure {
+				node_id: node_2_id,
+				is_permanent: true,
+			}),
+			short_channel_id: None,
+			retry: None,
+			error_code: None,
+			error_data: None,
+		});
+
+		assert_eq!(network_graph.read_only().nodes().len(), 0);
+		// Channels are also deleted because the associated node has been deleted
+		assert_eq!(network_graph.read_only().channels().len(), 0);
 	}
 
 	#[test]