86 lines
3.6 KiB
C++
86 lines
3.6 KiB
C++
#ifndef POKEMONSCRIPTTESTER_TESTFUNCTIONS_HPP
|
|
#define POKEMONSCRIPTTESTER_TESTFUNCTIONS_HPP
|
|
|
|
#include <angelscript.h>
|
|
#include "../RequirementFailed.hpp"
|
|
#include "../TestEnvironment.hpp"
|
|
|
|
class TestFunctions {
|
|
private:
|
|
struct RequirementData {
|
|
std::string FileName;
|
|
size_t Line;
|
|
};
|
|
|
|
static RequirementData GetRequirementData(asIScriptContext* ctx) {
|
|
auto filename = ctx->GetFunction()->GetScriptSectionName();
|
|
auto line = ctx->GetLineNumber();
|
|
return RequirementData{
|
|
.FileName = filename,
|
|
.Line = (size_t)line,
|
|
};
|
|
}
|
|
|
|
static bool Require(bool value) {
|
|
auto* ctx = asGetActiveContext();
|
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
|
|
env->TotalRequirements += 1;
|
|
if (!value) {
|
|
env->FailedRequirements += 1;
|
|
auto data = GetRequirementData(ctx);
|
|
throw RequirementFailed(data.FileName, data.Line);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
#define GetEqualityMessage(expected, actual) \
|
|
std::string eqMsg; \
|
|
{ \
|
|
std::ostringstream ss; \
|
|
ss << "Expected: '" << std::to_string(expected) << "', but was: '" << std::to_string(actual) << "'"; \
|
|
eqMsg = ss.str(); \
|
|
}
|
|
|
|
static bool RequireEqualsI32(i32 expected, i32 actual) {
|
|
auto* ctx = asGetActiveContext();
|
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
|
|
env->TotalRequirements += 1;
|
|
if (expected != actual) {
|
|
env->FailedRequirements += 1;
|
|
auto data = GetRequirementData(ctx);
|
|
GetEqualityMessage(expected, actual);
|
|
throw RequirementFailed(data.FileName, data.Line, eqMsg);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool RequireEqualsString(const std::string& expected, const std::string& actual) {
|
|
auto* ctx = asGetActiveContext();
|
|
TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
|
|
if (expected != actual) {
|
|
env->FailedRequirements += 1;
|
|
auto data = GetRequirementData(ctx);
|
|
std::string eqMsg;
|
|
{
|
|
std::ostringstream ss;
|
|
ss << "Expected: '" << expected << "', but was: '" << actual << "'";
|
|
eqMsg = ss.str();
|
|
}
|
|
throw RequirementFailed(data.FileName, data.Line, eqMsg);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public:
|
|
static void Register(AngelScriptResolver* scriptResolver) {
|
|
auto engine = scriptResolver->GetBuilder().GetEngine();
|
|
Ensure(engine->RegisterGlobalFunction("bool Require(bool expression)", asFUNCTION(Require), asCALL_CDECL) >= 0);
|
|
Ensure(engine->RegisterGlobalFunction("bool RequireEquals(int expected, int actual)",
|
|
asFUNCTION(RequireEqualsI32), asCALL_CDECL) >= 0);
|
|
Ensure(engine->RegisterGlobalFunction("bool RequireEquals(const string &in expected, const string &in actual)",
|
|
asFUNCTION(RequireEqualsString), asCALL_CDECL) >= 0);
|
|
}
|
|
};
|
|
|
|
#endif // POKEMONSCRIPTTESTER_TESTFUNCTIONS_HPP
|