Update L3 cases and nightly qn replicas (milvus-io#547)

Signed-off-by: ThreadDao <yufen.zong@zilliz.com>
ThreadDao · Aug 17, 2023 · b2aa07f · b2aa07f
1 parent 0c4a500
commit b2aa07f
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 5 deletions.
diff --git a/ci/milvus-operator/milvus-nightly.yaml b/ci/milvus-operator/milvus-nightly.yaml
@@ -119,7 +119,7 @@ spec:
   components:
     image: ${IMAGE_REPO}:${IMAGE_TAG}
     queryNode:
-      replicas: 2
+      replicas: 4
     dataNode:
       replicas: 1
     indexNode:

diff --git a/test/testcases/resource_group_test.go b/test/testcases/resource_group_test.go
@@ -22,6 +22,14 @@ const configQnNodes = int32(4)
 const newRgNode = int32(2)
 
 func resetRgs(t *testing.T, ctx context.Context, mc *base.MilvusClient) {
+	// release and drop all collections
+	collections, _ := mc.ListCollections(ctx)
+	for _, coll := range collections {
+		mc.ReleaseCollection(ctx, coll.Name)
+		err := mc.DropCollection(ctx, coll.Name)
+		common.CheckErr(t, err, true)
+	}
+
 	// reset resource groups
 	rgs, errList := mc.ListResourceGroups(ctx)
 	common.CheckErr(t, errList, true)
@@ -252,9 +260,9 @@ func TestTransferInvalidNodes(t *testing.T) {
 		errMsg   string
 	}
 	invalidNodes := []invalidNodesStruct{
-		{nodesNum: 0, errMsg: "transfer node num can't be"},
-		{nodesNum: -1, errMsg: "transfer node num can't be"},
-		{nodesNum: 99, errMsg: "failed to transfer node between resource group, err=nodes not enough"},
+		{nodesNum: 0, errMsg: "transfer node num can't be [0]"},
+		{nodesNum: -1, errMsg: "transfer node num can't be [-1]"},
+		{nodesNum: 99, errMsg: "nodes not enough"},
 	}
 	// transfer node
 	for _, invalidNode := range invalidNodes {
@@ -343,8 +351,27 @@ func TestTransferReplicas(t *testing.T) {
 	}
 	common.CheckResourceGroup(t, newRg, expRg)
 
+	// drop new rg that loaded collection
+	err := mc.DropResourceGroup(ctx, rgName)
+	common.CheckErr(t, err, false, "some replicas still loaded in resource group")
+
 	// search
-	// todo
+	sp, err := entity.NewIndexHNSWSearchParam(74)
+	searchRes, _ := mc.Search(
+		ctx, collName,
+		[]string{common.DefaultPartition},
+		"",
+		[]string{common.DefaultFloatFieldName},
+		common.GenSearchVectors(common.DefaultNq, common.DefaultDim, entity.FieldTypeFloatVector),
+		common.DefaultFloatVecFieldName,
+		entity.L2,
+		common.DefaultTopK,
+		sp,
+	)
+	// check search result contains search vector, which from all partitions
+	common.CheckErr(t, err, true)
+	common.CheckSearchResult(t, searchRes, common.DefaultNq, common.DefaultTopK)
+
 }
 
 // test transfer replica of not existed collection