mirror of
https://github.com/ThePhD/sol2.git
synced 2024-03-22 13:10:44 +08:00
581 lines
14 KiB
C++
581 lines
14 KiB
C++
#define SOL_CHECK_ARGUMENTS
|
|
|
|
#include <sol.hpp>
|
|
#include <catch.hpp>
|
|
|
|
#include <iterator>
|
|
#include <vector>
|
|
#include <list>
|
|
#include <map>
|
|
#include <unordered_map>
|
|
#include <set>
|
|
#include <unordered_set>
|
|
|
|
std::vector<int> test_table_return_one() {
|
|
return{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 };
|
|
}
|
|
|
|
std::vector<std::pair<std::string, int>> test_table_return_two() {
|
|
return{ { "one", 1 },{ "two", 2 },{ "three", 3 } };
|
|
}
|
|
|
|
std::map<std::string, std::string> test_table_return_three() {
|
|
return{ { "name", "Rapptz" },{ "friend", "ThePhD" },{ "project", "sol" } };
|
|
}
|
|
|
|
TEST_CASE("containers/returns", "make sure that even references to vectors are being serialized as tables") {
|
|
sol::state lua;
|
|
std::vector<int> v{ 1, 2, 3 };
|
|
lua.set_function("f", [&]() -> std::vector<int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
sol::object x = lua["x"];
|
|
sol::type xt = x.get_type();
|
|
REQUIRE(xt == sol::type::userdata);
|
|
sol::table t = x;
|
|
bool matching;
|
|
matching = t[1] == 1;
|
|
REQUIRE(matching);
|
|
matching = t[2] == 2;
|
|
REQUIRE(matching);
|
|
matching = t[3] == 3;
|
|
REQUIRE(matching);
|
|
}
|
|
|
|
TEST_CASE("containers/vector_roundtrip", "make sure vectors can be round-tripped") {
|
|
sol::state lua;
|
|
std::vector<int> v{ 1, 2, 3 };
|
|
lua.set_function("f", [&]() -> std::vector<int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
std::vector<int> x = lua["x"];
|
|
bool areequal = x == v;
|
|
REQUIRE(areequal);
|
|
}
|
|
|
|
TEST_CASE("containers/list_roundtrip", "make sure lists can be round-tripped") {
|
|
sol::state lua;
|
|
std::list<int> v{ 1, 2, 3 };
|
|
lua.set_function("f", [&]() -> std::list<int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
std::list <int> x = lua["x"];
|
|
bool areequal = x == v;
|
|
REQUIRE(areequal);
|
|
}
|
|
|
|
TEST_CASE("containers/map_roundtrip", "make sure maps can be round-tripped") {
|
|
sol::state lua;
|
|
std::map<std::string, int> v{ { "a", 1 },{ "b", 2 },{ "c", 3 } };
|
|
lua.set_function("f", [&]() -> std::map<std::string, int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
std::map<std::string, int> x = lua["x"];
|
|
bool areequal = x == v;
|
|
REQUIRE(areequal);
|
|
}
|
|
|
|
TEST_CASE("containers/unordered_map_roundtrip", "make sure unordered_maps can be round-tripped") {
|
|
sol::state lua;
|
|
std::unordered_map<std::string, int> v{ { "a", 1 },{ "b", 2 },{ "c", 3 } };
|
|
lua.set_function("f", [&]() -> std::unordered_map<std::string, int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
std::unordered_map<std::string, int> x = lua["x"];
|
|
bool areequal = x == v;
|
|
REQUIRE(areequal);
|
|
}
|
|
|
|
TEST_CASE("containers/unordered_set_roundtrip", "make sure unordered_sets can be round-tripped") {
|
|
sol::state lua;
|
|
std::unordered_set<int> v{ 1, 2, 3 };
|
|
lua.set_function("f", [&]() -> std::unordered_set<int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
std::unordered_set<int> x = lua["x"];
|
|
bool areequal = x == v;
|
|
REQUIRE(areequal);
|
|
}
|
|
|
|
TEST_CASE("containers/set_roundtrip", "make sure sets can be round-tripped") {
|
|
sol::state lua;
|
|
std::set<int> v{ 1, 2, 3 };
|
|
lua.set_function("f", [&]() -> std::set<int>& {
|
|
return v;
|
|
});
|
|
lua.script("x = f()");
|
|
std::set<int> x = lua["x"];
|
|
bool areequal = x == v;
|
|
REQUIRE(areequal);
|
|
}
|
|
|
|
TEST_CASE("containers/custom-usertype", "make sure container usertype metatables can be overridden") {
|
|
typedef std::unordered_map<int, int> bark;
|
|
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
lua.new_usertype<bark>("bark",
|
|
"something", [](const bark& b) {
|
|
INFO("It works: " << b.at(24));
|
|
},
|
|
"size", &bark::size,
|
|
"at", sol::resolve<const int&>(&bark::at),
|
|
"clear", &bark::clear
|
|
);
|
|
bark obj{ { 24, 50 } };
|
|
lua.set("a", &obj);
|
|
REQUIRE_NOTHROW(lua.script("assert(a:at(24) == 50)"));
|
|
REQUIRE_NOTHROW(lua.script("a:something()"));
|
|
lua.set("a", obj);
|
|
REQUIRE_NOTHROW(lua.script("assert(a:at(24) == 50)"));
|
|
REQUIRE_NOTHROW(lua.script("a:something()"));
|
|
}
|
|
|
|
TEST_CASE("containers/const-serialization-kvp", "make sure const keys / values are respected") {
|
|
typedef std::map<int, const int> bark;
|
|
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
bark obj{ { 24, 50 } };
|
|
lua.set("a", &obj);
|
|
REQUIRE_NOTHROW(lua.script("assert(a[24] == 50)"));
|
|
REQUIRE_THROWS(lua.script("a[24] = 51"));
|
|
REQUIRE_NOTHROW(lua.script("assert(a[24] == 50)"));
|
|
}
|
|
|
|
TEST_CASE("containers/basic-serialization", "make sure containers are turned into proper userdata and have basic hooks established") {
|
|
typedef std::vector<int> woof;
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
lua.set("b", woof{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30 });
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k = 1, #b do assert(k == b[k]) end")
|
|
);
|
|
woof w{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30 };
|
|
lua.set("b", w);
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k = 1, #b do assert(k == b[k]) end")
|
|
);
|
|
lua.set("b", &w);
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k = 1, #b do assert(k == b[k]) end")
|
|
);
|
|
lua.set("b", std::ref(w));
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k = 1, #b do assert(k == b[k]) end")
|
|
);
|
|
}
|
|
|
|
#if 0 // glibc is a fuccboi
|
|
TEST_CASE("containers/const-serialization", "make sure containers are turned into proper userdata and the basic hooks respect const-ness") {
|
|
typedef std::vector<const int> woof;
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
lua.set("b", woof{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30 });
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k, v in pairs(b) do assert(k == v) end");
|
|
);
|
|
REQUIRE_THROWS(lua.script("b[1] = 20"));
|
|
}
|
|
#endif // Fuck you, glibc
|
|
|
|
TEST_CASE("containers/table-serialization", "ensure types can be serialized as tables still") {
|
|
typedef std::vector<int> woof;
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
lua.set("b", sol::as_table(woof{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30 }));
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k, v in ipairs(b) do assert(k == v) end")
|
|
);
|
|
woof w{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30 };
|
|
lua.set("b", sol::as_table(w));
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k, v in ipairs(b) do assert(k == v) end")
|
|
);
|
|
lua.set("b", sol::as_table(&w));
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k, v in ipairs(b) do assert(k == v) end")
|
|
);
|
|
lua.set("b", sol::as_table(std::ref(w)));
|
|
REQUIRE_NOTHROW(
|
|
lua.script("for k, v in ipairs(b) do assert(k == v) end")
|
|
);
|
|
}
|
|
|
|
TEST_CASE("containers/const-correctness", "usertype metatable names should reasonably ignore const attributes") {
|
|
struct Vec {
|
|
int x, y, z;
|
|
};
|
|
|
|
sol::state lua;
|
|
lua.open_libraries(sol::lib::base);
|
|
lua.new_usertype<Vec>("Vec", "x", &Vec::x, "y", &Vec::y, "z", &Vec::z);
|
|
|
|
Vec vec;
|
|
vec.x = 1;
|
|
vec.y = 2;
|
|
vec.z = -3;
|
|
|
|
std::vector<Vec> foo;
|
|
foo.push_back(vec);
|
|
|
|
std::vector<Vec const *> bar;
|
|
bar.push_back(&vec);
|
|
|
|
lua.script(R"(
|
|
func = function(vecs)
|
|
for i = 1, #vecs do
|
|
vec = vecs[i]
|
|
print(i, ":", vec.x, vec.y, vec.z)
|
|
end
|
|
end
|
|
)");
|
|
|
|
REQUIRE_NOTHROW([&]{
|
|
lua["func"](foo);
|
|
lua["func"](bar);
|
|
}());
|
|
}
|
|
|
|
TEST_CASE("containers/arbitrary-creation", "userdata and tables should be usable from standard containers") {
|
|
sol::state lua;
|
|
lua.open_libraries(sol::lib::base);
|
|
lua.set_function("test_one", test_table_return_one);
|
|
lua.set_function("test_two", test_table_return_two);
|
|
lua.set_function("test_three", test_table_return_three);
|
|
|
|
REQUIRE_NOTHROW(lua.script("a = test_one()"));
|
|
REQUIRE_NOTHROW(lua.script("b = test_two()"));
|
|
REQUIRE_NOTHROW(lua.script("c = test_three()"));
|
|
|
|
REQUIRE_NOTHROW(lua.script("assert(#a == 10, 'error')"));
|
|
REQUIRE_NOTHROW(lua.script("assert(a[3] == 3, 'error')"));
|
|
REQUIRE_NOTHROW(lua.script("assert(b.one == 1, 'error')"));
|
|
REQUIRE_NOTHROW(lua.script("assert(b.three == 3, 'error')"));
|
|
REQUIRE_NOTHROW(lua.script("assert(c.name == 'Rapptz', 'error')"));
|
|
REQUIRE_NOTHROW(lua.script("assert(c.project == 'sol', 'error')"));
|
|
|
|
sol::table a = lua.get<sol::table>("a");
|
|
sol::table b = lua.get<sol::table>("b");
|
|
sol::table c = lua.get<sol::table>("c");
|
|
|
|
REQUIRE(a.size() == 10ULL);
|
|
REQUIRE(a.get<int>(3) == 3);
|
|
REQUIRE(b.get<int>("one") == 1);
|
|
REQUIRE(b.get<int>("three") == 3);
|
|
REQUIRE(c.get<std::string>("name") == "Rapptz");
|
|
REQUIRE(c.get<std::string>("project") == "sol");
|
|
}
|
|
|
|
TEST_CASE("containers/extra-functions", "make sure the manipulation functions are present and usable and working across various container types") {
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
|
|
lua.script(R"(
|
|
function g (x)
|
|
x:add(20)
|
|
end
|
|
|
|
function h (x)
|
|
x:add(20, 40)
|
|
end
|
|
|
|
function i (x)
|
|
x:clear()
|
|
end
|
|
|
|
function sf (x,v)
|
|
return x:find(v)
|
|
end
|
|
|
|
)");
|
|
|
|
// Have the function we
|
|
// just defined in Lua
|
|
sol::function g = lua["g"];
|
|
sol::function h = lua["h"];
|
|
sol::function i = lua["i"];
|
|
sol::function sf = lua["sf"];
|
|
|
|
// Set a global variable called
|
|
// "arr" to be a vector of 5 lements
|
|
lua["arr"] = std::vector<int>{ 2, 4, 6, 8, 10 };
|
|
lua["map"] = std::map<int, int>{ { 1 , 2 },{ 2, 4 },{ 3, 6 },{ 4, 8 },{ 5, 10 } };
|
|
lua["set"] = std::set<int>{ 2, 4, 6, 8, 10 };
|
|
std::vector<int>& arr = lua["arr"];
|
|
std::map<int, int>& map = lua["map"];
|
|
std::set<int>& set = lua["set"];
|
|
REQUIRE(arr.size() == 5);
|
|
REQUIRE(map.size() == 5);
|
|
REQUIRE(set.size() == 5);
|
|
|
|
g(lua["set"]);
|
|
g(lua["arr"]);
|
|
h(lua["map"]);
|
|
REQUIRE(arr.size() == 6);
|
|
REQUIRE(map.size() == 6);
|
|
REQUIRE(set.size() == 6);
|
|
|
|
{
|
|
int r = sf(set, 8);
|
|
REQUIRE(r == 8);
|
|
sol::object rn = sf(set, 9);
|
|
REQUIRE(rn == sol::nil);
|
|
}
|
|
|
|
{
|
|
int r = sf(map, 3);
|
|
REQUIRE(r == 6);
|
|
sol::object rn = sf(map, 9);
|
|
REQUIRE(rn == sol::nil);
|
|
}
|
|
|
|
i(lua["arr"]);
|
|
i(lua["map"]);
|
|
i(lua["set"]);
|
|
REQUIRE(arr.empty());
|
|
REQUIRE(map.empty());
|
|
REQUIRE(set.empty());
|
|
}
|
|
|
|
TEST_CASE("containers/usertype-transparency", "Make sure containers pass their arguments through transparently and push the results as references, not new values") {
|
|
class A {
|
|
public:
|
|
int a;
|
|
A(int b = 2) : a(b) {};
|
|
|
|
void func() { }
|
|
};
|
|
|
|
struct B {
|
|
|
|
B() {
|
|
for (std::size_t i = 0; i < 20; ++i) {
|
|
a_list.emplace_back(static_cast<int>(i));
|
|
}
|
|
}
|
|
|
|
std::vector<A> a_list;
|
|
};
|
|
|
|
sol::state lua;
|
|
lua.new_usertype<B>("B",
|
|
"a_list", &B::a_list
|
|
);
|
|
|
|
lua.script(R"(
|
|
b = B.new()
|
|
a_ref = b.a_list[2]
|
|
)");
|
|
|
|
B& b = lua["b"];
|
|
A& a_ref = lua["a_ref"];
|
|
REQUIRE(&b.a_list[1] == &a_ref);
|
|
REQUIRE(b.a_list[1].a == a_ref.a);
|
|
}
|
|
|
|
struct options {
|
|
static int livingcount;
|
|
static options* last;
|
|
options() {
|
|
++livingcount;
|
|
last = this;
|
|
INFO("constructor: " << this);
|
|
}
|
|
|
|
std::string output_help() {
|
|
last = this;
|
|
INFO("func: " << this);
|
|
return "";
|
|
}
|
|
|
|
void begin() {}
|
|
void end() {}
|
|
|
|
~options() {
|
|
last = this;
|
|
--livingcount;
|
|
}
|
|
};
|
|
|
|
options* options::last = nullptr;
|
|
int options::livingcount = 0;
|
|
|
|
struct machine {
|
|
options opt;
|
|
};
|
|
|
|
namespace sol {
|
|
template <>
|
|
struct is_container<options> : std::false_type {};
|
|
}
|
|
|
|
TEST_CASE("containers/is-container", "make sure the is_container trait behaves properly") {
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
|
|
lua.new_usertype<options>("options_type",
|
|
"output_help", &options::output_help
|
|
);
|
|
|
|
lua.new_usertype<machine>("machine_type",
|
|
"new", sol::no_constructor,
|
|
"opt", [](machine& m) { return &m.opt; },
|
|
"copy_opt", [](machine& m) { return m.opt; }
|
|
);
|
|
|
|
{
|
|
machine m;
|
|
lua["machine"] = &m;
|
|
|
|
lua.script(R"(
|
|
machine:opt():output_help()
|
|
)");
|
|
|
|
REQUIRE(options::last == &m.opt);
|
|
REQUIRE(options::livingcount == 1);
|
|
}
|
|
REQUIRE(options::livingcount == 0);
|
|
}
|
|
|
|
TEST_CASE("containers/readonly", "make sure readonly members are stored appropriately") {
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
|
|
struct bar {
|
|
int x = 24;
|
|
};
|
|
|
|
struct foo {
|
|
std::list<bar> seq;
|
|
};
|
|
|
|
lua.new_usertype<foo>(
|
|
"foo",
|
|
"seq", &foo::seq, // this one works
|
|
"readonly_seq", sol::readonly(&foo::seq) // this one does not work
|
|
);
|
|
lua["value"] = std::list<bar>{ {},{},{} };
|
|
|
|
lua.script(R"(
|
|
a = foo.new()
|
|
x = a.seq
|
|
a.seq = value
|
|
y = a.readonly_seq
|
|
)");
|
|
std::list<bar>& seqrefx = lua["x"];
|
|
std::list<bar>& seqrefy = lua["y"];
|
|
REQUIRE(&seqrefx == &seqrefy);
|
|
REQUIRE(seqrefx.size() == 3);
|
|
auto result = lua.do_string(R"(
|
|
a.readonly_seq = value;
|
|
)");
|
|
REQUIRE_FALSE(result.valid());
|
|
}
|
|
|
|
TEST_CASE("containers/to_args", "Test that the to_args abstractions works") {
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
|
|
lua.script("function f (a, b, c, d) print(a, b, c, d) return a, b, c, d end");
|
|
|
|
sol::function f = lua["f"];
|
|
int a, b, c, d;
|
|
|
|
std::vector<int> v2{ 3, 4 };
|
|
sol::tie(a, b, c, d) = f(1, 2, sol::as_args(v2));
|
|
REQUIRE(a == 1);
|
|
REQUIRE(b == 2);
|
|
REQUIRE(c == 3);
|
|
REQUIRE(d == 4);
|
|
|
|
std::set<int> v4{ 7, 6, 8, 5 };
|
|
sol::tie(a, b, c, d) = f(sol::as_args(v4));
|
|
REQUIRE(a == 5);
|
|
REQUIRE(b == 6);
|
|
REQUIRE(c == 7);
|
|
REQUIRE(d == 8);
|
|
|
|
int v3[] = { 10, 11, 12 };
|
|
sol::tie(a, b, c, d) = f(9, sol::as_args(v3));
|
|
REQUIRE(a == 9);
|
|
REQUIRE(b == 10);
|
|
REQUIRE(c == 11);
|
|
REQUIRE(d == 12);
|
|
|
|
}
|
|
|
|
TEST_CASE("containers/ipairs-test", "ensure that abstractions roundtrip properly and push nils to stop pairs / ipairs") {
|
|
struct thing {
|
|
int x = 20;
|
|
};
|
|
thing t{};
|
|
sol::state lua;
|
|
lua.open_libraries();
|
|
|
|
lua.set_function("f", [&t]() {
|
|
return std::vector<thing*>(5, &t);
|
|
});
|
|
|
|
lua.script(R"(
|
|
c = f()
|
|
)");
|
|
|
|
lua.script(R"(
|
|
check = {}
|
|
local i = 1
|
|
while c[i] do
|
|
check[i] = c[i]
|
|
i = i + 1
|
|
end
|
|
)");
|
|
sol::table c = lua["check"];
|
|
for (std::size_t i = 1; i < 6; ++i) {
|
|
thing& ct = c[i];
|
|
REQUIRE(&t == &ct);
|
|
REQUIRE(ct.x == 20);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("containers/append-idiom", "ensure the append-idiom works as intended") {
|
|
sol::state lua;
|
|
lua.open_libraries(sol::lib::base);
|
|
lua.script(
|
|
R"(
|
|
function f_fill(vec)
|
|
print("#vec in lua: " .. #vec)
|
|
for k = 1, #vec do
|
|
vec[k] = k
|
|
end
|
|
print("#vec in lua: " .. #vec)
|
|
end
|
|
function f_append(vec)
|
|
print("#vec in lua: " .. #vec)
|
|
vec[#vec] = -10456407
|
|
vec[#vec + 1] = -54
|
|
print("#vec in lua: " .. #vec)
|
|
end
|
|
)");
|
|
std::vector<int> fill_cmp{ 1, 2, 3 };
|
|
std::vector<int> append_cmp{ -1, -1, -10456407, -54 };
|
|
|
|
std::vector<int> vec1{ -1, -1, -1 };
|
|
std::vector<int> vec2{ -1, -1, -1 };
|
|
|
|
REQUIRE(vec1.size() == 3);
|
|
lua["f_fill"](vec1);
|
|
REQUIRE(vec1.size() == 3);
|
|
REQUIRE(vec1 == fill_cmp);
|
|
|
|
REQUIRE(vec2.size() == 3);
|
|
lua["f_append"](vec2);
|
|
REQUIRE(vec2.size() == 4);
|
|
REQUIRE(vec2 == append_cmp);
|
|
}
|