sleep: Fix whitespace issues. Don't panic when adding durations.

Refactor error propagation and use show_error! instead of USimpleError.
This commit is contained in:
Joining7943 2023-02-28 07:21:18 +01:00
parent 68f75af678
commit 71bbebdc76
2 changed files with 55 additions and 10 deletions

View file

@ -10,7 +10,7 @@ use std::time::Duration;
use uucore::{ use uucore::{
error::{UResult, USimpleError, UUsageError}, error::{UResult, USimpleError, UUsageError},
format_usage, help_about, help_section, help_usage, show, format_usage, help_about, help_section, help_usage, show_error,
}; };
use clap::{crate_version, Arg, ArgAction, Command}; use clap::{crate_version, Arg, ArgAction, Command};
@ -61,15 +61,17 @@ pub fn uu_app() -> Command {
fn sleep(args: &[&str]) -> UResult<()> { fn sleep(args: &[&str]) -> UResult<()> {
let mut arg_error = false; let mut arg_error = false;
let intervals = args.iter().map(|s| match uucore::parse_time::from_str(s) { let sleep_dur = args
Ok(result) => result, .iter()
Err(err) => { .filter_map(|input| {
arg_error = true; uucore::parse_time::from_str(input.trim()).ok().or_else(|| {
show!(USimpleError::new(1, err)); arg_error = true;
Duration::new(0, 0) show_error!("invalid time interval '{input}'");
} None
}); })
let sleep_dur = intervals.fold(Duration::ZERO, |acc, n| acc.saturating_add(n)); })
.fold(Duration::ZERO, |acc, n| acc.saturating_add(n));
if arg_error { if arg_error {
return Err(UUsageError::new(1, "")); return Err(UUsageError::new(1, ""));
}; };

View file

@ -1,3 +1,5 @@
use rstest::rstest;
// spell-checker:ignore dont // spell-checker:ignore dont
use crate::common::util::*; use crate::common::util::*;
@ -180,6 +182,47 @@ fn test_sleep_when_multiple_inputs_exceed_max_duration_then_no_error() {
.no_output(); .no_output();
} }
#[rstest]
#[case::whitespace_prefix(" 0.1s")]
#[case::multiple_whitespace_prefix(" 0.1s")]
#[case::whitespace_suffix("0.1s ")]
#[case::mixed_newlines_spaces_tabs("\n\t0.1s \n ")]
fn test_sleep_when_input_has_whitespace_then_no_error(#[case] input: &str) {
new_ucmd!()
.arg(input)
.timeout(Duration::from_secs(10))
.succeeds()
.no_output();
}
#[rstest]
#[case::only_space(" ")]
#[case::only_tab("\t")]
#[case::only_newline("\n")]
fn test_sleep_when_input_has_only_whitespace_then_error(#[case] input: &str) {
new_ucmd!()
.arg(input)
.timeout(Duration::from_secs(10))
.fails()
.usage_error(format!("invalid time interval '{input}'"));
}
#[test]
fn test_sleep_when_multiple_input_some_with_error_then_shows_all_errors() {
let expected = "invalid time interval 'abc'\n\
sleep: invalid time interval '1years'\n\
sleep: invalid time interval ' '";
// Even if one of the arguments is valid, but the rest isn't, we should still fail and exit early.
// So, the timeout of 10 seconds ensures we haven't executed `thread::sleep` with the only valid
// interval of `100000.0` seconds.
new_ucmd!()
.args(&["abc", "100000.0", "1years", " "])
.timeout(Duration::from_secs(10))
.fails()
.usage_error(expected);
}
#[test] #[test]
fn test_negative_interval() { fn test_negative_interval() {
new_ucmd!() new_ucmd!()