mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-04 21:09:06 +08:00
29 lines
582 B
Go
29 lines
582 B
Go
|
package proxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/milvus-io/milvus/internal/proto/planpb"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_createMilvusReducer(t *testing.T) {
|
||
|
|
n := &planpb.PlanNode{
|
||
|
|
Node: &planpb.PlanNode_Query{
|
||
|
|
Query: &planpb.QueryPlanNode{
|
||
|
|
IsCount: false,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
var r milvusReducer
|
||
|
|
|
||
|
|
r = createMilvusReducer(nil, nil, nil, nil, n, "")
|
||
|
|
_, ok := r.(*defaultLimitReducer)
|
||
|
|
assert.True(t, ok)
|
||
|
|
|
||
|
|
n.Node.(*planpb.PlanNode_Query).Query.IsCount = true
|
||
|
|
r = createMilvusReducer(nil, nil, nil, nil, n, "")
|
||
|
|
_, ok = r.(*cntReducer)
|
||
|
|
assert.True(t, ok)
|
||
|
|
}
|