milvus/internal/proxy/reducer_test.go
congqixia 709594f158
enhance: [2.5] Use v2 package name for pkg module (#40117)
Cherry-pick from master
pr: #39990
Related to #39095

https://go.dev/doc/modules/version-numbers

Update pkg version according to golang dep version convention

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2025-02-23 00:46:01 +08:00

32 lines
621 B
Go

package proxy
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
"github.com/milvus-io/milvus/pkg/v2/proto/planpb"
)
func Test_createMilvusReducer(t *testing.T) {
n := &planpb.PlanNode{
Node: &planpb.PlanNode_Query{
Query: &planpb.QueryPlanNode{
IsCount: false,
},
},
}
var r milvusReducer
ctx := context.Background()
r = createMilvusReducer(ctx, nil, nil, nil, n, "")
_, ok := r.(*defaultLimitReducer)
assert.True(t, ok)
n.Node.(*planpb.PlanNode_Query).Query.IsCount = true
r = createMilvusReducer(ctx, nil, nil, nil, n, "")
_, ok = r.(*cntReducer)
assert.True(t, ok)
}