Skip to content

Commit

Permalink
Add testing for pidfd_reaper
Browse files Browse the repository at this point in the history
Signed-off-by: Jiahao XU <[email protected]>
  • Loading branch information
NobodyXu committed Nov 15, 2023
1 parent 5591037 commit 64afdd0
Showing 1 changed file with 32 additions and 0 deletions.
32 changes: 32 additions & 0 deletions tokio/src/process/unix/pidfd_reaper.rs
Original file line number Diff line number Diff line change
Expand Up @@ -172,3 +172,35 @@ where
});
}
}

#[cfg(all(test, not(loom)))]
mod test {
use super::*;
use std::process::Command;

#[crate::test]
async fn test_pidfd_reaper_poll() {
let child = Command::new("true").spawn().unwrap();
let mut pidfd_reaper = PidfdReaper::new(child).unwrap();

let exit_status = pidfd_reaper.await.unwrap();
assert!(exit_status.success());
}

#[crate::test]
async fn test_pidfd_reaper_kill() {
let child = Command::new("sleep").arg("1800").spawn().unwrap();
let mut pidfd_reaper = PidfdReaper::new(child).unwrap();

pidfd_reaper.kill().unwrap();

let exit_status = pidfd_reaper.await.unwrap();
assert!(!exit_status.success());
}

#[crate::test]
async fn test_pidfd_reaper_drop() {
let child = Command::new("true").spawn().unwrap();
let mut pidfd_reaper = PidfdReaper::new(child).unwrap();
}
}

0 comments on commit 64afdd0

Please sign in to comment.