admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: update nightly cases (#43410)

Browse Source 
Signed-off-by: nico <cheng.yuan@zilliz.com>
This commit is contained in:
nico 2025-07-21 15:12:53 +08:00 committed by GitHub
parent c7b53ed43b
commit d451476a62
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 96 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
tests/python_client/milvus_client_v2/test_milvus_client_alias_v2.py
View File

@ -80,7 +80,8 @@ class TestMilvusClientV2AliasOperation(TestMilvusClientV2Base):
collection_name1 = cf.gen_unique_str("collection1")
index_params = self.prepare_index_params(client)[0]
index_params.add_index(field_name=default_vector_field_name, metric_type="L2")
self.create_collection(client, collection_name1, default_dim, consistency_level="Bounded", index_params=index_params)
self.create_collection(client, collection_name1, default_dim, consistency_level="Bounded",
index_params=index_params)
# 2. create alias and insert data
alias_name = cf.gen_unique_str(prefix)
@ -103,11 +104,11 @@ class TestMilvusClientV2AliasOperation(TestMilvusClientV2Base):
# 5. verify search using alias works on collection1
search_vectors = cf.gen_vectors(1, default_dim)
self.search(client, alias_name, search_vectors, limit=default_limit,
check_task=CheckTasks.check_search_results,
check_items={"enable_milvus_client_api": True,
"nq": len(search_vectors),
"pk_name": default_primary_key_field_name,
"limit": default_limit})
check_task=CheckTasks.check_search_results,
check_items={"enable_milvus_client_api": True,
"nq": len(search_vectors),
"pk_name": default_primary_key_field_name,
"limit": default_limit})
# 6. create collection2 with index and load
collection_name2 = cf.gen_unique_str("collection2")
@ -133,11 +134,11 @@ class TestMilvusClientV2AliasOperation(TestMilvusClientV2Base):
# 10. verify search using alias works on collection2
search_vectors = cf.gen_vectors(1, default_dim)
self.search(client, alias_name, search_vectors, limit=default_limit,
check_task=CheckTasks.check_search_results,
check_items={"enable_milvus_client_api": True,
"nq": len(search_vectors),
"pk_name": default_primary_key_field_name,
"limit": default_limit})
check_task=CheckTasks.check_search_results,
check_items={"enable_milvus_client_api": True,
"nq": len(search_vectors),
"pk_name": default_primary_key_field_name,
"limit": default_limit})
# 11. verify operations on collection1 still work
res1 = self.query(client, collection_name1, filter="", output_fields=["count(*)"])
@ -198,8 +199,8 @@ class TestMilvusClientV2AliasOperation(TestMilvusClientV2Base):
error = {ct.err_code: 0,
ct.err_msg: f"can't find collection[database=default][collection={alias_name}]"}
self.describe_collection(client, alias_name,
check_task=CheckTasks.err_res,
check_items=error)
check_task=CheckTasks.err_res,
check_items=error)
# 8. verify collection still exists and unchanged
assert self.has_collection(client, collection_name)[0]
@ -239,14 +240,82 @@ class TestMilvusClientV2AliasOperation(TestMilvusClientV2Base):
error = {ct.err_code: 1,
ct.err_msg: f"cannot drop the collection via alias = {alias_name}"}
self.drop_collection(client, alias_name,
check_task=CheckTasks.err_res,
check_items=error)
check_task=CheckTasks.err_res,
check_items=error)
# cleanup
self.drop_alias(client, alias_name)
self.drop_collection(client, collection_name)
assert not self.has_collection(client, collection_name)[0]
@pytest.mark.tags(CaseLabel.L2)
def test_milvus_client_v2_rename_back_old_alias(self):
"""
target: test collection operations using alias
method:
1. create collection with alias
2. drop the alias
3. rename collection to the dropped alias name
expected:
1. rename collection successfully
"""
client = self._client()
collection_name = cf.gen_unique_str("collection")
# 1. create collection
self.create_collection(client, collection_name, default_dim)
# 2. create alias
alias_name = cf.gen_unique_str(prefix)
self.create_alias(client, collection_name, alias_name)
# 3. drop the alias
self.drop_alias(client, alias_name)
assert self.list_aliases(client, collection_name)[0]["aliases"] == []
# 4. rename collection to the dropped alias name
self.rename_collection(client, collection_name, alias_name)
# cleanup
self.drop_collection(client, alias_name)
assert not self.has_collection(client, alias_name)[0]
@pytest.mark.tags(CaseLabel.L2)
def test_milvus_client_v2_rename_back_old_collection(self):
"""
target: test collection operations using alias
method:
1. create collection with alias
2. rename collection
3. rename back to old collection name
expected:
1. rename collection successfully
"""
client = self._client()
collection_name = cf.gen_unique_str("collection")
# 1. create collection
self.create_collection(client, collection_name, default_dim)
# 2. create alias
alias_name = cf.gen_unique_str(prefix)
self.create_alias(client, collection_name, alias_name)
# 3. rename collection
new_collection_name = cf.gen_unique_str("collection")
self.rename_collection(client, collection_name, new_collection_name)
# 4. rename back to old collection name
self.rename_collection(client, new_collection_name, collection_name)
assert self.has_collection(client, collection_name)[0]
assert not self.has_collection(client, new_collection_name)[0]
assert alias_name in self.list_aliases(client, collection_name)[0]["aliases"]
# cleanup
self.drop_alias(client, alias_name)
self.drop_collection(client, collection_name)
assert not self.has_collection(client, collection_name)[0]
class TestMilvusClientV2AliasOperationInvalid(TestMilvusClientV2Base):
""" Test cases of alias interface invalid operations"""
@ -276,8 +345,8 @@ class TestMilvusClientV2AliasOperationInvalid(TestMilvusClientV2Base):
error = {ct.err_code: 1,
ct.err_msg: f"{alias_name} is alias to another collection: {collection_name1}"}
self.create_alias(client, collection_name2, alias_name,
check_task=CheckTasks.err_res,
check_items=error)
check_task=CheckTasks.err_res,
check_items=error)
# cleanup
self.drop_alias(client, alias_name)
@ -306,8 +375,8 @@ class TestMilvusClientV2AliasOperationInvalid(TestMilvusClientV2Base):
error = {ct.err_code: 1600,
ct.err_msg: f"alias not found[database=default][alias={non_exist_alias}]"}
self.alter_alias(client, collection_name, non_exist_alias,
check_task=CheckTasks.err_res,
check_items=error)
check_task=CheckTasks.err_res,
check_items=error)
# 4. cleanup
self.drop_alias(client, alias_name)
@ -374,8 +443,8 @@ class TestMilvusClientV2AliasOperationInvalid(TestMilvusClientV2Base):
ct.err_msg: f"collection name [{alias_name}] conflicts with an existing alias,"
" please choose a unique name"}
self.create_collection(client, alias_name, default_dim, consistency_level="Bounded",
check_task=CheckTasks.err_res,
check_items=error)
check_task=CheckTasks.err_res,
check_items=error)
# cleanup
self.drop_alias(client, alias_name)
@ -447,8 +516,8 @@ class TestMilvusClientV2AliasOperationInvalid(TestMilvusClientV2Base):
error = {ct.err_code: 999,
ct.err_msg: f"cannot rename collection to an existing alias: {alias_name}"}
self.rename_collection(client, collection_name2, alias_name,
check_task=CheckTasks.err_res,
check_items=error)
check_task=CheckTasks.err_res,
check_items=error)
# cleanup
self.drop_alias(client, alias_name)

1
tests/python_client/testcases/async_milvus_client/test_collection_async.py
View File

@ -26,6 +26,7 @@ default_vector_field_name = "vector"
default_float_field_name = ct.default_float_field_name
default_string_field_name = ct.default_string_field_name
class TestAsyncMilvusClientCollectionInvalid(TestMilvusClientV2Base):
""" Test case of collection interface """

3
tests/python_client/testcases/test_utility.py
View File

@ -1145,6 +1145,7 @@ class TestUtilityBase(TestcaseBase):
assert collection_w_2.aliases[0] == alias_1
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.skip("move to test_milvus_client_v2_rename_back_old_collection")
def test_rename_back_old_collection(self):
"""
target: test rename collection function to single collection
@ -1175,6 +1176,7 @@ class TestUtilityBase(TestcaseBase):
assert collection_alias == collection_w.aliases
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.skip("move to test_milvus_client_v2_rename_back_old_alias")
def test_rename_back_old_alias(self):
"""
target: test rename collection function to single collection
@ -1190,6 +1192,7 @@ class TestUtilityBase(TestcaseBase):
# 3. create an alias
self.utility_wrap.create_alias(old_collection_name, alias)
collection_alias = collection_w.aliases
log.info(collection_alias)
# 4. drop the alias
self.utility_wrap.drop_alias(collection_alias[0])
# 5. rename collection to the dropped alias name