clap/tests/builder/validators.rs

#![allow(deprecated)]

use clap::{Arg, Command};

#[cfg(debug_assertions)]
#[test]
#[should_panic = "Argument 'test' has both `validator` and `validator_os` set which is not allowed"]
fn both_validator_and_validator_os() {
    let _ = Command::new("test")
        .arg(
            Arg::new("test")
                .validator(|val| val.parse::<u32>().map_err(|e| e.to_string()))
                .validator_os(|val| {
                    val.to_str()
                        .unwrap()
                        .parse::<u32>()
                        .map_err(|e| e.to_string())
                }),
        )
        .try_get_matches_from(&["cmd", "1"]);
}

#[test]
fn test_validator_fromstr_trait() {
    use std::str::FromStr;

    let matches = Command::new("test")
        .arg(Arg::new("from_str").validator(u32::from_str))
        .try_get_matches_from(&["cmd", "1234"])
        .expect("match failed");

    assert_eq!(
        matches.get_one::<String>("from_str"),
        Some(&"1234".to_owned())
    );
}

#[test]
fn test_validator_msg_newline() {
    let res = Command::new("test")
        .arg(Arg::new("test").validator(|val| val.parse::<u32>().map_err(|e| e.to_string())))
        .try_get_matches_from(&["cmd", "f"]);

    assert!(res.is_err());
    let err = res.unwrap_err();

    assert!(
        err.to_string()
            .contains("Invalid value \"f\" for '<test>': invalid digit found in string"),
        "{}",
        err
    );

    // This message is the only thing that gets printed -- make sure it ends with a newline
    let msg = format!("{}", err);
    assert!(msg.ends_with('\n'));
}

#[test]
fn stateful_validator() {
    let mut state = false;
    Command::new("test")
        .arg(Arg::new("test").validator(|val| {
            state = true;
            val.parse::<u32>().map_err(|e| e.to_string())
        }))
        .try_get_matches_from(&["cmd", "10"])
        .unwrap();

    assert!(state);
}