Merge branch 'master' of https://github.com/milvus-io/milvus into 231…

…2-import-debug
milvus-io · Jan 29, 2024 · 75c1265 · 75c1265
2 parents 68da88e + f92edc6
commit 75c1265
Show file tree

Hide file tree

Showing 5 changed files with 1,238 additions and 118 deletions.
diff --git a/tests/python_client/base/high_level_api_wrapper.py b/tests/python_client/base/high_level_api_wrapper.py
@@ -510,4 +510,17 @@ def list_aliases(self, client, timeout=None, check_task=None, check_items=None,
         check_result = ResponseChecker(res, func_name, check_task,
                                        check_items, check,
                                        **kwargs).run()
+        return res, check_result
+
+    @trace()
+    def using_database(self, client, db_name, timeout=None, check_task=None, check_items=None, **kwargs):
+        timeout = TIMEOUT if timeout is None else timeout
+        kwargs.update({"timeout": timeout})
+
+        func_name = sys._getframe().f_code.co_name
+        res, check = api_request([client.using_database, db_name], **kwargs)
+        check_result = ResponseChecker(res, func_name, check_task,
+                                       check_items, check,
+                                       db_name=db_name,
+                                       **kwargs).run()
         return res, check_result