PkmnLib/tests/ScriptTests/ScriptTypeTests/Library/ItemDataTests.cpp

#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"(
class testScript1 {
    bool testName(const Item@ i, const string &in name){ return i.Name == name; }
    bool testCategory(const Item@ i, ItemCategory category){ return i.Category == category; }
    bool testBattleCategory(const Item@ i, BattleItemCategory category){ return i.BattleCategory == category; }
    bool testPrice(const Item@ i, int price){ return i.Price == price; }
}
)"}};

static const char* _testLoadFunc(CreatureLib::Battling::ScriptResolver::ScriptCategory category, const char* name) {
    return _scripts[name];
}

struct ScriptData {
    AngelScriptScript* Script = nullptr;
    AngelScripResolver* Resolver = nullptr;
    asIScriptFunction* Func = nullptr;
    asIScriptContext* Context = nullptr;

    ~ScriptData() {
        Script->GetContextPool()->ReturnContextToPool(Context);
        delete Script;
    }
};

static AngelScripResolver* _resolverCache = nullptr;
static AngelScripResolver* GetScriptResolver(PkmnLib::Battling::BattleLibrary* mainLib){
    if (_resolverCache == nullptr){
        _resolverCache = dynamic_cast<AngelScripResolver*>(PkmnLib::Battling::BattleLibrary::CreateScriptResolver());
        _resolverCache->Initialize(mainLib);
        _resolverCache->SetCreateFunction(&_testLoadFunc);
        _resolverCache->CreateScript(AngelScripResolver::ScriptCategory::Attack ,"testScript1");
        _resolverCache->FinalizeModule();
    }
    return _resolverCache;
}

static ScriptData GetScript(PkmnLib::Battling::BattleLibrary* mainLib, const char* funcName) {
    auto lib = GetScriptResolver(mainLib);
    auto s = lib->LoadScript(AngelScripResolver::ScriptCategory::Creature, "testScript1");
    auto script = dynamic_cast<AngelScriptScript*>(s);
    auto ctxPool = script->GetContextPool();
    auto ctx = ctxPool->RequestContext();

    auto func = script->PrepareMethod(funcName, ctx);
    REQUIRE(func != nullptr);

    return {.Script = script, .Resolver = lib, .Func = func, .Context = ctx};
}

TEST_CASE("Validate Item Name in Script") {
    auto mainLib = TestLibrary::GetLibrary();
    auto data = GetScript(mainLib, "testName");

    auto item = mainLib->GetItemLibrary()->Get("testItem");
    data.Context->SetArgObject(0, const_cast<PkmnLib::Library::Item*>(item));
    auto name = item->GetName();
    data.Context->SetArgAddress(1, &name);

    REQUIRE(data.Context->Execute() == asEXECUTION_FINISHED);
    REQUIRE((bool)data.Context->GetReturnWord());
}

TEST_CASE("Validate Item Category in Script") {
    auto mainLib = TestLibrary::GetLibrary();
    auto data = GetScript(mainLib, "testCategory");

    auto item = mainLib->GetItemLibrary()->Get("testItem");
    data.Context->SetArgObject(0, const_cast<PkmnLib::Library::Item*>(item));
    data.Context->SetArgDWord(1, static_cast<int32_t >(item->GetCategory()));

    REQUIRE(data.Context->Execute() == asEXECUTION_FINISHED);
    REQUIRE((bool)data.Context->GetReturnWord());
}

TEST_CASE("Validate Item Battle Category in Script") {
    auto mainLib = TestLibrary::GetLibrary();
    auto data = GetScript(mainLib, "testBattleCategory");

    auto item = mainLib->GetItemLibrary()->Get("testItem");
    data.Context->SetArgObject(0, const_cast<PkmnLib::Library::Item*>(item));
    data.Context->SetArgDWord(1, static_cast<int32_t >(item->GetBattleCategory()));

    REQUIRE(data.Context->Execute() == asEXECUTION_FINISHED);
    REQUIRE((bool)data.Context->GetReturnWord());
}

TEST_CASE("Validate Item Price in Script") {
    auto mainLib = TestLibrary::GetLibrary();
    auto data = GetScript(mainLib, "testPrice");

    auto item = mainLib->GetItemLibrary()->Get("testItem");
    data.Context->SetArgObject(0, const_cast<PkmnLib::Library::Item*>(item));
    data.Context->SetArgDWord(1, static_cast<int32_t >(item->GetPrice()));

    REQUIRE(data.Context->Execute() == asEXECUTION_FINISHED);
    REQUIRE((bool)data.Context->GetReturnWord());
}




#endif