#include "catch.hpp"

#include "DualLookup.h"

#include <string>

TEST_CASE("all tests", "all") {
	DualLookup<int> container;
	DualLookup<std::string> string_container;

	SECTION ("allow add") {
		REQUIRE(container.add(1, 2));
		REQUIRE(string_container.add("1", "2"));

		SECTION ("no duplicates") {
			REQUIRE_FALSE(container.add(1, 2));
			REQUIRE_FALSE(container.add(2, 1));
			REQUIRE_FALSE(container.add(1, 3));
			REQUIRE_FALSE(container.add(2, 3));
			REQUIRE_FALSE(container.add(3, 1));
			REQUIRE_FALSE(container.add(3, 2));
			REQUIRE_FALSE(string_container.add("1", "2"));
			REQUIRE_FALSE(string_container.add("2", "1"));
			REQUIRE_FALSE(string_container.add("1", "3"));
			REQUIRE_FALSE(string_container.add("2", "3"));
			REQUIRE_FALSE(string_container.add("3", "1"));
			REQUIRE_FALSE(string_container.add("3", "2"));
		}

		SECTION ("searches") {
			bool        rtn;
			int         container_result;
			std::string string_container_result;

			SECTION ("found and not found") {
				REQUIRE(container.contains(1));
				REQUIRE(string_container.contains("1"));
				REQUIRE_FALSE(container.contains(10));
				REQUIRE_FALSE(string_container.contains("10"));
			}

			SECTION ("not found in container and unedited") {
				container_result = 10;
				string_container_result = "10";
				REQUIRE_FALSE(container.get(20, container_result));
				REQUIRE(container_result == 10);
				REQUIRE_FALSE(string_container.get("20", string_container_result));
				REQUIRE(string_container_result == "10");
			}

			SECTION ("in container and set") {
				container_result = 0;
				REQUIRE(container.get(1, container_result));
				REQUIRE(container_result == 2);
				REQUIRE(string_container.get("1", string_container_result));
				REQUIRE(string_container_result == "2");
			}
		}

		SECTION ("choosing result side") {
			int container_result;
			container.add(1, 2);

			container_result = 0;
			REQUIRE(container.get(1, container_result));
			CHECK(container_result == 2);
			container_result = 0;
			REQUIRE(container.get(1, container_result, DualLookup<int>::Type::OPPOSITE));
			CHECK(container_result == 2);
			container_result = 0;
			REQUIRE(container.get(1, container_result, DualLookup<int>::Type::EQUIVALENT));
			CHECK(container_result == 2);
			container_result = 0;
			REQUIRE(container.get(1, container_result, DualLookup<int>::Type::VALUE));
			CHECK(container_result == 1);

			container_result = 0;
			REQUIRE(container.get(2, container_result));
			CHECK(container_result == 1);
			container_result = 0;
			REQUIRE(container.get(2, container_result, DualLookup<int>::Type::OPPOSITE));
			CHECK(container_result == 1);
			container_result = 0;
			REQUIRE(container.get(2, container_result, DualLookup<int>::Type::EQUIVALENT));
			CHECK(container_result == 2);
			container_result = 0;
			REQUIRE(container.get(2, container_result, DualLookup<int>::Type::VALUE));
			CHECK(container_result == 1);
		}
	}
}