test: add timeouts to async tests

MystenLabs · May 21, 2024 · 1f55677 · 1f55677
1 parent fdea7e6
commit 1f55677
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/crates/scion/src/socket/udp.rs b/crates/scion/src/socket/udp.rs
@@ -427,12 +427,15 @@ mod tests {
     }
 
     macro_rules! async_test_case {
-        ($name:ident: $func:ident($($arg:expr),*)) => {
+        ($name:ident($timeout:expr): $func:ident($($arg:expr),*)) => {
             #[tokio::test]
             async fn $name() -> TestResult {
-                $func($($arg,)*).await
+                tokio::time::timeout(std::time::Duration::from_secs($timeout),$func($($arg,)*)).await.unwrap()
             }
         };
+        ($name:ident: $($tail:tt)*) => {
+            async_test_case!($name(1): $($tail)*);
+        };
     }
 
     const MESSAGE: &[u8] = b"Hello World! Hello World! Hello World!";