/* Copyright (C) 2022 Wildfire Games.
 * This file is part of 0 A.D.
 *
 * 0 A.D. is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * 0 A.D. is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with 0 A.D.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "lib/self_test.h"

#include "ps/GameSetup/CmdLineArgs.h"

#include <array>

class TestCmdLineArgs : public CxxTest::TestSuite
{
public:
	void test_has()
	{
		constexpr std::array<const char*, 2> argv = { "program", "-test2" };
		CmdLineArgs c(argv);
		TS_ASSERT(!c.Has("test1"));
		TS_ASSERT(c.Has("test2"));
		TS_ASSERT(!c.Has("test3"));
		TS_ASSERT(!c.Has(""));
	}

	void test_get()
	{
		constexpr std::array<const char*, 5> argv =
		{
			"program", "-test1=", "--test2=x", "-test3=-y=y-", "-=z"
		};
		CmdLineArgs c(argv);
		TS_ASSERT(!c.Has("program"));
		TS_ASSERT_STR_EQUALS(c.Get("test0"), "");
		TS_ASSERT_STR_EQUALS(c.Get("test1"), "");
		TS_ASSERT_STR_EQUALS(c.Get("test2"), "x");
		TS_ASSERT_STR_EQUALS(c.Get("test3"), "-y=y-");
		TS_ASSERT_STR_EQUALS(c.Get(""), "z");
	}

	void test_multiple()
	{
		constexpr std::array<const char*, 5> argv =
		{
			"program", "-test1=one", "--test1=two", "-test2=none", "-test1=three"
		};
		CmdLineArgs c(argv);

		TS_ASSERT_STR_EQUALS(c.Get("test1"), "one");
		TS_ASSERT_STR_EQUALS(c.Get("test2"), "none");

		std::vector<CStr> test1 = c.GetMultiple("test1");
		CStr expected1[] = { "one", "two", "three" };
		TS_ASSERT_VECTOR_EQUALS_ARRAY(test1, expected1);

		std::vector<CStr> test2 = c.GetMultiple("test2");
		CStr expected2[] = { "none" };
		TS_ASSERT_VECTOR_EQUALS_ARRAY(test2, expected2);

		std::vector<CStr> test3 = c.GetMultiple("test3");
		TS_ASSERT_EQUALS(test3.size(), (size_t)0);
	}

	void test_get_invalid()
	{
		constexpr std::array<const char*, 6> argv = {
			"-test1", "--test2", "test3-", " -test4", "--", "-=="
		};
		CmdLineArgs c(argv);

		TS_ASSERT(!c.Has("test1"));
		TS_ASSERT(c.Has("test2"));
		TS_ASSERT(!c.Has("test3"));
		TS_ASSERT(!c.Has("test4"));
	}

	void test_arg0()
	{
		constexpr std::array<const char*, 1> argv = { "program" };
		CmdLineArgs c(argv);
		TS_ASSERT_WSTR_EQUALS(c.GetArg0().string(), L"program");

		CmdLineArgs c2(PS::span<const char* const>{});
		TS_ASSERT_WSTR_EQUALS(c2.GetArg0().string(), L"");

		const std::array<const char*, 1> argv3 = { "ab/cd/ef/gh/../ij" };
		CmdLineArgs c3(argv3);
#if OS_WIN
		TS_ASSERT_WSTR_EQUALS(c3.GetArg0().string(), L"ab\\cd\\ef\\gh\\..\\ij");
#else
		TS_ASSERT_WSTR_EQUALS(c3.GetArg0().string(), L"ab/cd/ef/gh/../ij");
#endif
	}

	void test_get_without_names()
	{
		constexpr std::array<const char*, 6> argv =
		{
			"program", "test0", "-test1", "test2", "test3", "--test4=test5"
		};
		CmdLineArgs c(argv);
		TS_ASSERT(c.Has("test1"));
		TS_ASSERT_STR_EQUALS(c.Get("test4"), "test5");
		CStr expected_args[] = { "test0", "test2", "test3" };
		TS_ASSERT_VECTOR_EQUALS_ARRAY(c.GetArgsWithoutName(), expected_args);
	}
};