fix: ut failure (#32119)

issue: #30647

Signed-off-by: chyezh <chyezh@outlook.com>
This commit is contained in:
chyezh 2024-04-10 17:23:27 +08:00 committed by GitHub
parent dbab9c5096
commit 0be67e7f99
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -23,7 +23,6 @@ import (
"github.com/stretchr/testify/suite"
"github.com/milvus-io/milvus/internal/proto/querypb"
"github.com/milvus-io/milvus/pkg/util/typeutil"
)
type LeaderViewManagerSuite struct {
@ -134,14 +133,11 @@ func (suite *LeaderViewManagerSuite) TestGetByFilter() {
// Test WithReplica
for i, collectionID := range suite.collections {
replica := &Replica{
Replica: &querypb.Replica{
replica := newReplica(&querypb.Replica{
ID: int64(i),
CollectionID: collectionID,
Nodes: suite.nodes,
},
nodes: typeutil.NewUniqueSet(suite.nodes...),
}
})
views := suite.mgr.GetByFilter(WithReplica2LeaderView(replica))
suite.Len(views, 2)
}