summaryrefslogtreecommitdiff
path: root/argparse/src/test_many.rs
diff options
context:
space:
mode:
Diffstat (limited to 'argparse/src/test_many.rs')
-rw-r--r--argparse/src/test_many.rs95
1 files changed, 95 insertions, 0 deletions
diff --git a/argparse/src/test_many.rs b/argparse/src/test_many.rs
new file mode 100644
index 0000000..68e2cec
--- /dev/null
+++ b/argparse/src/test_many.rs
@@ -0,0 +1,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));
+}