mirror of
https://gitcode.com/JianFeeeee/LuaCangjia_api.git
synced 2026-09-19 16:42:48 +00:00
322 lines
10 KiB
C++
322 lines
10 KiB
C++
#include <gtest/gtest.h>
|
||
#include <cstdio>
|
||
#include <cstdlib>
|
||
#include <cstring>
|
||
#include <unistd.h>
|
||
#include <sys/stat.h>
|
||
#include <string>
|
||
#include <fstream>
|
||
#include <streambuf>
|
||
#include <stdexcept> // 新增：用于异常处理
|
||
|
||
extern "C" {
|
||
#include "lua_cj_api.h"
|
||
#include "errors.h"
|
||
}
|
||
|
||
// 辅助函数：创建临时目录
|
||
static std::string create_temp_dir() {
|
||
// 修正点：
|
||
// 1. mkdtemp 要求模板字符串最后6个字符必须是 'XXXXXX'
|
||
// 2. 使用 /tmp 目录确保路径存在，避免 "test/" 目录不存在导致的失败
|
||
char tmpl[] = "/tmp/lua_cj_test_XXXXXX";
|
||
char *dir = mkdtemp(tmpl);
|
||
|
||
// 修正点：检查 mkdtemp 是否成功，防止传入 NULL 给 std::string 导致崩溃
|
||
if (dir == nullptr) {
|
||
perror("mkdtemp failed");
|
||
throw std::runtime_error("Failed to create temp directory");
|
||
}
|
||
|
||
return std::string(dir);
|
||
}
|
||
|
||
// 全局计数器，用于验证回调是否被调用
|
||
static int callback_invoke_count = 0;
|
||
|
||
// 模拟输入/输出回调
|
||
static int mock_callback() {
|
||
callback_invoke_count++;
|
||
return 0;
|
||
}
|
||
|
||
class LuaCjApiTest : public ::testing::Test {
|
||
protected:
|
||
void* runner;
|
||
std::string temp_dir;
|
||
|
||
void SetUp() override {
|
||
temp_dir = create_temp_dir();
|
||
callback_invoke_count = 0; // 重置计数器
|
||
|
||
// 修改：传入 temp_dir 作为 pathio，确保 I/O 重定向路径有效
|
||
// 仓颉层封装中 pathio 和 pkgpath 都是可配置的，这里测试基础情况
|
||
runner = init_lua_runner(temp_dir.c_str(), NULL, mock_callback, mock_callback);
|
||
ASSERT_NE(runner, nullptr) << "init_lua_runner failed, errno=" << get_errno();
|
||
}
|
||
|
||
void TearDown() override {
|
||
if (runner) {
|
||
free_lua(runner);
|
||
runner = nullptr;
|
||
}
|
||
std::string cmd = "rm -rf " + temp_dir;
|
||
system(cmd.c_str());
|
||
}
|
||
|
||
void write_lua(const std::string& filename, const std::string& content) {
|
||
std::string path = temp_dir + "/" + filename;
|
||
FILE* f = fopen(path.c_str(), "w");
|
||
ASSERT_NE(f, nullptr);
|
||
fprintf(f, "%s", content.c_str());
|
||
fclose(f);
|
||
}
|
||
|
||
// 辅助函数：读取文件内容
|
||
std::string read_file(const std::string& filename) {
|
||
std::string path = temp_dir + "/" + filename;
|
||
std::ifstream t(path);
|
||
std::string str((std::istreambuf_iterator<char>(t)),
|
||
std::istreambuf_iterator<char>());
|
||
return str;
|
||
}
|
||
};
|
||
|
||
// 基础生命周期测试
|
||
TEST_F(LuaCjApiTest, InitFree) {
|
||
SUCCEED();
|
||
}
|
||
|
||
// 测试加载库
|
||
TEST_F(LuaCjApiTest, LoadLib) {
|
||
write_lua("mylib.lua", "function foo() return 42 end");
|
||
int ret = load_lib(runner, (temp_dir + "/mylib.lua").c_str(), "mylib");
|
||
EXPECT_EQ(ret, 0);
|
||
}
|
||
|
||
// 测试加载不存在的文件
|
||
TEST_F(LuaCjApiTest, LoadLibFileNotFound) {
|
||
int ret = load_lib(runner, "/no/such/file.lua", "bad");
|
||
EXPECT_EQ(ret, -1);
|
||
EXPECT_EQ(get_errno(), NAPI_LOAD_FILE_ERROR);
|
||
}
|
||
|
||
// 测试超过最大库数量限制
|
||
TEST_F(LuaCjApiTest, LoadLibOverflow) {
|
||
for (int i = 0; i < MAX_LUA_LIB; ++i) {
|
||
char name[20];
|
||
snprintf(name, sizeof(name), "lib%d", i);
|
||
std::string fname = std::string(name) + ".lua";
|
||
write_lua(fname, "return {}");
|
||
std::string path = temp_dir + "/" + fname;
|
||
int ret = load_lib(runner, path.c_str(), name);
|
||
EXPECT_EQ(ret, 0);
|
||
}
|
||
write_lua("extra.lua", "return {}");
|
||
std::string path = temp_dir + "/extra.lua";
|
||
int ret = load_lib(runner, path.c_str(), "extra");
|
||
EXPECT_EQ(ret, -1);
|
||
EXPECT_EQ(get_errno(), NAPI_LUALIB_LOAD_OVER_STACK);
|
||
}
|
||
|
||
// 测试卸载库
|
||
TEST_F(LuaCjApiTest, UnloadLib) {
|
||
write_lua("mylib.lua", "function foo() end");
|
||
std::string path = temp_dir + "/mylib.lua";
|
||
ASSERT_EQ(load_lib(runner, path.c_str(), "mylib"), 0);
|
||
int ret = unload_lib(runner, "mylib");
|
||
EXPECT_EQ(ret, 0);
|
||
}
|
||
|
||
// 测试卸载未加载的库
|
||
TEST_F(LuaCjApiTest, UnloadLibNotFound) {
|
||
int ret = unload_lib(runner, "nosuch");
|
||
EXPECT_EQ(ret, -1);
|
||
EXPECT_EQ(get_errno(), NAPI_UNLOADLIB_FAIL);
|
||
}
|
||
|
||
// 测试运行脚本
|
||
TEST_F(LuaCjApiTest, RunScript) {
|
||
write_lua("test.lua", "return 'hello'");
|
||
int ret = run(runner, (temp_dir + "/test.lua").c_str(), nullptr);
|
||
EXPECT_EQ(ret, 0);
|
||
char* res = getresult(runner);
|
||
EXPECT_STREQ(res, "hello");
|
||
}
|
||
|
||
// 测试带参数的脚本
|
||
TEST_F(LuaCjApiTest, RunScriptWithArg) {
|
||
// 修正点：Lua 中 ... 不能作为变量名。
|
||
// 使用 select(1, ...) 或者直接 return ...
|
||
// 这里使用 "return (...)" 返回第一个参数
|
||
write_lua("test.lua", "return (...)");
|
||
|
||
int ret = run(runner, (temp_dir + "/test.lua").c_str(), "world");
|
||
EXPECT_EQ(ret, 0);
|
||
char* res = getresult(runner);
|
||
EXPECT_STREQ(res, "world");
|
||
}
|
||
|
||
|
||
// 测试脚本语法错误
|
||
TEST_F(LuaCjApiTest, RunScriptSyntaxError) {
|
||
write_lua("bad.lua", "for i=1,10 print(i) end");
|
||
int ret = run(runner, (temp_dir + "/bad.lua").c_str(), NULL);
|
||
EXPECT_EQ(ret, -1);
|
||
EXPECT_EQ(get_errno(), NAPI_SCRIPT_ERROR);
|
||
char* res = getresult(runner);
|
||
EXPECT_NE(strstr(res, "'do' expected"), nullptr);
|
||
}
|
||
|
||
// 测试脚本运行时错误
|
||
TEST_F(LuaCjApiTest, RunScriptRuntimeError) {
|
||
write_lua("bad.lua", "error('oops')");
|
||
int ret = run(runner, (temp_dir + "/bad.lua").c_str(), nullptr);
|
||
EXPECT_EQ(ret, -1);
|
||
EXPECT_EQ(get_errno(), NAPI_SCRIPT_ERROR);
|
||
char* res = getresult(runner);
|
||
EXPECT_NE(strstr(res, "oops"), nullptr);
|
||
}
|
||
|
||
// 测试 cleanup 重置状态
|
||
TEST_F(LuaCjApiTest, Cleanup) {
|
||
write_lua("set.lua", "g = 123");
|
||
ASSERT_EQ(run(runner, (temp_dir + "/set.lua").c_str(), nullptr), 0);
|
||
int ret = cleanup(runner);
|
||
EXPECT_EQ(ret, 0);
|
||
write_lua("get.lua", "return tostring(g or 'nil')");
|
||
ret = run(runner, (temp_dir + "/get.lua").c_str(), nullptr);
|
||
EXPECT_EQ(ret, 0);
|
||
char* res = getresult(runner);
|
||
EXPECT_STREQ(res, "nil");
|
||
}
|
||
|
||
// 测试跨调用数据传递
|
||
TEST_F(LuaCjApiTest, CrossCallDataPassing) {
|
||
write_lua("set.lua", "data = {1,2,3}");
|
||
ASSERT_EQ(run(runner, (temp_dir + "/set.lua").c_str(), nullptr), 0);
|
||
write_lua("get.lua", "return table.concat(data, ',')");
|
||
ASSERT_EQ(run(runner, (temp_dir + "/get.lua").c_str(), nullptr), 0);
|
||
char* res = getresult(runner);
|
||
EXPECT_STREQ(res, "1,2,3");
|
||
}
|
||
|
||
|
||
// 测试多次加载卸载后栈索引正确
|
||
TEST_F(LuaCjApiTest, MultipleLoadUnload) {
|
||
for (int i = 0; i < 5; ++i) {
|
||
char name[20];
|
||
snprintf(name, sizeof(name), "L%d", i);
|
||
std::string fname = std::string(name) + ".lua";
|
||
write_lua(fname, "return " + std::to_string(i));
|
||
std::string path = temp_dir + "/" + fname;
|
||
ASSERT_EQ(load_lib(runner, path.c_str(), name), 0);
|
||
}
|
||
|
||
for (int i = 4; i >= 0; --i) {
|
||
char name[20];
|
||
snprintf(name, sizeof(name), "L%d", i);
|
||
ASSERT_EQ(unload_lib(runner, name), 0);
|
||
|
||
for (int j = 0; j < i; ++j) {
|
||
char libname[20], scriptname[20];
|
||
snprintf(libname, sizeof(libname), "L%d", j);
|
||
snprintf(scriptname, sizeof(scriptname), "check%d.lua", j);
|
||
std::string script = "return require '" + std::string(libname) + "'";
|
||
write_lua(scriptname, script);
|
||
int ret = run(runner, (temp_dir + "/" + scriptname).c_str(), nullptr);
|
||
EXPECT_EQ(ret, -1) << "Failed for lib " << libname << " after unloading " << name;
|
||
if (ret == 0) {
|
||
char expected[10];
|
||
snprintf(expected, sizeof(expected), "%d", j);
|
||
EXPECT_STREQ(getresult(runner), expected);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// ==================== 新增测试：针对仓颉封装覆盖的功能 ====================
|
||
|
||
// 测试 I/O 重定向：Print
|
||
TEST_F(LuaCjApiTest, RedirectPrint) {
|
||
write_lua("print_test.lua", "print('Hello Cangjie')");
|
||
int ret = run(runner, (temp_dir + "/print_test.lua").c_str(), nullptr);
|
||
EXPECT_EQ(ret, 0);
|
||
|
||
// 验证：回调函数应该被调用
|
||
EXPECT_GT(callback_invoke_count, 0) << "Output callback should be invoked by print";
|
||
|
||
// 验证：temp_dir 下应生成 output 文件
|
||
std::string content = read_file("output");
|
||
EXPECT_NE(content.find("Hello Cangjie"), std::string::npos);
|
||
}
|
||
|
||
// 测试 I/O 重定向：io.write
|
||
TEST_F(LuaCjApiTest, RedirectIoWrite) {
|
||
write_lua("write_test.lua", "io.write('Data from lua')");
|
||
int ret = run(runner, (temp_dir + "/write_test.lua").c_str(), nullptr);
|
||
EXPECT_EQ(ret, 0);
|
||
|
||
EXPECT_GT(callback_invoke_count, 0) << "Output callback should be invoked by io.write";
|
||
std::string content = read_file("output");
|
||
EXPECT_NE(content.find("Data from lua"), std::string::npos);
|
||
}
|
||
|
||
// 测试 I/O 重定向：io.read
|
||
TEST_F(LuaCjApiTest, RedirectIoRead) {
|
||
// 准备：模拟外部向 input 文件写入数据
|
||
std::string input_file = temp_dir + "/input";
|
||
FILE* f = fopen(input_file.c_str(), "w");
|
||
ASSERT_NE(f, nullptr);
|
||
fprintf(f, "MockInputData");
|
||
fclose(f);
|
||
|
||
// Lua 脚本读取输入并返回
|
||
write_lua("read_test.lua", "local s = io.read('*a'); return s");
|
||
|
||
int ret = run(runner, (temp_dir + "/read_test.lua").c_str(), nullptr);
|
||
EXPECT_EQ(ret, 0);
|
||
|
||
// 验证：输入回调被调用
|
||
EXPECT_GT(callback_invoke_count, 0) << "Input callback should be invoked by io.read";
|
||
|
||
// 验证：结果正确
|
||
char* res = getresult(runner);
|
||
EXPECT_STREQ(res, "MockInputData");
|
||
}
|
||
|
||
// 测试 pkgpath 配置是否生效
|
||
TEST_F(LuaCjApiTest, PackagePathConfig) {
|
||
// 创建子目录存放库
|
||
std::string lib_dir = temp_dir + "/mypkgs";
|
||
mkdir(lib_dir.c_str(), 0755);
|
||
|
||
// 在子目录创建库
|
||
std::string lib_path = lib_dir + "/mylib.lua";
|
||
FILE* f = fopen(lib_path.c_str(), "w");
|
||
fprintf(f, "return 'LoadedViaPkgPath'");
|
||
fclose(f);
|
||
|
||
// 销毁旧 runner
|
||
free_lua(runner);
|
||
runner = nullptr;
|
||
|
||
// 创建新 runner，配置 pkgpath
|
||
std::string pkg_path_pattern = lib_dir + "/?.lua";
|
||
runner = init_lua_runner(temp_dir.c_str(), pkg_path_pattern.c_str(), mock_callback, mock_callback);
|
||
ASSERT_NE(runner, nullptr);
|
||
|
||
// 脚本直接 require，不指定路径
|
||
write_lua("main.lua", "return require 'mylib'");
|
||
int ret = run(runner, (temp_dir + "/main.lua").c_str(), nullptr);
|
||
|
||
EXPECT_EQ(ret, 0) << "Require failed, maybe pkgpath not set correctly. Errno: " << get_errno();
|
||
char* res = getresult(runner);
|
||
EXPECT_STREQ(res, "LoadedViaPkgPath");
|
||
}
|
||
|
||
int main(int argc, char **argv) {
|
||
::testing::InitGoogleTest(&argc, argv);
|
||
return RUN_ALL_TESTS();
|
||
}
|