use crate::common::util::*; use regex::Regex; use std::os::unix::process::ExitStatusExt; use std::process::{Child, Command}; // A child process the tests will try to kill. struct Target { child: Child, killed: bool, } impl Target { // Creates a target that will naturally die after some time if not killed // fast enough. // This timeout avoids hanging failing tests. fn new() -> Target { Target { child: Command::new("sleep") .arg("30") .spawn() .expect("cannot spawn target"), killed: false, } } // Waits for the target to complete and returns the signal it received if any. fn wait_for_signal(&mut self) -> Option<i32> { let sig = self.child.wait().expect("cannot wait on target").signal(); self.killed = true; sig } fn pid(&self) -> u32 { self.child.id() } } impl Drop for Target { // Terminates this target to avoid littering test boxes with zombi processes // when a test fails after creating a target but before killing it. fn drop(&mut self) { if !self.killed { self.child.kill().expect("cannot kill target"); } } } #[test] fn test_kill_list_all_signals() { // Check for a few signals. Do not try to be comprehensive. new_ucmd!() .arg("-l") .succeeds() .stdout_contains("KILL") .stdout_contains("TERM") .stdout_contains("HUP"); } #[test] fn test_kill_list_all_signals_as_table() { // Check for a few signals. Do not try to be comprehensive. new_ucmd!() .arg("-t") .succeeds() .stdout_contains("KILL") .stdout_contains("TERM") .stdout_contains("HUP"); } #[test] fn test_kill_list_one_signal_from_name() { // Use SIGKILL because it is 9 on all unixes. new_ucmd!() .arg("-l") .arg("KILL") .succeeds() .stdout_matches(&Regex::new("\\b9\\b").unwrap()); } #[test] fn test_kill_set_bad_signal_name() { // spell-checker:disable-line new_ucmd!() .arg("-s") .arg("IAMNOTASIGNAL") // spell-checker:disable-line .fails() .stderr_contains("unknown signal"); } #[test] fn test_kill_with_default_signal() { let mut target = Target::new(); new_ucmd!().arg(format!("{}", target.pid())).succeeds(); assert_eq!(target.wait_for_signal(), Some(libc::SIGTERM)); } #[test] fn test_kill_with_signal_number_old_form() { let mut target = Target::new(); new_ucmd!() .arg("-9") .arg(format!("{}", target.pid())) .succeeds(); assert_eq!(target.wait_for_signal(), Some(9)); } #[test] fn test_kill_with_signal_number_new_form() { let mut target = Target::new(); new_ucmd!() .arg("-s") .arg("9") .arg(format!("{}", target.pid())) .succeeds(); assert_eq!(target.wait_for_signal(), Some(9)); } #[test] fn test_kill_with_signal_name_new_form() { let mut target = Target::new(); new_ucmd!() .arg("-s") .arg("KILL") .arg(format!("{}", target.pid())) .succeeds(); assert_eq!(target.wait_for_signal(), Some(libc::SIGKILL)); }