clap/tests/derive/occurrences.rs

#![cfg(feature = "unstable-grouped")]
use clap::Parser;

#[test]
fn test_vec_of_vec() {
    #[derive(Parser, Debug, PartialEq)]
    struct Opt {
        #[arg(short = 'p', num_args = 2)]
        points: Vec<Vec<i32>>,
    }

    assert_eq!(
        Opt {
            points: vec![vec![1, 2], vec![0, 0]]
        },
        Opt::try_parse_from(&["test", "-p", "1", "2", "-p", "0", "0"]).unwrap()
    );
}

#[test]
fn test_vec_of_vec_opt_out() {
    fn parser(s: &str) -> Result<Vec<Vec<String>>, std::convert::Infallible> {
        Ok(s.split(':')
            .map(|v| v.split(',').map(str::to_owned).collect())
            .collect())
    }

    #[derive(Parser, PartialEq, Debug)]
    struct Opt {
        #[arg(value_parser = parser, short = 'p')]
        arg: ::std::vec::Vec<Vec<String>>,
    }

    assert_eq!(
        Opt {
            arg: vec![vec!["1".into(), "2".into()], vec!["a".into(), "b".into()]],
        },
        Opt::try_parse_from(["test", "-p", "1,2:a,b"]).unwrap(),
    );
}

#[test]
fn test_vec_vec_empty() {
    #[derive(Parser, Debug, PartialEq)]
    struct Opt {
        #[arg(short = 'p', num_args = 2)]
        points: Vec<Vec<i32>>,
    }

    assert_eq!(
        Opt { points: vec![] },
        Opt::try_parse_from(&["test"]).unwrap()
    );
}

#[test]
fn test_option_vec_vec() {
    #[derive(Parser, Debug, PartialEq)]
    struct Opt {
        #[arg(short = 'p', num_args = 2)]
        points: Option<Vec<Vec<i32>>>,
    }

    assert_eq!(
        Opt {
            points: Some(vec![vec![1, 2], vec![3, 4]])
        },
        Opt::try_parse_from(&["test", "-p", "1", "2", "-p", "3", "4"]).unwrap()
    );
}

#[test]
fn test_option_vec_vec_empty() {
    #[derive(Parser, Debug, PartialEq)]
    struct Opt {
        #[arg(short = 'p', num_args = 2)]
        points: Option<Vec<Vec<i32>>>,
    }

    assert_eq!(
        Opt { points: None },
        Opt::try_parse_from(&["test"]).unwrap()
    );
}