[go: up one dir, main page]

structopt 0.1.3

Parse command line argument by defining a struct.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

// 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.

extern crate structopt;
#[macro_use]
extern crate structopt_derive;

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"])));
    assert_eq!(Opt { alice: true },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-a"])));
    assert_eq!(Opt { alice: true },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "--alice"])));
    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")]
        alice: u64,
    }

    assert_eq!(Opt { alice: 0 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test"])));
    assert_eq!(Opt { alice: 1 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-a"])));
    assert_eq!(Opt { alice: 2 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-a", "-a"])));
    assert_eq!(Opt { alice: 2 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-a", "--alice"])));
    assert_eq!(Opt { alice: 3 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-aaa"])));
    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")]
        bob: u64,
    }

    assert_eq!(Opt { alice: false, bob: 0 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test"])));
    assert_eq!(Opt { alice: true, bob: 0 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-a"])));
    assert_eq!(Opt { alice: true, bob: 0 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-a"])));
    assert_eq!(Opt { alice: false, bob: 1 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-b"])));
    assert_eq!(Opt { alice: true, bob: 1 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "--alice",  "--bob"])));
    assert_eq!(Opt { alice: true, bob: 4 },
               Opt::from_clap(Opt::clap().get_matches_from(&["test", "-bb", "-a", "-bb"])));
}