add tests for boxed link

shinolab · Jan 29, 2025 · 071a9a8 · 071a9a8
1 parent a881d71
commit 071a9a8
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 40 deletions.
diff --git a/autd3/src/async/controller/sender/mod.rs b/autd3/src/async/controller/sender/mod.rs
@@ -153,6 +153,7 @@ mod tests {
 
     use super::*;
 
+    #[derive(Default)]
     struct MockAsyncLink {
         pub is_open: bool,
         pub send_cnt: usize,
@@ -199,12 +200,8 @@ mod tests {
 
     #[tokio::test]
     async fn test_close() -> anyhow::Result<()> {
-        let mut link = MockAsyncLink {
-            is_open: true,
-            send_cnt: 0,
-            recv_cnt: 0,
-            down: false,
-        };
+        let mut link = MockAsyncLink::default();
+        link.open(&Geometry::new(Vec::new())).await?;
 
         assert!(link.is_open());
 
@@ -222,16 +219,12 @@ mod tests {
     #[cfg_attr(target_os = "windows", case(WaitableSleeper::new().unwrap()))]
     #[tokio::test]
     async fn test_send_receive(#[case] sleeper: impl AsyncSleep) {
-        let mut link = MockAsyncLink {
-            is_open: true,
-            send_cnt: 0,
-            recv_cnt: 0,
-            down: false,
-        };
+        let mut link = MockAsyncLink::default();
         let mut geometry = create_geometry(1);
         let mut tx = vec![];
         let mut rx = Vec::new();
 
+        assert!(link.open(&geometry).await.is_ok());
         let mut sender = Sender {
             link: &mut link,
             geometry: &mut geometry,
@@ -272,17 +265,13 @@ mod tests {
     #[cfg_attr(target_os = "windows", case(WaitableSleeper::new().unwrap()))]
     #[tokio::test]
     async fn test_wait_msg_processed(#[case] sleeper: impl AsyncSleep) {
-        let mut link = MockAsyncLink {
-            is_open: true,
-            send_cnt: 0,
-            recv_cnt: 0,
-            down: false,
-        };
+        let mut link = MockAsyncLink::default();
         let mut geometry = create_geometry(1);
         let mut tx = vec![TxMessage::new_zeroed(); 1];
         tx[0].header.msg_id = 2;
         let mut rx = vec![RxMessage::new(0, 0)];
 
+        assert!(link.open(&geometry).await.is_ok());
         let mut sender = Sender {
             link: &mut link,
             geometry: &mut geometry,

diff --git a/autd3/src/controller/group.rs b/autd3/src/controller/group.rs
@@ -321,7 +321,7 @@ mod tests {
                     option: Default::default(),
                 },
             ),
-        )? // GRCOV_EXCL_LINE
+        )?
         .send()?;
 
         assert_eq!(
@@ -421,7 +421,7 @@ mod tests {
                         option: Default::default(),
                     },
                 ),
-            )? // GRCOV_EXCL_LINE
+            )?
             .send()?;
 
         assert_eq!(

diff --git a/autd3/src/controller/mod.rs b/autd3/src/controller/mod.rs
@@ -401,7 +401,7 @@ pub(crate) mod tests {
                 config: 1. * Hz,
                 option: Default::default(),
             },
-        ))?; // GRCOV_EXCL_LINE
+        ))?;
 
         autd.iter().try_for_each(|dev| {
             assert_eq!(
@@ -563,6 +563,21 @@ pub(crate) mod tests {
         Ok(())
     }
 
+    #[test]
+    fn with_boxed_link() -> anyhow::Result<()> {
+        let link: Box<dyn Link> = Box::new(Audit::new(AuditOption::default()));
+        let mut autd = Controller::open([AUTD3::default()], link)?;
+
+        autd.send(Sine {
+            freq: 150. * Hz,
+            option: Default::default(),
+        })?;
+
+        autd.close()?;
+
+        Ok(())
+    }
+
     #[test]
     fn into_boxed_link_unsafe() -> anyhow::Result<()> {
         let option = SenderOption {
@@ -598,7 +613,7 @@ pub(crate) mod tests {
                 config: 1. * Hz,
                 option: Default::default(),
             },
-        ))?; // GRCOV_EXCL_LINE
+        ))?;
 
         let mut autd = unsafe { Controller::<Audit>::from_boxed_link(autd) };
 

diff --git a/autd3/src/controller/sender/mod.rs b/autd3/src/controller/sender/mod.rs
@@ -229,6 +229,7 @@ mod tests {
         assert_eq!(expect, mode.is_parallel(num_devices, threshold));
     }
 
+    #[derive(Default)]
     struct MockLink {
         pub is_open: bool,
         pub send_cnt: usize,
@@ -271,12 +272,8 @@ mod tests {
 
     #[test]
     fn test_close() -> anyhow::Result<()> {
-        let mut link = MockLink {
-            is_open: true,
-            send_cnt: 0,
-            recv_cnt: 0,
-            down: false,
-        };
+        let mut link = MockLink::default();
+        link.open(&Geometry::new(Vec::new()))?;
 
         assert!(link.is_open());
 
@@ -293,16 +290,12 @@ mod tests {
     #[cfg_attr(target_os = "windows", case(WaitableSleeper::new().unwrap()))]
     #[test]
     fn test_send_receive(#[case] sleeper: impl Sleep) {
-        let mut link = MockLink {
-            is_open: true,
-            send_cnt: 0,
-            recv_cnt: 0,
-            down: false,
-        };
+        let mut link = MockLink::default();
         let mut geometry = create_geometry(1);
         let mut tx = vec![];
         let mut rx = Vec::new();
 
+        assert!(link.open(&geometry).is_ok());
         let mut sender = Sender {
             link: &mut link,
             geometry: &mut geometry,
@@ -342,17 +335,13 @@ mod tests {
     #[cfg_attr(target_os = "windows", case(WaitableSleeper::new().unwrap()))]
     #[test]
     fn test_wait_msg_processed(#[case] sleeper: impl Sleep) {
-        let mut link = MockLink {
-            is_open: true,
-            send_cnt: 0,
-            recv_cnt: 0,
-            down: false,
-        };
+        let mut link = MockLink::default();
         let mut geometry = create_geometry(1);
         let mut tx = vec![TxMessage::new_zeroed(); 1];
         tx[0].header.msg_id = 2;
         let mut rx = vec![RxMessage::new(0, 0)];
 
+        assert!(link.open(&geometry).is_ok());
         let mut sender = Sender {
             link: &mut link,
             geometry: &mut geometry,