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

Update alias testcase (#8584) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/8590/head
zhuwenxing 2021-09-26 16:51:57 +08:00 committed by GitHub
parent 075d6880a4
commit c54f8064bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 81 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

108
tests/python_client/testcases/test_alias.py
View File

@ -1,3 +1,4 @@
from typing import Collection
import pytest import pytest
from base.client_base import TestcaseBase from base.client_base import TestcaseBase
@ -11,8 +12,15 @@ prefix = "alias"
exp_name = "name" exp_name = "name"
exp_schema = "schema" exp_schema = "schema"
default_schema = cf.gen_default_collection_schema() default_schema = cf.gen_default_collection_schema()
default_binary_schema = cf.gen_default_binary_collection_schema()
default_nb = ct.default_nb
default_nb_medium = ct.default_nb_medium
default_nq = ct.default_nq
default_dim = ct.default_dim
default_limit = ct.default_limit
default_search_exp = "int64 >= 0"
default_search_field = ct.default_float_vec_field_name
default_search_params = ct.default_search_params
class TestAliasParams(TestcaseBase): class TestAliasParams(TestcaseBase):
""" Test cases of alias interface parameters""" """ Test cases of alias interface parameters"""
pass pass
@ -29,7 +37,7 @@ class TestAliasOperation(TestcaseBase):
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_create_operation_default(self): def test_alias_create_operation_default(self):
""" """
target: test collection create alias target: test collection creating alias
method: method:
1.create a collection and create 10 partitions for it 1.create a collection and create 10 partitions for it
2.collection create a alias, then init a collection with this alias name but not create partitions 2.collection create a alias, then init a collection with this alias name but not create partitions
@ -48,15 +56,15 @@ class TestAliasOperation(TestcaseBase):
alias_name = cf.gen_unique_str(prefix) alias_name = cf.gen_unique_str(prefix)
collection_w.create_alias(alias_name) collection_w.create_alias(alias_name)
collection_alias = self.init_collection_wrap(name=alias_name, collection_alias, _ = self.collection_wrap.init_collection(name=alias_name,
check_task=CheckTasks.check_collection_property, check_task=CheckTasks.check_collection_property,
check_items={exp_name: alias_name, exp_schema: default_schema}) check_items={exp_name: alias_name, exp_schema: default_schema})
# assert collection is equal to alias # assert collection is equal to alias
assert [p.name for p in collection_w.partitions] == [p.name for p in collection_alias.partitions] assert [p.name for p in collection_w.partitions] == [p.name for p in collection_alias.partitions]
def test_alias_alter_operation_default(self): def test_alias_alter_operation_default(self):
""" """
target: test collection alter alias target: test collection altering alias
method: method:
1. create collection_1 with 10 partitions and its alias alias_a 1. create collection_1 with 10 partitions and its alias alias_a
2. create collection_2 with 5 partitions and its alias alias_b 2. create collection_2 with 5 partitions and its alias alias_b
@ -81,9 +89,9 @@ class TestAliasOperation(TestcaseBase):
alias_a_name = cf.gen_unique_str(prefix) alias_a_name = cf.gen_unique_str(prefix)
collection_1.create_alias(alias_a_name) collection_1.create_alias(alias_a_name)
collection_alias_a = self.init_collection_wrap(name=alias_a_name, collection_alias_a, _ = self.collection_wrap.init_collection(name=alias_a_name,
check_task=CheckTasks.check_collection_property, check_task=CheckTasks.check_collection_property,
check_items={exp_name: alias_a_name, exp_schema: default_schema}) check_items={exp_name: alias_a_name, exp_schema: default_schema})
assert [p.name for p in collection_1.partitions] == [p.name for p in collection_alias_a.partitions] assert [p.name for p in collection_1.partitions] == [p.name for p in collection_alias_a.partitions]
@ -102,9 +110,9 @@ class TestAliasOperation(TestcaseBase):
alias_b_name = cf.gen_unique_str(prefix) alias_b_name = cf.gen_unique_str(prefix)
collection_2.create_alias(alias_b_name) collection_2.create_alias(alias_b_name)
collection_alias_b = self.init_collection_wrap(name=alias_b_name, collection_alias_b, _ = self.collection_wrap.init_collection(name=alias_b_name,
check_task=CheckTasks.check_collection_property, check_task=CheckTasks.check_collection_property,
check_items={exp_name: alias_b_name, exp_schema: default_schema}) check_items={exp_name: alias_b_name, exp_schema: default_schema})
assert [p.name for p in collection_2.partitions] == [p.name for p in collection_alias_b.partitions] assert [p.name for p in collection_2.partitions] == [p.name for p in collection_alias_b.partitions]
@ -122,14 +130,14 @@ class TestAliasOperation(TestcaseBase):
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_drop_operation_default(self): def test_alias_drop_operation_default(self):
""" """
target: test collection drop alias target: test collection dropping alias
method: method:
1.create a collection with 10 partitons 1.create a collection with 10 partitons
2.collection create a alias 2.collection create a alias
3.collection drop the alias 3.collection drop the alias
expected: expected:
in step 2, collection is equal to alias in step 2, collection is equal to alias
in step 3, collection is not equal to alias any more in step 3, collection with alias name is exist
""" """
self._connect() self._connect()
c_name = cf.gen_unique_str("collection") c_name = cf.gen_unique_str("collection")
@ -144,19 +152,43 @@ class TestAliasOperation(TestcaseBase):
alias_name = cf.gen_unique_str(prefix) alias_name = cf.gen_unique_str(prefix)
collection_w.create_alias(alias_name) collection_w.create_alias(alias_name)
collection_alias = self.init_collection_wrap(name=alias_name, collection_alias, _ = self.collection_wrap.init_collection(name=alias_name,
check_task=CheckTasks.check_collection_property, check_task=CheckTasks.check_collection_property,
check_items={exp_name: alias_name, exp_schema: default_schema}) check_items={exp_name: alias_name, exp_schema: default_schema})
assert [p.name for p in collection_w.partitions] == [p.name for p in collection_alias.partitions] assert [p.name for p in collection_w.partitions] == [p.name for p in collection_alias.partitions]
collection_w.drop_alias(alias_name) collection_w.drop_alias(alias_name)
error = {ct.err_code: 0, ct.err_msg: f"Collection '{alias_name}' not exist, or you can pass in schema to create one"}
collection_alias, _ = self.collection_wrap.init_collection(name=alias_name,
check_task=CheckTasks.err_res,
check_items=error)
collection_alias = self.init_collection_wrap(name=alias_name, @pytest.mark.tags(CaseLabel.L1)
check_task=CheckTasks.check_collection_property, def test_alias_insert_data_default(self):
check_items={exp_name: alias_name, exp_schema: default_schema}) """
target: test alias inserting data
method:
1.create a collection with alias
2.collection insert data by alias
expected: inserting data by alias can work and the result is same as inserting data directly
assert [p.name for p in collection_w.partitions] != [p.name for p in collection_alias.partitions] """
self._connect()
c_name = cf.gen_unique_str("collection")
collection_w = self.init_collection_wrap(name=c_name, schema=default_schema,
check_task=CheckTasks.check_collection_property,
check_items={exp_name: c_name, exp_schema: default_schema})
alias_name = cf.gen_unique_str(prefix)
collection_w.create_alias(alias_name)
collection_alias, _ = self.collection_wrap.init_collection(name=alias_name,
check_task=CheckTasks.check_collection_property,
check_items={exp_name: alias_name, exp_schema: default_schema})
df = cf.gen_default_dataframe_data(ct.default_nb)
collection_alias.insert(data=df)
assert collection_w.num_entities == ct.default_nb
assert collection_alias.num_entities == ct.default_nb
class TestAliasOperationInvalid(TestcaseBase): class TestAliasOperationInvalid(TestcaseBase):
@ -165,7 +197,7 @@ class TestAliasOperationInvalid(TestcaseBase):
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_create_dupcation_alias(self): def test_alias_create_dupcation_alias(self):
""" """
target: test two collections create alias with same name target: test two collections creating alias with same name
method: method:
1.create a collection_1 with alias name alias_a 1.create a collection_1 with alias name alias_a
2.create a collection_2 also with alias name alias_a 2.create a collection_2 also with alias name alias_a
@ -193,7 +225,7 @@ class TestAliasOperationInvalid(TestcaseBase):
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_alter_not_exist_alias(self): def test_alias_alter_not_exist_alias(self):
""" """
target: test collection alter to alias which is not exist target: test collection altering to alias which is not exist
method: method:
1.create a collection with alias 1.create a collection with alias
2.collection alters to a alias name which is not exist 2.collection alters to a alias name which is not exist
@ -218,7 +250,7 @@ class TestAliasOperationInvalid(TestcaseBase):
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_drop_not_exist_alias(self): def test_alias_drop_not_exist_alias(self):
""" """
target: test collection drop alias which is not exist target: test collection dropping alias which is not exist
method: method:
1.create a collection with alias 1.create a collection with alias
2.collection drop alias which is not exist 2.collection drop alias which is not exist
@ -242,7 +274,7 @@ class TestAliasOperationInvalid(TestcaseBase):
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_drop_same_alias_twice(self): def test_alias_drop_same_alias_twice(self):
""" """
target: test collection drop same alias twice target: test collection dropping same alias twice
method: method:
1.create a collection with alias 1.create a collection with alias
2.collection drop alias 2.collection drop alias
@ -262,4 +294,26 @@ class TestAliasOperationInvalid(TestcaseBase):
error = {ct.err_code: 1, ct.err_msg: "Drop alias failed: alias does not exist"} error = {ct.err_code: 1, ct.err_msg: "Drop alias failed: alias does not exist"}
collection_w.drop_alias(alias_name, collection_w.drop_alias(alias_name,
check_task=CheckTasks.err_res, check_task=CheckTasks.err_res,
check_items=error) check_items=error)
def test_alias_cerate_dup_name_collection(self):
"""
target: test creating a collection with a same name as alias, but a different schema
method:
1.create a collection with alias
2.create a collection with same name as alias, but a different schema
expected: in step 2, create collection failed
"""
self._connect()
c_name = cf.gen_unique_str("collection")
collection_w = self.init_collection_wrap(name=c_name, schema=default_schema,
check_task=CheckTasks.check_collection_property,
check_items={exp_name: c_name, exp_schema: default_schema})
alias_name = cf.gen_unique_str(prefix)
collection_w.create_alias(alias_name)
error = {ct.err_code: 0, ct.err_msg: "The collection already exist, but the schema is not the same as the schema passed in"}
self.init_collection_wrap(alias_name, schema=default_binary_schema,
check_task=CheckTasks.err_res,
check_items=error)