clap/tests/arg_enum_basic.rs
2018-07-02 13:41:56 -04:00

54 lines
1.5 KiB
Rust

// Copyright 2018 Guillaume Pinot (@TeXitoi) <texitoi@texitoi.eu>,
// Kevin Knapp (@kbknapp) <kbknapp@gmail.com>, and
// Andrew Hobden (@hoverbear) <andrew@hoverbear.org>
//
// 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 clap;
#[macro_use]
extern crate clap_derive;
use clap::{App, Arg};
#[derive(ArgEnum, Debug, PartialEq)]
enum ArgChoice {
Foo,
Bar,
Baz,
}
#[test]
fn when_lowercase() {
let matches = App::new(env!("CARGO_PKG_NAME"))
.arg(
Arg::with_name("arg")
.required(true)
.takes_value(true)
.possible_values(&ArgChoice::variants()),
)
.get_matches_from_safe(vec!["", "foo"])
.unwrap();
let t = value_t!(matches.value_of("arg"), ArgChoice);
assert!(t.is_ok());
assert_eq!(t.unwrap(), ArgChoice::Foo);
}
#[test]
fn when_capitalized() {
let matches = App::new(env!("CARGO_PKG_NAME"))
.arg(
Arg::with_name("arg")
.required(true)
.takes_value(true)
.possible_values(&ArgChoice::variants()),
)
.get_matches_from_safe(vec!["", "Foo"])
.unwrap();
let t = value_t!(matches.value_of("arg"), ArgChoice);
assert!(t.is_ok());
assert_eq!(t.unwrap(), ArgChoice::Foo);
}