From 693dd911f3e9eb2a25cb410a8bcdd854a649837e Mon Sep 17 00:00:00 2001 From: ThreadDao Date: Fri, 24 Feb 2023 08:55:49 +0800 Subject: [PATCH] Add segment info and replicas in verify rg (#22366) Signed-off-by: ThreadDao --- .../testcases/test_chaos_resource_group.py | 26 ++++++++++++++++--- 1 file changed, 22 insertions(+), 4 deletions(-) diff --git a/tests/python_client/chaos/testcases/test_chaos_resource_group.py b/tests/python_client/chaos/testcases/test_chaos_resource_group.py index 0b2749af7..024d6b140 100644 --- a/tests/python_client/chaos/testcases/test_chaos_resource_group.py +++ b/tests/python_client/chaos/testcases/test_chaos_resource_group.py @@ -53,7 +53,8 @@ class TestChaosRG(TestcaseBase): log.info(f"[ResourceGroup] Create rg {rg_name} done") self.utility_wrap.transfer_node(source=ct.default_resource_group_name, target=rg_name, num_node=rg_info["available_node"]) - log.info(f'[ResourceGroup] Transfer {rg_info["available_node"]} nodes from {ct.default_resource_group_name} to {rg_name} done') + log.info( + f'[ResourceGroup] Transfer {rg_info["available_node"]} nodes from {ct.default_resource_group_name} to {rg_name} done') # verify RGs resource_groups, _ = self.utility_wrap.list_resource_groups() @@ -173,13 +174,30 @@ class TestChaosRG(TestcaseBase): log.info(f'[ResourceGroup] Rg of {rg_info["name"]} info is: {desc_rg_info}') # search - for coll_name in coll_name_1, coll_name_2: + for coll_name in coll_name_2, coll_name_1: + # get query segment info + segment, _ = self.utility_wrap.get_query_segment_info(coll_name) + log.info(f"{coll_name} query segment info: {segment}") + + # get replicas collection_w = self.init_collection_wrap(name=coll_name, active_trace=True) - for i in range(10): + replicas, _ = collection_w.get_replicas(check_task=ct.CheckTasks.check_nothing) + log.info(f"{coll_name} replicas: {replicas}") + + # search + for i in range(100): search_vectors = cf.gen_vectors(ct.default_nq, ct.default_dim) search_params = {"metric_type": "L2", "params": {"ef": 64}} search_res, _ = collection_w.search(data=search_vectors, anns_field=ct.default_float_vec_field_name, param=search_params, limit=ct.default_limit, expr="int64 >= 0") assert len(search_res) == ct.default_nq - assert len(search_res[0]) == ct.default_limit \ No newline at end of file + assert len(search_res[0]) == ct.default_limit + + # show query segment info finally + segment_2, _ = self.utility_wrap.get_query_segment_info(coll_name) + log.info(f"{coll_name} query segment info: {segment_2}") + + # show replicas finally + replicas_2, _ = collection_w.get_replicas(check_task=ct.CheckTasks.check_nothing) + log.info(f"{coll_name} replicas: {replicas_2}") \ No newline at end of file -- GitLab