@@ -138,6 +138,7 @@ class TestcaseBase(Base):
138
138
Additional methods;
139
139
Public methods that can be used for test cases.
140
140
"""
141
+ client = None
141
142
142
143
def _connect (self , enable_milvus_client_api = False ):
143
144
""" Add a connection and create the connect """
@@ -152,6 +153,7 @@ def _connect(self, enable_milvus_client_api=False):
152
153
self .connection_wrap .connect (alias = DefaultConfig .DEFAULT_USING ,uri = uri ,token = cf .param_info .param_token )
153
154
res , is_succ = self .connection_wrap .MilvusClient (uri = uri ,
154
155
token = cf .param_info .param_token )
156
+ self .client = MilvusClient (uri = uri , token = cf .param_info .param_token )
155
157
else :
156
158
if cf .param_info .param_user and cf .param_info .param_password :
157
159
res , is_succ = self .connection_wrap .connect (alias = DefaultConfig .DEFAULT_USING ,
@@ -165,6 +167,8 @@ def _connect(self, enable_milvus_client_api=False):
165
167
host = cf .param_info .param_host ,
166
168
port = cf .param_info .param_port )
167
169
170
+ uri = "http://" + cf .param_info .param_host + ":" + str (cf .param_info .param_port )
171
+ self .client = MilvusClient (uri = uri , token = cf .param_info .param_token )
168
172
server_version = utility .get_server_version ()
169
173
log .info (f"server version: { server_version } " )
170
174
return res
@@ -183,7 +187,7 @@ def get_tokens_by_analyzer(self, text, analyzer_params):
183
187
res = client .run_analyzer (text , analyzer_params , with_detail = True , with_hash = True )
184
188
tokens = [r ['token' ] for r in res .tokens ]
185
189
return tokens
186
-
190
+
187
191
188
192
# def init_async_milvus_client(self):
189
193
# uri = cf.param_info.param_uri or f"http://{cf.param_info.param_host}:{cf.param_info.param_port}"
0 commit comments