add load_collection case with partition (#4507)

* add load_collection with partition
Signed-off-by: Nzw <zw@milvus.io>

* add filebeat in CI
Signed-off-by: Nzw <zw@milvus.io>

* remove CI on ubuntu
Signed-off-by: Nzw <zw@milvus.io>
Co-authored-by: Nzw <zw@milvus.io>

ci/jenkins/Jenkinsfile

@@ -40,7 +40,7 @@ pipeline {
                 axes {
                     axis {
                         name 'OS_NAME'
-                        values 'ubuntu18.04', 'centos7'
+                        values 'centos7'
                     }
 
                     axis {

ci/jenkins/step/shardsDevNightlyTest.groovy

@@ -10,9 +10,9 @@ timeout(time: 180, unit: 'MINUTES') {
             try {
                 dir ('charts/milvus') {
                     if ("${BINARY_VERSION}" == "CPU") {
-                        sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP --set image.resources.requests.memory=8Gi --set image.resources.requests.cpu=2.0 --set image.resources.limits.memory=12Gi --set image.resources.limits.cpu=4.0 -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
+                        sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP --set image.resources.requests.memory=8Gi --set image.resources.requests.cpu=2.0 --set image.resources.limits.memory=12Gi --set image.resources.limits.cpu=4.0 -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml -f ci/filebeat/values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
                     } else {
-                        sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set gpu.enabled=true --set readonly.gpu.enabled=true --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
+                        sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set gpu.enabled=true --set readonly.gpu.enabled=true --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml -f ci/filebeat/values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
                     }
                 }
             } catch (exc) {

tests/milvus_python_test/collection/test_collection.py

@@ -864,6 +864,48 @@ class TestCollection:
         status = connect.load_collection(collection_name)
         assert not status.OK()
 
+    @pytest.mark.level(1)
+    def test_load_collection_partition(self, connect, collection):
+        partition_name = gen_unique_str()
+        status, ids = connect.insert(collection, vectors)
+        assert status.OK()
+        status = connect.create_partition(collection, partition_name)
+        status = connect.load_collection(collection, partition_tags=[partition_name])
+        assert status.OK()
+
+    @pytest.mark.level(1)
+    def test_load_collection_partitions(self, connect, collection):
+        partition_names = []
+        for i in range(2):
+            name = gen_unique_str()
+            partition_names.append(name)
+            status = connect.create_partition(collection, name)
+            assert status.OK()
+        status, ids = connect.insert(collection, vectors)
+        status = connect.load_collection(collection, partition_tags=partition_names)
+        assert status.OK()
+
+    @pytest.mark.level(1)
+    def test_load_collection_partition_not_existed(self, connect, collection):
+        partition_name = gen_unique_str()
+        status, ids = connect.insert(collection, vectors)
+        assert status.OK()
+        status = connect.load_collection(collection, partition_tags=[partition_name])
+        assert not status.OK()
+
+    @pytest.mark.level(1)
+    def test_load_collection_partition_invalid_string(self, connect, collection):
+        partition_name = "invalid string"
+        status, ids = connect.insert(collection, vectors)
+        assert status.OK()
+        status = connect.load_collection(collection, partition_tags=[partition_name])
+        assert not status.OK()
+
+    @pytest.mark.level(1)
+    def test_load_collection_partition_None(self, connect, collection):
+        status = connect.load_collection(collection, partition_tags=None)
+        assert status.OK()
+
     @pytest.mark.level(2)
     def test_load_collection_not_existed_ip(self, connect, ip_collection, get_simple_index):
         index_param = get_simple_index["index_param"]