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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
use parser::ArgumentParser;
use super::{List, Store, Collect};
use test_parser::{check_ok};
fn pos_list(args: &[&str]) -> (isize, Vec<isize>) {
let mut val1 = 1;
let mut val2 = Vec::new();
{
let mut ap = ArgumentParser::new();
ap.refer(&mut val1).add_argument("v1", Store, "The value 1");
ap.refer(&mut val2).add_argument("v2", List, "The list of vals");
check_ok(&ap, args);
}
return (val1, val2);
}
#[test]
fn test_pos_list() {
assert_eq!(pos_list(&["./argparse_test", "10"]), (10, vec!()));
assert_eq!(pos_list(&["./argparse_test", "11", "21"]), (11, vec!(21)));
assert_eq!(pos_list(&["./argparse_test", "10", "20", "30"]),
(10, vec!(20, 30)));
}
fn pos_collect(args: &[&str]) -> Vec<isize> {
let mut lst = Vec::new();
{
let mut ap = ArgumentParser::new();
ap.refer(&mut lst)
.add_argument("v", Collect, "The list of vals");
check_ok(&ap, args);
}
return lst;
}
#[test]
fn test_pos_collect() {
assert_eq!(pos_collect(&["./argparse_test", "10"]), vec!(10));
assert_eq!(pos_collect(&["./argparse_test", "11", "21"]), vec!(11, 21));
assert_eq!(pos_collect(&["./argparse_test", "10", "20", "30"]),
vec!(10, 20, 30));
}
#[test]
#[should_panic]
fn wrong_type() {
pos_collect(&["./argparse_test", "10", "20", "test"]);
}
fn collect(args: &[&str]) -> Vec<isize> {
let mut lst = Vec::new();
{
let mut ap = ArgumentParser::new();
ap.refer(&mut lst).add_option(&["-a", "--add"], Collect,
"The list of vals");
check_ok(&ap, args);
}
return lst;
}
#[test]
fn test_collect() {
assert_eq!(collect(&["./argparse_test", "-a10"]), vec!(10));
assert_eq!(collect(&["./argparse_test", "--add=11", "-a", "21"]),
vec!(11, 21));
assert_eq!(collect(&["./argparse_test",
"-a", "10", "--add=20", "--add", "30"]), vec!(10, 20, 30));
}
#[test]
#[should_panic]
fn test_extra() {
collect(&["./argparse_test", "-a", "10", "20", "30"]);
}
fn list(args: &[&str]) -> Vec<isize> {
let mut vec = Vec::new();
{
let mut ap = ArgumentParser::new();
ap.refer(&mut vec).add_option(&["-a", "--add"], List,
"The list of vals");
check_ok(&ap, args);
}
return vec;
}
#[test]
#[should_panic]
fn test_list() {
assert_eq!(list(&["./argparse_test", "-a10"]), vec!(10));
assert_eq!(list(&["./argparse_test", "--add", "11", "21"]), vec!(11, 21));
assert_eq!(list(&["./argparse_test", "-a", "10", "20", "30"]),
vec!(10, 20, 30));
assert_eq!(list(&["./argparse_test", "10", "20", "30"]), vec!(10, 20, 30));
}
|
