f no-std test

Author: dunxen

lightning/src/routing/gossip.rs

@@ -2438,57 +2438,59 @@ mod tests {
 		// Nodes are also deleted because there are no associated channels anymore
 		assert_eq!(network_graph.read_only().nodes().len(), 0);
 
-		// Get a new network graph since we don't want to track removed nodes in this test
-		let network_graph = NetworkGraph::new(genesis_hash, &logger);
+		#[cfg(feature = "std")]
+		{
+			// Get a new network graph since we don't want to track removed nodes in this test
+			let network_graph = NetworkGraph::new(genesis_hash, &logger);
 
-		// 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());
+			// 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,
-		});
+			// 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());
+			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,
-		});
+			// 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,
+			});
 
-		println!("{:?}", network_graph.read_only().nodes());
-		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);
+			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]