fix integration test

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
weiliu1031 · Dec 19, 2024 · 731cb0a · 731cb0a
1 parent 59a23ec
commit 731cb0a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/internal/querycoordv2/services.go b/internal/querycoordv2/services.go
@@ -1240,7 +1240,7 @@ func (s *Server) UpdateLoadConfig(ctx context.Context, req *querypb.UpdateLoadCo
 	jobs := make([]job.Job, 0, len(req.GetCollectionIDs()))
 	for _, collectionID := range req.GetCollectionIDs() {
 		collection := s.meta.GetCollection(collectionID)
-		if collection == nil{
+		if collection == nil {
 			err := merr.WrapErrCollectionNotLoaded(collectionID)
 			log.Warn("failed to update load config", zap.Error(err))
 			continue

diff --git a/tests/integration/replicas/load/load_test.go b/tests/integration/replicas/load/load_test.go
@@ -921,7 +921,6 @@ func (s *LoadTestSuite) TestDynamicUpdateLoadConfigs_OnLoadingCollection() {
 	}, 30*time.Second, time.Second)
 
 	// trigger collection loading, and modify collection's load config during loading
-	s.loadCollection(collectionName, dbName, 3, rgs[:3])
 	loadStatus, err := s.Cluster.Proxy.LoadCollection(ctx, &milvuspb.LoadCollectionRequest{
 		DbName:         dbName,
 		CollectionName: collectionName,
@@ -930,7 +929,6 @@ func (s *LoadTestSuite) TestDynamicUpdateLoadConfigs_OnLoadingCollection() {
 	})
 	s.NoError(err)
 	s.True(merr.Ok(loadStatus))
-	s.loadCollection(collectionName, dbName, 3, rgs[:3])
 	loadStatus, err = s.Cluster.Proxy.LoadCollection(ctx, &milvuspb.LoadCollectionRequest{
 		DbName:         dbName,
 		CollectionName: collectionName,
@@ -939,7 +937,6 @@ func (s *LoadTestSuite) TestDynamicUpdateLoadConfigs_OnLoadingCollection() {
 	})
 	s.NoError(err)
 	s.True(merr.Ok(loadStatus))
-	s.loadCollection(collectionName, dbName, 3, rgs[:3])
 	loadStatus, err = s.Cluster.Proxy.LoadCollection(ctx, &milvuspb.LoadCollectionRequest{
 		DbName:         dbName,
 		CollectionName: collectionName,