Skip to content

Commit 398414d

Browse files
yongpengli-zyongpengli-z
authored
add testcases abort enable mmap (#1043) (#1044)
Signed-off-by: yongpengli-z <[email protected]>
1 parent bb995a5 commit 398414d

File tree

2 files changed

+91
-1
lines changed

2 files changed

+91
-1
lines changed

‎tests/milvustestv2/src/test/java/com/zilliz/milvustestv2/collection/DescribeCollectionTest.java‎

Lines changed: 32 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,23 @@
22

33
import com.zilliz.milvustestv2.common.BaseTest;
44
import com.zilliz.milvustestv2.common.CommonData;
5+
import com.zilliz.milvustestv2.common.CommonFunction;
6+
import io.milvus.grpc.CollectionSchema;
7+
import io.milvus.param.Constant;
8+
import io.milvus.v2.common.DataType;
9+
import io.milvus.v2.service.collection.request.AlterCollectionReq;
10+
import io.milvus.v2.service.collection.request.CreateCollectionReq;
511
import io.milvus.v2.service.collection.request.DescribeCollectionReq;
12+
import io.milvus.v2.service.collection.request.DropCollectionReq;
613
import io.milvus.v2.service.collection.response.DescribeCollectionResp;
714
import org.testng.Assert;
815
import org.testng.annotations.Test;
916

17+
import java.util.ArrayList;
18+
import java.util.HashMap;
19+
import java.util.List;
20+
import java.util.Map;
21+
1022
/**
1123
* @Author yongpeng.li
1224
* @Date 2024/2/19 11:11
@@ -18,6 +30,25 @@ public void describeCollection() {
1830
DescribeCollectionResp describeCollectionResp = milvusClientV2.describeCollection(DescribeCollectionReq.builder()
1931
.collectionName(CommonData.defaultFloatVectorCollection)
2032
.build());
21-
Assert.assertEquals(describeCollectionResp.getVectorFieldNames().get(0),CommonData.fieldFloatVector);
33+
Assert.assertEquals(describeCollectionResp.getVectorFieldNames().get(0), CommonData.fieldFloatVector);
34+
}
35+
36+
@Test(description = "Describe collection when mmap field", groups = {"Smoke"})
37+
public void describeMMapCollection() {
38+
String newCollection = CommonFunction.createNewCollection(CommonData.dim, null, DataType.FloatVector);
39+
Map<String, String> map = new HashMap<String, String>() {{
40+
put(Constant.MMAP_ENABLED, "true");
41+
}};
42+
milvusClientV2.alterCollection(AlterCollectionReq.builder()
43+
.collectionName(newCollection)
44+
.properties(map).build());
45+
DescribeCollectionResp describeCollectionResp = milvusClientV2.describeCollection(DescribeCollectionReq.builder()
46+
.collectionName(newCollection)
47+
.build());
48+
Assert.assertTrue(describeCollectionResp.getProperties().containsKey(Constant.MMAP_ENABLED));
49+
Assert.assertTrue(describeCollectionResp.getProperties().get(Constant.MMAP_ENABLED).equalsIgnoreCase("true"));
50+
milvusClientV2.dropCollection(DropCollectionReq.builder().collectionName(newCollection).build());
51+
52+
2253
}
2354
}

‎tests/milvustestv2/src/test/java/com/zilliz/milvustestv2/index/AlterIndexTest.java‎

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
package com.zilliz.milvustestv2.index;
2+
3+
import com.zilliz.milvustestv2.common.BaseTest;
4+
import com.zilliz.milvustestv2.common.CommonData;
5+
import com.zilliz.milvustestv2.common.CommonFunction;
6+
import io.milvus.param.Constant;
7+
import io.milvus.v2.common.DataType;
8+
import io.milvus.v2.service.collection.request.DropCollectionReq;
9+
import io.milvus.v2.service.index.request.AlterIndexReq;
10+
import io.milvus.v2.service.index.request.DescribeIndexReq;
11+
import io.milvus.v2.service.index.request.ListIndexesReq;
12+
import io.milvus.v2.service.index.response.DescribeIndexResp;
13+
import io.milvus.v2.service.vector.request.DeleteReq;
14+
import org.testng.Assert;
15+
import org.testng.annotations.AfterClass;
16+
import org.testng.annotations.BeforeClass;
17+
import org.testng.annotations.Test;
18+
19+
import java.util.HashMap;
20+
import java.util.List;
21+
import java.util.Map;
22+
23+
public class AlterIndexTest extends BaseTest {
24+
private String collectionName;
25+
26+
@BeforeClass(alwaysRun = true)
27+
public void initTest() {
28+
collectionName = CommonFunction.createNewCollection(CommonData.dim, null, DataType.FloatVector);
29+
CommonFunction.createIndex(collectionName, DataType.FloatVector);
30+
}
31+
32+
@AfterClass(alwaysRun = true)
33+
public void destroyTestData() {
34+
milvusClientV2.dropCollection(DropCollectionReq.builder()
35+
.collectionName(collectionName).build());
36+
}
37+
38+
@Test(description = "alter index mmap", groups = {"Smoke"})
39+
public void alterIndexMMapTest() {
40+
List<String> strings = milvusClientV2.listIndexes(ListIndexesReq.builder()
41+
.collectionName(collectionName).build());
42+
System.out.println(strings);
43+
Map<String, String> stringMap = new HashMap<>();
44+
stringMap.put(Constant.MMAP_ENABLED, "true");
45+
milvusClientV2.alterIndex(AlterIndexReq.builder()
46+
.indexName(strings.get(0))
47+
.collectionName(collectionName)
48+
.properties(stringMap)
49+
.build());
50+
DescribeIndexResp describeIndexResp = milvusClientV2.describeIndex(DescribeIndexReq.builder()
51+
.collectionName(collectionName)
52+
.indexName(strings.get(0)).build());
53+
System.out.println(describeIndexResp);
54+
Assert.assertTrue(describeIndexResp.getIndexDescriptions().get(0).getExtraParams().get(Constant.MMAP_ENABLED).equalsIgnoreCase("true"));
55+
56+
}
57+
58+
59+
}

0 commit comments

Comments
 (0)