141 lines
4.2 KiB
C++
141 lines
4.2 KiB
C++
#ifdef TESTS_BUILD
|
|
#include <catch.hpp>
|
|
#include "../src/Script.hpp"
|
|
|
|
using namespace Porygon;
|
|
|
|
TEST_CASE( "Define script function", "[integration]" ) {
|
|
Script* script = Script::Create("function add(number a, number b) a + b end");
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
auto variable = script->GetVariable(u"add");
|
|
REQUIRE(variable != nullptr);
|
|
REQUIRE(variable->GetTypeClass() == TypeClass::Function);
|
|
delete script;
|
|
}
|
|
|
|
TEST_CASE( "Define script function and call", "[integration]" ) {
|
|
Script* script = Script::Create("function add(number a, number b) result = a + b end add(1, 2)");
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
auto variable = script->GetVariable(u"add");
|
|
REQUIRE(variable != nullptr);
|
|
REQUIRE(variable->GetTypeClass() == TypeClass::Function);
|
|
auto result = script->GetVariable(u"result");
|
|
REQUIRE(result->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(result->EvaluateInteger() == 3);
|
|
delete script;
|
|
}
|
|
|
|
TEST_CASE( "Define script function and call multiple times", "[integration]" ) {
|
|
Script* script = Script::Create("result = 0 function add(number a) result = result + a end add(1) add(4)");
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
auto variable = script->GetVariable(u"add");
|
|
REQUIRE(variable != nullptr);
|
|
REQUIRE(variable->GetTypeClass() == TypeClass::Function);
|
|
auto result = script->GetVariable(u"result");
|
|
REQUIRE(result->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(result->EvaluateInteger() == 5);
|
|
delete script;
|
|
}
|
|
|
|
TEST_CASE( "Define script function and call from extern", "[integration]" ) {
|
|
Script* script = Script::Create("result = 0 function add(number a) result = result + a end");
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
|
|
REQUIRE(script->HasFunction(u"add"));
|
|
auto toAddVal = new IntegerEvalValue(5);
|
|
script->CallFunction(u"add", {toAddVal});
|
|
delete toAddVal;
|
|
toAddVal = new IntegerEvalValue(6);
|
|
script->CallFunction(u"add", {toAddVal});
|
|
delete toAddVal;
|
|
|
|
auto result = script->GetVariable(u"result");
|
|
REQUIRE(result->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(result->EvaluateInteger() == 11);
|
|
delete script;
|
|
}
|
|
|
|
TEST_CASE( "Define script function and return", "[integration]" ) {
|
|
Script* script = Script::Create(
|
|
"val = 0\n"
|
|
"function add(number a, number b) \n"
|
|
"return a + b \n"
|
|
"val = val + 1\n"
|
|
"end");
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
|
|
REQUIRE(script->HasFunction(u"add"));
|
|
auto toAddVal = new IntegerEvalValue(5);
|
|
auto toAddVal2 = new IntegerEvalValue(6);
|
|
auto result = script->CallFunction(u"add", {toAddVal, toAddVal2});
|
|
delete toAddVal;
|
|
delete toAddVal2;
|
|
|
|
REQUIRE(result->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(result->EvaluateInteger() == 11);
|
|
|
|
auto variable = script->GetVariable(u"val");
|
|
REQUIRE(variable->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(variable->EvaluateInteger() == 0);
|
|
|
|
delete script;
|
|
}
|
|
|
|
TEST_CASE( "Functions can call themselves", "[integration]" ) {
|
|
Script* script = Script::Create(
|
|
R"(
|
|
val = 0
|
|
function add()
|
|
if val < 5 then
|
|
val = val + 1
|
|
add()
|
|
end
|
|
end
|
|
)");
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
script->CallFunction(u"add", {});
|
|
|
|
auto variable = script->GetVariable(u"val");
|
|
REQUIRE(variable->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(variable->EvaluateInteger() == 5);
|
|
|
|
delete script;
|
|
}
|
|
|
|
TEST_CASE( "Functions respect scope", "[integration]" ) {
|
|
Script* script = Script::Create(
|
|
R"(
|
|
function test()
|
|
function foo()
|
|
local a = 10
|
|
end
|
|
if true then
|
|
if true then
|
|
local a = 50
|
|
foo()
|
|
result = a
|
|
end
|
|
end
|
|
end
|
|
test()
|
|
)"
|
|
);
|
|
REQUIRE(!script->Diagnostics -> HasErrors());
|
|
script->Evaluate();
|
|
|
|
auto variable = script->GetVariable(u"result");
|
|
REQUIRE(variable->GetTypeClass() == TypeClass::Number);
|
|
REQUIRE(variable->EvaluateInteger() == 50);
|
|
|
|
delete script;
|
|
}
|
|
|
|
|
|
#endif
|