clap/tests/arguments.rs
Brandon W Maister ccb04a30cc Add StructOpt::from_iter_safe method (#98)
I would like to be able to test argument parsing, which requires an error
value, not a `process::exit()`.
2018-04-20 15:02:12 +02:00

87 lines
2.7 KiB
Rust

// Copyright 2018 Guillaume Pinot (@TeXitoi) <texitoi@texitoi.eu>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
#[macro_use]
extern crate structopt;
use structopt::StructOpt;
use structopt::clap;
#[test]
fn required_argument() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
arg: i32,
}
assert_eq!(Opt { arg: 42 }, Opt::from_iter(&["test", "42"]));
assert!(Opt::clap().get_matches_from_safe(&["test"]).is_err());
assert!(Opt::clap().get_matches_from_safe(&["test", "42", "24"]).is_err());
}
#[test]
fn optional_argument() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
arg: Option<i32>,
}
assert_eq!(Opt { arg: Some(42) }, Opt::from_iter(&["test", "42"]));
assert_eq!(Opt { arg: None }, Opt::from_iter(&["test"]));
assert!(Opt::clap().get_matches_from_safe(&["test", "42", "24"]).is_err());
}
#[test]
fn argument_with_default() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
#[structopt(default_value = "42")]
arg: i32,
}
assert_eq!(Opt { arg: 24 }, Opt::from_iter(&["test", "24"]));
assert_eq!(Opt { arg: 42 }, Opt::from_iter(&["test"]));
assert!(Opt::clap().get_matches_from_safe(&["test", "42", "24"]).is_err());
}
#[test]
fn argument_with_raw_default() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
#[structopt(raw(default_value = r#""42""#))]
arg: i32,
}
assert_eq!(Opt { arg: 24 }, Opt::from_iter(&["test", "24"]));
assert_eq!(Opt { arg: 42 }, Opt::from_iter(&["test"]));
assert!(Opt::clap().get_matches_from_safe(&["test", "42", "24"]).is_err());
}
#[test]
fn arguments() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
arg: Vec<i32>,
}
assert_eq!(Opt { arg: vec![24] }, Opt::from_iter(&["test", "24"]));
assert_eq!(Opt { arg: vec![] }, Opt::from_iter(&["test"]));
assert_eq!(Opt { arg: vec![24, 42] }, Opt::from_iter(&["test", "24", "42"]));
}
#[test]
fn arguments_safe() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
arg: Vec<i32>,
}
assert_eq!(Opt { arg: vec![24] }, Opt::from_iter_safe(&["test", "24"]).unwrap());
assert_eq!(Opt { arg: vec![] }, Opt::from_iter_safe(&["test"]).unwrap());
assert_eq!(Opt { arg: vec![24, 42] }, Opt::from_iter_safe(&["test", "24", "42"]).unwrap());
assert_eq!(
clap::ErrorKind::ValueValidation,
Opt::from_iter_safe(&["test", "NOPE"]).err().unwrap().kind,
);
}