From f6514a233d46f5d49257eaa2af6efa31bc59eb7c Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Mon, 8 Nov 2021 20:48:34 +0800 Subject: [PATCH] Add load_balance utility wrapper (#11449) Signed-off-by: zhuwenxing --- tests/python_client/base/utility_wrapper.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/python_client/base/utility_wrapper.py b/tests/python_client/base/utility_wrapper.py index 5f542c20db..f1fcaa6d55 100644 --- a/tests/python_client/base/utility_wrapper.py +++ b/tests/python_client/base/utility_wrapper.py @@ -107,3 +107,14 @@ class ApiUtilityWrapper: check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ, timeout=timeout, using=using).run() return res, check_result + + def load_balance(self, src_node_id, dst_node_ids, sealed_segment_ids, 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.load_balance, src_node_id, dst_node_ids, + sealed_segment_ids, timeout, using]) + check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ, + timeout=timeout, using=using).run() + return res, check_result