Add tests for getting node announcements

Author: naumenkogs

lightning/src/ln/router.rs

@@ -2451,5 +2451,89 @@ mod tests {
 		assert_eq!(next_announcements.len(), 0);
 	}
 
+	#[test]
+	fn getting_next_node_announcements() {
+		let (secp_ctx, _, router) = create_router();
+		let secret_key = &SecretKey::from_slice(&hex::decode("0202020202020202020202020202020202020202020202020202020202020202").unwrap()[..]).unwrap();
+		let public_key = PublicKey::from_secret_key(&secp_ctx, secret_key);
+		let secret_key2 = &SecretKey::from_slice(&hex::decode("0202020202020202020202020202020202020202020202020202020202020201").unwrap()[..]).unwrap();
+		let public_key2 = PublicKey::from_secret_key(&secp_ctx, secret_key2);
+
+		let zero_hash = Sha256dHash::hash(&[0; 32]);
+		let channel_key = NetworkMap::get_key(0, zero_hash.clone());
+		// No nodes yet.
+		let next_announcements = router.get_next_node_announcements(None, 10);
+		assert_eq!(next_announcements.len(), 0);
+		
+		{
+			let mut network = router.network_map.write().unwrap();
+			network.nodes.insert(public_key.clone(), NodeInfo {
+				channels: vec!(channel_key),
+				lowest_inbound_channel_fee_base_msat: u32::max_value(),
+				lowest_inbound_channel_fee_proportional_millionths: 0,
+				features: NodeFeatures::empty(),
+				last_update: Some(1),
+				rgb: [0; 3],
+				alias: [0; 32],
+				addresses: Vec::new(),
+				announcement_message: None,
+			});
+		}
+
+		// Nodes without announcements are not returned.
+		let next_announcements = router.get_next_node_announcements(None, 2);
+		assert_eq!(next_announcements.len(), 0);
 
+		{
+			let mut network = router.network_map.write().unwrap();
+
+			let unsigned_announcement = UnsignedNodeAnnouncement {
+				features: NodeFeatures::supported(),
+				timestamp: 10,
+				node_id: public_key,
+				rgb: [0; 3],
+				alias: [0; 32],
+				addresses: Vec::new(),
+				excess_address_data: Vec::new(),
+				excess_data: Vec::new(),
+			};
+			let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_announcement.encode()[..])[..]);
+			let valid_announcement = NodeAnnouncement {
+				signature: secp_ctx.sign(&msghash, secret_key),
+				contents: unsigned_announcement.clone()
+			};
+
+			network.nodes.insert(public_key.clone(), NodeInfo {
+				channels: vec!(channel_key),
+				lowest_inbound_channel_fee_base_msat: u32::max_value(),
+				lowest_inbound_channel_fee_proportional_millionths: 0,
+				features: NodeFeatures::empty(),
+				last_update: Some(1),
+				rgb: [0; 3],
+				alias: [0; 32],
+				addresses: Vec::new(),
+				announcement_message: Some(valid_announcement.clone()),
+			});
+	
+			network.nodes.insert(public_key2.clone(), NodeInfo {
+				channels: vec!(channel_key),
+				lowest_inbound_channel_fee_base_msat: u32::max_value(),
+				lowest_inbound_channel_fee_proportional_millionths: 0,
+				features: NodeFeatures::empty(),
+				last_update: Some(1),
+				rgb: [0; 3],
+				alias: [0; 32],
+				addresses: Vec::new(),
+				announcement_message: Some(valid_announcement.clone()),
+			});
+
+		}
+
+		let next_announcements = router.get_next_node_announcements(None, 3);
+		assert_eq!(next_announcements.len(), 2);
+
+		// Skip the first node.
+		let next_announcements = router.get_next_node_announcements(Some(&public_key2), 2);
+		assert_eq!(next_announcements.len(), 1);
+	}
 }