Test UdpSender

seamlik · Mar 16, 2024 · a071de3 · a071de3
1 parent 5e0c70e
commit a071de3
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/main/src/network/udp_sender.rs b/main/src/network/udp_sender.rs
@@ -63,3 +63,31 @@ impl UdpSender for TokioUdpSender {
         Self::send_unicast(unicast_address, data).boxed()
     }
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+    use std::net::Ipv6Addr;
+
+    #[tokio::test]
+    async fn send_multicast() {
+        crate::test::init();
+
+        let address = SocketAddrV6::new(crate::get_discovery_ip(), 1, 0, 0);
+        TokioUdpSender
+            .send_multicast(address, Arc::new([1]))
+            .await
+            .unwrap();
+    }
+
+    #[tokio::test]
+    async fn send_unicast() {
+        crate::test::init();
+
+        let address = SocketAddrV6::new(Ipv6Addr::LOCALHOST, 1, 0, 0);
+        TokioUdpSender
+            .send_unicast(address, Arc::new([1]))
+            .await
+            .unwrap();
+    }
+}