milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[test]Add database for chaos test (#26636) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/26668/head
zhuwenxing 2023-08-29 14:28:40 +08:00 committed by GitHub
parent 69495bd23d
commit cb34edde88
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/python_client/chaos/testcases/test_data_persistence.py
View File

@ -18,8 +18,17 @@ class TestDataPersistence(TestcaseBase):
log.info("skip drop collection")
@pytest.mark.tags(CaseLabel.L3)
def test_milvus_default(self):
# create
@pytest.mark.parametrize("db_name", ["default", "prod"])
def test_milvus_default(self, db_name):
self._connect()
# create database if not exist
dbs, _ = self.database_wrap.list_database()
log.info(f"all database: {dbs}")
if db_name not in dbs:
log.info(f"create database {db_name}")
self.database_wrap.create_database(db_name)
self.database_wrap.using_database(db_name)
# create collection
name = "Hello_Milvus"
t0 = time.time()
collection_w = self.init_collection_wrap(name=name, active_trace=True)
@ -101,4 +110,4 @@ class TestDataPersistence(TestcaseBase):
t0 = time.time()
res, _ = collection_w.query(term_expr)
tt = time.time() - t0
log.info(f"assert query result {len(res)}: {tt}")
log.info(f"assert query result {len(res)}: {tt}")