未验证 提交 3b9716bb 编写于 作者: Y yanliang567 提交者: GitHub

Update tests for resource group (#22867)

Signed-off-by: Nyanliang567 <yanliang.qiao@zilliz.com>
上级 7cdcaeb1
...@@ -576,7 +576,6 @@ class TestTransferNode(TestcaseBase): ...@@ -576,7 +576,6 @@ class TestTransferNode(TestcaseBase):
check_task=CheckTasks.check_rg_property, check_task=CheckTasks.check_rg_property,
check_items=default_rg_info) check_items=default_rg_info)
@pytest.mark.xfail(reason="issue #22224")
@pytest.mark.tags(CaseLabel.MultiQueryNodes) @pytest.mark.tags(CaseLabel.MultiQueryNodes)
@pytest.mark.parametrize("num_node", [0, 99, -1, 0.5, True, "str"]) @pytest.mark.parametrize("num_node", [0, 99, -1, 0.5, True, "str"])
def test_transfer_node_with_wrong_num_node(self, num_node): def test_transfer_node_with_wrong_num_node(self, num_node):
...@@ -1374,7 +1373,7 @@ class TestResourceGroupMultiNodes(TestcaseBase): ...@@ -1374,7 +1373,7 @@ class TestResourceGroupMultiNodes(TestcaseBase):
# transfer 1/2/3 replicas from default rg to rgA # transfer 1/2/3 replicas from default rg to rgA
self.utility_wrap.transfer_replica(source=ct.default_resource_group_name, target=rgA_name, self.utility_wrap.transfer_replica(source=ct.default_resource_group_name, target=rgA_name,
collection_name=collection_w.name, collection_name=collection_w.name,
num_replica=1) num_replica=num_replica)
# verify search succ # verify search succ
nq = 5 nq = 5
vectors = [[random.random() for _ in range(dim)] for _ in range(nq)] vectors = [[random.random() for _ in range(dim)] for _ in range(nq)]
...@@ -1399,7 +1398,7 @@ class TestResourceGroupMultiNodes(TestcaseBase): ...@@ -1399,7 +1398,7 @@ class TestResourceGroupMultiNodes(TestcaseBase):
# transfer 1/2/3 replicas from rgA to default rg # transfer 1/2/3 replicas from rgA to default rg
self.utility_wrap.transfer_replica(source=rgA_name, target=ct.default_resource_group_name, self.utility_wrap.transfer_replica(source=rgA_name, target=ct.default_resource_group_name,
collection_name=collection_w.name, collection_name=collection_w.name,
num_replica=1) num_replica=num_replica)
# verify search succ # verify search succ
collection_w.search(vectors[:nq], collection_w.search(vectors[:nq],
ct.default_float_vec_field_name, ct.default_float_vec_field_name,
...@@ -1642,9 +1641,9 @@ class TestResourceGroupMultiNodes(TestcaseBase): ...@@ -1642,9 +1641,9 @@ class TestResourceGroupMultiNodes(TestcaseBase):
4. load the collection into rgA and rgB 4. load the collection into rgA and rgB
verify search succ verify search succ
5. transfer replica from rgB to default rg 5. transfer replica from rgB to default rg
verify search succ verify fail
6. transfer node from rgA to default rgA 6. transfer node from rgA to default rgA
verify fail with error verify succ
7. transfer node from rgB to default rgB 7. transfer node from rgB to default rgB
8. verify search succ 8. verify search succ
""" """
...@@ -1694,10 +1693,14 @@ class TestResourceGroupMultiNodes(TestcaseBase): ...@@ -1694,10 +1693,14 @@ class TestResourceGroupMultiNodes(TestcaseBase):
"limit": ct.default_limit} "limit": ct.default_limit}
) )
# transfer the replica from rgB to default rg # 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, self.utility_wrap.transfer_replica(source=rgB_name,
target=ct.default_resource_group_name, target=ct.default_resource_group_name,
collection_name=collection_w.name, collection_name=collection_w.name,
num_replica=1) num_replica=1,
check_task=CheckTasks.err_res,
check_items=error)
time.sleep(1) time.sleep(1)
collection_w.search(vectors[:nq], collection_w.search(vectors[:nq],
...@@ -1711,13 +1714,9 @@ class TestResourceGroupMultiNodes(TestcaseBase): ...@@ -1711,13 +1714,9 @@ class TestResourceGroupMultiNodes(TestcaseBase):
) )
# transfer all nodes back to default rg # 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, self.utility_wrap.transfer_node(source=rgA_name,
target=ct.default_resource_group_name, target=ct.default_resource_group_name,
num_node=2, num_node=2)
check_task=CheckTasks.err_res,
check_items=error)
self.utility_wrap.transfer_node(source=rgB_name, self.utility_wrap.transfer_node(source=rgB_name,
target=ct.default_resource_group_name, target=ct.default_resource_group_name,
num_node=2) num_node=2)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册