Add test for registration from Lua code

mousebyte · Jul 21, 2023 · 3842a57 · 3842a57
1 parent 8147488
commit 3842a57
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/tests/assets/RegistersItself.moon b/tests/assets/RegistersItself.moon
@@ -0,0 +1,6 @@
+import register from require "lcl"
+
+class RegistersItself
+    foo: => "I registered myself!"
+
+register RegistersItself
diff --git a/tests/basicfunctions.cpp b/tests/basicfunctions.cpp
@@ -75,6 +75,22 @@ TEST_SUITE("Basic Functionality") {
             REQUIRE(luaC_uclass(L, -1) == NULL);
         }
 
+        SUBCASE("Registration from Lua") {
+            moonL_dofile(L, "RegistersItself.moon");
+            LCL_CHECKSTACK(1);
+            CHECK(luaC_isclass(L, -1));
+            lua_pop(L, 1);
+
+            luaC_construct(L, 0, "RegistersItself");
+            LCL_CHECKSTACK(1);
+            REQUIRE(luaC_isinstance(L, -1, "RegistersItself"));
+
+            luaC_mcall(L, "foo", 0, 1);
+            LCL_CHECKSTACK(2);
+            REQUIRE(lua_type(L, -1) == LUA_TSTRING);
+            REQUIRE(String(lua_tostring(L, -1)) == "I registered myself!");
+        }
+
         LCL_TEST_END
     }
 

diff --git a/tests/tests.hpp b/tests/tests.hpp
@@ -5,10 +5,12 @@ extern "C" {
 #include <moonauxlib.h>
 }
 
-#define LCL_TEST_BEGIN              \
-    using doctest::String;          \
-    lua_State *L = luaL_newstate(); \
-    luaL_openlibs(L);
+#define LCL_TEST_BEGIN                       \
+    using doctest::String;                   \
+    lua_State *L = luaL_newstate();          \
+    luaL_openlibs(L);                        \
+    luaL_requiref(L, "lcl", luaopen_lcl, 0); \
+    lua_pop(L, 1);
 
 #define LCL_TEST_END lua_close(L);