Jelajahi Sumber

modify testcase for 2.1.0-beta4 (#355)

Signed-off-by: yongpengli-z <yongpeng.li@zilliz.com>
yongpengli-z 2 tahun lalu
induk
melakukan
221a01e75b

+ 1 - 1
tests/milvustest/allure-results/environment.properties

@@ -1,3 +1,3 @@
-milvus.url=10.98.0.8
+milvus.url=10.100.31.102
 milvus.version=2.1.0
 milvus-jdk-java.version=2.1.0

+ 21 - 17
tests/milvustest/src/test/java/com/zilliz/milvustest/index/DropIndexTest.java

@@ -47,7 +47,7 @@ public class DropIndexTest extends BaseTest {
     milvusClient.dropIndex(
         DropIndexParam.newBuilder()
             .withCollectionName(collection)
-            .withIndexName(CommonData.defaultVectorField)
+            .withIndexName(CommonData.defaultIndex)
             .build());
     milvusClient.dropCollection(
         DropCollectionParam.newBuilder().withCollectionName(collection).build());
@@ -60,64 +60,68 @@ public class DropIndexTest extends BaseTest {
         milvusClient.dropIndex(
             DropIndexParam.newBuilder()
                 .withCollectionName(collection)
-                .withIndexName(CommonData.defaultVectorField)
+                .withIndexName(CommonData.defaultIndex)
                 .build());
     Assert.assertEquals(rpcStatusR.getStatus().intValue(), 0);
     Assert.assertEquals(rpcStatusR.getData().getMsg(), "Success");
   }
 
   @Severity(SeverityLevel.MINOR)
-  @Test(description = "drop index without field name", expectedExceptions = ParamException.class)
-  public void dropIndexWithoutFieldName() {
+  @Test(description = "drop index without index name")
+  public void dropIndexWithoutIndexName() {
     R<RpcStatus> rpcStatusR =
         milvusClient.dropIndex(DropIndexParam.newBuilder().withCollectionName(collection).build());
+    Assert.assertEquals(rpcStatusR.getStatus().intValue(),25);
+    Assert.assertTrue(rpcStatusR.getException().getMessage().contains("Index doesn't exist"));
   }
 
   @Severity(SeverityLevel.NORMAL)
-  @Test(description = "drop index with error field name")
-  public void dropIndexWithErrorFieldName() {
+  @Test(description = "drop index with error index name")
+  public void dropIndexWithErrorIndexName() {
     R<RpcStatus> rpcStatusR =
         milvusClient.dropIndex(
             DropIndexParam.newBuilder()
                 .withCollectionName(collection)
                 .withIndexName("book_id")
                 .build());
-    Assert.assertEquals(rpcStatusR.getStatus().intValue(), 0);
-    Assert.assertEquals(rpcStatusR.getData().getMsg(), "Success");
+    Assert.assertEquals(rpcStatusR.getStatus().intValue(), 25);
+    Assert.assertTrue(rpcStatusR.getException().getMessage().contains( "Index doesn't exist"));
   }
 
   @Severity(SeverityLevel.NORMAL)
-  @Test(description = "drop index with nonexistent field name")
-  public void dropIndexWithNonexistentFieldName() {
+  @Test(description = "drop index with nonexistent index name")
+  public void dropIndexWithNonexistentIndexName() {
     R<RpcStatus> rpcStatusR =
         milvusClient.dropIndex(
             DropIndexParam.newBuilder()
                 .withCollectionName(collection)
                 .withIndexName("nonexistent")
                 .build());
-    Assert.assertEquals(rpcStatusR.getStatus().intValue(), 1);
+    Assert.assertEquals(rpcStatusR.getStatus().intValue(), 25);
     Assert.assertEquals(
         rpcStatusR.getException().getMessage(),
-        "DropIndex failed: collection " + collection + " doesn't have filed nonexistent");
+        "Index doesn't exist");
   }
 
   @Severity(SeverityLevel.MINOR)
   @Test(
       description = "describe index after drop index",
-      dependsOnMethods = "dropIndexSuccess",
-      enabled = false)
+      dependsOnMethods = "dropIndexSuccess")
   public void describeIndexAfterDropIndex() {
     R<DescribeIndexResponse> describeIndexResponseR =
         milvusClient.describeIndex(
             DescribeIndexParam.newBuilder()
                 .withCollectionName(collection)
-                .withIndexName(CommonData.defaultVectorField)
+                .withIndexName(CommonData.defaultIndex)
                 .build());
-    Assert.assertEquals(describeIndexResponseR.getStatus().intValue(), 0);
+    Assert.assertEquals(describeIndexResponseR.getStatus().intValue(), 25);
+    Assert.assertTrue(
+            describeIndexResponseR.getException().getMessage().contains(
+            "index not exist"));
   }
 
   @Severity(SeverityLevel.NORMAL)
-  @Test(description = "create index after drop index", dependsOnMethods = "dropIndexSuccess")
+  @Test(description = "create index after drop index", dependsOnMethods = {"dropIndexSuccess","describeIndexAfterDropIndex"})
   public void createIndexAfterDrop(){
     R<RpcStatus> indexR = milvusClient.createIndex(
             CreateIndexParam.newBuilder()

+ 5 - 15
tests/milvustest/src/test/java/com/zilliz/milvustest/index/GetIndexBuildProgressTest.java

@@ -14,7 +14,6 @@ import io.milvus.param.index.CreateIndexParam;
 import io.milvus.param.index.DescribeIndexParam;
 import io.milvus.param.index.DropIndexParam;
 import io.milvus.param.index.GetIndexBuildProgressParam;
-import io.milvus.response.DescIndexResponseWrapper;
 import io.qameta.allure.*;
 import org.testng.Assert;
 import org.testng.annotations.AfterClass;
@@ -48,7 +47,7 @@ public class GetIndexBuildProgressTest extends BaseTest {
     milvusClient.dropIndex(
         DropIndexParam.newBuilder()
             .withCollectionName(collection)
-            .withIndexName(CommonData.defaultVectorField)
+            .withIndexName(CommonData.defaultIndex)
             .build());
     milvusClient.dropCollection(
         DropCollectionParam.newBuilder().withCollectionName(collection).build());
@@ -58,21 +57,12 @@ public class GetIndexBuildProgressTest extends BaseTest {
   @Test(description = "Get Index Build Progress")
   @Issue("https://github.com/milvus-io/milvus-sdk-java/issues/299")
   public void getIndexBuildProgressTest() {
-    R<DescribeIndexResponse> describeIndexResponseR =
-        milvusClient.describeIndex(
-            DescribeIndexParam.newBuilder()
-                .withCollectionName(collection)
-                .withIndexName(CommonData.defaultVectorField)
-                .build());
-    Assert.assertEquals(describeIndexResponseR.getStatus().intValue(), 0);
-    DescribeIndexResponse describeIndexResponse = describeIndexResponseR.getData();
-    DescIndexResponseWrapper descIndexResponseWrapper =
-        new DescIndexResponseWrapper(describeIndexResponse);
-    System.out.println(descIndexResponseWrapper);
-
     R<GetIndexBuildProgressResponse> indexBuildProgress =
         milvusClient.getIndexBuildProgress(
-            GetIndexBuildProgressParam.newBuilder().withCollectionName(collection).build());
+            GetIndexBuildProgressParam.newBuilder()
+                .withCollectionName(collection)
+                .withIndexName(CommonData.defaultIndex)
+                .build());
     Assert.assertEquals(indexBuildProgress.getStatus().intValue(), 0);
     Assert.assertEquals(indexBuildProgress.getData().getIndexedRows(), 0);
   }

+ 11 - 8
tests/milvustest/src/test/java/com/zilliz/milvustest/index/GetIndexStateTest.java

@@ -4,6 +4,7 @@ import com.zilliz.milvustest.common.BaseTest;
 import com.zilliz.milvustest.common.CommonData;
 import com.zilliz.milvustest.common.CommonFunction;
 import io.milvus.grpc.GetIndexStateResponse;
+import io.milvus.grpc.IndexState;
 import io.milvus.param.IndexType;
 import io.milvus.param.MetricType;
 import io.milvus.param.R;
@@ -35,7 +36,9 @@ public class GetIndexStateTest extends BaseTest {
                 .withMetricType(MetricType.L2)
                 .withIndexType(IndexType.IVF_FLAT)
                 .withExtraParam(CommonData.defaultExtraParam)
-                .withSyncMode(Boolean.FALSE)
+                .withSyncMode(Boolean.TRUE)
+                .withSyncWaitingInterval(500L)
+                .withSyncWaitingTimeout(30L)
                 .build());
     softAssert.assertEquals(indexR.getStatus().intValue(), 0);
     softAssert.assertAll();
@@ -46,24 +49,24 @@ public class GetIndexStateTest extends BaseTest {
     milvusClient.dropIndex(
         DropIndexParam.newBuilder()
             .withCollectionName(collection)
-            .withIndexName(CommonData.defaultVectorField)
+            .withIndexName(CommonData.defaultIndex)
             .build());
     milvusClient.dropCollection(
         DropCollectionParam.newBuilder().withCollectionName(collection).build());
   }
 
   @Severity(SeverityLevel.BLOCKER)
-  @Test(description = "Get index state of default collection")
+  @Test(description = "Get index state ")
   @Issue("https://github.com/milvus-io/milvus-sdk-java/issues/302")
-  public void getIndexStateDefault() {
+  public void getIndexState() {
     R<GetIndexStateResponse> getIndexStateResponseR =
         milvusClient.getIndexState(
             GetIndexStateParam.newBuilder()
                 .withCollectionName(collection)
-                .withIndexName(CommonData.defaultVectorField)
+                .withIndexName(CommonData.defaultIndex)
                 .build());
     Assert.assertEquals(getIndexStateResponseR.getStatus().intValue(), 0);
-    Assert.assertEquals(getIndexStateResponseR.getData(), 0);
+    Assert.assertEquals(getIndexStateResponseR.getData().getState(), IndexState.Finished);
   }
 
   @Severity(SeverityLevel.NORMAL)
@@ -73,7 +76,7 @@ public class GetIndexStateTest extends BaseTest {
         milvusClient.getIndexState(
             GetIndexStateParam.newBuilder()
                 .withCollectionName("NonexistentCollection")
-                .withIndexName(CommonData.defaultVectorField)
+                .withIndexName(CommonData.defaultIndex)
                 .build());
     Assert.assertEquals(getIndexStateResponseR.getStatus().intValue(), 1);
     Assert.assertTrue(getIndexStateResponseR.getException().getMessage().contains("not found"));
@@ -87,7 +90,7 @@ public class GetIndexStateTest extends BaseTest {
         milvusClient.getIndexState(
             GetIndexStateParam.newBuilder()
                 .withCollectionName(newCollection)
-                .withIndexName(CommonData.defaultVectorField)
+                .withIndexName(CommonData.defaultIndex)
                 .build());
     Assert.assertEquals(getIndexStateResponseR.getStatus().intValue(), 1);
     Assert.assertTrue(

+ 3 - 3
tests/milvustest/src/test/java/com/zilliz/milvustest/insert/InsertAsyncTest.java

@@ -113,7 +113,7 @@ public class InsertAsyncTest extends BaseTest {
                             .build());
     R<MutationResult> mutationResultR = rListenableFuture.get();
     Assert.assertEquals(mutationResultR.getStatus().intValue(), -5);
-    Assert.assertTrue(mutationResultR.getException().getMessage().contains("The field: VectorFieldAutoTest is not provided"));
+    Assert.assertTrue(mutationResultR.getException().getMessage().contains("The field: VectorFieldAutoTest is not provided."));
   }
 
 
@@ -148,7 +148,7 @@ public class InsertAsyncTest extends BaseTest {
                             .build());
     R<MutationResult> mutationResultR = rListenableFuture.get();
     Assert.assertEquals(mutationResultR.getStatus().intValue(), -5);
-    Assert.assertTrue(mutationResultR.getException().getMessage().contains("book_id data type doesn't match the collection schema"));
+    Assert.assertTrue(mutationResultR.getException().getMessage().contains("'book_id': Int64 field value type must be Long"));
   }
 
   @Severity(SeverityLevel.NORMAL)
@@ -182,7 +182,7 @@ public class InsertAsyncTest extends BaseTest {
                             .build());
     R<MutationResult> mutationResultR = rListenableFuture.get();
     Assert.assertEquals(mutationResultR.getStatus().intValue(), -5);
-    Assert.assertTrue(mutationResultR.getException().getMessage().contains("word_count data type doesn't match the collection schema"));
+    Assert.assertTrue(mutationResultR.getException().getMessage().contains("'word_count': Int64 field value type must be Long"));
   }
 
   @Severity(SeverityLevel.NORMAL)

+ 2 - 2
tests/milvustest/src/test/java/com/zilliz/milvustest/insert/InsertTest.java

@@ -143,7 +143,7 @@ public class InsertTest extends BaseTest {
                             .withFields(fields)
                             .build());
     Assert.assertEquals(mutationResultR.getStatus().intValue(), -5);
-    Assert.assertTrue(mutationResultR.getException().getMessage().contains("book_id data type doesn't match the collection schema"));
+    Assert.assertTrue(mutationResultR.getException().getMessage().contains("'book_id': Int64 field value type must be Long"));
   }
 
   @Severity(SeverityLevel.NORMAL)
@@ -176,7 +176,7 @@ public class InsertTest extends BaseTest {
                             .withFields(fields)
                             .build());
     Assert.assertEquals(mutationResultR.getStatus().intValue(), -5);
-    Assert.assertTrue(mutationResultR.getException().getMessage().contains("word_count data type doesn't match the collection schema"));
+    Assert.assertTrue(mutationResultR.getException().getMessage().contains("'word_count': Int64 field value type must be Long"));
   }
 
   @Severity(SeverityLevel.NORMAL)

+ 14 - 15
tests/milvustest/src/test/java/com/zilliz/milvustest/partition/CreatePartitionTest.java

@@ -7,11 +7,11 @@ import com.zilliz.milvustest.util.MathUtil;
 import io.milvus.grpc.QueryResults;
 import io.milvus.param.R;
 import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.DropCollectionParam;
 import io.milvus.param.collection.LoadCollectionParam;
 import io.milvus.param.dml.InsertParam;
 import io.milvus.param.dml.QueryParam;
 import io.milvus.param.partition.CreatePartitionParam;
-import io.milvus.param.partition.DropPartitionParam;
 import io.milvus.response.QueryResultsWrapper;
 import io.qameta.allure.Epic;
 import io.qameta.allure.Feature;
@@ -30,19 +30,18 @@ import java.util.List;
 @Feature("CreatePartition")
 public class CreatePartitionTest extends BaseTest {
   private String partition;
+  private String collection;
 
   @BeforeClass(description = "init partition Name")
   public void createPartitionTest() {
+    collection=CommonFunction.createNewCollection();
     partition = "partition_" + MathUtil.getRandomString(10);
   }
 
   @AfterClass(description = "delete partition after test")
-  public void deletePartition() {
-    milvusClient.dropPartition(
-        DropPartitionParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
-            .withPartitionName(partition)
-            .build());
+  public void deleteCollection() {
+    milvusClient.dropCollection(
+        DropCollectionParam.newBuilder().withCollectionName(collection).build());
   }
 
   @Severity(SeverityLevel.BLOCKER)
@@ -50,7 +49,7 @@ public class CreatePartitionTest extends BaseTest {
   public void createPartition() {
     CreatePartitionParam createPartitionParam =
         CreatePartitionParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionName(partition)
             .build();
     R<RpcStatus> rpcStatusR = milvusClient.createPartition(createPartitionParam);
@@ -63,7 +62,7 @@ public class CreatePartitionTest extends BaseTest {
   public void createPartitionRepeatedly() {
     CreatePartitionParam createPartitionParam =
             CreatePartitionParam.newBuilder()
-                    .withCollectionName(CommonData.defaultCollection)
+                    .withCollectionName(collection)
                     .withPartitionName(partition)
                     .build();
     R<RpcStatus> rpcStatusR = milvusClient.createPartition(createPartitionParam);
@@ -77,7 +76,7 @@ public class CreatePartitionTest extends BaseTest {
   @Severity(SeverityLevel.NORMAL)
   public void queryFromEmptyPartition() {
     milvusClient.loadCollection(
-        LoadCollectionParam.newBuilder().withCollectionName(CommonData.defaultCollection).build());
+        LoadCollectionParam.newBuilder().withCollectionName(collection).build());
     String SEARCH_PARAM = "book_id in [2,4,6,8]";
     List<String> outFields = Arrays.asList("book_id", "word_count");
     List<String> partitions =
@@ -88,14 +87,14 @@ public class CreatePartitionTest extends BaseTest {
         };
     QueryParam queryParam =
         QueryParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionNames(partitions)
             .withOutFields(outFields)
             .withExpr(SEARCH_PARAM)
             .build();
     R<QueryResults> queryResultsR = milvusClient.query(queryParam);
     Assert.assertEquals(queryResultsR.getStatus().intValue(), 26);
-    Assert.assertEquals(queryResultsR.getException().getMessage(), "emptly collection");
+    Assert.assertTrue(queryResultsR.getException().getMessage().contains("empty collection or improper expression"));
   }
 
   @Test(
@@ -106,12 +105,12 @@ public class CreatePartitionTest extends BaseTest {
     List<InsertParam.Field> fields = CommonFunction.generateData(100);
     milvusClient.insert(
         InsertParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionName(partition)
             .withFields(fields)
             .build());
     milvusClient.loadCollection(
-        LoadCollectionParam.newBuilder().withCollectionName(CommonData.defaultCollection).withSyncLoad(Boolean.TRUE)
+        LoadCollectionParam.newBuilder().withCollectionName(collection).withSyncLoad(Boolean.TRUE)
                 .withSyncLoadWaitingInterval(1000L).withSyncLoadWaitingTimeout(300L).build());
     Thread.sleep(5000);
     String SEARCH_PARAM = "book_id in [2,4,6,8]";
@@ -124,7 +123,7 @@ public class CreatePartitionTest extends BaseTest {
         };
     QueryParam queryParam =
         QueryParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionNames(partitions)
             .withOutFields(outFields)
             .withExpr(SEARCH_PARAM)

+ 18 - 8
tests/milvustest/src/test/java/com/zilliz/milvustest/partition/LoadPartitionsTest.java

@@ -2,9 +2,11 @@ package com.zilliz.milvustest.partition;
 
 import com.zilliz.milvustest.common.BaseTest;
 import com.zilliz.milvustest.common.CommonData;
+import com.zilliz.milvustest.common.CommonFunction;
 import com.zilliz.milvustest.util.MathUtil;
 import io.milvus.param.R;
 import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.DropCollectionParam;
 import io.milvus.param.partition.CreatePartitionParam;
 import io.milvus.param.partition.DropPartitionParam;
 import io.milvus.param.partition.LoadPartitionsParam;
@@ -24,21 +26,23 @@ import java.util.List;
 @Epic("Partition")
 @Feature("LoadPartitions")
 public class LoadPartitionsTest extends BaseTest {
+  private String collection;
   private String partition;
   private String partition2;
 
   @BeforeClass(description = "init partition Name")
   public void createPartitionTest() {
+    collection= CommonFunction.createNewCollection();
     partition = "partition_" + MathUtil.getRandomString(10);
     partition2 = "partition_" + MathUtil.getRandomString(10);
     milvusClient.createPartition(
         CreatePartitionParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionName(partition)
             .build());
     milvusClient.createPartition(
             CreatePartitionParam.newBuilder()
-                    .withCollectionName(CommonData.defaultCollection)
+                    .withCollectionName(collection)
                     .withPartitionName(partition2)
                     .build());
   }
@@ -47,14 +51,15 @@ public class LoadPartitionsTest extends BaseTest {
   public void deletePartition() {
     milvusClient.dropPartition(
         DropPartitionParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionName(partition)
             .build());
     milvusClient.dropPartition(
             DropPartitionParam.newBuilder()
-                    .withCollectionName(CommonData.defaultCollection)
+                    .withCollectionName(collection)
                     .withPartitionName(partition2)
                     .build());
+    milvusClient.dropCollection(DropCollectionParam.newBuilder().withCollectionName(collection).build());
   }
 
   @Severity(SeverityLevel.BLOCKER)
@@ -70,7 +75,7 @@ public class LoadPartitionsTest extends BaseTest {
     R<RpcStatus> rpcStatusR =
         milvusClient.loadPartitions(
             LoadPartitionsParam.newBuilder()
-                .withCollectionName(CommonData.defaultCollection)
+                .withCollectionName(collection)
                 .withPartitionNames(partitionNames)
                 .build());
     Assert.assertEquals(rpcStatusR.getStatus().intValue(), 0);
@@ -90,7 +95,7 @@ public class LoadPartitionsTest extends BaseTest {
     R<RpcStatus> rpcStatusR =
         milvusClient.loadPartitions(
             LoadPartitionsParam.newBuilder()
-                .withCollectionName(CommonData.defaultCollection)
+                .withCollectionName(collection)
                 .withPartitionNames(partitionNames)
                 .build());
     Assert.assertEquals(rpcStatusR.getStatus().intValue(), 1);
@@ -105,9 +110,14 @@ public class LoadPartitionsTest extends BaseTest {
     // release first
     milvusClient.releasePartitions(
         ReleasePartitionsParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .addPartitionName(partition)
             .build());
+    milvusClient.releasePartitions(
+            ReleasePartitionsParam.newBuilder()
+                    .withCollectionName(collection)
+                    .addPartitionName(partition2)
+                    .build());
     // load
     List<String> partitionNames =
         new ArrayList<String>() {
@@ -118,7 +128,7 @@ public class LoadPartitionsTest extends BaseTest {
     R<RpcStatus> rpcStatusR =
         milvusClient.loadPartitions(
             LoadPartitionsParam.newBuilder()
-                .withCollectionName(CommonData.defaultCollection)
+                .withCollectionName(collection)
                 .withPartitionNames(partitionNames)
                 .build());
     Assert.assertEquals(rpcStatusR.getStatus().intValue(), 0);

+ 34 - 26
tests/milvustest/src/test/java/com/zilliz/milvustest/partition/ReleasePartitionsTest.java

@@ -1,11 +1,12 @@
 package com.zilliz.milvustest.partition;
 
 import com.zilliz.milvustest.common.BaseTest;
-import com.zilliz.milvustest.common.CommonData;
+import com.zilliz.milvustest.common.CommonFunction;
 import com.zilliz.milvustest.util.MathUtil;
 import io.milvus.grpc.QueryResults;
 import io.milvus.param.R;
 import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.DropCollectionParam;
 import io.milvus.param.dml.QueryParam;
 import io.milvus.param.partition.CreatePartitionParam;
 import io.milvus.param.partition.DropPartitionParam;
@@ -28,18 +29,20 @@ import java.util.List;
 @Feature("ReleasePartitions")
 public class ReleasePartitionsTest extends BaseTest {
   public String partition;
+  public String collection;
 
   @BeforeClass(description = "load partition")
   public void initPartition() {
+    collection = CommonFunction.createNewCollection();
     partition = "partition_" + MathUtil.getRandomString(10);
     milvusClient.createPartition(
         CreatePartitionParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionName(partition)
             .build());
     milvusClient.loadPartitions(
         LoadPartitionsParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .addPartitionName(partition)
             .build());
   }
@@ -48,9 +51,11 @@ public class ReleasePartitionsTest extends BaseTest {
   public void deletePartition() {
     milvusClient.dropPartition(
         DropPartitionParam.newBuilder()
-            .withCollectionName(CommonData.defaultCollection)
+            .withCollectionName(collection)
             .withPartitionName(partition)
             .build());
+    milvusClient.dropCollection(
+        DropCollectionParam.newBuilder().withCollectionName(collection).build());
   }
 
   @Severity(SeverityLevel.BLOCKER)
@@ -59,48 +64,51 @@ public class ReleasePartitionsTest extends BaseTest {
     R<RpcStatus> rpcStatusR =
         milvusClient.releasePartitions(
             ReleasePartitionsParam.newBuilder()
-                .withCollectionName(CommonData.defaultCollection)
+                .withCollectionName(collection)
                 .addPartitionName(partition)
                 .build());
     Assert.assertEquals(rpcStatusR.getStatus().intValue(), 0);
     Assert.assertEquals(rpcStatusR.getData().getMsg(), "Success");
   }
 
-  @Test(
-          description = "query from partition after release ",
-          dependsOnMethods = "releasePartition")
+  @Test(description = "query from partition after release ", dependsOnMethods = "releasePartition")
   @Severity(SeverityLevel.NORMAL)
   public void queryAfterReleasePartition() {
     String SEARCH_PARAM = "book_id in [2,4,6,8]";
     List<String> outFields = Arrays.asList("book_id", "word_count");
     List<String> partitions =
-            new ArrayList<String>() {
-              {
-                add(partition);
-              }
-            };
+        new ArrayList<String>() {
+          {
+            add(partition);
+          }
+        };
     QueryParam queryParam =
-            QueryParam.newBuilder()
-                    .withCollectionName(CommonData.defaultCollection)
-                    .withPartitionNames(partitions)
-                    .withOutFields(outFields)
-                    .withExpr(SEARCH_PARAM)
-                    .build();
+        QueryParam.newBuilder()
+            .withCollectionName(collection)
+            .withPartitionNames(partitions)
+            .withOutFields(outFields)
+            .withExpr(SEARCH_PARAM)
+            .build();
     R<QueryResults> queryResultsR = milvusClient.query(queryParam);
     Assert.assertEquals(queryResultsR.getStatus().intValue(), 1);
-    Assert.assertTrue(queryResultsR.getException().getMessage().contains("checkIfLoaded failed when query"));
+    Assert.assertTrue(
+        queryResultsR.getException().getMessage().contains("checkIfLoaded failed when query"));
   }
 
   @Severity(SeverityLevel.NORMAL)
   @Test(description = "release nonexistent partition")
   public void releaseNonexistentPartition() {
     R<RpcStatus> rpcStatusR =
-            milvusClient.releasePartitions(
-                    ReleasePartitionsParam.newBuilder()
-                            .withCollectionName(CommonData.defaultCollection)
-                            .addPartitionName("NonexistentPartition")
-                            .build());
+        milvusClient.releasePartitions(
+            ReleasePartitionsParam.newBuilder()
+                .withCollectionName(collection)
+                .addPartitionName("NonexistentPartition")
+                .build());
     Assert.assertEquals(rpcStatusR.getStatus().intValue(), 1);
-    Assert.assertTrue(rpcStatusR.getException().getMessage().contains("partitionID of partitionName:NonexistentPartition can not be find"));
+    Assert.assertTrue(
+        rpcStatusR
+            .getException()
+            .getMessage()
+            .contains("partitionID of partitionName:NonexistentPartition can not be find"));
   }
 }

+ 3 - 3
tests/milvustest/src/test/java/com/zilliz/milvustest/query/QueryAsyncTest.java

@@ -550,7 +550,7 @@ public class QueryAsyncTest extends BaseTest {
   }
 
   @Test(
-          description = "Int PK and  float vector query after delete data ",
+          description = "Int PK and float vector query after delete data ",
           dataProvider = "providerPartition")
   @Severity(SeverityLevel.NORMAL)
   public void intPKAndFloatVectorAfterDelete(Boolean usePart) throws ExecutionException, InterruptedException {
@@ -627,9 +627,9 @@ public class QueryAsyncTest extends BaseTest {
             queryResultsR.getException().getMessage(), "Expression cannot be null or empty");
   }
 
-  @Test(description = "query return empty ")
+  @Test(description = "query async return empty ")
   @Severity(SeverityLevel.MINOR)
-  public void queryReturnEmpty()throws ExecutionException, InterruptedException  {
+  public void queryAsyncReturnEmpty()throws ExecutionException, InterruptedException  {
     String SEARCH_PARAM = "book_id in [-1,-2,-3,-4]";
     List<String> outFields = Arrays.asList("book_id", "word_count");
     QueryParam queryParam =

+ 4 - 2
tests/milvustest/src/test/java/com/zilliz/milvustest/query/QueryTest.java

@@ -554,6 +554,8 @@ public class QueryTest extends BaseTest {
     Thread.sleep(2000);
     String SEARCH_PARAM = "book_id in [22,24,26,28]";
     List<String> outFields = Arrays.asList("book_id", "word_count");
+    milvusClient.loadCollection(LoadCollectionParam.newBuilder()
+            .withCollectionName(CommonData.defaultCollection).build());
     QueryParam queryParam =
         QueryParam.newBuilder()
             .withCollectionName(CommonData.defaultCollection)
@@ -569,7 +571,7 @@ public class QueryTest extends BaseTest {
   }
 
   @Test(
-      description = "String PK and  float vector query after delete data ",
+      description = "String PK and float vector query after delete data ",
       dataProvider = "providerPartition",
       dependsOnMethods = {"stringPKAndFloatVectorAfterInsert", "stringPKAndFloatVectorAfterUpdate"})
   @Severity(SeverityLevel.NORMAL)
@@ -626,7 +628,7 @@ public class QueryTest extends BaseTest {
             .withExpr(SEARCH_PARAM)
             .build();
     R<QueryResults> queryResultsR = milvusClient.query(queryParam);
-    Assert.assertEquals(queryResultsR.getStatus().intValue(), 25);
+    Assert.assertEquals(queryResultsR.getStatus().intValue(), 26);
     Assert.assertEquals(
         queryResultsR.getException().getMessage(), "empty collection or improper expression");
   }

+ 5 - 2
tests/milvustest/src/test/java/com/zilliz/milvustest/search/SearchAsyncTest.java

@@ -717,6 +717,7 @@ public class SearchAsyncTest extends BaseTest {
             .withVectors(book_intro_array)
             .withVectorFieldName(CommonData.defaultVectorField)
             .withParams(SEARCH_PARAM)
+            .withExpr("book_id == 9999")
             .build();
     ListenableFuture<R<SearchResults>> rListenableFuture = milvusClient.searchAsync(searchParam);
     try {
@@ -854,6 +855,7 @@ public class SearchAsyncTest extends BaseTest {
             .withVectors(book_intro_array)
             .withVectorFieldName(CommonData.defaultVectorField)
             .withParams(SEARCH_PARAM)
+            .withExpr("book_name == \"" + newBookName + "\"")
             .build();
     ListenableFuture<R<SearchResults>> rListenableFuture = milvusClient.searchAsync(searchParam);
     try {
@@ -871,7 +873,7 @@ public class SearchAsyncTest extends BaseTest {
 
   @Severity(SeverityLevel.NORMAL)
   @Test(
-      description = "string PK and float vector search async after insert the entity",
+      description = "string PK and binary vector search async after insert the entity",
       dataProvider = "providerPartition")
   public void stringPKAndBinaryVectorSearchAsyncAfterInsertNewEntity(Boolean usePart)
       throws InterruptedException {
@@ -994,6 +996,7 @@ public class SearchAsyncTest extends BaseTest {
             .withVectors(book_intro_array)
             .withVectorFieldName(CommonData.defaultVectorField)
             .withParams(SEARCH_PARAM)
+            .withExpr("book_id =="+id)
             .build();
     ListenableFuture<R<SearchResults>> rListenableFuture = milvusClient.searchAsync(searchParam);
     try {
@@ -1157,7 +1160,7 @@ public class SearchAsyncTest extends BaseTest {
 
   @Severity(SeverityLevel.NORMAL)
   @Test(
-      description = "string PK and float vector search async after update the entity",
+      description = "string PK and binary vector search async after update the entity",
       dataProvider = "providerPartition",
       dependsOnMethods = "stringPKAndBinaryVectorSearchAsyncAfterInsertNewEntity")
   public void stringPKAndBinaryVectorSearchAsyncAfterUpdateNewEntity(Boolean usePart)

+ 3 - 1
tests/milvustest/src/test/java/com/zilliz/milvustest/search/SearchTest.java

@@ -744,6 +744,7 @@ public class SearchTest extends BaseTest {
             .withVectors(book_intro_array)
             .withVectorFieldName(CommonData.defaultVectorField)
             .withParams(SEARCH_PARAM)
+            .withExpr("book_id == 9999")
             .build();
     R<SearchResults> searchResultsR = milvusClient.search(searchParam);
     Assert.assertEquals(searchResultsR.getStatus().intValue(), 0);
@@ -873,7 +874,7 @@ public class SearchTest extends BaseTest {
             .withVectors(book_intro_array)
             .withVectorFieldName(CommonData.defaultVectorField)
             .withParams(SEARCH_PARAM)
-                .withExpr("book_name == \"" + newBookName + "\"")
+            .withExpr("book_name == \"" + newBookName + "\"")
             .build();
     R<SearchResults> searchResultsR = milvusClient.search(searchParam);
     Assert.assertEquals(searchResultsR.getStatus().intValue(), 0);
@@ -1007,6 +1008,7 @@ public class SearchTest extends BaseTest {
             .withVectors(book_intro_array)
             .withVectorFieldName(CommonData.defaultVectorField)
             .withParams(SEARCH_PARAM)
+            .withExpr("book_id == "+id)
             .build();
     R<SearchResults> searchResultsR = milvusClient.search(searchParam);
     Assert.assertEquals(searchResultsR.getStatus().intValue(), 0);

+ 1 - 1
tests/milvustest/src/test/java/resources/run.properties

@@ -1,4 +1,4 @@
-milvusHost=10.98.0.8
+milvusHost=10.100.31.102
 milvusPort=19530
 #enviroment
 milvusV=2.1.0

+ 2 - 1
tests/milvustest/testngAll.xml

@@ -1,6 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd">
-<suite name="Milvus Java SDK test Suite" parallel="classes" thread-count="1">
+<suite name="Milvus Java SDK test Suite" parallel="classes" thread-count="6">
+
     <!--folder-->
     <test name="connection">
         <packages>