add testcases for random sampling (#1333)

Signed-off-by: yongpengli-z <yongpeng.li@zilliz.com>

Author: yongpengli-z

.github/workflows/java_sdk_ci_test.yaml

@@ -28,7 +28,7 @@ jobs:
         run: |
           echo "build jar"
           git submodule update --init
-          mvn clean versions:set -DnewVersion=2.4.0
+          mvn clean versions:set -DnewVersion=2.5.5
           mvn clean install -Dmaven.test.skip=true
 
 #      - name: Test

tests/milvustestv2/pom.xml

@@ -70,7 +70,7 @@
         <dependency>
             <groupId>io.milvus</groupId>
             <artifactId>milvus-sdk-java</artifactId>
-            <version>2.4.0</version>
+            <version>2.5.5</version>
         </dependency>
         <dependency>
             <groupId>com.google.protobuf</groupId>

tests/milvustestv2/src/main/java/com/zilliz/milvustestv2/common/BaseTest.java

@@ -5,6 +5,8 @@
 import com.zilliz.milvustestv2.config.ConnectInfoConfig;
 import com.zilliz.milvustestv2.params.FieldParam;
 import com.zilliz.milvustestv2.utils.PropertyFilesUtil;
+import io.milvus.client.MilvusServiceClient;
+import io.milvus.param.ConnectParam;
 import io.milvus.param.MetricType;
 import io.milvus.v2.client.ConnectConfig;
 import io.milvus.v2.client.MilvusClientV2;

tests/milvustestv2/src/main/java/com/zilliz/milvustestv2/common/CommonFunction.java

@@ -742,11 +742,11 @@ public static List<JsonObject> generateSimpleNullData(long startId, long num, in
         for (long i = startId; i < (num + startId); i++) {
             JsonObject row = new JsonObject();
             row.addProperty(CommonData.fieldInt64, i);
-            row.addProperty(CommonData.fieldInt32, (String) null);
-            row.addProperty(CommonData.fieldInt16, (int) i % 32767);
-            row.addProperty(CommonData.fieldInt8, (short) i % 127);
-            row.addProperty(CommonData.fieldBool, i % 2 == 0);
-            if (i % 2 == 1) {
+            if (i % 2 == 0) {
+                row.addProperty(CommonData.fieldInt32, (int) i % 32767);
+                row.addProperty(CommonData.fieldInt16, (int) i % 32767);
+                row.addProperty(CommonData.fieldInt8, (short) i % 127);
+                row.addProperty(CommonData.fieldBool, i % 3 == 0);
                 row.addProperty(CommonData.fieldDouble, (double) i);
                 row.addProperty(CommonData.fieldVarchar, "Str" + i);
                 row.addProperty(CommonData.fieldFloat, (float) i);
@@ -774,15 +774,15 @@ public static List<JsonObject> generateSimpleNullData(long startId, long num, in
             }
 
             JsonObject json = new JsonObject();
-            json.addProperty(CommonData.fieldInt64, (int) i % 32767);
-            json.addProperty(CommonData.fieldInt32, (int) i % 32767);
-            json.addProperty(CommonData.fieldDouble, (double) i);
-            json.add(CommonData.fieldArray, gson.toJsonTree(Arrays.asList(i, i + 1, i + 2)));
-            json.addProperty(CommonData.fieldBool, i % 2 == 0);
-            if (i % 2 == 1) {
+            if (i % 2 == 0) {
+                json.addProperty(CommonData.fieldInt64, (int) i % 32767);
+                json.addProperty(CommonData.fieldInt32, (int) i % 32767);
+                json.addProperty(CommonData.fieldDouble, (double) i);
+                json.add(CommonData.fieldArray, gson.toJsonTree(Arrays.asList(i, i + 1, i + 2)));
+                json.addProperty(CommonData.fieldBool, i % 3 == 0);
                 json.addProperty(CommonData.fieldVarchar, "Str" + i);
+                json.addProperty(CommonData.fieldFloat, (float) i);
             }
-            json.addProperty(CommonData.fieldFloat, (float) i);
             row.add(CommonData.fieldJson, json);
             jsonList.add(row);
         }
@@ -1542,7 +1542,7 @@ public static List<List<String>> providerBatchFiles(String collection, BulkFileT
         CreateCollectionReq.CollectionSchema collectionSchema = describeCollectionResp.getCollectionSchema();
         RemoteBulkWriter remoteBulkWriter = buildRemoteBulkWriter(collectionSchema, bulkFileType);
         List<JsonObject> jsonObjects = CommonFunction.genCommonData(collection, count);
-        jsonObjects.forEach(x->{
+        jsonObjects.forEach(x -> {
             try {
                 remoteBulkWriter.appendRow(x);
             } catch (IOException | InterruptedException e) {

tests/milvustestv2/src/test/java/com/zilliz/milvustestv2/alias/AlterAliasTest.java

@@ -39,8 +39,9 @@ public void providerCollection() {
     @AfterClass(alwaysRun = true)
     public void cleanTestData(){
         milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(newCollectionName).build());
-        milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(newCollectionName2).build());
+        // remove alias before drop collection
         milvusClientV2.dropAlias(DropAliasReq.builder().alias(aliasName).build());
+        milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(newCollectionName2).build());
     }
 
     @Test(description = "Alter alias test",groups = {"Smoke"})

tests/milvustestv2/src/test/java/com/zilliz/milvustestv2/alias/CreateAliasTest.java

@@ -10,6 +10,7 @@
 import io.milvus.v2.service.collection.request.DropCollectionReq;
 import io.milvus.v2.service.index.request.CreateIndexReq;
 import io.milvus.v2.service.utility.request.CreateAliasReq;
+import io.milvus.v2.service.utility.request.DropAliasReq;
 import io.milvus.v2.service.utility.request.ListAliasesReq;
 import io.milvus.v2.service.utility.response.ListAliasResp;
 import io.milvus.v2.service.vector.request.InsertReq;
@@ -49,6 +50,7 @@ public void providerCollection(){
 
     @AfterClass(alwaysRun = true)
     public void cleanTestData(){
+        milvusClientV2.dropAlias(DropAliasReq.builder().alias(aliasName).build());
         milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(newCollectionName).build());
     }