#ifdef TESTS_BUILD
#include "../../extern/catch.hpp"
#include "../../src/ScriptResolving/AngelScript/AngelScripResolver.hpp"
#include "../TestLibrary/TestLibrary.hpp"

static std::unordered_map<const char*, const char*> _scripts =
    std::unordered_map<const char*, const char*>{{"testScript1", R"(
namespace Pokemon{
[Pokemon effect=testScript1]
class testScript1 {
    int add(int a, int b) {
        return a + b;
    }

    void testFunc2(const Species@ s){
        if (s.Name != "testSpecies2"){
            throw("err");
        }
    }
}
}
)"}};

TEST_CASE("Get a script resolver, initialize it, then delete it") {
    auto lib = PkmnLib::Battling::BattleLibrary::CreateScriptResolver();
    lib->Initialize(TestLibrary::GetLibrary());
    delete lib;
}

TEST_CASE("Get a script resolver, set script load function, load script, then build module") {
    auto lib = dynamic_cast<AngelScripResolver*>(PkmnLib::Battling::BattleLibrary::CreateScriptResolver());
    lib->Initialize(TestLibrary::GetLibrary());
    lib->CreateScript("testScript1" , _scripts["testScript1"]);
    lib->FinalizeModule();
    delete lib;
}

TEST_CASE("Build script resolver, then create object") {
    auto lib = dynamic_cast<AngelScripResolver*>(PkmnLib::Battling::BattleLibrary::CreateScriptResolver());
    lib->Initialize(TestLibrary::GetLibrary());
    lib->CreateScript("testScript1" , _scripts["testScript1"]);
    lib->FinalizeModule();

    auto obj = lib->LoadScript(ScriptCategory::Creature, "testScript1"_cnc);

    delete obj;
    delete lib;
}

TEST_CASE("Build script resolver, create object, invoke addition method") {
    auto lib = dynamic_cast<AngelScripResolver*>(PkmnLib::Battling::BattleLibrary::CreateScriptResolver());
    lib->Initialize(TestLibrary::GetLibrary());
    lib->CreateScript("testScript1" , _scripts["testScript1"]);
    lib->FinalizeModule();

    auto obj =
        dynamic_cast<AngelScriptScript*>(lib->LoadScript(ScriptCategory::Creature, "testScript1"_cnc));
    REQUIRE(obj != nullptr);
    auto ctxPool = obj->GetContextPool();
    auto ctx = ctxPool->RequestContext();

    auto func = obj->PrepareMethod("add"_cnc, ctx);
    REQUIRE(func != nullptr);

    ctx->SetArgDWord(0, 5);
    ctx->SetArgDWord(1, 100);

    auto result = ctx->Execute();
    if (result == asEXECUTION_EXCEPTION) {
        FAIL(ctx->GetExceptionString());
    }
    CHECK(result == asEXECUTION_FINISHED);

    auto returnValue = ctx->GetReturnDWord();
    REQUIRE(returnValue == 105);

    ctxPool->ReturnContextToPool(ctx);
    delete obj;
    delete lib;
}

#endif