clap/tests/flags.rs

80 lines
3.2 KiB
Rust
Raw Normal View History

2017-02-07 00:41:16 +00:00
// Copyright (c) 2017 Guillaume Pinot <texitoi(a)texitoi.eu>
//
// This work is free. You can redistribute it and/or modify it under
// the terms of the Do What The Fuck You Want To Public License,
// Version 2, as published by Sam Hocevar. See the COPYING file for
// more details.
#[macro_use]
extern crate structopt;
2017-02-07 00:41:16 +00:00
use structopt::StructOpt;
#[test]
fn unique_flag() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
#[structopt(short = "a", long = "alice")]
alice: bool,
}
assert_eq!(Opt { alice: false },
Opt::from_clap(&Opt::clap().get_matches_from(&["test"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: true },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-a"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: true },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "--alice"])));
2017-02-07 00:41:16 +00:00
assert!(Opt::clap().get_matches_from_safe(&["test", "-i"]).is_err());
assert!(Opt::clap().get_matches_from_safe(&["test", "-a", "foo"]).is_err());
assert!(Opt::clap().get_matches_from_safe(&["test", "-a", "-a"]).is_err());
assert!(Opt::clap().get_matches_from_safe(&["test", "-a", "--alice"]).is_err());
}
#[test]
fn multiple_flag() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
#[structopt(short = "a", long = "alice", parse(from_occurrences))]
2017-02-07 00:41:16 +00:00
alice: u64,
#[structopt(short = "b", long = "bob", parse(from_occurrences))]
bob: u8,
2017-02-07 00:41:16 +00:00
}
assert_eq!(Opt { alice: 0, bob: 0 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test"])));
assert_eq!(Opt { alice: 1, bob: 0 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-a"])));
assert_eq!(Opt { alice: 2, bob: 0 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-a", "-a"])));
assert_eq!(Opt { alice: 2, bob: 2 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-a", "--alice", "-bb"])));
assert_eq!(Opt { alice: 3, bob: 1 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-aaa", "--bob"])));
2017-02-07 00:41:16 +00:00
assert!(Opt::clap().get_matches_from_safe(&["test", "-i"]).is_err());
assert!(Opt::clap().get_matches_from_safe(&["test", "-a", "foo"]).is_err());
}
#[test]
fn combined_flags() {
#[derive(StructOpt, PartialEq, Debug)]
struct Opt {
#[structopt(short = "a", long = "alice")]
alice: bool,
#[structopt(short = "b", long = "bob", parse(from_occurrences))]
2017-02-07 00:41:16 +00:00
bob: u64,
}
assert_eq!(Opt { alice: false, bob: 0 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: true, bob: 0 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-a"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: true, bob: 0 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-a"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: false, bob: 1 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-b"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: true, bob: 1 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "--alice", "--bob"])));
2017-02-07 00:41:16 +00:00
assert_eq!(Opt { alice: true, bob: 4 },
Opt::from_clap(&Opt::clap().get_matches_from(&["test", "-bb", "-a", "-bb"])));
2017-02-07 00:41:16 +00:00
}