+
+#[cfg(test)]
+mod tests {
+ use super::{ChannelFeatures, InitFeatures, NodeFeatures};
+
+ #[test]
+ fn sanity_test_known_features() {
+ assert!(!ChannelFeatures::known().requires_unknown_bits());
+ assert!(!ChannelFeatures::known().supports_unknown_bits());
+ assert!(!InitFeatures::known().requires_unknown_bits());
+ assert!(!InitFeatures::known().supports_unknown_bits());
+ assert!(!NodeFeatures::known().requires_unknown_bits());
+ assert!(!NodeFeatures::known().supports_unknown_bits());
+
+ assert!(InitFeatures::known().supports_upfront_shutdown_script());
+ assert!(NodeFeatures::known().supports_upfront_shutdown_script());
+ assert!(!InitFeatures::known().requires_upfront_shutdown_script());
+ assert!(!NodeFeatures::known().requires_upfront_shutdown_script());
+
+ assert!(InitFeatures::known().supports_gossip_queries());
+ assert!(NodeFeatures::known().supports_gossip_queries());
+ assert!(!InitFeatures::known().requires_gossip_queries());
+ assert!(!NodeFeatures::known().requires_gossip_queries());
+
+ assert!(InitFeatures::known().supports_data_loss_protect());
+ assert!(NodeFeatures::known().supports_data_loss_protect());
+ assert!(!InitFeatures::known().requires_data_loss_protect());
+ assert!(!NodeFeatures::known().requires_data_loss_protect());
+
+ assert!(InitFeatures::known().supports_variable_length_onion());
+ assert!(NodeFeatures::known().supports_variable_length_onion());
+ assert!(!InitFeatures::known().requires_variable_length_onion());
+ assert!(!NodeFeatures::known().requires_variable_length_onion());
+
+ assert!(InitFeatures::known().supports_static_remote_key());
+ assert!(NodeFeatures::known().supports_static_remote_key());
+ assert!(InitFeatures::known().requires_static_remote_key());
+ assert!(NodeFeatures::known().requires_static_remote_key());
+
+ assert!(InitFeatures::known().supports_payment_secret());
+ assert!(NodeFeatures::known().supports_payment_secret());
+ assert!(!InitFeatures::known().requires_payment_secret());
+ assert!(!NodeFeatures::known().requires_payment_secret());
+
+ assert!(InitFeatures::known().supports_basic_mpp());
+ assert!(NodeFeatures::known().supports_basic_mpp());
+ assert!(!InitFeatures::known().requires_basic_mpp());
+ assert!(!NodeFeatures::known().requires_basic_mpp());
+
+ assert!(InitFeatures::known().supports_shutdown_anysegwit());
+ assert!(NodeFeatures::known().supports_shutdown_anysegwit());
+
+ let mut init_features = InitFeatures::known();
+ assert!(init_features.initial_routing_sync());
+ init_features.clear_initial_routing_sync();
+ assert!(!init_features.initial_routing_sync());
+ }
+
+ #[test]
+ fn sanity_test_unknown_bits() {
+ let mut features = ChannelFeatures::empty();
+ assert!(!features.requires_unknown_bits());
+ assert!(!features.supports_unknown_bits());
+
+ features.set_required_unknown_bits();
+ assert!(features.requires_unknown_bits());
+ assert!(features.supports_unknown_bits());
+
+ features.clear_unknown_bits();
+ assert!(!features.requires_unknown_bits());
+ assert!(!features.supports_unknown_bits());
+
+ features.set_optional_unknown_bits();
+ assert!(!features.requires_unknown_bits());
+ assert!(features.supports_unknown_bits());
+ }
+
+ #[test]
+ fn convert_to_context_with_relevant_flags() {
+ let init_features = InitFeatures::known().clear_upfront_shutdown_script().clear_gossip_queries();
+ assert!(init_features.initial_routing_sync());
+ assert!(!init_features.supports_upfront_shutdown_script());
+ assert!(!init_features.supports_gossip_queries());
+
+ let node_features: NodeFeatures = init_features.to_context();
+ {
+ // Check that the flags are as expected:
+ // - option_data_loss_protect
+ // - var_onion_optin | static_remote_key (req) | payment_secret
+ // - basic_mpp
+ // - opt_shutdown_anysegwit
+ assert_eq!(node_features.flags.len(), 4);
+ assert_eq!(node_features.flags[0], 0b00000010);
+ assert_eq!(node_features.flags[1], 0b10010010);
+ assert_eq!(node_features.flags[2], 0b00000010);
+ assert_eq!(node_features.flags[3], 0b00001000);
+ }
+
+ // Check that cleared flags are kept blank when converting back:
+ // - initial_routing_sync was not applicable to NodeContext
+ // - upfront_shutdown_script was cleared before converting
+ // - gossip_queries was cleared before converting
+ let features: InitFeatures = node_features.to_context_internal();
+ assert!(!features.initial_routing_sync());
+ assert!(!features.supports_upfront_shutdown_script());
+ assert!(!init_features.supports_gossip_queries());
+ }
+}