|
@@ -1,12 +1,16 @@
|
|
package com.zilliz.milvustestv2.collection;
|
|
package com.zilliz.milvustestv2.collection;
|
|
|
|
|
|
|
|
+import com.google.common.collect.Lists;
|
|
import com.zilliz.milvustestv2.common.BaseTest;
|
|
import com.zilliz.milvustestv2.common.BaseTest;
|
|
import com.zilliz.milvustestv2.common.CommonData;
|
|
import com.zilliz.milvustestv2.common.CommonData;
|
|
import com.zilliz.milvustestv2.common.CommonFunction;
|
|
import com.zilliz.milvustestv2.common.CommonFunction;
|
|
-import io.milvus.common.resourcegroup.ResourceGroupConfig;
|
|
|
|
|
|
+import com.zilliz.milvustestv2.utils.MathUtil;
|
|
|
|
+import io.milvus.common.clientenum.FunctionType;
|
|
|
|
+import io.milvus.param.Constant;
|
|
import io.milvus.v2.common.DataType;
|
|
import io.milvus.v2.common.DataType;
|
|
import io.milvus.v2.common.IndexParam;
|
|
import io.milvus.v2.common.IndexParam;
|
|
import io.milvus.v2.service.collection.request.*;
|
|
import io.milvus.v2.service.collection.request.*;
|
|
|
|
+import io.milvus.v2.service.collection.response.DescribeCollectionResp;
|
|
import io.milvus.v2.service.collection.response.ListCollectionsResp;
|
|
import io.milvus.v2.service.collection.response.ListCollectionsResp;
|
|
import io.milvus.v2.service.vector.response.SearchResp;
|
|
import io.milvus.v2.service.vector.response.SearchResp;
|
|
import org.testng.Assert;
|
|
import org.testng.Assert;
|
|
@@ -14,10 +18,7 @@ import org.testng.annotations.AfterClass;
|
|
import org.testng.annotations.DataProvider;
|
|
import org.testng.annotations.DataProvider;
|
|
import org.testng.annotations.Test;
|
|
import org.testng.annotations.Test;
|
|
|
|
|
|
-import javax.ws.rs.ext.Provider;
|
|
|
|
-import java.util.ArrayList;
|
|
|
|
-import java.util.Collections;
|
|
|
|
-import java.util.List;
|
|
|
|
|
|
+import java.util.*;
|
|
|
|
|
|
/**
|
|
/**
|
|
* @Author yongpeng.li
|
|
* @Author yongpeng.li
|
|
@@ -25,19 +26,26 @@ import java.util.List;
|
|
*/
|
|
*/
|
|
|
|
|
|
public class CreateCollectionTest extends BaseTest {
|
|
public class CreateCollectionTest extends BaseTest {
|
|
- String simpleCollection="simpleCollection";
|
|
|
|
- String repeatCollection="repeatCollection";
|
|
|
|
- String collectionNameWithIndex="collectionNameWithIndex";
|
|
|
|
|
|
+ String simpleCollection = "simpleCollection";
|
|
|
|
+ String repeatCollection = "repeatCollection";
|
|
|
|
+ String collectionNameWithIndex = "collectionNameWithIndex";
|
|
|
|
+ String collectionNameWithNull = "collectionNameWithNull";
|
|
|
|
+ String collectionNameWithDefault = "collectionNameWithDefault";
|
|
|
|
+ String collectionNameWithNullAndDefault = "collectionNameWithNullAndDefault";
|
|
|
|
+
|
|
@AfterClass(alwaysRun = true)
|
|
@AfterClass(alwaysRun = true)
|
|
- public void cleanTestData(){
|
|
|
|
|
|
+ public void cleanTestData() {
|
|
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(simpleCollection).build());
|
|
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(simpleCollection).build());
|
|
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(repeatCollection).build());
|
|
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(repeatCollection).build());
|
|
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collectionNameWithIndex).build());
|
|
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collectionNameWithIndex).build());
|
|
|
|
+ milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collectionNameWithNull).build());
|
|
|
|
+ milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collectionNameWithDefault).build());
|
|
|
|
+ milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collectionNameWithNullAndDefault).build());
|
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
@DataProvider(name = "VectorTypeList")
|
|
@DataProvider(name = "VectorTypeList")
|
|
- public Object[][] providerVectorType(){
|
|
|
|
|
|
+ public Object[][] providerVectorType() {
|
|
return new Object[][]{
|
|
return new Object[][]{
|
|
{DataType.FloatVector},
|
|
{DataType.FloatVector},
|
|
{DataType.BinaryVector},
|
|
{DataType.BinaryVector},
|
|
@@ -46,18 +54,20 @@ public class CreateCollectionTest extends BaseTest {
|
|
{DataType.SparseFloatVector},
|
|
{DataType.SparseFloatVector},
|
|
};
|
|
};
|
|
}
|
|
}
|
|
|
|
+
|
|
@Test(description = "Create simple collection success", groups = {"Smoke"})
|
|
@Test(description = "Create simple collection success", groups = {"Smoke"})
|
|
- public void createSimpleCollectionSuccess(){
|
|
|
|
|
|
+ public void createSimpleCollectionSuccess() {
|
|
milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
- .collectionName(simpleCollection)
|
|
|
|
- .dimension(CommonData.dim)
|
|
|
|
- .autoID(false)
|
|
|
|
|
|
+ .collectionName(simpleCollection)
|
|
|
|
+ .dimension(CommonData.dim)
|
|
|
|
+ .autoID(false)
|
|
.build());
|
|
.build());
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
Assert.assertTrue(listCollectionsResp.getCollectionNames().contains("simpleCollection"));
|
|
Assert.assertTrue(listCollectionsResp.getCollectionNames().contains("simpleCollection"));
|
|
}
|
|
}
|
|
|
|
+
|
|
@Test(description = "Create duplicate collection", groups = {"Smoke"})
|
|
@Test(description = "Create duplicate collection", groups = {"Smoke"})
|
|
- public void createDuplicateSimpleCollection(){
|
|
|
|
|
|
+ public void createDuplicateSimpleCollection() {
|
|
milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
.collectionName(repeatCollection)
|
|
.collectionName(repeatCollection)
|
|
.dimension(CommonData.dim)
|
|
.dimension(CommonData.dim)
|
|
@@ -66,7 +76,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
try {
|
|
try {
|
|
milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
.collectionName(repeatCollection)
|
|
.collectionName(repeatCollection)
|
|
- .dimension(CommonData.dim+1)
|
|
|
|
|
|
+ .dimension(CommonData.dim + 1)
|
|
.autoID(true)
|
|
.autoID(true)
|
|
.build());
|
|
.build());
|
|
} catch (Exception e) {
|
|
} catch (Exception e) {
|
|
@@ -75,69 +85,69 @@ public class CreateCollectionTest extends BaseTest {
|
|
}
|
|
}
|
|
|
|
|
|
@Test(description = "Create collection with index params,will auto load", groups = {"Smoke"})
|
|
@Test(description = "Create collection with index params,will auto load", groups = {"Smoke"})
|
|
- public void createCollectionWithIndexParams(){
|
|
|
|
- CreateCollectionReq.FieldSchema fieldInt64=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ public void createCollectionWithIndexParams() {
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt64 = CreateCollectionReq.FieldSchema.builder()
|
|
.autoID(false)
|
|
.autoID(false)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.isPrimaryKey(true)
|
|
.isPrimaryKey(true)
|
|
.name(CommonData.fieldInt64)
|
|
.name(CommonData.fieldInt64)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt32=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt32 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int32)
|
|
.dataType(DataType.Int32)
|
|
.name(CommonData.fieldInt32)
|
|
.name(CommonData.fieldInt32)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt16=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt16 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int16)
|
|
.dataType(DataType.Int16)
|
|
.name(CommonData.fieldInt16)
|
|
.name(CommonData.fieldInt16)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt8=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt8 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int8)
|
|
.dataType(DataType.Int8)
|
|
.name(CommonData.fieldInt8)
|
|
.name(CommonData.fieldInt8)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldDouble=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldDouble = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Double)
|
|
.dataType(DataType.Double)
|
|
.name(CommonData.fieldDouble)
|
|
.name(CommonData.fieldDouble)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldArray=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldArray = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Array)
|
|
.dataType(DataType.Array)
|
|
.name(CommonData.fieldArray)
|
|
.name(CommonData.fieldArray)
|
|
.elementType(DataType.Int64)
|
|
.elementType(DataType.Int64)
|
|
.maxCapacity(1000)
|
|
.maxCapacity(1000)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldBool=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldBool = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Bool)
|
|
.dataType(DataType.Bool)
|
|
.name(CommonData.fieldBool)
|
|
.name(CommonData.fieldBool)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldVarchar=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldVarchar = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.VarChar)
|
|
.dataType(DataType.VarChar)
|
|
.name(CommonData.fieldVarchar)
|
|
.name(CommonData.fieldVarchar)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.maxLength(1000)
|
|
.maxLength(1000)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloat=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloat = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Float)
|
|
.dataType(DataType.Float)
|
|
.name(CommonData.fieldFloat)
|
|
.name(CommonData.fieldFloat)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldJson=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldJson = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.JSON)
|
|
.dataType(DataType.JSON)
|
|
.name(CommonData.fieldJson)
|
|
.name(CommonData.fieldJson)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloatVector=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloatVector = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.FloatVector)
|
|
.dataType(DataType.FloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.dimension(CommonData.dim)
|
|
.dimension(CommonData.dim)
|
|
.build();
|
|
.build();
|
|
|
|
|
|
- List<CreateCollectionReq.FieldSchema> fieldSchemaList=new ArrayList<>();
|
|
|
|
|
|
+ List<CreateCollectionReq.FieldSchema> fieldSchemaList = new ArrayList<>();
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt16);
|
|
fieldSchemaList.add(fieldInt16);
|
|
@@ -149,7 +159,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
- CreateCollectionReq.CollectionSchema collectionSchema= CreateCollectionReq.CollectionSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.CollectionSchema collectionSchema = CreateCollectionReq.CollectionSchema.builder()
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.build();
|
|
.build();
|
|
IndexParam indexParam = IndexParam.builder()
|
|
IndexParam indexParam = IndexParam.builder()
|
|
@@ -170,14 +180,14 @@ public class CreateCollectionTest extends BaseTest {
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithIndex));
|
|
Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithIndex));
|
|
//insert
|
|
//insert
|
|
- CommonFunction.generateDefaultData(0,100,CommonData.dim,DataType.FloatVector);
|
|
|
|
|
|
+ CommonFunction.generateDefaultData(0, 100, CommonData.dim, DataType.FloatVector);
|
|
// search
|
|
// search
|
|
SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithIndex);
|
|
SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithIndex);
|
|
- Assert.assertEquals(searchResp.getSearchResults().size(),10);
|
|
|
|
|
|
+ Assert.assertEquals(searchResp.getSearchResults().size(), 10);
|
|
}
|
|
}
|
|
|
|
|
|
- @Test(description="create collection with different vector ", groups={"Smoke"}, dataProvider="VectorTypeList")
|
|
|
|
- public void createCollectionWithDifferentVector(DataType vectorType){
|
|
|
|
|
|
+ @Test(description = "create collection with different vector ", groups = {"Smoke"}, dataProvider = "VectorTypeList")
|
|
|
|
+ public void createCollectionWithDifferentVector(DataType vectorType) {
|
|
String newCollection = CommonFunction.createNewCollection(CommonData.dim, null, vectorType);
|
|
String newCollection = CommonFunction.createNewCollection(CommonData.dim, null, vectorType);
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(newCollection));
|
|
Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(newCollection));
|
|
@@ -185,38 +195,38 @@ public class CreateCollectionTest extends BaseTest {
|
|
}
|
|
}
|
|
|
|
|
|
@Test(description = "Create collection with Null Data, will auto load", groups = {"Smoke"})
|
|
@Test(description = "Create collection with Null Data, will auto load", groups = {"Smoke"})
|
|
- public void createCollectionWithNullData(){
|
|
|
|
- CreateCollectionReq.FieldSchema fieldInt64=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ public void createCollectionWithNullData() {
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt64 = CreateCollectionReq.FieldSchema.builder()
|
|
.autoID(false)
|
|
.autoID(false)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.isPrimaryKey(true)
|
|
.isPrimaryKey(true)
|
|
.name(CommonData.fieldInt64)
|
|
.name(CommonData.fieldInt64)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt32=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt32 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int32)
|
|
.dataType(DataType.Int32)
|
|
.name(CommonData.fieldInt32)
|
|
.name(CommonData.fieldInt32)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt16=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt16 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int16)
|
|
.dataType(DataType.Int16)
|
|
.name(CommonData.fieldInt16)
|
|
.name(CommonData.fieldInt16)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt8=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt8 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int8)
|
|
.dataType(DataType.Int8)
|
|
.name(CommonData.fieldInt8)
|
|
.name(CommonData.fieldInt8)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldDouble=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldDouble = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Double)
|
|
.dataType(DataType.Double)
|
|
.name(CommonData.fieldDouble)
|
|
.name(CommonData.fieldDouble)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldArray=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldArray = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Array)
|
|
.dataType(DataType.Array)
|
|
.name(CommonData.fieldArray)
|
|
.name(CommonData.fieldArray)
|
|
.elementType(DataType.Int64)
|
|
.elementType(DataType.Int64)
|
|
@@ -224,39 +234,39 @@ public class CreateCollectionTest extends BaseTest {
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldBool=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldBool = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Bool)
|
|
.dataType(DataType.Bool)
|
|
.name(CommonData.fieldBool)
|
|
.name(CommonData.fieldBool)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldVarchar=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldVarchar = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.VarChar)
|
|
.dataType(DataType.VarChar)
|
|
.name(CommonData.fieldVarchar)
|
|
.name(CommonData.fieldVarchar)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.maxLength(1000)
|
|
.maxLength(1000)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloat=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloat = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Float)
|
|
.dataType(DataType.Float)
|
|
.name(CommonData.fieldFloat)
|
|
.name(CommonData.fieldFloat)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldJson=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldJson = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.JSON)
|
|
.dataType(DataType.JSON)
|
|
.name(CommonData.fieldJson)
|
|
.name(CommonData.fieldJson)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloatVector=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloatVector = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.FloatVector)
|
|
.dataType(DataType.FloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.dimension(CommonData.dim)
|
|
.dimension(CommonData.dim)
|
|
.build();
|
|
.build();
|
|
|
|
|
|
- List<CreateCollectionReq.FieldSchema> fieldSchemaList=new ArrayList<>();
|
|
|
|
|
|
+ List<CreateCollectionReq.FieldSchema> fieldSchemaList = new ArrayList<>();
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt16);
|
|
fieldSchemaList.add(fieldInt16);
|
|
@@ -268,7 +278,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
- CreateCollectionReq.CollectionSchema collectionSchema= CreateCollectionReq.CollectionSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.CollectionSchema collectionSchema = CreateCollectionReq.CollectionSchema.builder()
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.build();
|
|
.build();
|
|
IndexParam indexParam = IndexParam.builder()
|
|
IndexParam indexParam = IndexParam.builder()
|
|
@@ -279,7 +289,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
.build();
|
|
.build();
|
|
CreateCollectionReq createCollectionReq = CreateCollectionReq.builder()
|
|
CreateCollectionReq createCollectionReq = CreateCollectionReq.builder()
|
|
.collectionSchema(collectionSchema)
|
|
.collectionSchema(collectionSchema)
|
|
- .collectionName(collectionNameWithIndex)
|
|
|
|
|
|
+ .collectionName(collectionNameWithNull)
|
|
.enableDynamicField(false)
|
|
.enableDynamicField(false)
|
|
.indexParams(Collections.singletonList(indexParam))
|
|
.indexParams(Collections.singletonList(indexParam))
|
|
.numShards(1)
|
|
.numShards(1)
|
|
@@ -287,85 +297,85 @@ public class CreateCollectionTest extends BaseTest {
|
|
BaseTest.milvusClientV2.createCollection(createCollectionReq);
|
|
BaseTest.milvusClientV2.createCollection(createCollectionReq);
|
|
|
|
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
- Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithIndex));
|
|
|
|
|
|
+ Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithNull));
|
|
//insert
|
|
//insert
|
|
- CommonFunction.generateDefaultData(0,100,CommonData.dim,DataType.FloatVector);
|
|
|
|
|
|
+ CommonFunction.generateDefaultData(0, 100, CommonData.dim, DataType.FloatVector);
|
|
// search
|
|
// search
|
|
- SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithIndex);
|
|
|
|
- Assert.assertEquals(searchResp.getSearchResults().size(),10);
|
|
|
|
|
|
+ SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithNull);
|
|
|
|
+ Assert.assertEquals(searchResp.getSearchResults().size(), 10);
|
|
}
|
|
}
|
|
|
|
|
|
@Test(description = "Create collection with Default Data, will auto load", groups = {"Smoke"})
|
|
@Test(description = "Create collection with Default Data, will auto load", groups = {"Smoke"})
|
|
- public void createCollectionWithDefaultData(){
|
|
|
|
- CreateCollectionReq.FieldSchema fieldInt64=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ public void createCollectionWithDefaultData() {
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt64 = CreateCollectionReq.FieldSchema.builder()
|
|
.autoID(false)
|
|
.autoID(false)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.isPrimaryKey(true)
|
|
.isPrimaryKey(true)
|
|
.name(CommonData.fieldInt64)
|
|
.name(CommonData.fieldInt64)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt32=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt32 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int32)
|
|
.dataType(DataType.Int32)
|
|
.name(CommonData.fieldInt32)
|
|
.name(CommonData.fieldInt32)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.defaultValue(CommonData.defaultValueInt)
|
|
.defaultValue(CommonData.defaultValueInt)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt16=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt16 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int16)
|
|
.dataType(DataType.Int16)
|
|
.name(CommonData.fieldInt16)
|
|
.name(CommonData.fieldInt16)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt8=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt8 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int8)
|
|
.dataType(DataType.Int8)
|
|
.name(CommonData.fieldInt8)
|
|
.name(CommonData.fieldInt8)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldDouble=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldDouble = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Double)
|
|
.dataType(DataType.Double)
|
|
.name(CommonData.fieldDouble)
|
|
.name(CommonData.fieldDouble)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.defaultValue(CommonData.defaultValueDouble)
|
|
.defaultValue(CommonData.defaultValueDouble)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldArray=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldArray = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Array)
|
|
.dataType(DataType.Array)
|
|
.name(CommonData.fieldArray)
|
|
.name(CommonData.fieldArray)
|
|
.elementType(DataType.Int64)
|
|
.elementType(DataType.Int64)
|
|
.maxCapacity(1000)
|
|
.maxCapacity(1000)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldBool=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldBool = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Bool)
|
|
.dataType(DataType.Bool)
|
|
.name(CommonData.fieldBool)
|
|
.name(CommonData.fieldBool)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.defaultValue(CommonData.defaultValueBool)
|
|
.defaultValue(CommonData.defaultValueBool)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldVarchar=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldVarchar = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.VarChar)
|
|
.dataType(DataType.VarChar)
|
|
.name(CommonData.fieldVarchar)
|
|
.name(CommonData.fieldVarchar)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.maxLength(1000)
|
|
.maxLength(1000)
|
|
.defaultValue(CommonData.defaultValueString)
|
|
.defaultValue(CommonData.defaultValueString)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloat=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloat = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Float)
|
|
.dataType(DataType.Float)
|
|
.name(CommonData.fieldFloat)
|
|
.name(CommonData.fieldFloat)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.defaultValue(CommonData.defaultValueFloat)
|
|
.defaultValue(CommonData.defaultValueFloat)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldJson=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldJson = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.JSON)
|
|
.dataType(DataType.JSON)
|
|
.name(CommonData.fieldJson)
|
|
.name(CommonData.fieldJson)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloatVector=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloatVector = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.FloatVector)
|
|
.dataType(DataType.FloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.dimension(CommonData.dim)
|
|
.dimension(CommonData.dim)
|
|
.build();
|
|
.build();
|
|
|
|
|
|
- List<CreateCollectionReq.FieldSchema> fieldSchemaList=new ArrayList<>();
|
|
|
|
|
|
+ List<CreateCollectionReq.FieldSchema> fieldSchemaList = new ArrayList<>();
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt16);
|
|
fieldSchemaList.add(fieldInt16);
|
|
@@ -377,7 +387,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
- CreateCollectionReq.CollectionSchema collectionSchema= CreateCollectionReq.CollectionSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.CollectionSchema collectionSchema = CreateCollectionReq.CollectionSchema.builder()
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.build();
|
|
.build();
|
|
IndexParam indexParam = IndexParam.builder()
|
|
IndexParam indexParam = IndexParam.builder()
|
|
@@ -388,7 +398,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
.build();
|
|
.build();
|
|
CreateCollectionReq createCollectionReq = CreateCollectionReq.builder()
|
|
CreateCollectionReq createCollectionReq = CreateCollectionReq.builder()
|
|
.collectionSchema(collectionSchema)
|
|
.collectionSchema(collectionSchema)
|
|
- .collectionName(collectionNameWithIndex)
|
|
|
|
|
|
+ .collectionName(collectionNameWithDefault)
|
|
.enableDynamicField(false)
|
|
.enableDynamicField(false)
|
|
.indexParams(Collections.singletonList(indexParam))
|
|
.indexParams(Collections.singletonList(indexParam))
|
|
.numShards(1)
|
|
.numShards(1)
|
|
@@ -396,65 +406,65 @@ public class CreateCollectionTest extends BaseTest {
|
|
BaseTest.milvusClientV2.createCollection(createCollectionReq);
|
|
BaseTest.milvusClientV2.createCollection(createCollectionReq);
|
|
|
|
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
- Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithIndex));
|
|
|
|
|
|
+ Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithDefault));
|
|
//insert
|
|
//insert
|
|
- CommonFunction.generateDefaultData(0,100,CommonData.dim,DataType.FloatVector);
|
|
|
|
|
|
+ CommonFunction.generateDefaultData(0, 100, CommonData.dim, DataType.FloatVector);
|
|
// search
|
|
// search
|
|
- SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithIndex);
|
|
|
|
- Assert.assertEquals(searchResp.getSearchResults().size(),10);
|
|
|
|
|
|
+ SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithDefault);
|
|
|
|
+ Assert.assertEquals(searchResp.getSearchResults().size(), 10);
|
|
}
|
|
}
|
|
|
|
|
|
@Test(description = "Create collection with nullable and Default Data, will auto load", groups = {"Smoke"})
|
|
@Test(description = "Create collection with nullable and Default Data, will auto load", groups = {"Smoke"})
|
|
- public void createCollectionWithNullAndDefaultData(){
|
|
|
|
- CreateCollectionReq.FieldSchema fieldInt64=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ public void createCollectionWithNullAndDefaultData() {
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt64 = CreateCollectionReq.FieldSchema.builder()
|
|
.autoID(false)
|
|
.autoID(false)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.dataType(io.milvus.v2.common.DataType.Int64)
|
|
.isPrimaryKey(true)
|
|
.isPrimaryKey(true)
|
|
.name(CommonData.fieldInt64)
|
|
.name(CommonData.fieldInt64)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt32=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt32 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int32)
|
|
.dataType(DataType.Int32)
|
|
.name(CommonData.fieldInt32)
|
|
.name(CommonData.fieldInt32)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueInt)
|
|
.defaultValue(CommonData.defaultValueInt)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt16=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt16 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int16)
|
|
.dataType(DataType.Int16)
|
|
.name(CommonData.fieldInt16)
|
|
.name(CommonData.fieldInt16)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldInt8=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt8 = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Int8)
|
|
.dataType(DataType.Int8)
|
|
.name(CommonData.fieldInt8)
|
|
.name(CommonData.fieldInt8)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.defaultValue(CommonData.defaultValueShort)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldDouble=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldDouble = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Double)
|
|
.dataType(DataType.Double)
|
|
.name(CommonData.fieldDouble)
|
|
.name(CommonData.fieldDouble)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueDouble)
|
|
.defaultValue(CommonData.defaultValueDouble)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldArray=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldArray = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Array)
|
|
.dataType(DataType.Array)
|
|
.name(CommonData.fieldArray)
|
|
.name(CommonData.fieldArray)
|
|
.elementType(DataType.Int64)
|
|
.elementType(DataType.Int64)
|
|
.maxCapacity(1000)
|
|
.maxCapacity(1000)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldBool=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldBool = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Bool)
|
|
.dataType(DataType.Bool)
|
|
.name(CommonData.fieldBool)
|
|
.name(CommonData.fieldBool)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueBool)
|
|
.defaultValue(CommonData.defaultValueBool)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldVarchar=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldVarchar = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.VarChar)
|
|
.dataType(DataType.VarChar)
|
|
.name(CommonData.fieldVarchar)
|
|
.name(CommonData.fieldVarchar)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
@@ -462,26 +472,26 @@ public class CreateCollectionTest extends BaseTest {
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueString)
|
|
.defaultValue(CommonData.defaultValueString)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloat=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloat = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.Float)
|
|
.dataType(DataType.Float)
|
|
.name(CommonData.fieldFloat)
|
|
.name(CommonData.fieldFloat)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.isNullable(true)
|
|
.isNullable(true)
|
|
.defaultValue(CommonData.defaultValueFloat)
|
|
.defaultValue(CommonData.defaultValueFloat)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldJson=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldJson = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.JSON)
|
|
.dataType(DataType.JSON)
|
|
.name(CommonData.fieldJson)
|
|
.name(CommonData.fieldJson)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.build();
|
|
.build();
|
|
- CreateCollectionReq.FieldSchema fieldFloatVector=CreateCollectionReq.FieldSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldFloatVector = CreateCollectionReq.FieldSchema.builder()
|
|
.dataType(DataType.FloatVector)
|
|
.dataType(DataType.FloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.name(CommonData.fieldFloatVector)
|
|
.isPrimaryKey(false)
|
|
.isPrimaryKey(false)
|
|
.dimension(CommonData.dim)
|
|
.dimension(CommonData.dim)
|
|
.build();
|
|
.build();
|
|
|
|
|
|
- List<CreateCollectionReq.FieldSchema> fieldSchemaList=new ArrayList<>();
|
|
|
|
|
|
+ List<CreateCollectionReq.FieldSchema> fieldSchemaList = new ArrayList<>();
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt64);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt32);
|
|
fieldSchemaList.add(fieldInt16);
|
|
fieldSchemaList.add(fieldInt16);
|
|
@@ -493,7 +503,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldJson);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldVarchar);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
fieldSchemaList.add(fieldFloatVector);
|
|
- CreateCollectionReq.CollectionSchema collectionSchema= CreateCollectionReq.CollectionSchema.builder()
|
|
|
|
|
|
+ CreateCollectionReq.CollectionSchema collectionSchema = CreateCollectionReq.CollectionSchema.builder()
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.fieldSchemaList(fieldSchemaList)
|
|
.build();
|
|
.build();
|
|
IndexParam indexParam = IndexParam.builder()
|
|
IndexParam indexParam = IndexParam.builder()
|
|
@@ -504,7 +514,7 @@ public class CreateCollectionTest extends BaseTest {
|
|
.build();
|
|
.build();
|
|
CreateCollectionReq createCollectionReq = CreateCollectionReq.builder()
|
|
CreateCollectionReq createCollectionReq = CreateCollectionReq.builder()
|
|
.collectionSchema(collectionSchema)
|
|
.collectionSchema(collectionSchema)
|
|
- .collectionName(collectionNameWithIndex)
|
|
|
|
|
|
+ .collectionName(collectionNameWithNullAndDefault)
|
|
.enableDynamicField(false)
|
|
.enableDynamicField(false)
|
|
.indexParams(Collections.singletonList(indexParam))
|
|
.indexParams(Collections.singletonList(indexParam))
|
|
.numShards(1)
|
|
.numShards(1)
|
|
@@ -512,12 +522,82 @@ public class CreateCollectionTest extends BaseTest {
|
|
BaseTest.milvusClientV2.createCollection(createCollectionReq);
|
|
BaseTest.milvusClientV2.createCollection(createCollectionReq);
|
|
|
|
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
ListCollectionsResp listCollectionsResp = milvusClientV2.listCollections();
|
|
- Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithIndex));
|
|
|
|
|
|
+ Assert.assertTrue(listCollectionsResp.getCollectionNames().contains(collectionNameWithNullAndDefault));
|
|
//insert
|
|
//insert
|
|
- CommonFunction.generateDefaultData(0,100,CommonData.dim,DataType.FloatVector);
|
|
|
|
|
|
+ CommonFunction.generateDefaultData(0, 100, CommonData.dim, DataType.FloatVector);
|
|
// search
|
|
// search
|
|
- SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithIndex);
|
|
|
|
- Assert.assertEquals(searchResp.getSearchResults().size(),10);
|
|
|
|
|
|
+ SearchResp searchResp = CommonFunction.defaultSearch(collectionNameWithNullAndDefault);
|
|
|
|
+ Assert.assertEquals(searchResp.getSearchResults().size(), 10);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+ @Test(description = "Create collection with properties", groups = {"Smoke"})
|
|
|
|
+ public void createCollectionWithProperties() {
|
|
|
|
+ String collection = "a"+MathUtil.getRandomString(10);
|
|
|
|
+ Map<String, String> map = new HashMap<String, String>() {{
|
|
|
|
+ put(Constant.MMAP_ENABLED, "true");
|
|
|
|
+ }};
|
|
|
|
+ CreateCollectionReq.CollectionSchema collectionSchema = CommonFunction.providerCollectionSchema(CommonData.dim, DataType.FloatVector);
|
|
|
|
+ milvusClientV2.createCollection(CreateCollectionReq.builder()
|
|
|
|
+ .collectionName(collection)
|
|
|
|
+ .properties(map)
|
|
|
|
+ .collectionSchema(collectionSchema)
|
|
|
|
+ .build());
|
|
|
|
+ DescribeCollectionResp describeCollectionResp = milvusClientV2.describeCollection(DescribeCollectionReq.builder()
|
|
|
|
+ .collectionName(collection)
|
|
|
|
+ .build());
|
|
|
|
+ Map<String, String> properties = describeCollectionResp.getProperties();
|
|
|
|
+ Assert.assertTrue(properties.containsKey(Constant.MMAP_ENABLED));
|
|
|
|
+ Assert.assertTrue(properties.get(Constant.MMAP_ENABLED).equalsIgnoreCase("true"));
|
|
|
|
+ milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collection).build());
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ @Test(description = "Create collection with functions", groups = {"Smoke"})
|
|
|
|
+ public void createCollectionWithFunctions() {
|
|
|
|
+ String collection = "a"+MathUtil.getRandomString(10);
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldInt64 = CreateCollectionReq.FieldSchema.builder()
|
|
|
|
+ .autoID(false)
|
|
|
|
+ .dataType(io.milvus.v2.common.DataType.Int64)
|
|
|
|
+ .isPrimaryKey(true)
|
|
|
|
+ .name(CommonData.fieldInt64)
|
|
|
|
+ .build();
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldVarchar = CreateCollectionReq.FieldSchema.builder()
|
|
|
|
+ .dataType(DataType.VarChar)
|
|
|
|
+ .isPrimaryKey(false)
|
|
|
|
+ .name(CommonData.fieldVarchar)
|
|
|
|
+ .enableAnalyzer(true)
|
|
|
|
+ .build();
|
|
|
|
+ CreateCollectionReq.FieldSchema fieldVector = CreateCollectionReq.FieldSchema.builder()
|
|
|
|
+ .dataType(DataType.SparseFloatVector)
|
|
|
|
+ .isPrimaryKey(false)
|
|
|
|
+// .dimension(CommonData.dim)
|
|
|
|
+ .name(CommonData.fieldSparseVector)
|
|
|
|
+ .build();
|
|
|
|
+
|
|
|
|
+ List<CreateCollectionReq.FieldSchema> fieldSchemaList = new ArrayList<>();
|
|
|
|
+ fieldSchemaList.add(fieldInt64);
|
|
|
|
+ fieldSchemaList.add(fieldVarchar);
|
|
|
|
+ fieldSchemaList.add(fieldVector);
|
|
|
|
+ List<CreateCollectionReq.Function> functions = new ArrayList<>();
|
|
|
|
+ functions.add(CreateCollectionReq.Function.builder()
|
|
|
|
+ .name("varcharToFloatVector")
|
|
|
|
+ .functionType(FunctionType.BM25)
|
|
|
|
+ .inputFieldNames(Lists.newArrayList(CommonData.fieldVarchar))
|
|
|
|
+ .outputFieldNames(Lists.newArrayList(CommonData.fieldSparseVector))
|
|
|
|
+ .build());
|
|
|
|
+ CreateCollectionReq.CollectionSchema schema = CreateCollectionReq.CollectionSchema.builder()
|
|
|
|
+ .fieldSchemaList(fieldSchemaList)
|
|
|
|
+ .functionList(functions)
|
|
|
|
+ .build();
|
|
|
|
+
|
|
|
|
+ milvusClientV2.createCollection(CreateCollectionReq.builder().collectionName(collection)
|
|
|
|
+ .collectionSchema(schema)
|
|
|
|
+ .numShards(1)
|
|
|
|
+ .enableDynamicField(false)
|
|
|
|
+ .description("collection desc")
|
|
|
|
+ .build());
|
|
|
|
+
|
|
|
|
+ milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(collection).build());
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+
|
|
}
|
|
}
|