Browse Source

modify testcase for index (#441)

Signed-off-by: yongpengli-z <yongpeng.li@zilliz.com>
yongpengli-z 2 years ago
parent
commit
bbe11422f8

+ 1 - 1
tests/milvustest/src/test/java/com/zilliz/milvustest/collection/GetCollectionStatisticsTest.java

@@ -30,7 +30,7 @@ public class GetCollectionStatisticsTest extends BaseTest {
     Assert.assertEquals(respCollectionStatistics.getStatus().intValue(), 0);
     GetCollStatResponseWrapper wrapperCollectionStatistics =
         new GetCollStatResponseWrapper(respCollectionStatistics.getData());
-    Assert.assertEquals(wrapperCollectionStatistics.getRowCount(), 2000);
+    Assert.assertEquals(wrapperCollectionStatistics.getRowCount(), 0);
     System.out.println(wrapperCollectionStatistics);
   }
 

+ 12 - 2
tests/milvustest/src/test/java/com/zilliz/milvustest/index/DropIndexTest.java

@@ -4,11 +4,13 @@ import com.zilliz.milvustest.common.BaseTest;
 import com.zilliz.milvustest.common.CommonData;
 import com.zilliz.milvustest.common.CommonFunction;
 import io.milvus.exception.ParamException;
+import io.milvus.grpc.DescribeCollectionResponse;
 import io.milvus.grpc.DescribeIndexResponse;
 import io.milvus.param.IndexType;
 import io.milvus.param.MetricType;
 import io.milvus.param.R;
 import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.DescribeCollectionParam;
 import io.milvus.param.collection.DropCollectionParam;
 import io.milvus.param.index.CreateIndexParam;
 import io.milvus.param.index.DescribeIndexParam;
@@ -115,7 +117,7 @@ public class DropIndexTest extends BaseTest {
     Assert.assertEquals(describeIndexResponseR.getStatus().intValue(), 25);
     Assert.assertTrue(
             describeIndexResponseR.getException().getMessage().contains(
-            "index not exist"));
+            "index doesn't exist"));
   }
 
   @Severity(SeverityLevel.NORMAL)
@@ -133,6 +135,14 @@ public class DropIndexTest extends BaseTest {
                     .build());
     Assert.assertEquals(indexR.getStatus().intValue(),0);
     Assert.assertEquals(indexR.getData().getMsg(),"Success");
-
+    R<DescribeIndexResponse> describeIndexResponseR =
+            milvusClient.describeIndex(
+                    DescribeIndexParam.newBuilder()
+                            .withCollectionName(collection)
+                            .withIndexName(CommonData.defaultIndex)
+                            .build());
+    logger.info(describeIndexResponseR.getData().toString());
+    R<DescribeCollectionResponse> describeCollectionResponseR = milvusClient.describeCollection(DescribeCollectionParam.newBuilder().withCollectionName(collection).build());
+    logger.info(describeCollectionResponseR.getData().getSchema().getFieldsList().toString());
   }
 }

+ 4 - 4
tests/milvustest/src/test/java/com/zilliz/milvustest/index/IndexLoadTest.java

@@ -118,12 +118,12 @@ public class IndexLoadTest extends BaseTest {
         Assert.assertEquals(rpcStatusR2.getStatus().intValue(), 0);
         Assert.assertEquals(rpcStatusR2.getData().getMsg(), "Success");
 
-        // 4. create index
+      /*  // 4. create index
         R<RpcStatus> rpcStatusR3 = milvusClient.createIndex(createIndexParams);
         System.out.println("Create index " + rpcStatusR);
         Assert.assertEquals(rpcStatusR3.getStatus().intValue(), 1);
         Assert.assertTrue(rpcStatusR3.getMessage().contains("create index failed, collection is loaded, please release it first"));
-
+*/
         // 5. drop index
         R<RpcStatus> rpcStatusR4 = milvusClient.dropIndex(
                         DropIndexParam.newBuilder()
@@ -194,12 +194,12 @@ public class IndexLoadTest extends BaseTest {
         Assert.assertEquals(rpcStatusR2.getStatus().intValue(), 0);
         Assert.assertEquals(rpcStatusR2.getData().getMsg(), "Success");
 
-        // 4. create index
+      /*  // 4. create index
         R<RpcStatus> rpcStatusR3 = milvusClient.createIndex(createIndexParams);
         System.out.println("Create index " + rpcStatusR);
         Assert.assertEquals(rpcStatusR3.getStatus().intValue(), 1);
         Assert.assertTrue(rpcStatusR3.getMessage().contains("create index failed, collection is loaded, please release it first"));
-
+*/
         // 5. drop index
         R<RpcStatus> rpcStatusR4 = milvusClient.dropIndex(
                 DropIndexParam.newBuilder()

+ 4 - 0
tests/milvustest/src/test/java/com/zilliz/milvustest/other/GetReplicasTest.java

@@ -4,6 +4,7 @@ import com.zilliz.milvustest.common.BaseTest;
 import com.zilliz.milvustest.common.CommonData;
 import io.milvus.grpc.GetReplicasResponse;
 import io.milvus.param.R;
+import io.milvus.param.RpcStatus;
 import io.milvus.param.collection.LoadCollectionParam;
 import io.milvus.param.collection.ReleaseCollectionParam;
 import io.milvus.param.control.GetReplicasParam;
@@ -39,6 +40,9 @@ public class GetReplicasTest extends BaseTest {
   @Test(description = "Returns the collection's replica information",groups = {"Smoke"})
   @Severity(SeverityLevel.BLOCKER)
   public void getReplicasTest() {
+    R<RpcStatus> rpcStatusR = milvusClient.loadCollection(LoadCollectionParam.newBuilder()
+            .withCollectionName(CommonData.defaultCollection).build());
+    logger.info(rpcStatusR.toString());
     R<GetReplicasResponse> getReplicasResponseR =
         milvusClient.getReplicas(
             GetReplicasParam.newBuilder().withCollectionName(CommonData.defaultCollection).build());