test: add test_provider

Cargo.toml

@@ -31,6 +31,7 @@ yaml-merge-keys = { version = "0.5.1", features = ["serde_yaml"] }
 serde_json = "1.0"
 serde = { version = "1.0.119", features = ["derive"] }
 rd-interface = { path = "./rabbit-digger/rd-interface/", version = "0.4" }
+rd-std = { path = "./rabbit-digger/rd-std/", version = "0.1" }
 cfg-if = "1.0"
 reqwest = { version = "0.11.4", features = ["native-tls-vendored", "socks"] }
 parking_lot = "0.12.0"

protocol/obfs/Cargo.toml

@@ -6,7 +6,7 @@ edition = "2021"
 
 [dependencies]
 rd-interface = { path = "../../rabbit-digger/rd-interface/", version = "0.4" }
-# rd-interface = "0.3"
+rd-std = { path = "../../rabbit-digger/rd-std/", version = "0.1" }
 serde = "1.0"
 tokio = "1.0"
 pin-project-lite = "0.2.6"

protocol/obfs/src/obfs_net.rs

@@ -70,3 +70,31 @@ impl ITcpListener for ObfsTcpListener {
         self.0.local_addr().await
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use rd_std::tests::{assert_net_provider, ProviderCapability, TestNet};
+
+    use super::*;
+
+    #[test]
+    fn test_provider() {
+        let net = TestNet::new().into_dyn();
+
+        let obfs = ObfsNet::new(ObfsNetConfig {
+            net: NetRef::new_with_value("test".into(), net.clone()),
+            obfs_type: ObfsType::Plain(Default::default()),
+        })
+        .unwrap()
+        .into_dyn();
+
+        assert_net_provider(
+            &obfs,
+            ProviderCapability {
+                tcp_connect: true,
+                tcp_bind: true,
+                ..Default::default()
+            },
+        );
+    }
+}

protocol/obfs/src/plain.rs

@@ -2,7 +2,7 @@ use crate::Obfs;
 use rd_interface::{prelude::*, Address, Result, TcpStream};
 
 #[rd_config]
-#[derive(Debug)]
+#[derive(Debug, Default)]
 pub struct Plain;
 
 impl Obfs for Plain {

protocol/rpc/src/net.rs

@@ -148,3 +148,34 @@ impl INet for RpcNet {
         Some(self)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use rd_interface::IntoAddress;
+    use rd_std::tests::{assert_net_provider, ProviderCapability, TestNet};
+
+    use super::*;
+
+    #[test]
+    fn test_provider() {
+        let net = TestNet::new().into_dyn();
+
+        let rpc = RpcNet::new(
+            net,
+            "127.0.0.1:1234".into_address().unwrap(),
+            true,
+            Codec::Cbor,
+        )
+        .into_dyn();
+
+        assert_net_provider(
+            &rpc,
+            ProviderCapability {
+                tcp_connect: true,
+                tcp_bind: true,
+                udp_bind: true,
+                lookup_host: true,
+            },
+        );
+    }
+}

protocol/ss/src/client.rs

@@ -107,3 +107,34 @@ impl INet for SSNet {
         Some(self)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use rd_interface::IntoAddress;
+    use rd_std::tests::{assert_net_provider, ProviderCapability, TestNet};
+
+    use super::*;
+
+    #[test]
+    fn test_provider() {
+        let net = TestNet::new().into_dyn();
+
+        let ss = SSNet::new(SSNetConfig {
+            server: "127.0.0.1:1234".into_address().unwrap(),
+            password: "password".to_string(),
+            udp: false,
+            cipher: Cipher::AES_128_CCM,
+            net: NetRef::new_with_value("test".into(), net),
+        })
+        .into_dyn();
+
+        assert_net_provider(
+            &ss,
+            ProviderCapability {
+                tcp_connect: true,
+                udp_bind: true,
+                ..Default::default()
+            },
+        );
+    }
+}

protocol/trojan/src/client.rs

@@ -155,3 +155,37 @@ impl INet for TrojanNet {
         Some(self)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use rd_interface::IntoAddress;
+    use rd_std::tests::{assert_net_provider, ProviderCapability, TestNet};
+
+    use super::*;
+
+    #[test]
+    fn test_provider() {
+        let net = TestNet::new().into_dyn();
+
+        let trojan = TrojanNet::new(TrojanNetConfig {
+            net: NetRef::new_with_value("test".into(), net),
+            server: "127.0.0.1:1234".into_address().unwrap(),
+            password: "password".to_string(),
+            sni: None,
+            skip_cert_verify: false,
+            websocket: None,
+            handshake_timeout: None,
+        })
+        .unwrap()
+        .into_dyn();
+
+        assert_net_provider(
+            &trojan,
+            ProviderCapability {
+                tcp_connect: true,
+                udp_bind: true,
+                ..Default::default()
+            },
+        );
+    }
+}

src/select.rs

@@ -64,3 +64,33 @@ pub fn init(registry: &mut Registry) -> Result<()> {
     registry.add_net::<SelectNet>();
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use rd_interface::IntoDyn;
+    use rd_std::tests::{assert_net_provider, ProviderCapability, TestNet};
+
+    use super::*;
+
+    #[test]
+    fn test_provider() {
+        let net = NetRef::new_with_value("test".into(), TestNet::new().into_dyn());
+
+        let select = SelectNet::new(SelectNetConfig {
+            selected: net.clone(),
+            list: vec![net],
+        })
+        .unwrap()
+        .into_dyn();
+
+        assert_net_provider(
+            &select,
+            ProviderCapability {
+                tcp_connect: true,
+                tcp_bind: true,
+                udp_bind: true,
+                lookup_host: true,
+            },
+        );
+    }
+}