diff --git a/tests/python_client/base/client_base.py b/tests/python_client/base/client_base.py index 60d9e0a4bf..0886fc4673 100644 --- a/tests/python_client/base/client_base.py +++ b/tests/python_client/base/client_base.py @@ -150,10 +150,13 @@ class TestcaseBase(Base): uri = cf.param_info.param_uri else: uri = "http://" + cf.param_info.param_host + ":" + str(cf.param_info.param_port) - self.connection_wrap.connect(alias=DefaultConfig.DEFAULT_USING,uri=uri,token=cf.param_info.param_token) - res, is_succ = self.connection_wrap.MilvusClient(uri=uri, - token=cf.param_info.param_token) - self.client = MilvusClient(uri=uri, token=cf.param_info.param_token) + if cf.param_info.param_token: + token = cf.param_info.param_token + else: + token = f"{ct.default_user}:{ct.default_password}" + self.connection_wrap.connect(alias=DefaultConfig.DEFAULT_USING, uri=uri, token=token) + res, is_succ = self.connection_wrap.MilvusClient(uri=uri, token=token) + self.client = MilvusClient(uri=uri, token=token) else: if cf.param_info.param_user and cf.param_info.param_password: res, is_succ = self.connection_wrap.connect(alias=DefaultConfig.DEFAULT_USING, @@ -168,27 +171,29 @@ class TestcaseBase(Base): port=cf.param_info.param_port) uri = "http://" + cf.param_info.param_host + ":" + str(cf.param_info.param_port) - self.client = MilvusClient(uri=uri, token=cf.param_info.param_token) + if cf.param_info.param_token: + token = cf.param_info.param_token + else: + token = f"{ct.default_user}:{ct.default_password}" + self.client = MilvusClient(uri=uri, token=token) server_version = utility.get_server_version() log.info(f"server version: {server_version}") return res - def get_tokens_by_analyzer(self, text, analyzer_params): if cf.param_info.param_uri: uri = cf.param_info.param_uri else: uri = "http://" + cf.param_info.param_host + ":" + str(cf.param_info.param_port) - - client = MilvusClient( - uri = uri, + if cf.param_info.param_token: token = cf.param_info.param_token - ) + else: + token = f"{ct.default_user}:{ct.default_password}" + self.client = MilvusClient(uri=uri, token=token) res = client.run_analyzer(text, analyzer_params, with_detail=True, with_hash=True) tokens = [r['token'] for r in res.tokens] return tokens - # def init_async_milvus_client(self): # uri = cf.param_info.param_uri or f"http://{cf.param_info.param_host}:{cf.param_info.param_port}" # kwargs = {