From 86496efa7eda776d2a45be812a0adeb2e77571b7 Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Mon, 27 Sep 2021 19:52:02 +0800 Subject: [PATCH] Add alias test case for utility (#8711) Signed-off-by: zhuwenxing --- tests/python_client/testcases/test_alias.py | 78 +++++++++++++++++++++ 1 file changed, 78 insertions(+) diff --git a/tests/python_client/testcases/test_alias.py b/tests/python_client/testcases/test_alias.py index 43e4ccbd35..36c86635c7 100644 --- a/tests/python_client/testcases/test_alias.py +++ b/tests/python_client/testcases/test_alias.py @@ -327,6 +327,84 @@ class TestAliasOperation(TestcaseBase): """ pass + @pytest.mark.tags(CaseLabel.L1) + def test_alias_called_by_utility_has_collection(self): + """ + target: test utility has collection by alias + method: + 1.create collection with alias + 2.call has_collection function with alias as param + expected: result is True + """ + + 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}) + res, _ = self.utility_wrap.has_collection(alias_name) + + assert res == True + + @pytest.mark.tags(CaseLabel.L1) + def test_alias_called_by_utility_drop_collection(self): + """ + target: test utility drop collection by alias + method: + 1.create collection with alias + 2.call drop_collection function with alias as param + expected: collection is dropped + """ + + 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}) + assert self.utility_wrap.has_collection(c_name)[0] + self.utility_wrap.drop_collection(alias_name) + assert not self.utility_wrap.has_collection(c_name)[0] + + @pytest.mark.tags(CaseLabel.L1) + def test_alias_called_by_utility_has_partition(self): + """ + target: test utility has partition by alias + method: + 1.create collection with partition and alias + 2.call has_partition function with alias as param + expected: result is True + """ + + 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}) + partition_name = cf.gen_unique_str("partition") + self.init_partition_wrap(collection_w, partition_name) + + 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}) + res, _ = self.utility_wrap.has_partition(alias_name, partition_name) + + assert res == True + + class TestAliasOperationInvalid(TestcaseBase):