milvus/tests/python_client/deploy/testcases/test_get_all_collections.py
zhuwenxing a1d3030867
[test] fix get collection (#23099)
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
2023-03-29 15:22:01 +08:00

28 lines
1.0 KiB
Python

import json
import pytest
from base.client_base import TestcaseBase
from deploy.common import get_deploy_test_collections
from common.common_type import CaseLabel
from utils.util_log import test_log as log
class TestGetCollections(TestcaseBase):
""" Test case of getting all collections """
@pytest.mark.tags(CaseLabel.L3)
def test_get_collections_by_prefix(self,):
self._connect()
all_collections = self.utility_wrap.list_collections()[0]
all_collections = [c_name for c_name in all_collections if "deploy_test" in c_name]
log.info(f"find {len(all_collections)} collections:")
log.info(all_collections)
data = {
"all": all_collections,
}
with open("/tmp/ci_logs/deploy_test_all_collections.json", "w") as f:
f.write(json.dumps(data))
log.info(f"write {len(all_collections)} collections to /tmp/ci_logs/deploy_test_all_collections.json")
collections_in_json = get_deploy_test_collections()
assert len(all_collections) == len(collections_in_json)