From fa6b9f89da12e43f029f89004456aa1d3da3c23a Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Mon, 8 Nov 2021 20:46:26 +0800 Subject: [PATCH] Add get_query_segment_info utility wrapper (#11447) Signed-off-by: zhuwenxing --- tests/python_client/base/utility_wrapper.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/python_client/base/utility_wrapper.py b/tests/python_client/base/utility_wrapper.py index c78a434a22..5f542c20db 100644 --- a/tests/python_client/base/utility_wrapper.py +++ b/tests/python_client/base/utility_wrapper.py @@ -14,6 +14,14 @@ class ApiUtilityWrapper: ut = utility + def get_query_segment_info(self, collection_name, timeout=None, using="default", check_task=None, check_items=None): + timeout = TIMEOUT if timeout is None else timeout + func_name = sys._getframe().f_code.co_name + res, is_succ = api_request([self.ut.get_query_segment_info, collection_name, timeout, using]) + check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ, + collection_name=collection_name, timeout=timeout, using=using).run() + return res, check_result + def loading_progress(self, collection_name, partition_names=None, using="default", check_task=None, check_items=None): func_name = sys._getframe().f_code.co_name