sol2/tests/run_time/source/basic.cpp
2022-06-25 04:00:53 -04:00

730 lines
21 KiB
C++

// sol2
// The MIT License (MIT)
// Copyright (c) 2013-2022 Rapptz, ThePhD and contributors
// Permission is hereby granted, free of charge, to any person obtaining a copy of
// this software and associated documentation files (the "Software"), to deal in
// the Software without restriction, including without limitation the rights to
// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
// the Software, and to permit persons to whom the Software is furnished to do so,
// subject to the following conditions:
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#include "sol_test.hpp"
#include <catch2/catch_all.hpp>
#include <fstream>
#include <iostream>
#include <map>
#include <unordered_map>
#include <vector>
inline namespace sol2_tests_basic {
bool func_opt_ret_bool(sol::optional<int> i) {
if (i) {
INFO(i.value());
}
else {
INFO("optional isn't set");
}
return true;
}
struct base1 {
int a1 = 250;
};
struct base2 {
int a2 = 500;
};
struct simple : base1 { };
struct complex : base1, base2 { };
struct CardAction {
int value;
};
} // namespace sol2_tests_basic
SOL_BASE_CLASSES(complex, base1, base2);
SOL_BASE_CLASSES(simple, base1);
SOL_DERIVED_CLASSES(base1, simple, complex);
SOL_DERIVED_CLASSES(base2, complex);
TEST_CASE("table/traversal", "ensure that we can chain requests and tunnel down into a value if we desire") {
sol::state lua;
int begintop = 0, endtop = 0;
sol::function scriptload = lua.load("t1 = {t2 = {t3 = 24}};");
scriptload();
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
int traversex24 = lua.traverse_get<int>("t1", "t2", "t3");
REQUIRE(traversex24 == 24);
}
REQUIRE(begintop == endtop);
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
int x24 = lua["t1"]["t2"]["t3"];
REQUIRE(x24 == 24);
}
REQUIRE(begintop == endtop);
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
lua["t1"]["t2"]["t3"] = 64;
}
REQUIRE(begintop == endtop);
{
int traversex64 = lua.traverse_get<int>("t1", "t2", "t3");
REQUIRE(traversex64 == 64);
}
REQUIRE(begintop == endtop);
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
int x64 = lua["t1"]["t2"]["t3"];
REQUIRE(x64 == 64);
}
REQUIRE(begintop == endtop);
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
lua.traverse_set("t1", "t2", "t3", 13);
}
REQUIRE(begintop == endtop);
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
int traversex13 = lua.traverse_get<int>("t1", "t2", "t3");
REQUIRE(traversex13 == 13);
}
REQUIRE(begintop == endtop);
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
int x13 = lua["t1"]["t2"]["t3"];
REQUIRE(x13 == 13);
}
REQUIRE(begintop == endtop);
}
TEST_CASE("simple/set", "Check if the set works properly.") {
sol::state lua;
int begintop = 0, endtop = 0;
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
lua.set("a", 9);
}
REQUIRE(begintop == endtop);
{
auto result = lua.safe_script("if a ~= 9 then error('wrong value') end", sol::script_pass_on_error);
REQUIRE(result.valid());
}
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
lua.set("d", "hello");
}
REQUIRE(begintop == endtop);
{
auto result = lua.safe_script("if d ~= 'hello' then error('expected \\'hello\\', got '.. tostring(d)) end", sol::script_pass_on_error);
REQUIRE(result.valid());
}
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
lua.set("e", std::string("hello"), "f", true);
}
REQUIRE(begintop == endtop);
{
auto result = lua.safe_script("if d ~= 'hello' then error('expected \\'hello\\', got '.. tostring(d)) end", sol::script_pass_on_error);
REQUIRE(result.valid());
}
{
auto result = lua.safe_script("if f ~= true then error('wrong value') end", sol::script_pass_on_error);
REQUIRE(result.valid());
}
}
TEST_CASE("simple/get", "Tests if the get function works properly.") {
sol::state lua;
int begintop = 0, endtop = 0;
lua.safe_script("a = 9");
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
auto a = lua.get<int>("a");
REQUIRE(a == 9.0);
}
REQUIRE(begintop == endtop);
lua.safe_script("b = nil");
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
REQUIRE_NOTHROW(lua.get<sol::lua_nil_t>("b"));
}
REQUIRE(begintop == endtop);
lua.safe_script("d = 'hello'");
lua.safe_script("e = true");
{
test_stack_guard g(lua.lua_state(), begintop, endtop);
std::string d;
bool e;
std::tie(d, e) = lua.get<std::string, bool>("d", "e");
REQUIRE(d == "hello");
REQUIRE(e == true);
}
REQUIRE(begintop == endtop);
}
TEST_CASE("simple/set and get global integer", "Tests if the get function works properly with global integers") {
sol::state lua;
lua[1] = 25.4;
lua.safe_script("b = 1");
double a = lua.get<double>(1);
double b = lua.get<double>("b");
REQUIRE(a == 25.4);
REQUIRE(b == 1);
}
TEST_CASE("simple/get_or", "check if table.get_or works correctly") {
sol::state lua;
auto bob_table = lua.create_table("bob");
bob_table.set("is_set", 42);
int is_set = bob_table.get_or("is_set", 3);
int is_not_set = bob_table.get_or("is_not_set", 22);
REQUIRE(is_set == 42);
REQUIRE(is_not_set == 22);
lua["joe"] = 55.6;
double bark = lua.get_or<double>("joe", 60);
REQUIRE(bark == 55.6);
}
TEST_CASE("simple/table_proxy get_or", "check if table_proxy.get_or works correctly") {
sol::state lua;
auto bob_table = lua.create_table("bob");
bob_table.set("is_set", 42);
int is_set = bob_table["is_set"].get_or(3);
int is_not_set = bob_table["is_not_set"].get_or(22);
REQUIRE(is_set == 42);
REQUIRE(is_not_set == 22);
lua["joe"] = 55.6;
double bark = lua["joe"].get_or<double>(60);
REQUIRE(bark == 55.6);
}
TEST_CASE("simple/addition", "check if addition works and can be gotten through lua.get and lua.set") {
sol::state lua;
lua.set("b", 0.2);
lua.safe_script("c = 9 + b");
auto c = lua.get<double>("c");
REQUIRE(c == 9.2);
}
TEST_CASE("simple/if", "check if if statements work through lua") {
sol::state lua;
std::string program = "if true then f = 0.1 else f = 'test' end";
lua.safe_script(program);
auto f = lua.get<double>("f");
REQUIRE(f == 0.1);
REQUIRE((f == lua["f"]));
}
TEST_CASE("negative/basic errors", "Check if error handling works correctly") {
sol::state lua;
auto result = lua.safe_script("nil[5]", sol::script_pass_on_error);
REQUIRE_FALSE(result.valid());
}
TEST_CASE("libraries", "Check if we can open libraries") {
sol::state lua;
REQUIRE_NOTHROW(lua.open_libraries(sol::lib::base, sol::lib::os));
}
TEST_CASE("libraries2", "Check if we can open ALL the libraries") {
sol::state lua;
REQUIRE_NOTHROW(lua.open_libraries(sol::lib::base,
sol::lib::bit32,
sol::lib::coroutine,
sol::lib::debug,
sol::lib::ffi,
sol::lib::jit,
sol::lib::math,
sol::lib::os,
sol::lib::package,
sol::lib::string,
sol::lib::table));
}
TEST_CASE("interop/null-to-nil-and-back",
"nil should be the given type when a pointer from C++ is returned as nullptr, and nil should result in nullptr in connected C++ code") {
sol::state lua;
lua.open_libraries(sol::lib::base);
lua.set_function("lol", []() -> int* { return nullptr; });
lua.set_function("rofl", [](int* x) { INFO(x); });
REQUIRE_NOTHROW(
lua.safe_script("x = lol()\n"
"rofl(x)\n"
"assert(x == nil)"));
}
TEST_CASE("object/conversions", "make sure all basic reference types can be made into objects") {
sol::state lua;
lua.open_libraries(sol::lib::base);
struct d { };
lua.safe_script("function f () print('bark') end");
lua["d"] = d {};
lua["l"] = static_cast<void*>(nullptr);
sol::table t = lua.create_table();
sol::table t2(lua, sol::create);
sol::thread th = sol::thread::create(lua);
sol::function f = lua["f"];
sol::protected_function pf = lua["f"];
sol::userdata ud = lua["d"];
sol::lightuserdata lud = lua["l"];
sol::environment env(lua, sol::create);
sol::object ot(t);
sol::object ot2(t2);
sol::object oteq = ot;
sol::object oth(th);
sol::object of(f);
sol::object opf(pf);
sol::object od(ud);
sol::object ol(lud);
sol::object oenv(env);
auto oni = sol::make_object(lua, 50);
auto ond = sol::make_object(lua, 50.0);
std::string somestring = "look at this text isn't it nice";
auto osl = sol::make_object(lua, "Bark bark bark");
auto os = sol::make_object(lua, somestring);
auto omn = sol::make_object(lua, sol::lua_nil);
REQUIRE(ot.get_type() == sol::type::table);
REQUIRE(ot2.get_type() == sol::type::table);
REQUIRE(oteq.get_type() == sol::type::table);
REQUIRE(oth.get_type() == sol::type::thread);
REQUIRE(of.get_type() == sol::type::function);
REQUIRE(opf.get_type() == sol::type::function);
REQUIRE(od.get_type() == sol::type::userdata);
REQUIRE(ol.get_type() == sol::type::lightuserdata);
REQUIRE(oni.get_type() == sol::type::number);
REQUIRE(ond.get_type() == sol::type::number);
REQUIRE(osl.get_type() == sol::type::string);
REQUIRE(os.get_type() == sol::type::string);
REQUIRE(omn.get_type() == sol::type::lua_nil);
REQUIRE(oenv.get_type() == sol::type::table);
}
TEST_CASE("object/main_* conversions", "make sure all basic reference types can be made into objects") {
sol::state lua;
lua.open_libraries(sol::lib::base);
struct d { };
lua.safe_script("function f () print('bark') end");
lua["d"] = d {};
lua["l"] = static_cast<void*>(nullptr);
sol::main_table t = lua.create_table();
sol::main_table t2(lua, sol::create);
sol::thread th = sol::thread::create(lua);
sol::main_function f = lua["f"];
sol::main_protected_function pf = lua["f"];
sol::main_userdata ud = lua["d"];
sol::main_lightuserdata lud = lua["l"];
sol::main_environment env(lua, sol::create);
sol::main_object ot(t);
sol::main_object ot2(t2);
sol::main_object oteq = ot;
sol::main_object oth(th);
sol::main_object of(f);
sol::main_object opf(pf);
sol::main_object od(ud);
sol::main_object ol(lud);
sol::main_object oenv(env);
auto oni = sol::make_object(lua, 50);
auto ond = sol::make_object(lua, 50.0);
std::string somestring = "look at this text isn't it nice";
auto osl = sol::make_object(lua, "Bark bark bark");
auto os = sol::make_object(lua, somestring);
auto omn = sol::make_object(lua, sol::lua_nil);
REQUIRE(ot.get_type() == sol::type::table);
REQUIRE(ot2.get_type() == sol::type::table);
REQUIRE(oteq.get_type() == sol::type::table);
REQUIRE(oth.get_type() == sol::type::thread);
REQUIRE(of.get_type() == sol::type::function);
REQUIRE(opf.get_type() == sol::type::function);
REQUIRE(od.get_type() == sol::type::userdata);
REQUIRE(ol.get_type() == sol::type::lightuserdata);
REQUIRE(oni.get_type() == sol::type::number);
REQUIRE(ond.get_type() == sol::type::number);
REQUIRE(osl.get_type() == sol::type::string);
REQUIRE(os.get_type() == sol::type::string);
REQUIRE(omn.get_type() == sol::type::lua_nil);
REQUIRE(oenv.get_type() == sol::type::table);
}
TEST_CASE("feature/indexing overrides", "make sure index functions can be overridden on types") {
struct PropertySet {
sol::object get_property_lua(const char* name, sol::this_state s) {
auto& var = props[name];
return sol::make_object(s, var);
}
void set_property_lua(const char* name, sol::stack_object object) {
props[name] = object.as<std::string>();
}
std::unordered_map<std::string, std::string> props;
};
struct DynamicObject {
PropertySet& get_dynamic_props() {
return dynamic_props;
}
PropertySet dynamic_props;
};
sol::state lua;
lua.open_libraries(sol::lib::base);
sol::usertype<PropertySet> utps = lua.new_usertype<PropertySet>("PropertySet");
utps[sol::meta_function::new_index] = &PropertySet::set_property_lua;
utps[sol::meta_function::index] = &PropertySet::get_property_lua;
sol::usertype<DynamicObject> utdo = lua.new_usertype<DynamicObject>("DynamicObject");
utdo["props"] = sol::property(&DynamicObject::get_dynamic_props);
auto result = lua.safe_script(R"__(
obj = DynamicObject:new()
obj.props.name = 'test name'
print('name = ' .. obj.props.name)
)__",
sol::script_pass_on_error);
REQUIRE(result.valid());
std::string name = lua["obj"]["props"]["name"];
REQUIRE(name == "test name");
}
TEST_CASE("features/indexing numbers", "make sure indexing functions can be override on usertypes") {
class vector {
public:
double data[3];
vector() : data { 0, 0, 0 } {
}
double& operator[](int i) {
return data[i];
}
static double my_index(vector& v, int i) {
return v[i];
}
static void my_new_index(vector& v, int i, double x) {
v[i] = x;
}
};
sol::state lua;
lua.open_libraries(sol::lib::base);
lua.new_usertype<vector>(
"vector", sol::constructors<sol::types<>>(), sol::meta_function::index, &vector::my_index, sol::meta_function::new_index, &vector::my_new_index);
lua.safe_script(
"v = vector.new()\n"
"print(v[1])\n"
"v[2] = 3\n"
"print(v[2])\n");
vector& v = lua["v"];
REQUIRE(v[0] == 0.0);
REQUIRE(v[1] == 0.0);
REQUIRE(v[2] == 3.0);
}
TEST_CASE("features/multiple inheritance", "Ensure that multiple inheritance works as advertised") {
sol::state lua;
lua.open_libraries(sol::lib::base);
lua.new_usertype<base1>("base1", "a1", &base1::a1);
lua.new_usertype<base2>("base2", "a2", &base2::a2);
lua.new_usertype<simple>("simple", "a1", &simple::a1, sol::base_classes, sol::bases<base1>());
lua.new_usertype<complex>("complex", "a1", &complex::a1, "a2", &complex::a2, sol::base_classes, sol::bases<base1, base2>());
lua.safe_script(
"c = complex.new()\n"
"s = simple.new()\n"
"b1 = base1.new()\n"
"b2 = base1.new()\n");
base1* sb1 = lua["s"];
REQUIRE(sb1 != nullptr);
REQUIRE(sb1->a1 == 250);
base1* cb1 = lua["c"];
base2* cb2 = lua["c"];
REQUIRE(cb1 != nullptr);
REQUIRE(cb2 != nullptr);
REQUIRE(cb1->a1 == 250);
REQUIRE(cb2->a2 == 500);
}
TEST_CASE("regressions/std::ref", "Ensure that std::reference_wrapper<> isn't considered as a function by using unwrap_unqualified_t trait") {
struct base1 {
int a1 = 250;
};
sol::state lua;
base1 v;
lua["vp"] = &v;
lua["vr"] = std::ref(v);
base1* vp = lua["vp"];
base1& vr = lua["vr"];
REQUIRE(vp != nullptr);
REQUIRE(vp == &v);
REQUIRE(vp->a1 == 250);
REQUIRE(vr.a1 == 250);
v.a1 = 568;
REQUIRE(vp->a1 == 568);
REQUIRE(vr.a1 == 568);
}
TEST_CASE("optional/left out args", "Make sure arguments can be left out of optional without tanking miserably") {
sol::state lua;
lua.open_libraries(sol::lib::base);
// sol::optional needs an argument no matter what?
lua.set_function("func_opt_ret_bool", func_opt_ret_bool);
REQUIRE_NOTHROW([&] {
lua.safe_script(R"(
func_opt_ret_bool(42)
func_opt_ret_bool()
print('ok')
)");
}());
}
TEST_CASE("optional/engaged versus unengaged", "solidify semantics for an engaged and unengaged optional") {
sol::state lua;
lua.open_libraries(sol::lib::base);
lua.set_function("f", [](sol::optional<sol::object> optional_arg) {
if (optional_arg) {
return true;
}
return false;
});
auto valid0 = lua.safe_script("assert(not f())", sol::script_pass_on_error);
REQUIRE(valid0.valid());
auto valid1 = lua.safe_script("assert(not f(nil))", sol::script_pass_on_error);
REQUIRE(valid1.valid());
auto valid2 = lua.safe_script("assert(f(1))", sol::script_pass_on_error);
REQUIRE(valid2.valid());
auto valid3 = lua.safe_script("assert(f('hi'))", sol::script_pass_on_error);
REQUIRE(valid3.valid());
}
TEST_CASE("pusher/constness", "Make sure more types can handle being const and junk") {
struct Foo {
Foo(const sol::function& f) : _f(f) {
}
const sol::function& _f;
const sol::function& f() const {
return _f;
}
};
sol::state lua;
lua.new_usertype<Foo>("Foo", sol::call_constructor, sol::no_constructor, "f", &Foo::f);
lua["func"] = []() { return 20; };
sol::function f = lua["func"];
lua["foo"] = Foo(f);
Foo& foo = lua["foo"];
int x = foo.f()();
REQUIRE(x == 20);
}
TEST_CASE("compilation/const regression", "make sure constness in tables is respected all the way down") {
struct State {
public:
State() {
this->state_.globals()["state"] = this;
}
sol::state state_;
};
State state;
State* s = state.state_.globals()["state"];
REQUIRE(s == &state);
}
TEST_CASE("numbers/integers", "make sure integers are detectable on most platforms") {
sol::state lua;
lua["a"] = 50; // int
lua["b"] = 50.5; // double
sol::object a = lua["a"];
sol::object b = lua["b"];
bool a_is_int = a.is<int>();
bool a_is_double = a.is<double>();
bool b_is_int = b.is<int>();
bool b_is_double = b.is<double>();
REQUIRE(a_is_int);
REQUIRE(a_is_double);
// TODO: will this fail on certain lower Lua versions?
REQUIRE_FALSE(b_is_int);
REQUIRE(b_is_double);
}
TEST_CASE("object/is", "test whether or not the is abstraction works properly for a user-defined type") {
struct thing { };
SECTION("stack_object") {
sol::state lua;
lua.open_libraries(sol::lib::base);
lua.set_function("is_thing", [](sol::stack_object obj) { return obj.is<thing>(); });
lua["a"] = thing {};
{
auto result = lua.safe_script("assert(is_thing(a))", sol::script_pass_on_error);
REQUIRE(result.valid());
}
}
SECTION("object") {
sol::state lua;
lua.open_libraries(sol::lib::base);
lua.set_function("is_thing", [](sol::object obj) { return obj.is<thing>(); });
lua["a"] = thing {};
{
auto result = lua.safe_script("assert(is_thing(a))", sol::script_pass_on_error);
REQUIRE(result.valid());
}
}
}
TEST_CASE("object/base_of_things", "make sure that object is the base of things and can be sliced / returned safely") {
SECTION("reference") {
sol::state lua;
lua.open_libraries(sol::lib::base, sol::lib::coroutine);
lua["ud"] = base1 {};
lua["f1"] = [](sol::object o) -> sol::object { return o; };
lua["f2"] = [](sol::table o) -> sol::object { return o; };
lua["f3"] = [](sol::thread o) -> sol::object { return o; };
lua["f4"] = [](sol::unsafe_function o) -> sol::object { return o; };
lua["f5"] = [](sol::protected_function o) -> sol::object { return o; };
lua["f6"] = [](sol::userdata o) -> sol::object { return o; };
auto result1 = lua.safe_script("f1(2)", sol::script_pass_on_error);
REQUIRE(result1.valid());
auto result2 = lua.safe_script("f2({})", sol::script_pass_on_error);
REQUIRE(result2.valid());
auto result3 = lua.safe_script("f3(coroutine.create( function () end ) )", sol::script_pass_on_error);
REQUIRE(result3.valid());
auto result4 = lua.safe_script("f4( function () end )", sol::script_pass_on_error);
REQUIRE(result4.valid());
auto result5 = lua.safe_script("f5( function () end )", sol::script_pass_on_error);
REQUIRE(result5.valid());
auto result6 = lua.safe_script("f6(ud)", sol::script_pass_on_error);
REQUIRE(result6.valid());
}
SECTION("stack_reference") {
sol::state lua;
lua.open_libraries(sol::lib::base, sol::lib::coroutine);
lua["ud"] = base1 {};
lua["f1"] = [](sol::stack_object o) -> sol::stack_object { return o; };
lua["f2"] = [](sol::stack_table o) -> sol::stack_object { return std::move(o); };
lua["f3"] = [](sol::stack_thread o) -> sol::stack_object { return std::move(o); };
lua["f4"] = [](sol::stack_unsafe_function o) -> sol::stack_object { return std::move(o); };
lua["f5"] = [](sol::stack_protected_function o) -> sol::stack_object { return std::move(o); };
lua["f6"] = [](sol::stack_userdata o) -> sol::stack_object { return std::move(o); };
auto result1 = lua.safe_script("f1(2)", sol::script_pass_on_error);
REQUIRE(result1.valid());
auto result2 = lua.safe_script("f2({})", sol::script_pass_on_error);
REQUIRE(result2.valid());
auto result3 = lua.safe_script("f3(coroutine.create( function () end ) )", sol::script_pass_on_error);
REQUIRE(result3.valid());
auto result4 = lua.safe_script("f4( function () end )", sol::script_pass_on_error);
REQUIRE(result4.valid());
auto result5 = lua.safe_script("f5( function () end )", sol::script_pass_on_error);
REQUIRE(result5.valid());
auto result6 = lua.safe_script("f6(ud)", sol::script_pass_on_error);
REQUIRE(result6.valid());
}
}
TEST_CASE("object/pop-based reference conversions", "Make sure conversions going through .as<T>() also have support for reference") {
sol::state lua;
lua["on_execute_impl"] = []() { return std::make_unique<CardAction>(CardAction { 5 }); };
sol::object obj = lua["on_execute_impl"]();
REQUIRE(obj.valid());
auto& ptr = obj.as<std::unique_ptr<CardAction>&>();
REQUIRE(ptr->value == 5);
}