mirror of
https://github.com/uutils/coreutils
synced 2024-12-13 23:02:38 +00:00
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:
parent
68f75af678
commit
71bbebdc76
2 changed files with 55 additions and 10 deletions
|
@ -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, ""));
|
||||||
};
|
};
|
||||||
|
|
|
@ -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!()
|
||||||
|
|
Loading…
Reference in a new issue