#ifdef TESTS_BUILD

#include <utility>
#include "../../../extern/catch.hpp"
#include "../../../src/Battling/ScriptHandling/ScriptSet.hpp"

using namespace CreatureLib;
using namespace CreatureLib::Battling;

class TestScript : public Script {
private:
    ArbUt::CaseInsensitiveConstString _name;

public:
    explicit TestScript(std::string name) : _name(std::move(name)){};
    const ArbUt::CaseInsensitiveConstString& GetName() const noexcept override { return _name; }
};

TEST_CASE("Empty script set count == 0", "[Battling, Scripting]") {
    auto set = ScriptSet();
    REQUIRE(set.Count() == 0);
}

TEST_CASE("Add script to script set", "[Battling, Scripting]") {
    auto set = ScriptSet();
    auto s = new TestScript("foobar");
    set.Add(s);
    REQUIRE(set.Count() == 1);
}

TEST_CASE("Add script to script set, then retrieve it", "[Battling, Scripting]") {
    auto set = ScriptSet();
    auto s = new TestScript("foobar");
    set.Add(s);
    REQUIRE(set.Count() == 1);
    auto get = set.GetIterator()->At(0);
    REQUIRE(get->GetName() == "foobar");
}

TEST_CASE("Add two scripts to script set", "[Battling, Scripting]") {
    auto set = ScriptSet();
    auto s = new TestScript("foobar");
    auto s2 = new TestScript("foobar2");
    set.Add(s);
    set.Add(s2);
    REQUIRE(set.Count() == 2);
}

TEST_CASE("Add two scripts to script set, then retrieve them", "[Battling, Scripting]") {
    auto set = ScriptSet();
    auto s = new TestScript("foobar");
    auto s2 = new TestScript("foobar2");
    set.Add(s);
    set.Add(s2);
    REQUIRE(set.Count() == 2);
    auto get1 = set.GetIterator()->At(0);
    auto get2 = set.GetIterator()->At(1);
    REQUIRE(get1->GetName() == "foobar");
    REQUIRE(get2->GetName() == "foobar2");
}

TEST_CASE("Add script to script set, then remove it", "[Battling, Scripting]") {
    auto set = ScriptSet();
    auto s = new TestScript("foobar");
    set.Add(s);
    REQUIRE(set.Count() == 1);
    set.Remove("foobar"_cnc.GetHash());
    REQUIRE(set.Count() == 0);
    auto it = set.GetIterator();
    REQUIRE(it->Count() == 0);
}

TEST_CASE("Add two scripts to script set, then remove them", "[Battling, Scripting]") {
    auto set = ScriptSet();
    auto s = new TestScript("foobar");
    auto s2 = new TestScript("foobar2");
    set.Add(s);
    set.Add(s2);
    REQUIRE(set.Count() == 2);
    set.Remove("foobar"_cnc.GetHash());
    REQUIRE(set.Count() == 1);
    auto it = set.GetIterator();
    REQUIRE(it->At(0)->GetName() == "foobar2");
}

#endif