Add basic tests for statime-netptp

Author: davidv1992

statime-netptp/Cargo.toml

@@ -10,7 +10,14 @@ description.workspace = true
 publish.workspace = true
 rust-version.workspace = true
 
+[features]
+default = ["privileged_tests"]
+privileged_tests = []
+
 [dependencies]
 timestamped-socket.workspace = true
 tokio = { workspace = true, features = ["sync"] }
 tracing.workspace = true
+
+[dev-dependencies]
+tokio = { workspace = true, features = ["rt", "macros"] }

statime-netptp/src/addresses/ipv4.rs

@@ -16,8 +16,14 @@ use crate::{
     addresses::{BoundInterface, SealedPtpAddressFamily},
 };
 
+#[cfg(not(test))]
 const EVENT_PORT: u16 = 319;
+#[cfg(not(test))]
 const GENERAL_PORT: u16 = 320;
+#[cfg(test)]
+const EVENT_PORT: u16 = 4319;
+#[cfg(test)]
+const GENERAL_PORT: u16 = 4320;
 
 type RecvTimestampFuture =
     dyn Future<Output = Result<(SendTimestampToken, FullTimestampData)>> + Sync + Send + 'static;

statime-netptp/src/addresses/ipv6.rs

@@ -16,8 +16,14 @@ use crate::{
     addresses::{BoundInterface, SealedPtpAddressFamily},
 };
 
+#[cfg(not(test))]
 const EVENT_PORT: u16 = 319;
+#[cfg(not(test))]
 const GENERAL_PORT: u16 = 320;
+#[cfg(test)]
+const EVENT_PORT: u16 = 4319;
+#[cfg(test)]
+const GENERAL_PORT: u16 = 4320;
 
 type RecvTimestampFuture =
     dyn Future<Output = Result<(SendTimestampToken, FullTimestampData)>> + Sync + Send + 'static;

statime-netptp/src/lib.rs

@@ -380,3 +380,6 @@ enum TimestampSource {
     System,
     Hardware,
 }
+
+#[cfg(all(test, feature = "privileged_tests"))]
+mod test;

statime-netptp/src/test.rs

@@ -0,0 +1,49 @@
+use std::net::{Ipv4Addr, Ipv6Addr};
+
+use crate::NetworkManager;
+
+#[tokio::test]
+async fn test_ipv4() {
+    let manager = NetworkManager::<Ipv4Addr>::new().unwrap();
+    let general = manager.open_general();
+    let mut socket = general.listen_socket();
+
+    socket
+        .send_general(&[1, 2, 3, 4], None, Ipv4Addr::LOCALHOST)
+        .await
+        .unwrap();
+    let result = socket.recv().await.unwrap();
+    assert_eq!(&*result.bytes_read, [1, 2, 3, 4].as_slice());
+
+    let ts = socket
+        .send_event(&[5, 6, 7, 8], None, Ipv4Addr::LOCALHOST)
+        .await
+        .unwrap();
+    assert!(ts.is_some());
+    let result = socket.recv().await.unwrap();
+    assert_eq!(&*result.bytes_read, [5, 6, 7, 8].as_slice());
+    assert!(result.timestamp.is_some());
+}
+
+#[tokio::test]
+async fn test_ipv6() {
+    let manager = NetworkManager::<Ipv6Addr>::new().unwrap();
+    let general = manager.open_general();
+    let mut socket = general.listen_socket();
+
+    socket
+        .send_general(&[1, 2, 3, 4], None, Ipv6Addr::LOCALHOST)
+        .await
+        .unwrap();
+    let result = socket.recv().await.unwrap();
+    assert_eq!(&*result.bytes_read, [1, 2, 3, 4].as_slice());
+
+    let ts = socket
+        .send_event(&[5, 6, 7, 8], None, Ipv6Addr::LOCALHOST)
+        .await
+        .unwrap();
+    assert!(ts.is_some());
+    let result = socket.recv().await.unwrap();
+    assert_eq!(&*result.bytes_read, [5, 6, 7, 8].as_slice());
+    assert!(result.timestamp.is_some());
+}