clap/tests/utils.rs

#![allow(unused_imports, dead_code)]

use std::io::{Cursor, Write};
use std::str;

use regex::Regex;

use clap::{App, Arg, ArgGroup};

fn compare<S, S2>(l: S, r: S2) -> bool
where
    S: AsRef<str>,
    S2: AsRef<str>,
{
    let re = Regex::new("\x1b[^m]*m").unwrap();
    // Strip out any mismatching \r character on windows that might sneak in on either side
    let ls = l.as_ref().trim().replace("\r", "");
    let rs = r.as_ref().trim().replace("\r", "");
    let left = re.replace_all(&*ls, "");
    let right = re.replace_all(&*rs, "");
    let b = left == right;
    if !b {
        println!();
        println!("--> left");
        println!("{}", left);
        println!("--> right");
        println!("{}", right);
        println!("--")
    }
    b
}

pub fn compare_output(l: App, args: &str, right: &str, stderr: bool) -> bool {
    let mut buf = Cursor::new(Vec::with_capacity(50));
    let res = l.try_get_matches_from(args.split(' ').collect::<Vec<_>>());
    let err = res.unwrap_err();
    write!(&mut buf, "{}", err).unwrap();
    let content = buf.into_inner();
    let left = String::from_utf8(content).unwrap();
    assert_eq!(
        stderr,
        err.use_stderr(),
        "Should Use STDERR failed. Should be {} but is {}",
        stderr,
        err.use_stderr()
    );
    compare(left, right)
}

pub fn compare_output2(l: App, args: &str, right1: &str, right2: &str, stderr: bool) -> bool {
    let mut buf = Cursor::new(Vec::with_capacity(50));
    let res = l.try_get_matches_from(args.split(' ').collect::<Vec<_>>());
    let err = res.unwrap_err();
    write!(&mut buf, "{}", err).unwrap();
    let content = buf.into_inner();
    let left = String::from_utf8(content).unwrap();
    assert_eq!(
        stderr,
        err.use_stderr(),
        "Should Use STDERR failed. Should be {} but is {}",
        stderr,
        err.use_stderr()
    );
    compare(&*left, right1) || compare(&*left, right2)
}

// Legacy tests from the python script days

pub fn complex_app() -> App<'static> {
    let opt3_vals = ["fast", "slow"];
    let pos3_vals = ["vi", "emacs"];

    App::new("clap-test")
        .version("v1.4.8")
        .about("tests clap library")
        .author("Kevin K. <kbknapp@gmail.com>")
        .arg("-o --option=[opt]... 'tests options'")
        .arg("[positional] 'tests positionals'")
        .arg(Arg::from("-f --flag... 'tests flags'").global(true))
        .args(&[
            Arg::from("[flag2] -F 'tests flags with exclusions'")
                .conflicts_with("flag")
                .requires("long-option-2"),
            Arg::from("--long-option-2 [option2] 'tests long options with exclusions'")
                .conflicts_with("option")
                .requires("positional2"),
            Arg::from("[positional2] 'tests positionals with exclusions'"),
            Arg::from("-O --Option [option3] 'specific vals'").possible_values(&opt3_vals),
            Arg::from("[positional3]... 'tests specific values'").possible_values(&pos3_vals),
            Arg::from("--multvals [one] [two] 'Tests multiple values, not mult occs'"),
            Arg::from("--multvalsmo... [one] [two] 'Tests multiple values, and mult occs'"),
            Arg::from("--minvals2 [minvals]... 'Tests 2 min vals'").min_values(2),
            Arg::from("--maxvals3 [maxvals]... 'Tests 3 max vals'").max_values(3),
        ])
        .subcommand(
            App::new("subcmd")
                .about("tests subcommands")
                .version("0.1")
                .author("Kevin K. <kbknapp@gmail.com>")
                .arg("-o --option [scoption]... 'tests options'")
                .arg("-s --subcmdarg [subcmdarg] 'tests other args'")
                .arg("[scpositional] 'tests positionals'"),
        )
}