Add tests for getting next channel announcements

Author: naumenkogs

lightning/src/ln/router.rs

@@ -2332,5 +2332,124 @@ mod tests {
 	 	// router.handle_htlc_fail_channel_update(&node_failure_msg);
 	}
 
+	#[test]
+	fn getting_next_channel_announcements() {
+		let (secp_ctx, our_id, 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 zero_hash = Sha256dHash::hash(&[0; 32]);
+		let channel_id = 0;
+		let channel_time = 100;
+		let channel_key = NetworkMap::get_key(channel_id, zero_hash.clone());
+
+		// No channels yet.
+		let next_announcements = router.get_next_channel_announcements(channel_key, 10);
+		assert_eq!(next_announcements.len(), 0);
+		
+		{
+			let mut network = router.network_map.write().unwrap();
+			network.channels.insert(channel_key, ChannelInfo {
+				features: ChannelFeatures::empty(),
+				one_to_two: DirectionalChannelInfo {
+					src_node_id: public_key.clone(),
+					last_update: channel_time,
+					enabled: false,
+					cltv_expiry_delta: u16::max_value(),
+					htlc_minimum_msat: 0,
+					fee_base_msat: u32::max_value(),
+					fee_proportional_millionths: u32::max_value(),
+					last_update_message: None,
+				}, two_to_one: DirectionalChannelInfo {
+					src_node_id: our_id.clone(),
+					last_update: channel_time,
+					enabled: true,
+					cltv_expiry_delta: 0,
+					htlc_minimum_msat: 0,
+					fee_base_msat: u32::max_value(),
+					fee_proportional_millionths: 0,
+					last_update_message: None,
+				},
+				announcement_message: None,
+			});
+		}
+
+		// Channels without announcements are not returned.
+		let next_announcements = router.get_next_channel_announcements(channel_key, 2);
+		assert_eq!(next_announcements.len(), 0);
+
+		{
+			let mut network = router.network_map.write().unwrap();
+
+			let unsigned_channel_update = UnsignedChannelUpdate {
+				chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
+				short_channel_id: channel_id,
+				timestamp: channel_time + 100,
+				flags: 0,
+				cltv_expiry_delta: 144,
+				htlc_minimum_msat: 1000000,
+				fee_base_msat: 10000,
+				fee_proportional_millionths: 20,
+				excess_data: Vec::new()
+			};
+			let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_channel_update.encode()[..])[..]);
+			let valid_channel_update = ChannelUpdate {
+				signature: secp_ctx.sign(&msghash, secret_key),
+				contents: unsigned_channel_update.clone()
+			};
+
+			let unsigned_announcement = UnsignedChannelAnnouncement {
+				features: ChannelFeatures::supported(),
+				chain_hash: genesis_block(Network::Testnet).header.bitcoin_hash(),
+				short_channel_id: 0,
+				node_id_1: our_id,
+				node_id_2: our_id,
+				bitcoin_key_1: public_key,
+				bitcoin_key_2: public_key,
+				excess_data: Vec::new(),
+			};
+
+			let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_announcement.encode()[..])[..]);
+			let announcement = ChannelAnnouncement {
+				node_signature_1: secp_ctx.sign(&msghash, secret_key),
+				node_signature_2: secp_ctx.sign(&msghash, secret_key),
+				bitcoin_signature_1: secp_ctx.sign(&msghash, secret_key),
+				bitcoin_signature_2: secp_ctx.sign(&msghash, secret_key),
+				contents: unsigned_announcement.clone(),
+			};
+
+			network.channels.insert(channel_key, ChannelInfo {
+				features: ChannelFeatures::empty(),
+				one_to_two: DirectionalChannelInfo {
+					src_node_id: public_key.clone(),
+					last_update: channel_time,
+					enabled: false,
+					cltv_expiry_delta: u16::max_value(),
+					htlc_minimum_msat: 0,
+					fee_base_msat: u32::max_value(),
+					fee_proportional_millionths: u32::max_value(),
+					last_update_message: Some(valid_channel_update.clone()),
+				}, two_to_one: DirectionalChannelInfo {
+					src_node_id: our_id.clone(),
+					last_update: channel_time,
+					enabled: true,
+					cltv_expiry_delta: 0,
+					htlc_minimum_msat: 0,
+					fee_base_msat: u32::max_value(),
+					fee_proportional_millionths: 0,
+					last_update_message: Some(valid_channel_update.clone()),
+				},
+				announcement_message: Some(announcement.clone()),
+			});
+		}
+
+		let next_announcements = router.get_next_channel_announcements(channel_key, 2);
+		assert_eq!(next_announcements.len(), 1);
+
+		// Further starting point have no channels after it
+		let next_announcements = router.get_next_channel_announcements(channel_key + 1000, 2);
+		assert_eq!(next_announcements.len(), 0);
+	}
+
 
 }