|
|
package matcher
|
|
|
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
"testing"
|
|
|
|
|
|
"github.com/noahlann/nnet/internal/codec"
|
|
|
"github.com/noahlann/nnet/internal/request"
|
|
|
"github.com/noahlann/nnet/internal/response"
|
|
|
ctxpkg "github.com/noahlann/nnet/pkg/context"
|
|
|
routerpkg "github.com/noahlann/nnet/pkg/router"
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
)
|
|
|
|
|
|
func TestFrameDataMatcher(t *testing.T) {
|
|
|
// 创建测试Context
|
|
|
conn := &mockConnection{id: "test-conn"}
|
|
|
|
|
|
// 创建包含JSON数据的请求（使用float64以匹配JSON解析后的类型）
|
|
|
data := map[string]interface{}{
|
|
|
"user": map[string]interface{}{
|
|
|
"id": float64(1), // JSON解析后数字是float64类型
|
|
|
"name": "test",
|
|
|
},
|
|
|
}
|
|
|
jsonData, _ := json.Marshal(data)
|
|
|
req := request.New(jsonData, nil)
|
|
|
// 使用AsRequestSetter设置Data（模拟已解析的JSON数据）
|
|
|
if setter := request.AsRequestSetter(req); setter != nil {
|
|
|
setter.SetData(data)
|
|
|
}
|
|
|
|
|
|
codecRegistry := codec.NewRegistry()
|
|
|
resp := response.New(conn, nil, codecRegistry, "json")
|
|
|
ctx := ctxpkg.New(nil, conn, req, resp)
|
|
|
|
|
|
// 创建帧数据匹配器
|
|
|
matcher := NewFrameDataMatcher("user.id", "==", float64(1))
|
|
|
|
|
|
// 测试匹配
|
|
|
input := routerpkg.MatchInput{
|
|
|
Raw: jsonData,
|
|
|
DataBytes: jsonData,
|
|
|
Data: data,
|
|
|
}
|
|
|
assert.True(t, matcher.Match(input, ctx), "Expected match for user.id == 1")
|
|
|
}
|
|
|
|
|
|
func TestFrameDataMatcherNotMatch(t *testing.T) {
|
|
|
conn := &mockConnection{id: "test-conn"}
|
|
|
|
|
|
data := map[string]interface{}{
|
|
|
"user": map[string]interface{}{
|
|
|
"id": float64(2), // JSON解析后数字是float64类型
|
|
|
"name": "test",
|
|
|
},
|
|
|
}
|
|
|
jsonData, _ := json.Marshal(data)
|
|
|
req := request.New(jsonData, nil)
|
|
|
if setter := request.AsRequestSetter(req); setter != nil {
|
|
|
setter.SetData(data)
|
|
|
}
|
|
|
|
|
|
codecRegistry := codec.NewRegistry()
|
|
|
resp := response.New(conn, nil, codecRegistry, "json")
|
|
|
ctx := ctxpkg.New(nil, conn, req, resp)
|
|
|
|
|
|
matcher := NewFrameDataMatcher("user.id", "==", float64(1))
|
|
|
input := routerpkg.MatchInput{
|
|
|
Raw: jsonData,
|
|
|
DataBytes: jsonData,
|
|
|
Data: data,
|
|
|
}
|
|
|
assert.False(t, matcher.Match(input, ctx), "Expected no match")
|
|
|
}
|
|
|
|
|
|
func TestFrameDataMatcherPriority(t *testing.T) {
|
|
|
matcher := NewFrameDataMatcher("path", "==", "value")
|
|
|
assert.Equal(t, 150, matcher.Priority(), "Expected priority to be 150")
|
|
|
}
|