mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 06:55:27 +08:00
1dabbfc835
Add a part of the test case for the Collection interface See also: #5224 #5231 Signed-off-by: ThreadDao yufen.zong@zilliz.com
120 lines
4.4 KiB
Python
120 lines
4.4 KiB
Python
import pytest
|
|
from base.client_request import ApiReq
|
|
from utils.util_log import test_log as log
|
|
from common.common_type import *
|
|
from common.common_func import *
|
|
|
|
default_schema = gen_default_collection_schema()
|
|
|
|
|
|
class TestCollectionParams(ApiReq):
|
|
""" Test case of collection interface """
|
|
|
|
def teardown_method(self):
|
|
if self.collection.collection is not None:
|
|
self.collection.drop()
|
|
|
|
@pytest.fixture(
|
|
scope="function",
|
|
params=get_invalid_strs
|
|
)
|
|
def get_invalid_string(self, request):
|
|
yield request.param
|
|
|
|
# #5224
|
|
@pytest.mark.tags(CaseLabel.L3)
|
|
def test_collection(self):
|
|
"""
|
|
target: test collection with default schema
|
|
method: create collection with default schema
|
|
expected: assert collection property
|
|
"""
|
|
self._connect()
|
|
c_name = get_unique_str
|
|
collection, _ = self.collection.collection_init(c_name, data=None, schema=default_schema)
|
|
assert collection.name == c_name
|
|
assert collection.description == default_schema.description
|
|
assert collection.schema == default_schema
|
|
assert collection.is_empty
|
|
assert collection.num_entities == 0
|
|
assert collection.primary_field is None
|
|
assert c_name in self.utility.list_collections()
|
|
|
|
def test_collection_empty_name(self):
|
|
"""
|
|
target: test collection with empty name
|
|
method: create collection with a empty name
|
|
expected: raise exception
|
|
"""
|
|
self._connect()
|
|
c_name = ""
|
|
ex, check = self.collection.collection_init(c_name, schema=default_schema)
|
|
assert "value is illegal" in str(ex)
|
|
|
|
def test_collection_invalid_name(self, get_invalid_string):
|
|
"""
|
|
target: test collection with invalid name
|
|
method: create collection with invalid name
|
|
expected: raise exception
|
|
"""
|
|
self._connect()
|
|
c_name = get_invalid_string
|
|
ex, check = self.collection.collection_init(c_name, schema=default_schema)
|
|
assert "invalid" or "illegal" in str(ex)
|
|
|
|
# #5231 TODO
|
|
def test_collection_dup_name(self):
|
|
"""
|
|
target: test collection with dup name
|
|
method: create collection with dup name and none schema and data
|
|
expected: collection properties consistent
|
|
"""
|
|
self._connect()
|
|
c_name = get_unique_str
|
|
collection, _ = self.collection.collection_init(c_name, data=None, schema=default_schema)
|
|
assert collection.name == c_name
|
|
dup_collection, _ = self.collection.collection_init(c_name)
|
|
assert c_name, c_name in self.utility.list_collections()
|
|
assert collection.name == dup_collection.name
|
|
# log.debug(collection.schema)
|
|
# log.debug(dup_collection.schema)
|
|
# assert collection.schema == dup_collection.schema
|
|
|
|
def test_collection_dup_name_new_schema(self):
|
|
"""
|
|
target: test collection with dup name and new schema
|
|
method: 1.create collection with default schema 2. collection with dup name and new schema
|
|
expected: raise exception
|
|
"""
|
|
self._connect()
|
|
c_name = get_unique_str
|
|
collection, _ = self.collection.collection_init(c_name, data=None, schema=default_schema)
|
|
assert collection.name == c_name
|
|
fields = [gen_int64_field()]
|
|
schema = gen_collection_schema(fields=fields)
|
|
ex, _ = self.collection.collection_init(c_name, schema=schema)
|
|
assert "The collection already exist, but the schema isnot the same as the passed in" in str(ex)
|
|
|
|
|
|
class TestCollectionOperation(ApiReq):
|
|
"""
|
|
******************************************************************
|
|
The following cases are used to test collection interface operations
|
|
******************************************************************
|
|
"""
|
|
|
|
def test_collection_without_connection(self):
|
|
"""
|
|
target: test collection without connection
|
|
method: 1.create collection after connection removed
|
|
expected: raise exception
|
|
"""
|
|
self._connect()
|
|
self.connection.remove_connection(default_alias)
|
|
res_list = self.connection.list_connections()
|
|
assert len(res_list) == 0
|
|
c_name = get_unique_str
|
|
ex, check = self.collection.collection_init(c_name, schema=default_schema)
|
|
assert "no connection" in str(ex)
|
|
assert self.collection is None
|