Initial work on also doing battle stuff in tests.

src/Tester/AngelScript/TestFunctions.hpp

@@ -0,0 +1,77 @@
+#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() {
+        auto* ctx = asGetActiveContext();
+        TestEnvironment* env = static_cast<TestEnvironment*>(ctx->GetUserData());
+        env->AssertionCount += 1;
+        auto filename = ctx->GetFunction()->GetScriptSectionName();
+        auto line = ctx->GetLineNumber();
+        return RequirementData{
+            .FileName = filename,
+            .Line = (size_t)line,
+        };
+    }
+
+    static bool Require(bool value) {
+        if (!value) {
+            auto data = GetRequirementData();
+            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) {
+        if (expected != actual) {
+            auto data = GetRequirementData();
+            GetEqualityMessage(expected, actual);
+            throw RequirementFailed(data.FileName, data.Line, eqMsg);
+        }
+        return true;
+    }
+
+    static bool RequireEqualsString(const std::string& expected, const std::string& actual) {
+        if (expected != actual) {
+            auto data = GetRequirementData();
+            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