test(lua): 添加 api_shared_dict 和 engine 测试覆盖

- api_shared_dict_test.go: 测试 shared_dict API 功能
- engine_test.go: 测试 Lua 引擎初始化和执行

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
xfy 2026-04-20 08:27:24 +08:00
parent 28be9e7e66
commit 7ef16f2be1
2 changed files with 988 additions and 0 deletions

View File

@ -0,0 +1,527 @@
// Package lua 提供 ngx.shared API 扩展测试,覆盖 Lua API 函数的更多场景
package lua
import (
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
// TestSharedDictLuaReplace 测试 dict:replace
func TestSharedDictLuaReplace(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("testdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
// 测试 replace 存在的 key
err = L.DoString(`
local dict = ngx.shared.DICT("testdict")
-- 先设置一个值
dict:set("mykey", "old_value")
-- replace 存在的 key 应该成功
local ok, err = dict:replace("mykey", "new_value")
assert(ok == true, "replace existing key should succeed: " .. tostring(err))
-- 验证值已更新
local val, _ = dict:get("mykey")
assert(val == "new_value", "value should be updated")
`)
require.NoError(t, err)
// 测试 replace 不存在的 key
err = L.DoString(`
local dict = ngx.shared.DICT("testdict")
-- replace 不存在的 key 应该失败
local ok, err = dict:replace("nonexistent", "value")
assert(ok == false, "replace nonexistent key should fail")
assert(err == "not found", "error should be 'not found', got: " .. tostring(err))
`)
require.NoError(t, err)
}
// TestSharedDictLuaReplaceWithTTL 测试 dict:replace 带 TTL
func TestSharedDictLuaReplaceWithTTL(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("ttldict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("ttldict")
dict:set("ttlkey", "original")
-- replace TTL0.1
local ok, err = dict:replace("ttlkey", "replaced", 0.1)
assert(ok == true, "replace with TTL should succeed: " .. tostring(err))
`)
require.NoError(t, err)
// 等待过期
time.Sleep(150 * time.Millisecond)
// 验证过期
err = L.DoString(`
local dict = ngx.shared.DICT("ttldict")
local val, err = dict:get("ttlkey")
assert(val == nil, "expired key should return nil")
`)
require.NoError(t, err)
}
// TestSharedDictLuaIndexAccess 测试 dict["key"] 索引访问
func TestSharedDictLuaIndexAccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("idxdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("idxdict")
-- 通过方法设置值
dict:set("foo", "bar")
-- 通过索引方式读取
local val = dict["foo"]
assert(val == "bar", "index access should work")
-- 通过 __newindex 设置值
dict["newkey"] = "newvalue"
local v = dict:get("newkey")
assert(v == "newvalue", "__newindex should work")
`)
require.NoError(t, err)
}
// TestSharedDictLuaIndexNotFound 测试索引访问不存在的 key
func TestSharedDictLuaIndexNotFound(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("idxdict2", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("idxdict2")
-- 读取不存在的 key 应该返回 nil
local val = dict["nonexistent"]
assert(val == nil, "nonexistent key should return nil")
`)
require.NoError(t, err)
}
// TestSharedDictLuaIndexMethodAccess 测试索引访问方法名
func TestSharedDictLuaIndexMethodAccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("metdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("metdict")
-- 通过索引访问方法
local getFn = dict["get"]
assert(type(getFn) == "function", "get should be a function")
local setFn = dict["set"]
assert(type(setFn) == "function", "set should be a function")
`)
require.NoError(t, err)
}
// TestSharedDictLuaFlushExpired 测试 dict:flush_expired
func TestSharedDictLuaFlushExpired(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("flushdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
// 设置一些带 TTL 的条目和一个永久的
err = L.DoString(`
local dict = ngx.shared.DICT("flushdict")
dict:set("exp1", "v1", 0.05)
dict:set("exp2", "v2", 0.05)
dict:set("perm", "permanent")
`)
require.NoError(t, err)
// 立即清除应该没有过期条目
err = L.DoString(`
local dict = ngx.shared.DICT("flushdict")
local count = dict:flush_expired()
assert(count == 0, "should have 0 expired immediately")
`)
require.NoError(t, err)
// 等待过期
time.Sleep(100 * time.Millisecond)
// 现在应该有 2 个过期条目
err = L.DoString(`
local dict = ngx.shared.DICT("flushdict")
local count = dict:flush_expired()
assert(count == 2, "should have 2 expired after wait")
-- 永久条目应该还在
local val, _ = dict:get("perm")
assert(val == "permanent", "permanent key should still exist")
`)
require.NoError(t, err)
}
// TestSharedDictLuaGetKeys 测试 dict:get_keys
func TestSharedDictLuaGetKeys(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("keysdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("keysdict")
dict:set("k1", "v1")
dict:set("k2", "v2")
dict:set("k3", "v3")
local keys = dict:get_keys()
assert(type(keys) == "table", "get_keys should return a table")
-- 当前实现返回空表
`)
require.NoError(t, err)
}
// TestSharedDictLuaSize 测试 dict:size
func TestSharedDictLuaSize(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("sizedict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("sizedict")
-- 初始为 0
assert(dict:size() == 0, "initial size should be 0")
dict:set("a", "1")
dict:set("b", "2")
dict:set("c", "3")
assert(dict:size() >= 3, "size should be at least 3")
`)
require.NoError(t, err)
}
// TestSharedDictLuaFreeSpace 测试 dict:free_space
func TestSharedDictLuaFreeSpace(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("freedict", 50)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("freedict")
-- 初始 free_space 应该等于 maxItems
local free = dict:free_space()
assert(free == 50, "initial free_space should be 50, got: " .. tostring(free))
-- 添加条目后 free_space 减少
dict:set("key1", "value1")
local free2 = dict:free_space()
assert(free2 == 49, "free_space should be 49 after 1 item, got: " .. tostring(free2))
`)
require.NoError(t, err)
}
// TestSharedDictLuaFlushAll 测试 dict:flush_all
func TestSharedDictLuaFlushAll(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("flushalldict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("flushalldict")
dict:set("a", "1")
dict:set("b", "2")
assert(dict:size() >= 2)
-- flush_all 清空所有
dict:flush_all()
assert(dict:size() == 0, "size should be 0 after flush_all")
`)
require.NoError(t, err)
}
// TestSharedDictLuaDictNotFound 测试请求不存在的 shared dict
func TestSharedDictLuaDictNotFound(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict, err = ngx.shared.DICT("nonexistent_dict")
assert(dict == nil, "nonexistent dict should return nil")
assert(err ~= nil, "should return error message")
`)
require.NoError(t, err)
}
// TestSharedDictLuaToString 测试 dict 的 tostring
func TestSharedDictLuaToString(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("tostringdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("tostringdict")
local s = tostring(dict)
assert(type(s) == "string", "tostring should return a string")
assert(string.match(s, "ngx.shared.dict"), "should contain 'ngx.shared.dict'")
`)
require.NoError(t, err)
}
// TestSharedDictLuaIncrNonNumber 测试 incr 对非数值
func TestSharedDictLuaIncrNonNumber(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("incrdict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
// 设置一个非数值
dict := engine.SharedDictManager().GetDict("incrdict")
dict.Set("notnum", "hello", 0)
err = L.DoString(`
local dict = ngx.shared.DICT("incrdict")
local val, err = dict:incr("notnum", 1)
-- 非数值 incr 返回 nil
assert(val == nil, "incr non-number should return nil")
`)
require.NoError(t, err)
}
// TestSharedDictLuaAddMultipleKeys 测试批量添加
func TestSharedDictLuaAddMultipleKeys(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("multiadd", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("multiadd")
-- 批量添加
for i = 1, 10 do
local ok, err = dict:add("key_" .. i, "val_" .. i)
assert(ok == true, "add should succeed: " .. tostring(err))
end
-- 验证所有 key 都存在
for i = 1, 10 do
local val, _ = dict:get("key_" .. i)
assert(val == "val_" .. i, "key_" .. i .. " should have correct value")
end
assert(dict:size() == 10, "size should be 10")
`)
require.NoError(t, err)
}
// TestSharedDictEngineAPI 测试通过 Engine API 创建共享字典
func TestSharedDictEngineAPI(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 通过 Engine 的 CreateSharedDict 方法创建
dict := engine.CreateSharedDict("enginedict", 50)
require.NotNil(t, dict)
// 验证可以通过 SharedDictManager 获取
dict2 := engine.SharedDictManager().GetDict("enginedict")
assert.Equal(t, dict, dict2, "should return same dict")
// 再次调用 CreateSharedDict 应返回已存在的
dict3 := engine.CreateSharedDict("enginedict", 100)
assert.Equal(t, dict, dict3, "should return existing dict regardless of maxItems")
}
// TestSharedDictLuaSetWithTTL 测试 set 带 TTL
func TestSharedDictLuaSetWithTTL(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("setttldict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("setttldict")
-- set TTL0.1
local ok, err = dict:set("ttlkey", "temp_value", 0.1)
assert(ok == true, "set with TTL should succeed: " .. tostring(err))
-- 立即获取应该成功
local val, exp = dict:get("ttlkey")
assert(val == "temp_value", "value should be correct")
assert(exp == 0 or exp == nil, "should not be expired yet")
`)
require.NoError(t, err)
// 等待过期
time.Sleep(150 * time.Millisecond)
err = L.DoString(`
local dict = ngx.shared.DICT("setttldict")
local val, err = dict:get("ttlkey")
assert(val == nil, "expired key should return nil")
assert(err == "expired", "error should be 'expired', got: " .. tostring(err))
`)
require.NoError(t, err)
}
// TestSharedDictLuaAddWithTTL 测试 add 带 TTL
func TestSharedDictLuaAddWithTTL(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("addttldict", 100)
L := engine.L
ngx := L.NewTable()
L.SetGlobal("ngx", ngx)
RegisterSharedDictAPI(L, engine.SharedDictManager(), ngx)
err = L.DoString(`
local dict = ngx.shared.DICT("addttldict")
-- add TTL
local ok, err = dict:add("addkey", "temp", 0.1)
assert(ok == true, "add with TTL should succeed: " .. tostring(err))
`)
require.NoError(t, err)
time.Sleep(150 * time.Millisecond)
err = L.DoString(`
local dict = ngx.shared.DICT("addttldict")
local val, err = dict:get("addkey")
assert(val == nil, "expired key should return nil")
`)
require.NoError(t, err)
}
// TestSharedDictClose 测试 SharedDictManager.Close
func TestSharedDictClose(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
_ = engine.CreateSharedDict("closedict", 100)
// Close 引擎会清理 sharedDictManager
engine.Close()
// Close 后不应该 panic即使再次 Close
engine.Close()
}

461
internal/lua/engine_test.go Normal file
View File

@ -0,0 +1,461 @@
// Package lua 提供 LuaEngine 测试,覆盖协程创建和管理、调度器、回调队列
package lua
import (
"strings"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/valyala/fasthttp"
glua "github.com/yuin/gopher-lua"
"github.com/yuin/gopher-lua/parse"
)
// engineCodeToProtoForTest 编译 Lua 代码为 FunctionProto测试辅助函数
func engineCodeToProtoForTest(src string) (*glua.FunctionProto, error) {
chunk, err := parse.Parse(strings.NewReader(src), "<test>")
if err != nil {
return nil, err
}
return glua.Compile(chunk, "<test>")
}
// TestNewEngineNilConfig 测试 NewEngine 使用 nil config 时使用默认配置
func TestNewEngineNilConfig(t *testing.T) {
engine, err := NewEngine(nil)
require.NoError(t, err)
defer engine.Close()
assert.NotNil(t, engine.L)
assert.NotNil(t, engine.codeCache)
assert.NotNil(t, engine.sharedDictManager)
assert.NotNil(t, engine.timerManager)
assert.NotNil(t, engine.locationManager)
assert.NotNil(t, engine.ctx)
assert.NotNil(t, engine.cancel)
}
// TestEngineCloseMultiple 测试多次 Close 不 panic
func TestEngineCloseMultiple(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
engine.Close()
engine.Close() // 第二次不应该 panic
}
// TestEngineCloseScheduler 测试关闭调度器
func TestEngineCloseScheduler(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 初始化调度器
err = engine.InitSchedulerLState()
require.NoError(t, err)
// 关闭调度器
engine.CloseScheduler()
// 再次关闭不应该 panic
engine.CloseScheduler()
}
// TestEngineNewCoroutine 测试创建协程
func TestEngineNewCoroutine(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
var req fasthttp.Request
req.Header.SetMethod("GET")
req.Header.SetRequestURI("/test")
ctx := &fasthttp.RequestCtx{}
ctx.Init(&req, nil, nil)
coro, err := engine.NewCoroutine(ctx)
require.NoError(t, err)
require.NotNil(t, coro)
assert.NotNil(t, coro.Co)
assert.NotNil(t, coro.Engine)
assert.Equal(t, ctx, coro.RequestCtx)
coro.Close()
}
// TestEngineNewCoroutineNilContext 测试创建带 nil 请求上下文的协程
func TestEngineNewCoroutineNilContext(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
coro, err := engine.NewCoroutine(nil)
require.NoError(t, err)
require.NotNil(t, coro)
assert.Nil(t, coro.RequestCtx)
coro.Close()
}
// TestEngineActiveCoroutines 测试活跃协程计数
func TestEngineActiveCoroutines(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 初始应该为 0
assert.Equal(t, int32(0), engine.ActiveCoroutines())
// 创建一个协程
coro, err := engine.NewCoroutine(nil)
require.NoError(t, err)
assert.Equal(t, int32(1), engine.ActiveCoroutines())
// 关闭协程
coro.Close()
assert.Equal(t, int32(0), engine.ActiveCoroutines())
}
// TestEngineCoroutinePoolWarmup 测试协程池预热
func TestEngineCoroutinePoolWarmup(t *testing.T) {
config := DefaultConfig()
config.CoroutinePoolWarmup = 10
engine, err := NewEngine(config)
require.NoError(t, err)
defer engine.Close()
// 预热后池中应该有 10 个对象
// 直接验证 engine 创建成功即可,预热是内部实现
assert.NotNil(t, engine)
}
// TestEngineStatsAfterOperations 测试引擎统计信息在操作后更新
func TestEngineStatsAfterOperations(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 创建并关闭多个协程
for range 5 {
coro, err := engine.NewCoroutine(nil)
require.NoError(t, err)
coro.Close()
}
stats := engine.Stats()
assert.Equal(t, uint64(5), stats.CoroutinesCreated)
assert.Equal(t, uint64(5), stats.CoroutinesClosed)
}
// TestEngineMaxCoroutinesExceeded 测试超过最大并发协程限制
func TestEngineMaxCoroutinesExceeded(t *testing.T) {
config := &Config{
MaxConcurrentCoroutines: 2,
MaxExecutionTime: 5 * time.Second,
}
engine, err := NewEngine(config)
require.NoError(t, err)
defer engine.Close()
// 创建 2 个协程
coro1, err := engine.NewCoroutine(nil)
require.NoError(t, err)
coro2, err := engine.NewCoroutine(nil)
require.NoError(t, err)
// 第 3 个应该失败
coro3, err := engine.NewCoroutine(nil)
assert.Error(t, err)
assert.Nil(t, coro3)
assert.Contains(t, err.Error(), "max concurrent coroutines exceeded")
coro1.Close()
coro2.Close()
}
// TestEngineNewCoroutineFails 测试 NewThread 返回 nil 的情况
// 这个场景在实际中很难触发,我们验证引擎在正常情况下不会返回 nil
func TestEngineNewCoroutineSuccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 创建大量协程,验证稳定性
for range 100 {
coro, err := engine.NewCoroutine(nil)
require.NoError(t, err)
require.NotNil(t, coro.Co)
coro.Close()
}
}
// TestEngineCodeCacheAccess 测试 CodeCache 访问器
func TestEngineCodeCacheAccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
cache := engine.CodeCache()
require.NotNil(t, cache)
// 编译一段脚本
proto, err := cache.GetOrCompileInline("return 1 + 1")
require.NoError(t, err)
require.NotNil(t, proto)
}
// TestEngineSharedDictManagerAccess 测试 SharedDictManager 访问器
func TestEngineSharedDictManagerAccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
mgr := engine.SharedDictManager()
require.NotNil(t, mgr)
// 创建共享字典
dict := engine.CreateSharedDict("test", 100)
require.NotNil(t, dict)
// 通过 manager 获取
dict2 := mgr.GetDict("test")
assert.Equal(t, dict, dict2)
}
// TestEngineTimerManagerAccess 测试 TimerManager 访问器
func TestEngineTimerManagerAccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
mgr := engine.TimerManager()
require.NotNil(t, mgr)
}
// TestEngineLocationManagerAccess 测试 LocationManager 访问器
func TestEngineLocationManagerAccess(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
mgr := engine.LocationManager()
require.NotNil(t, mgr)
// 注册一个 location
mgr.Register("/test", func(ctx *fasthttp.RequestCtx) {
ctx.SetStatusCode(200)
})
// 验证已注册
_, err2 := mgr.Capture(&fasthttp.RequestCtx{}, "/test", nil)
assert.NoError(t, err2)
}
// TestEngineSchedulerLoop 测试调度器循环处理回调
func TestEngineSchedulerLoop(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 初始化调度器
err = engine.InitSchedulerLState()
require.NoError(t, err)
// 创建一个简单的回调函数并加入队列
proto, err := engineCodeToProtoForTest("return 42")
require.NoError(t, err)
entry := &CallbackEntry{
proto: proto,
args: []glua.LValue{},
}
ok := engine.EnqueueCallback(entry)
assert.True(t, ok, "enqueue should succeed")
// 给调度器一些时间处理
time.Sleep(50 * time.Millisecond)
// 关闭调度器
engine.CloseScheduler()
}
// TestEngineEnqueueCallbackFull 测试回调队列满时入队失败
func TestEngineEnqueueCallbackFull(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
err = engine.InitSchedulerLState()
require.NoError(t, err)
proto, err := engineCodeToProtoForTest("return 1")
require.NoError(t, err)
// 填满回调队列(默认 1024 容量)
full := false
for range 1024 {
if !engine.EnqueueCallback(&CallbackEntry{proto: proto, args: []glua.LValue{}}) {
full = true
break
}
}
// 在正常环境下 1024 个应该能填满队列
// 最后一个应该失败
last := engine.EnqueueCallback(&CallbackEntry{proto: proto, args: []glua.LValue{}})
// 可能为 false队列满或 true如果调度器已经开始消费
// 不强制断言,因为调度器可能在消费
_ = full
_ = last
engine.CloseScheduler()
}
// TestEngineExecuteCallbackNilScheduler 测试 executeCallback 时 schedulerLState 为 nil
func TestEngineExecuteCallbackNilScheduler(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 不调用 InitSchedulerLStateschedulerLState 为 nil
// executeCallback 会在 schedulerLState == nil 时直接返回
engine.executeCallback(&CallbackEntry{
proto: nil,
args: []glua.LValue{},
})
// 不应 panic
}
// TestEngineExecuteCallbackPanicRecovery 测试 executeCallback 中 panic 的恢复
func TestEngineExecuteCallbackPanicRecovery(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
err = engine.InitSchedulerLState()
require.NoError(t, err)
// 传入 nil protoexecuteCallback 内部应该不会 panic
// 因为 recover() 会捕获
engine.executeCallback(&CallbackEntry{
proto: nil,
args: []glua.LValue{},
})
engine.CloseScheduler()
}
// TestEngineSchedulerLoopExitOnClose 测试调度器在引擎关闭时退出
func TestEngineSchedulerLoopExitOnClose(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
err = engine.InitSchedulerLState()
require.NoError(t, err)
// 关闭引擎(会触发 cancel 信号)
engine.Close()
// 给调度器一些时间退出
time.Sleep(50 * time.Millisecond)
// 再次关闭不应该 panic
engine.Close()
}
// TestEngineSchedulerLoopExitOnChannelClose 测试调度器在回调队列关闭时退出
func TestEngineSchedulerLoopExitOnChannelClose(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
err = engine.InitSchedulerLState()
require.NoError(t, err)
// 直接关闭调度器(关闭 callbackQueue channel
engine.CloseScheduler()
// 给调度器一些时间退出
time.Sleep(50 * time.Millisecond)
engine.Close()
}
// TestEngineCoroutineExecutionContext 测试协程的执行上下文和超时控制
func TestEngineCoroutineExecutionContext(t *testing.T) {
config := &Config{
MaxConcurrentCoroutines: 100,
MaxExecutionTime: 100 * time.Millisecond,
}
engine, err := NewEngine(config)
require.NoError(t, err)
defer engine.Close()
coro, err := engine.NewCoroutine(nil)
require.NoError(t, err)
// 验证执行上下文已设置
assert.NotNil(t, coro.ExecutionContext)
assert.NotNil(t, coro.Cancel)
coro.Close()
}
// TestEngineReleaseCoroutineNilSafety 测试 releaseCoroutine 对 nil 的安全处理
func TestEngineReleaseCoroutineNilSafety(t *testing.T) {
engine, err := NewEngine(DefaultConfig())
require.NoError(t, err)
defer engine.Close()
// 释放 nil 协程不应 panic
engine.releaseCoroutine(nil)
}
// TestEngineCoroutinePoolReuse 测试协程池复用
func TestEngineCoroutinePoolReuse(t *testing.T) {
engine, err := NewEngine(&Config{
MaxConcurrentCoroutines: 1000,
MaxExecutionTime: 5 * time.Second,
CoroutinePoolWarmup: 5,
})
require.NoError(t, err)
defer engine.Close()
// 创建并释放多次
for range 10 {
coro, err := engine.NewCoroutine(nil)
require.NoError(t, err)
coro.Close()
}
stats := engine.Stats()
assert.Equal(t, uint64(10), stats.CoroutinesCreated)
assert.Equal(t, uint64(10), stats.CoroutinesClosed)
}
// TestEngineConfigOverride 测试配置覆盖
func TestEngineConfigOverride(t *testing.T) {
config := &Config{
MaxConcurrentCoroutines: 500,
MaxExecutionTime: 10 * time.Second,
CodeCacheSize: 2000,
CodeCacheTTL: 5 * time.Minute,
CoroutineStackSize: 64,
MinimizeStackMemory: true,
}
engine, err := NewEngine(config)
require.NoError(t, err)
defer engine.Close()
assert.Equal(t, 500, engine.maxCoroutines)
assert.Equal(t, 10*time.Second, config.MaxExecutionTime)
}