mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package segments
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/milvus-io/milvus/internal/proto/internalpb"
|
|
"github.com/milvus-io/milvus/internal/proto/querypb"
|
|
)
|
|
|
|
type ReducerFactorySuite struct {
|
|
suite.Suite
|
|
ir internalReducer
|
|
sr segCoreReducer
|
|
ok bool
|
|
}
|
|
|
|
func (suite *ReducerFactorySuite) SetupTest() {}
|
|
|
|
func (suite *ReducerFactorySuite) TearDownTest() {}
|
|
|
|
func TestReducerFactorySuite(t *testing.T) {
|
|
suite.Run(t, new(ReducerFactorySuite))
|
|
}
|
|
|
|
func (suite *ReducerFactorySuite) TestCreateInternalReducer() {
|
|
req := &querypb.QueryRequest{
|
|
Req: &internalpb.RetrieveRequest{
|
|
IsCount: false,
|
|
},
|
|
}
|
|
|
|
suite.ir = CreateInternalReducer(req, nil)
|
|
_, suite.ok = suite.ir.(*defaultLimitReducer)
|
|
suite.True(suite.ok)
|
|
|
|
req.Req.IsCount = true
|
|
|
|
suite.ir = CreateInternalReducer(req, nil)
|
|
_, suite.ok = suite.ir.(*cntReducer)
|
|
suite.True(suite.ok)
|
|
|
|
req.GetReq().IsCount = false
|
|
req.GetReq().IterationExtensionReduceRate = 10
|
|
req.GetReq().Limit = 10
|
|
suite.ir = CreateInternalReducer(req, nil)
|
|
extReducer, typeOk := suite.ir.(*extensionLimitReducer)
|
|
suite.True(typeOk)
|
|
suite.Equal(int64(100), extReducer.extendedLimit)
|
|
}
|
|
|
|
func (suite *ReducerFactorySuite) TestCreateSegCoreReducer() {
|
|
|
|
req := &querypb.QueryRequest{
|
|
Req: &internalpb.RetrieveRequest{
|
|
IsCount: false,
|
|
},
|
|
}
|
|
|
|
suite.sr = CreateSegCoreReducer(req, nil)
|
|
_, suite.ok = suite.sr.(*defaultLimitReducerSegcore)
|
|
suite.True(suite.ok)
|
|
|
|
req.Req.IsCount = true
|
|
suite.sr = CreateSegCoreReducer(req, nil)
|
|
_, suite.ok = suite.sr.(*cntReducerSegCore)
|
|
suite.True(suite.ok)
|
|
|
|
req.GetReq().IsCount = false
|
|
req.GetReq().IterationExtensionReduceRate = 10
|
|
req.GetReq().Limit = 10
|
|
suite.sr = CreateSegCoreReducer(req, nil)
|
|
extReducer, typeOk := suite.sr.(*extensionLimitSegcoreReducer)
|
|
suite.True(typeOk)
|
|
suite.Equal(int64(100), extReducer.extendedLimit)
|
|
}
|