// Copyright 2018 David Roundy <roundyd@physics.oregonstate.edu>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

#[macro_use]
extern crate auto_args;

use auto_args::AutoArgs;

#[test]
fn required_option() {
    #[derive(AutoArgs, PartialEq, Debug)]
    struct Opt {
        arg: i32,
    }
    #[derive(AutoArgs, PartialEq, Debug)]
    struct SuperOpt {
        arg: Opt,
        other: String,
    }
    println!("help: {}", SuperOpt::help());
    assert!(SuperOpt::help().contains("--arg"));
    assert!(SuperOpt::help().contains("--arg-arg"));

    assert_eq!(
        SuperOpt {
            arg: Opt { arg: 7 },
            other: "hello".to_string()
        },
        SuperOpt::from_iter(&["", "--arg-arg", "7", "--other", "hello"]).unwrap()
    );

    assert!(SuperOpt::from_iter(&["", "--arg"]).is_err());
}

#[test]
fn required_option_with_flattened_name() {
    #[derive(AutoArgs, PartialEq, Debug)]
    struct Opt {
        arg: i32,
    }
    #[derive(AutoArgs, PartialEq, Debug)]
    struct SuperOpt {
        _arg: Opt,
        other: String,
    }
    println!("help: {}", SuperOpt::help());
    assert!(SuperOpt::help().contains("--arg "));

    assert_eq!(
        SuperOpt {
            _arg: Opt { arg: 7 },
            other: "hello".to_string()
        },
        SuperOpt::from_iter(&["", "--arg", "7", "--other", "hello"]).unwrap()
    );
}

#[test]
fn optional_option() {
    #[derive(AutoArgs, PartialEq, Debug)]
    struct Foo {
        arg1: u32,
        arg2: i32,
    }
    #[derive(AutoArgs, PartialEq, Debug)]
    struct SuperOpt {
        _arg: Option<Foo>,
        other: String,
    }
    println!("help: {}", SuperOpt::help());
    assert!(SuperOpt::help().contains("--arg1 "));
    assert!(SuperOpt::help().contains("--arg2 "));

    assert_eq!(
        SuperOpt {
            _arg: Some(Foo { arg1: 37, arg2: -3 }),
            other: "hello".to_string()
        },
        SuperOpt::from_iter(&["", "--arg1", "37", "--arg2=-3", "--other", "hello"]).unwrap()
    );

    assert_eq!(
        SuperOpt {
            _arg: None,
            other: "hello".to_string()
        },
        SuperOpt::from_iter(&["", "--other", "hello"]).unwrap()
    );
}