|
@@ -2,21 +2,42 @@ package com.zilliz.milvustest.collection;
|
|
|
|
|
|
import com.zilliz.milvustest.common.BaseTest;
|
|
import com.zilliz.milvustest.common.BaseTest;
|
|
import com.zilliz.milvustest.common.CommonData;
|
|
import com.zilliz.milvustest.common.CommonData;
|
|
|
|
+import com.zilliz.milvustest.common.CommonFunction;
|
|
import io.milvus.grpc.GetCollectionStatisticsResponse;
|
|
import io.milvus.grpc.GetCollectionStatisticsResponse;
|
|
import io.milvus.param.R;
|
|
import io.milvus.param.R;
|
|
|
|
+import io.milvus.param.collection.DropCollectionParam;
|
|
import io.milvus.param.collection.GetCollectionStatisticsParam;
|
|
import io.milvus.param.collection.GetCollectionStatisticsParam;
|
|
|
|
+import io.milvus.param.dml.InsertParam;
|
|
import io.milvus.response.GetCollStatResponseWrapper;
|
|
import io.milvus.response.GetCollStatResponseWrapper;
|
|
import io.qameta.allure.Epic;
|
|
import io.qameta.allure.Epic;
|
|
import io.qameta.allure.Feature;
|
|
import io.qameta.allure.Feature;
|
|
import io.qameta.allure.Severity;
|
|
import io.qameta.allure.Severity;
|
|
import io.qameta.allure.SeverityLevel;
|
|
import io.qameta.allure.SeverityLevel;
|
|
import org.testng.Assert;
|
|
import org.testng.Assert;
|
|
|
|
+import org.testng.annotations.AfterClass;
|
|
|
|
+import org.testng.annotations.BeforeClass;
|
|
import org.testng.annotations.Test;
|
|
import org.testng.annotations.Test;
|
|
|
|
|
|
|
|
+import java.util.List;
|
|
|
|
+
|
|
@Epic("Collection")
|
|
@Epic("Collection")
|
|
@Feature("GetCollectionStatistics")
|
|
@Feature("GetCollectionStatistics")
|
|
public class GetCollectionStatisticsTest extends BaseTest {
|
|
public class GetCollectionStatisticsTest extends BaseTest {
|
|
|
|
|
|
|
|
+ public String commonCollection;
|
|
|
|
+
|
|
|
|
+ @BeforeClass(description = "Create collection before test",alwaysRun=true)
|
|
|
|
+ public void provideCollectionName() {
|
|
|
|
+ String newCollection = CommonFunction.createNewCollection();
|
|
|
|
+ commonCollection = newCollection;
|
|
|
|
+ List<InsertParam.Field> fields = CommonFunction.generateData(2000);
|
|
|
|
+ CommonFunction.insertDataIntoCollection(newCollection,fields);
|
|
|
|
+ }
|
|
|
|
+ @AfterClass(description = "drop collection before test",alwaysRun=true)
|
|
|
|
+ public void dropCollection() {
|
|
|
|
+ milvusClient.dropCollection(DropCollectionParam.newBuilder().withCollectionName(commonCollection).build());
|
|
|
|
+ }
|
|
|
|
+
|
|
@Severity(SeverityLevel.BLOCKER)
|
|
@Severity(SeverityLevel.BLOCKER)
|
|
@Test(description = "Shows the statistics information of a collection.",groups = {"Smoke"})
|
|
@Test(description = "Shows the statistics information of a collection.",groups = {"Smoke"})
|
|
public void getCollectionStatisticsInfo() {
|
|
public void getCollectionStatisticsInfo() {
|
|
@@ -24,7 +45,7 @@ public class GetCollectionStatisticsTest extends BaseTest {
|
|
milvusClient
|
|
milvusClient
|
|
.getCollectionStatistics( // Return the statistics information of the collection.
|
|
.getCollectionStatistics( // Return the statistics information of the collection.
|
|
GetCollectionStatisticsParam.newBuilder()
|
|
GetCollectionStatisticsParam.newBuilder()
|
|
- .withCollectionName(CommonData.defaultCollection)
|
|
|
|
|
|
+ .withCollectionName(commonCollection)
|
|
.withFlush(false)
|
|
.withFlush(false)
|
|
.build());
|
|
.build());
|
|
Assert.assertEquals(respCollectionStatistics.getStatus().intValue(), 0);
|
|
Assert.assertEquals(respCollectionStatistics.getStatus().intValue(), 0);
|
|
@@ -41,7 +62,7 @@ public class GetCollectionStatisticsTest extends BaseTest {
|
|
milvusClient
|
|
milvusClient
|
|
.getCollectionStatistics( // Return the statistics information of the collection.
|
|
.getCollectionStatistics( // Return the statistics information of the collection.
|
|
GetCollectionStatisticsParam.newBuilder()
|
|
GetCollectionStatisticsParam.newBuilder()
|
|
- .withCollectionName(CommonData.defaultCollection)
|
|
|
|
|
|
+ .withCollectionName(commonCollection)
|
|
.withFlush(true)
|
|
.withFlush(true)
|
|
.build());
|
|
.build());
|
|
Assert.assertEquals(respCollectionStatistics.getStatus().intValue(), 0);
|
|
Assert.assertEquals(respCollectionStatistics.getStatus().intValue(), 0);
|