1
0
mirror of https://gitlab.com/OpenMW/openmw.git synced 2025-01-29 09:32:45 +00:00
2021-10-30 10:42:11 +02:00

536 lines
12 KiB
C++

#include <gtest/gtest.h>
#include <sstream>
#include "test_utils.hpp"
namespace
{
struct MWScriptTest : public ::testing::Test
{
MWScriptTest() : mErrorHandler(), mParser(mErrorHandler, mCompilerContext) {}
std::optional<CompiledScript> compile(const std::string& scriptBody, bool shouldFail = false)
{
mParser.reset();
mErrorHandler.reset();
std::istringstream input(scriptBody);
Compiler::Scanner scanner(mErrorHandler, input, mCompilerContext.getExtensions());
scanner.scan(mParser);
if(mErrorHandler.isGood())
{
std::vector<Interpreter::Type_Code> code;
mParser.getCode(code);
return CompiledScript(code, mParser.getLocals());
}
else if(!shouldFail)
logErrors();
return {};
}
void logErrors()
{
for(const auto& [error, loc] : mErrorHandler.getErrors())
{
std::cout << error;
if(loc.mLine)
std::cout << " at line" << loc.mLine << " column " << loc.mColumn << " (" << loc.mLiteral << ")";
std::cout << "\n";
}
}
void registerExtensions()
{
Compiler::registerExtensions(mExtensions);
mCompilerContext.setExtensions(&mExtensions);
}
void run(const CompiledScript& script, TestInterpreterContext& context)
{
mInterpreter.run(&script.mByteCode[0], static_cast<int>(script.mByteCode.size()), context);
}
void installOpcode(int code, Interpreter::Opcode0* opcode)
{
mInterpreter.installSegment5(code, opcode);
}
protected:
void SetUp() override
{
Interpreter::installOpcodes(mInterpreter);
}
void TearDown() override {}
private:
TestErrorHandler mErrorHandler;
TestCompilerContext mCompilerContext;
Compiler::FileParser mParser;
Compiler::Extensions mExtensions;
Interpreter::Interpreter mInterpreter;
};
const std::string sScript1 = R"mwscript(Begin basic_logic
; Comment
short one
short two
set one to two
if ( one == two )
set one to 1
elseif ( two == 1 )
set one to 2
else
set one to 3
endif
while ( one < two )
set one to ( one + 1 )
endwhile
End)mwscript";
const std::string sScript2 = R"mwscript(Begin addtopic
AddTopic "OpenMW Unit Test"
End)mwscript";
const std::string sScript3 = R"mwscript(Begin math
short a
short b
short c
short d
short e
set b to ( a + 1 )
set c to ( a - 1 )
set d to ( b * c )
set e to ( d / a )
End)mwscript";
// https://forum.openmw.org/viewtopic.php?f=6&t=2262
const std::string sScript4 = R"mwscript(Begin scripting_once_again
player -> addSpell "fire_bite", 645
PositionCell "Rabenfels, Taverne" 4480.000 3968.000 15820.000 0
End)mwscript";
const std::string sIssue1430 = R"mwscript(Begin issue1430
short var
If ( menumode == 1 )
Player->AddItem "fur_boots", 1
Player->Equip "iron battle axe", 1
player->addspell "fire bite", 645
player->additem "ring_keley", 1,
endif
End)mwscript";
const std::string sIssue1767 = R"mwscript(Begin issue1767
player->GetPcRank "temple"
End)mwscript";
const std::string sIssue3006 = R"mwscript(Begin issue3006
short a
if ( a == 1 )
set a to 2
else set a to 3
endif
End)mwscript";
const std::string sIssue3725 = R"mwscript(Begin issue3725
onactivate
if onactivate
; do something
endif
End)mwscript";
const std::string sIssue3846 = R"mwscript(Begin issue3846
Addtopic -spells...
Addtopic -magicka...
End)mwscript";
const std::string sIssue4061 = R"mwscript(Begin 01_Rz_neuvazhay-koryto2
End)mwscript";
const std::string sIssue4451 = R"mwscript(Begin, GlassDisplayScript
;[Script body]
End, GlassDisplayScript)mwscript";
const std::string sIssue4597 = R"mwscript(Begin issue4597
short a
short b
short c
short d
set c to 0
set d to 0
if ( a <> b )
set c to ( c + 1 )
endif
if ( a << b )
set c to ( c + 1 )
endif
if ( a < b )
set c to ( c + 1 )
endif
if ( a >< b )
set d to ( d + 1 )
endif
if ( a >> b )
set d to ( d + 1 )
endif
if ( a > b )
set d to ( d + 1 )
endif
End)mwscript";
const std::string sIssue4598 = R"mwscript(Begin issue4598
StartScript kal_S_Pub_Jejubãr_Faraminos
End)mwscript";
const std::string sIssue4803 = R"mwscript(
--
+-Begin issue4803
End)mwscript";
const std::string sIssue4867 = R"mwscript(Begin issue4867
float PcMagickaMult : The gameplay setting fPcBaseMagickaMult - 1.0000
End)mwscript";
const std::string sIssue4888 = R"mwscript(Begin issue4888
if (player->GameHour == 10)
set player->GameHour to 20
endif
End)mwscript";
const std::string sIssue5087 = R"mwscript(Begin Begin
player->sethealth 0
stopscript Begin
End Begin)mwscript";
const std::string sIssue5097 = R"mwscript(Begin issue5097
setscale "0.3"
End)mwscript";
const std::string sIssue5345 = R"mwscript(Begin issue5345
StartScript DN_MinionDrain_s"
End)mwscript";
const std::string sIssue6066 = R"mwscript(Begin issue6066
addtopic "return"
End)mwscript";
const std::string sIssue6282 = R"mwscript(Begin 11AA_LauraScript7.5
End)mwscript";
const std::string sIssue6363 = R"mwscript(Begin issue6363
short 1
if ( "1" == 1 )
PositionCell 0 1 2 3 4 5 "Morrowland"
endif
set 1 to 42
End)mwscript";
TEST_F(MWScriptTest, mwscript_test_invalid)
{
EXPECT_THROW(compile("this is not a valid script", true), Compiler::SourceException);
}
TEST_F(MWScriptTest, mwscript_test_compilation)
{
EXPECT_FALSE(!compile(sScript1));
}
TEST_F(MWScriptTest, mwscript_test_no_extensions)
{
EXPECT_THROW(compile(sScript2, true), Compiler::SourceException);
}
TEST_F(MWScriptTest, mwscript_test_function)
{
registerExtensions();
bool failed = true;
if(auto script = compile(sScript2))
{
class AddTopic : public Interpreter::Opcode0
{
bool& mFailed;
public:
AddTopic(bool& failed) : mFailed(failed) {}
void execute(Interpreter::Runtime& runtime)
{
const auto topic = runtime.getStringLiteral(runtime[0].mInteger);
runtime.pop();
mFailed = false;
EXPECT_EQ(topic, "OpenMW Unit Test");
}
};
installOpcode(Compiler::Dialogue::opcodeAddTopic, new AddTopic(failed));
TestInterpreterContext context;
run(*script, context);
}
if(failed)
{
FAIL();
}
}
TEST_F(MWScriptTest, mwscript_test_math)
{
if(auto script = compile(sScript3))
{
struct Algorithm
{
int a;
int b;
int c;
int d;
int e;
void run(int input)
{
a = input;
b = a + 1;
c = a - 1;
d = b * c;
e = d / a;
}
void test(const TestInterpreterContext& context) const
{
EXPECT_EQ(a, context.getLocalShort(0));
EXPECT_EQ(b, context.getLocalShort(1));
EXPECT_EQ(c, context.getLocalShort(2));
EXPECT_EQ(d, context.getLocalShort(3));
EXPECT_EQ(e, context.getLocalShort(4));
}
} algorithm;
TestInterpreterContext context;
for(int i = 1; i < 1000; ++i)
{
context.setLocalShort(0, i);
run(*script, context);
algorithm.run(i);
algorithm.test(context);
}
}
else
{
FAIL();
}
}
TEST_F(MWScriptTest, mwscript_test_forum_thread)
{
registerExtensions();
EXPECT_FALSE(!compile(sScript4));
}
TEST_F(MWScriptTest, mwscript_test_1430)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue1430));
}
TEST_F(MWScriptTest, mwscript_test_1767)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue1767));
}
TEST_F(MWScriptTest, mwscript_test_3006)
{
if(auto script = compile(sIssue3006))
{
TestInterpreterContext context;
context.setLocalShort(0, 0);
run(*script, context);
EXPECT_EQ(context.getLocalShort(0), 0);
context.setLocalShort(0, 1);
run(*script, context);
EXPECT_EQ(context.getLocalShort(0), 2);
}
else
{
FAIL();
}
}
TEST_F(MWScriptTest, mwscript_test_3725)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue3725));
}
TEST_F(MWScriptTest, mwscript_test_3846)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue3846));
}
TEST_F(MWScriptTest, mwscript_test_4061)
{
EXPECT_FALSE(!compile(sIssue4061));
}
TEST_F(MWScriptTest, mwscript_test_4451)
{
EXPECT_FALSE(!compile(sIssue4451));
}
TEST_F(MWScriptTest, mwscript_test_4597)
{
if(auto script = compile(sIssue4597))
{
TestInterpreterContext context;
for(int a = 0; a < 100; ++a)
{
for(int b = 0; b < 100; ++b)
{
context.setLocalShort(0, a);
context.setLocalShort(1, b);
run(*script, context);
if(a < b)
EXPECT_EQ(context.getLocalShort(2), 3);
else
EXPECT_EQ(context.getLocalShort(2), 0);
if(a > b)
EXPECT_EQ(context.getLocalShort(3), 3);
else
EXPECT_EQ(context.getLocalShort(3), 0);
}
}
}
else
{
FAIL();
}
}
TEST_F(MWScriptTest, mwscript_test_4598)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue4598));
}
TEST_F(MWScriptTest, mwscript_test_4803)
{
EXPECT_FALSE(!compile(sIssue4803));
}
TEST_F(MWScriptTest, mwscript_test_4867)
{
EXPECT_FALSE(!compile(sIssue4867));
}
TEST_F(MWScriptTest, mwscript_test_4888)
{
EXPECT_FALSE(!compile(sIssue4888));
}
TEST_F(MWScriptTest, mwscript_test_5087)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue5087));
}
TEST_F(MWScriptTest, mwscript_test_5097)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue5097));
}
TEST_F(MWScriptTest, mwscript_test_5345)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue5345));
}
TEST_F(MWScriptTest, mwscript_test_6066)
{
registerExtensions();
EXPECT_FALSE(!compile(sIssue6066));
}
TEST_F(MWScriptTest, mwscript_test_6282)
{
EXPECT_FALSE(!compile(sIssue6282));
}
TEST_F(MWScriptTest, mwscript_test_6363)
{
registerExtensions();
if(const auto script = compile(sIssue6363))
{
class PositionCell : public Interpreter::Opcode0
{
bool& mRan;
public:
PositionCell(bool& ran) : mRan(ran) {}
void execute(Interpreter::Runtime& runtime)
{
mRan = true;
}
};
bool ran = false;
installOpcode(Compiler::Transformation::opcodePositionCell, new PositionCell(ran));
TestInterpreterContext context;
context.setLocalShort(0, 0);
run(*script, context);
EXPECT_FALSE(ran);
ran = false;
context.setLocalShort(0, 1);
run(*script, context);
EXPECT_TRUE(ran);
}
else
{
FAIL();
}
}
}