/*
 * .============.
 * // M A K E / \
 * // C++ DEV / \
 * // E A S Y / \/ \
 * ++ ----------. \/\ .
 * \\ \ \ /\ /
 * \\ \ \ /
 * \\ \ \ /
 * -============'
 *
 * Copyright (c) 2018 Hevake and contributors, all rights reserved.
 *
 * This file is part of cpp-tbox (https://github.com/cpp-main/cpp-tbox)
 * Use of this source code is governed by MIT license that can be found
 * in the LICENSE file in the root of the source tree. All contributing
 * project authors may be found in the CONTRIBUTORS.md file in the root
 * of the source tree.
 */
#include <gtest/gtest.h>

#include <tbox/base/defines.h>
#include "argument_parser.h"

using namespace tbox::util;

TEST(ArgumentParser, Fail)
{
 const char* argv[] = {"test_fail", "-x"};
 ArgumentParser parser(
 [&](char short_opt, const std::string &long_opt, ArgumentParser::OptionValue& opt_value) {
 (void)short_opt;
 (void)long_opt;
 (void)opt_value;
 return false;
 }
 );

 ASSERT_FALSE(parser.parse(NUMBER_OF_ARRAY(argv), argv));
}

TEST(ArgumentParser, LongOptionKeyAndValue)
{
 const char* argv[] = {"test_long_opt_key_and_value", "--key=123", "-n"};
 ArgumentParser parser(
 [&](char short_opt, const std::string &long_opt, ArgumentParser::OptionValue& opt_value) {
 if (short_opt == 0) {
 EXPECT_EQ(long_opt, "key");
 EXPECT_TRUE(opt_value.valid());
 EXPECT_EQ(opt_value.get(), "123");
 } else {
 EXPECT_EQ(short_opt, 'n');
 EXPECT_FALSE(opt_value.valid());
 }
 return true;
 }
 );
 ASSERT_TRUE(parser.parse(NUMBER_OF_ARRAY(argv), argv));
}

TEST(ArgumentParser, LongOptionKeyAndValueWithQuot)
{
 const char* argv[] = {"any", R"(--name='Hevake Lee')", R"(--address="Shenzhen China")"};
 bool has_name = false;
 bool has_address = false;
 ArgumentParser parser(
 [&](char short_opt, const std::string &long_opt, ArgumentParser::OptionValue& opt_value) {
 if (long_opt == "name") {
 EXPECT_TRUE(opt_value.valid());
 EXPECT_EQ(opt_value.get(), "Hevake Lee");
 has_name = true;
 } else if (long_opt == "address"){
 EXPECT_TRUE(opt_value.valid());
 EXPECT_EQ(opt_value.get(), "Shenzhen China");
 has_address = true;
 }
 (void)short_opt;
 return true;
 }
 );
 ASSERT_TRUE(parser.parse(NUMBER_OF_ARRAY(argv), argv));
 EXPECT_TRUE(has_name);
 EXPECT_TRUE(has_address);
}

TEST(ArgumentParser, All)
{
 const char* argv[] = {"test_all", "-ab", "123", "-x", "--set", "xyz", "--run", "--key=value"};
 ArgumentParser parser(
 [&](char short_opt, const std::string &long_opt, ArgumentParser::OptionValue& opt_value) {
 if (short_opt == 'a') {
 EXPECT_FALSE(opt_value.valid());
 } else if (short_opt == 'b') {
 EXPECT_TRUE(opt_value.valid());
 EXPECT_EQ(opt_value.get(), "123");
 } else if (short_opt == 'x') {
 EXPECT_TRUE(opt_value.valid());
 } else if (long_opt == "set") {
 EXPECT_TRUE(opt_value.valid());
 EXPECT_EQ(opt_value.get(), "xyz");
 } else if (long_opt == "run") {
 EXPECT_TRUE(opt_value.valid());
 } else if (long_opt == "key") {
 EXPECT_TRUE(opt_value.valid());
 EXPECT_EQ(opt_value.get(), "value");
 } else {
 return false;
 }
 return true;
 }
 );

 ASSERT_TRUE(parser.parse(NUMBER_OF_ARRAY(argv), argv));
}