tests: Extend ch-remote helper to support optional single argument

Signed-off-by: Rob Bradford <robert.bradford@intel.com>
This commit is contained in:
Rob Bradford 2020-03-11 16:48:48 +00:00 committed by Sebastien Boeuf
parent 9a7d9c9465
commit 5c3ce9dd00

View File

@ -462,12 +462,15 @@ mod tests {
assert!(status.success()); assert!(status.success());
} }
fn remote_command(api_socket: &str, command: &str) -> bool { fn remote_command(api_socket: &str, command: &str, arg: Option<&str>) -> bool {
Command::new("target/release/ch-remote") let mut cmd = Command::new("target/release/ch-remote");
.args(&[&format!("--api-socket={}", api_socket), command]) cmd.args(&[&format!("--api-socket={}", api_socket), command]);
.status()
.expect("Failed to launch ch-remote") if let Some(arg) = arg {
.success() cmd.arg(arg);
}
cmd.status().expect("Failed to launch ch-remote").success()
} }
fn resize_command( fn resize_command(
@ -2794,10 +2797,10 @@ mod tests {
aver!(tb, guest.get_entropy().unwrap_or_default() >= 900); aver!(tb, guest.get_entropy().unwrap_or_default() >= 900);
// We now pause the VM // We now pause the VM
aver!(tb, remote_command(&api_socket, "pause")); aver!(tb, remote_command(&api_socket, "pause", None));
// Check pausing again fails // Check pausing again fails
aver!(tb, !remote_command(&api_socket, "pause")); aver!(tb, !remote_command(&api_socket, "pause", None));
thread::sleep(std::time::Duration::new(2, 0)); thread::sleep(std::time::Duration::new(2, 0));
@ -2814,10 +2817,10 @@ mod tests {
); );
// Resume the VM // Resume the VM
aver!(tb, remote_command(&api_socket, "resume")); aver!(tb, remote_command(&api_socket, "resume", None));
// Check resuming again fails // Check resuming again fails
aver!(tb, !remote_command(&api_socket, "resume")); aver!(tb, !remote_command(&api_socket, "resume", None));
thread::sleep(std::time::Duration::new(2, 0)); thread::sleep(std::time::Duration::new(2, 0));