xfy
a4a820ab24
添加 location 子请求实现： - LocationManager: location handler 注册与管理 - ngx.location.capture: 发起同步子请求 - 支持 method/body/headers 参数配置 - 返回 status/body/headers 结果结构 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
119 lines
2.8 KiB
Go
119 lines
2.8 KiB
Go
// Package lua 提供 Lua 脚本嵌入能力
|
|
package lua
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
func TestLocationManagerRegister(t *testing.T) {
|
|
manager := NewLocationManager()
|
|
require.NotNil(t, manager)
|
|
|
|
// 注册 location
|
|
handler := func(ctx *fasthttp.RequestCtx) {
|
|
ctx.WriteString("test response")
|
|
}
|
|
manager.Register("/test", handler)
|
|
|
|
// 验证注册成功
|
|
manager.mu.Lock()
|
|
_, ok := manager.handlers["/test"]
|
|
manager.mu.Unlock()
|
|
assert.True(t, ok)
|
|
}
|
|
|
|
func TestLocationManagerCapture(t *testing.T) {
|
|
manager := NewLocationManager()
|
|
|
|
// 注册 location
|
|
handler := func(ctx *fasthttp.RequestCtx) {
|
|
ctx.SetStatusCode(200)
|
|
ctx.SetBodyString("hello from subrequest")
|
|
ctx.Response.Header.Set("X-Custom", "value")
|
|
}
|
|
manager.Register("/api/sub", handler)
|
|
|
|
// 创建父请求上下文
|
|
parentCtx := &fasthttp.RequestCtx{}
|
|
parentCtx.Request.SetRequestURI("/parent")
|
|
|
|
// 执行子请求
|
|
result, err := manager.Capture(parentCtx, "/api/sub", nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
|
|
assert.Equal(t, 200, result.Status)
|
|
assert.Equal(t, "hello from subrequest", string(result.Body))
|
|
assert.Equal(t, "value", result.Headers["X-Custom"])
|
|
}
|
|
|
|
func TestLocationManagerCaptureNotFound(t *testing.T) {
|
|
manager := NewLocationManager()
|
|
|
|
parentCtx := &fasthttp.RequestCtx{}
|
|
|
|
// 执行不存在的 location
|
|
result, err := manager.Capture(parentCtx, "/notexist", nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
|
|
assert.Equal(t, 404, result.Status)
|
|
}
|
|
|
|
func TestLocationManagerCaptureWithOptions(t *testing.T) {
|
|
manager := NewLocationManager()
|
|
|
|
// 注册 location
|
|
handler := func(ctx *fasthttp.RequestCtx) {
|
|
ctx.SetStatusCode(200)
|
|
ctx.WriteString("method: " + string(ctx.Method()) + ", body: " + string(ctx.PostBody()))
|
|
}
|
|
manager.Register("/echo", handler)
|
|
|
|
parentCtx := &fasthttp.RequestCtx{}
|
|
parentCtx.Request.SetRequestURI("/parent")
|
|
|
|
// 使用自定义选项
|
|
opts := map[string]interface{}{
|
|
"method": "POST",
|
|
"body": "test body",
|
|
}
|
|
|
|
result, err := manager.Capture(parentCtx, "/echo", opts)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
|
|
assert.Equal(t, 200, result.Status)
|
|
assert.Contains(t, string(result.Body), "method: POST")
|
|
assert.Contains(t, string(result.Body), "body: test body")
|
|
}
|
|
|
|
func TestLocationLuaAPI(t *testing.T) {
|
|
engine, err := NewEngine(DefaultConfig())
|
|
require.NoError(t, err)
|
|
defer engine.Close()
|
|
|
|
L := engine.L
|
|
|
|
// 注册 ngx.location API
|
|
ngx := L.NewTable()
|
|
L.SetGlobal("ngx", ngx)
|
|
RegisterLocationAPI(L, engine.LocationManager(), ngx)
|
|
|
|
// 测试 ngx.location.capture
|
|
err = L.DoString(`
|
|
-- 创建模拟的 location 结果
|
|
local result = ngx.location.capture("/test")
|
|
|
|
-- 验证结果结构
|
|
assert(result ~= nil)
|
|
assert(result.status ~= nil)
|
|
assert(result.body ~= nil)
|
|
`)
|
|
require.NoError(t, err)
|
|
}
|