mirror of
https://github.com/clap-rs/clap
synced 2024-12-14 23:02:31 +00:00
fc359e3c46
The example code.
~~~rust
use clap::{App, Arg};
fn main() {
let matches = App::new("My Super Program")
.arg(
Arg::with_name("verbose")
.help("Sets the level of verbosity")
.short('v')
.long("verbose")
.takes_value(false)
.multiple_occurrences(true)
.env("VERBOSE"),
)
.get_matches();
match matches.occurrences_of("verbose") {
0 => println!("0 No verbose info"),
1 => println!("1 Some verbose info"),
2 => println!("2 Tons of verbose info"),
3 | _ => println!("3 >= Don't be crazy"),
}
}
~~~
It code use multiple_occurrences with env.
But it do not work.
`env` method set require take value.
It result see under.
~~~console
% cargo run -- -v
Finished dev [unoptimized + debuginfo] target(s) in 0.01s
Running `target/debug/foo -v`
error: The argument '--verbose <verbose>...' requires a value but none was supplied
USAGE:
foo [OPTIONS]
For more information try --help
~~~
And, structopt or clap_derive may be create similar code.
So I am confused by structopt.
This to fix code small.
132 lines
4.4 KiB
Rust
132 lines
4.4 KiB
Rust
extern crate clap;
|
|
|
|
use clap::{App, Arg, ArgSettings};
|
|
|
|
#[test]
|
|
fn multiple_occurrences_of_flags_long() {
|
|
let m = App::new("mo_flags_long")
|
|
.arg(
|
|
Arg::from("--multflag 'allowed multiple flag'")
|
|
.setting(ArgSettings::MultipleOccurrences),
|
|
)
|
|
.arg(Arg::from("--flag 'disallowed multiple flag'"))
|
|
.get_matches_from(vec!["", "--multflag", "--flag", "--multflag"]);
|
|
assert!(m.is_present("multflag"));
|
|
assert_eq!(m.occurrences_of("multflag"), 2);
|
|
assert!(m.is_present("flag"));
|
|
assert_eq!(m.occurrences_of("flag"), 1)
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_occurrences_of_flags_short() {
|
|
let m = App::new("mo_flags_short")
|
|
.arg(
|
|
Arg::from("-m --multflag 'allowed multiple flag'")
|
|
.setting(ArgSettings::MultipleOccurrences),
|
|
)
|
|
.arg(Arg::from("-f --flag 'disallowed multiple flag'"))
|
|
.get_matches_from(vec!["", "-m", "-f", "-m"]);
|
|
assert!(m.is_present("multflag"));
|
|
assert_eq!(m.occurrences_of("multflag"), 2);
|
|
assert!(m.is_present("flag"));
|
|
assert_eq!(m.occurrences_of("flag"), 1);
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_occurrences_of_flags_mixed() {
|
|
let m = App::new("mo_flags_mixed")
|
|
.arg(
|
|
Arg::from("-m, --multflag1 'allowed multiple flag'")
|
|
.setting(ArgSettings::MultipleOccurrences),
|
|
)
|
|
.arg(
|
|
Arg::from("-n, --multflag2 'another allowed multiple flag'")
|
|
.setting(ArgSettings::MultipleOccurrences),
|
|
)
|
|
.arg(Arg::from("-f, --flag 'disallowed multiple flag'"))
|
|
.get_matches_from(vec![
|
|
"",
|
|
"-m",
|
|
"-f",
|
|
"-n",
|
|
"--multflag1",
|
|
"-m",
|
|
"--multflag2",
|
|
]);
|
|
assert!(m.is_present("multflag1"));
|
|
assert_eq!(m.occurrences_of("multflag1"), 3);
|
|
assert!(m.is_present("multflag2"));
|
|
assert_eq!(m.occurrences_of("multflag2"), 2);
|
|
assert!(m.is_present("flag"));
|
|
assert_eq!(m.occurrences_of("flag"), 1);
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_occurrences_of_flags_large_quantity() {
|
|
let args: Vec<&str> = vec![""]
|
|
.into_iter()
|
|
.chain(vec!["-m"; 1024].into_iter())
|
|
.collect();
|
|
let m = App::new("mo_flags_larg_qty")
|
|
.arg(
|
|
Arg::from("-m --multflag 'allowed multiple flag'")
|
|
.setting(ArgSettings::MultipleOccurrences),
|
|
)
|
|
.get_matches_from(args);
|
|
assert!(m.is_present("multflag"));
|
|
assert_eq!(m.occurrences_of("multflag"), 1024);
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_occurrences_of_before_env() {
|
|
let app = App::new("mo_before_env").arg(
|
|
Arg::with_name("verbose")
|
|
.env("VERBOSE")
|
|
.short('v')
|
|
.long("verbose")
|
|
.takes_value(false)
|
|
.multiple_occurrences(true),
|
|
);
|
|
|
|
let m = app.clone().try_get_matches_from(vec![""]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 0);
|
|
|
|
let m = app.clone().try_get_matches_from(vec!["", "-v"]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 1);
|
|
|
|
let m = app.clone().try_get_matches_from(vec!["", "-vv"]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 2);
|
|
let m = app.clone().try_get_matches_from(vec!["", "-vvv"]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 3);
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_occurrences_of_after_env() {
|
|
let app = App::new("mo_after_env").arg(
|
|
Arg::with_name("verbose")
|
|
.short('v')
|
|
.long("verbose")
|
|
.takes_value(false)
|
|
.multiple_occurrences(true)
|
|
.env("VERBOSE"),
|
|
);
|
|
|
|
let m = app.clone().try_get_matches_from(vec![""]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 0);
|
|
|
|
let m = app.clone().try_get_matches_from(vec!["", "-v"]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 1);
|
|
|
|
let m = app.clone().try_get_matches_from(vec!["", "-vv"]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 2);
|
|
let m = app.clone().try_get_matches_from(vec!["", "-vvv"]);
|
|
assert!(m.is_ok(), "{}", m.unwrap_err().message);
|
|
assert_eq!(m.unwrap().occurrences_of("verbose"), 3);
|
|
}
|