test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-01 03:18:29 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Update tests for resource group (#22867)

Browse Source 
Signed-off-by: yanliang567 <yanliang.qiao@zilliz.com>
This commit is contained in:
yanliang567 2023-04-01 10:02:23 +08:00 committed by GitHub
parent 7cdcaeb18d
commit 3b9716bbec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/python_client/testcases/test_resourcegroup.py
View File

@ -576,7 +576,6 @@ class TestTransferNode(TestcaseBase):
check_task=CheckTasks.check_rg_property,
check_items=default_rg_info)
@pytest.mark.xfail(reason="issue #22224")
@pytest.mark.tags(CaseLabel.MultiQueryNodes)
@pytest.mark.parametrize("num_node", [0, 99, -1, 0.5, True, "str"])
def test_transfer_node_with_wrong_num_node(self, num_node):
@ -1374,7 +1373,7 @@ class TestResourceGroupMultiNodes(TestcaseBase):
# transfer 1/2/3 replicas from default rg to rgA
self.utility_wrap.transfer_replica(source=ct.default_resource_group_name, target=rgA_name,
collection_name=collection_w.name,
num_replica=1)
num_replica=num_replica)
# verify search succ
nq = 5
vectors = [[random.random() for _ in range(dim)] for _ in range(nq)]
@ -1399,7 +1398,7 @@ class TestResourceGroupMultiNodes(TestcaseBase):
# transfer 1/2/3 replicas from rgA to default rg
self.utility_wrap.transfer_replica(source=rgA_name, target=ct.default_resource_group_name,
collection_name=collection_w.name,
num_replica=1)
num_replica=num_replica)
# verify search succ
collection_w.search(vectors[:nq],
ct.default_float_vec_field_name,
@ -1642,9 +1641,9 @@ class TestResourceGroupMultiNodes(TestcaseBase):
4. load the collection into rgA and rgB
verify search succ
5. transfer replica from rgB to default rg
verify search succ
verify fail
6. transfer node from rgA to default rgA
verify fail with error
verify succ
7. transfer node from rgB to default rgB
8. verify search succ
"""
@ -1694,10 +1693,14 @@ class TestResourceGroupMultiNodes(TestcaseBase):
"limit": ct.default_limit}
)
# transfer the replica from rgB to default rg
error = {ct.err_code: 999,
ct.err_msg: "transfer replica will cause replica loaded in both default rg and other rg"}
self.utility_wrap.transfer_replica(source=rgB_name,
target=ct.default_resource_group_name,
collection_name=collection_w.name,
num_replica=1)
num_replica=1,
check_task=CheckTasks.err_res,
check_items=error)
time.sleep(1)
collection_w.search(vectors[:nq],
@ -1711,13 +1714,9 @@ class TestResourceGroupMultiNodes(TestcaseBase):
)
# transfer all nodes back to default rg
error = {ct.err_code: 999,
ct.err_msg: f"can't transfer node, cause the resource group[{rgA_name}] and the resource group[{ct.default_resource_group_name}] loaded same collection"}
self.utility_wrap.transfer_node(source=rgA_name,
target=ct.default_resource_group_name,
num_node=2,
check_task=CheckTasks.err_res,
check_items=error)
num_node=2)
self.utility_wrap.transfer_node(source=rgB_name,
target=ct.default_resource_group_name,
num_node=2)