#ifdef TESTS_BUILD

#include "../../../src/Battling/ScriptHandling/ScriptSource.hpp"
#include <utility>
#include "../../../extern/catch.hpp"
#include "../../../src/Battling/ScriptHandling/ScriptAggregator.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; }

    void TestMethod(int& runCount) { runCount++; }
};

class ScriptSourceWithScriptPtr : public ScriptSource {
public:
    std::unique_ptr<Script> ScriptPtr = nullptr;

protected:
    size_t ScriptCount() const override { return 1; }
    void GetActiveScripts(ArbUt::List<ScriptWrapper>& scripts) override {
        scripts.Append(ScriptWrapper::FromScript(&ScriptPtr));
    }
};

class ScriptSourceWithScriptSet : public ScriptSource {
public:
    ScriptSet Set;

protected:
    size_t ScriptCount() const override { return 1; }
    void GetActiveScripts(ArbUt::List<ScriptWrapper>& scripts) override {
        scripts.Append(ScriptWrapper::FromSet(&Set));
    }
};

TEST_CASE("Script source with unset script ptr.", "[Battling, Scripting]") {
    auto source = ScriptSourceWithScriptPtr();
    auto scripts = source.GetScriptIterator();
    auto first = scripts.GetNext();
    CHECK(first == nullptr);
}

TEST_CASE("Script source with script ptr being set.", "[Battling, Scripting]") {
    auto source = ScriptSourceWithScriptPtr();
    source.ScriptPtr = std::make_unique<TestScript>("foobar");
    auto scripts = source.GetScriptIterator();
    auto first = scripts.GetNext();
    CHECK(first != nullptr);
}

TEST_CASE("Script source with script ptr being set after first iteration.", "[Battling, Scripting]") {
    auto source = ScriptSourceWithScriptPtr();
    auto scripts = source.GetScriptIterator();
    auto first = scripts.GetNext();
    CHECK(first == nullptr);
    source.ScriptPtr = std::make_unique<TestScript>("foobar");
    scripts = source.GetScriptIterator();
    first = scripts.GetNext();
    CHECK(first != nullptr);
}

TEST_CASE("Script source with empty script set.", "[Battling, Scripting]") {
    auto source = ScriptSourceWithScriptSet();
    auto scripts = source.GetScriptIterator();
    auto first = scripts.GetNext();
    CHECK(first == nullptr);
}

TEST_CASE("Script source with single item script set.", "[Battling, Scripting]") {
    auto source = ScriptSourceWithScriptSet();
    auto s = new TestScript("foobar");
    source.Set.Add(s);
    auto scripts = source.GetScriptIterator();
    auto first = scripts.GetNextNotNull();
    CHECK(first != nullptr);
    CHECK(first->GetName() == "foobar");
}

TEST_CASE("Script source with multiple item script set.", "[Battling, Scripting]") {
    auto source = ScriptSourceWithScriptSet();
    auto s = new TestScript("foobar");
    auto s2 = new TestScript("foobar2");
    source.Set.Add(s);
    source.Set.Add(s2);
    auto scripts = source.GetScriptIterator();
    auto first = scripts.GetNextNotNull();
    REQUIRE(first != nullptr);
    CHECK(first->GetName() == "foobar");
    auto second = scripts.GetNextNotNull();
    REQUIRE(second != nullptr);
    CHECK(second->GetName() == "foobar2");
}

#endif