add a unit test for fork.

youki-dev · Oct 21, 2021 · 41cbf8e · 41cbf8e
1 parent ea3d37d
commit 41cbf8e
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/process/fork.rs b/src/process/fork.rs
@@ -22,3 +22,23 @@ pub fn container_fork<F: FnOnce() -> Result<()>>(cb: F) -> Result<Pid> {
         }
     }
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+    use anyhow::{bail, Result};
+    use nix::sys::wait::{waitpid, WaitStatus};
+
+    #[test]
+    fn test_container_fork() -> Result<()> {
+        let pid = container_fork(|| Ok(()))?;
+        match waitpid(pid, None).expect("wait pid failed.") {
+            WaitStatus::Exited(p, status) => {
+                assert_eq!(pid, p);
+                assert_eq!(status, 0);
+                Ok(())
+            }
+            _ => bail!("test failed"),
+        }
+    }
+}