drivers: udp: server: Add udpin prefix

Signed-off-by: Patrick José Pereira <[email protected]>
bluerobotics · Aug 26, 2024 · cb34830 · cb34830
1 parent 8731fcb
commit cb34830
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 16 deletions.
diff --git a/src/cli.rs b/src/cli.rs
@@ -123,23 +123,26 @@ mod tests {
     #[test]
     fn test_endpoints() {
         let endpoints = vec![
-            "serial:/dev/ttyS0:9600",
-            "serial:COM1:115200",
-            "tcpc:10.0.0.1:4000",
-            //"tcpc:localhost:7000",
-            "tcpc:127.0.0.1:7000",
-            "tcps:0.0.0.0:5000",
-            "tcps:192.168.1.10:6000",
-            "udpb:192.168.0.255:3000",
-            "udpb:255.255.255.255:9999",
-            "udpc:127.0.0.1:1234",
-            "udpc:192.168.1.100:8080",
-            "udps:0.0.0.0:5000",
-            "udps:192.168.1.5:6789",
+            ("serial:/dev/ttyS0:9600", false),
+            ("serial:COM1:115200", false),
+            ("tcpc:10.0.0.1:4000", true),
+            ("tcpc:127.0.0.1:7000", true),
+            ("tcps:0.0.0.0:5000", true),
+            ("tcps:192.168.1.10:6000", true),
+            ("udpb:192.168.0.255:3000", false),
+            ("udpb:255.255.255.255:9999", false),
+            ("udpc:127.0.0.1:1234", true),
+            ("udpc:192.168.1.100:8080", true),
+            ("udpout:127.0.0.1:1234", true),
+            ("udpout:192.168.1.100:8080", true),
+            ("udps:0.0.0.0:5000", true),
+            ("udps:192.168.1.5:6789", true),
+            ("udpin:0.0.0.0:5000", true),
+            ("udpin:192.168.1.5:6789", true),
         ];
 
-        for endpoint in endpoints {
-            dbg!(endpoints_parser(endpoint).is_ok());
+        for (endpoint, result) in endpoints {
+            assert_eq!(endpoints_parser(endpoint).is_ok(), result);
         }
     }
 }
diff --git a/src/drivers/udp/server.rs b/src/drivers/udp/server.rs
@@ -168,7 +168,11 @@ impl Driver for UdpServer {
 pub struct UdpServerExt;
 impl DriverExt for UdpServerExt {
     fn valid_schemes(&self) -> Vec<String> {
-        vec!["udps".to_string(), "udpserver".to_string()]
+        vec![
+            "udps".to_string(),
+            "udpserver".to_string(),
+            "udpin".to_string(),
+        ]
     }
 
     fn create_endpoint_from_url(&self, url: &url::Url) -> Option<Arc<dyn Driver>> {