diff --git a/internal/querycoordv2/meta/leader_view_manager_test.go b/internal/querycoordv2/meta/leader_view_manager_test.go index 1cdde5cb0e..ac67cd5806 100644 --- a/internal/querycoordv2/meta/leader_view_manager_test.go +++ b/internal/querycoordv2/meta/leader_view_manager_test.go @@ -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) }