Merge pull request #3541 from plaflamme/fix-argenum-defaults

fix: multiple default values combined with `ArgEnum` / `PossibleValues`
This commit is contained in:
Ed Page 2022-03-07 10:39:08 -06:00 committed by GitHub
commit 8170754287
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 59 additions and 9 deletions

View file

@ -675,15 +675,29 @@ fn assert_defaults<'d>(
for default_os in defaults { for default_os in defaults {
if let Some(default_s) = default_os.to_str() { if let Some(default_s) = default_os.to_str() {
if !arg.possible_vals.is_empty() { if !arg.possible_vals.is_empty() {
assert!( if let Some(delim) = arg.get_value_delimiter() {
arg.possible_vals.iter().any(|possible_val| { for part in default_s.split(delim) {
possible_val.matches(default_s, arg.is_ignore_case_set()) assert!(
}), arg.possible_vals.iter().any(|possible_val| {
"Argument `{}`'s {}={} doesn't match possible values", possible_val.matches(part, arg.is_ignore_case_set())
arg.name, }),
field, "Argument `{}`'s {}={} doesn't match possible values",
default_s arg.name,
); field,
part
)
}
} else {
assert!(
arg.possible_vals.iter().any(|possible_val| {
possible_val.matches(default_s, arg.is_ignore_case_set())
}),
"Argument `{}`'s {}={} doesn't match possible values",
arg.name,
field,
default_s
);
}
} }
if let Some(validator) = arg.validator.as_ref() { if let Some(validator) = arg.validator.as_ref() {

View file

@ -457,3 +457,39 @@ fn option_vec_type() {
); );
assert!(Opt::try_parse_from(&["", "-a", "fOo"]).is_err()); assert!(Opt::try_parse_from(&["", "-a", "fOo"]).is_err());
} }
#[test]
fn vec_type_default_value() {
#[derive(clap::ArgEnum, PartialEq, Debug, Clone)]
enum ArgChoice {
Foo,
Bar,
Baz,
}
#[derive(Parser, PartialEq, Debug)]
struct Opt {
#[clap(
arg_enum,
short,
long,
default_value = "foo,bar",
value_delimiter = ','
)]
arg: Vec<ArgChoice>,
}
assert_eq!(
Opt {
arg: vec![ArgChoice::Foo, ArgChoice::Bar]
},
Opt::try_parse_from(&[""]).unwrap()
);
assert_eq!(
Opt {
arg: vec![ArgChoice::Foo, ArgChoice::Baz]
},
Opt::try_parse_from(&["", "-a", "foo,baz"]).unwrap()
);
}