Test Bucket Notifications - first draft

Signed-off-by: Sagi Hirshfeld <[email protected]>

ocs_ci/ocs/amq.py

@@ -161,9 +161,10 @@ def setup_amq_cluster_operator(self, namespace=constants.AMQ_NAMESPACE):
                     log.warn(
                         "Some amq leftovers are present, please cleanup the cluster"
                     )
-                    pytest.skip(
-                        "AMQ leftovers are present needs to cleanup the cluster"
-                    )
+                    # TODO: Check why this skip was here in the first place
+                    # pytest.skip(
+                    #     "AMQ leftovers are present needs to cleanup the cluster"
+                    # )
         time.sleep(30)
         #  Check strimzi-cluster-operator pod created
         if self.is_amq_pod_running(pod_pattern="cluster-operator", expected_pods=1):

ocs_ci/ocs/constants.py

@@ -438,6 +438,7 @@
 DEFAULT_NOOBAA_BACKINGSTORE = "noobaa-default-backing-store"
 DEFAULT_NOOBAA_BUCKETCLASS = "noobaa-default-bucket-class"
 DEFAULT_MCG_BUCKET_LOGS_PVC = "noobaa-bucket-logging-pvc"
+DEFAULT_MCG_BUCKET_NOTIFS_PVC = "noobaa-bucket-notifications-pvc"
 CUSTOM_MCG_LABEL = "custom=mcg-label"
 NOOBAA_RESOURCE_NAME = "noobaa"
 NOOBAA_DB_PVC_NAME = "db-noobaa-db-pg-0"
@@ -619,7 +620,9 @@
 NOOBAA_OPERATOR_DEPLOYMENT = "noobaa-operator"
 NOOBAA_ENDPOINT_DEPLOYMENT = "noobaa-endpoint"
 NOOBAA_DB_STATEFULSET = "noobaa-db-pg"
+NOOBAA_DB_POD = "noobaa-db-pg-0"
 NOOBAA_CORE_STATEFULSET = "noobaa-core"
+NOOBAA_CORE_POD = "noobaa-core-0"
 
 # Noobaa db secret
 NOOBAA_DB_SECRET = "noobaa-db"

ocs_ci/ocs/resources/bucket_notifications_manager.py

@@ -0,0 +1,324 @@
+import json
+import logging
+import os
+import tempfile
+import time
+
+from ocs_ci.framework import config
+from ocs_ci.helpers.helpers import (
+    craft_s3_command,
+    create_unique_resource_name,
+    default_storage_class,
+)
+from ocs_ci.ocs import constants
+from ocs_ci.ocs.cluster import CephCluster
+from ocs_ci.ocs.ocp import OCP
+from ocs_ci.ocs.amq import AMQ
+from ocs_ci.ocs.exceptions import CommandFailed
+from ocs_ci.ocs.resources.pod import (
+    Pod,
+    get_noobaa_endpoint_pods,
+    get_pods_having_label,
+    wait_for_pods_to_be_running,
+)
+
+logger = logging.getLogger(__name__)
+
+NOTIFS_YAML_PATH_NB_CR = "/spec/bucketNotifications"
+
+
+class BucketNotificationsManager:
+    """
+    A class to manage the MCG bucket notifications feature
+    """
+
+    @property
+    def nb_config_resource(self):
+        """
+        Return the NooBaa configuration resource
+        Note that this might change in the future.
+
+        Returns:
+            ocs_ci.ocs.ocp.OCP: OCP instance of the NooBaa configuration resource
+        """
+        return OCP(
+            kind="noobaa",
+            namespace=config.ENV_DATA["cluster_namespace"],
+            resource_name="noobaa",
+        )
+
+    def __init__(self):
+        self.amq = AMQ()
+        self.kafka_topics = []
+        self.conn_secrets = []
+        self.cur_logs_pvc = constants.DEFAULT_MCG_BUCKET_NOTIFS_PVC
+
+    def setup_kafka(self):
+        """
+        TODO
+        """
+        # Get sc
+        sc = default_storage_class(interface_type=constants.CEPHBLOCKPOOL)
+
+        # Deploy amq cluster
+        self.amq.setup_amq_cluster(sc.name)
+
+    def cleanup_kafka(self):
+        for topic in self.kafka_topics:
+            topic.delete()
+        self.kafka_topics = []
+        self.amq.cleanup()
+
+    def enable_bucket_notifs_on_cr(self, notifs_pvc=None):
+        """
+        Set the bucket notifications feature on the NooBaa CR
+
+        Args:
+            notifs_pvc(str|optional): Name of a provided PVC for MCG to use for
+                                      intermediate logging of the events.
+            Note:
+                If not provided, a PVC will be automatically be created
+                by MCG when first enabling the feature.
+        """
+        logger.info("Enabling bucket notifications on the NooBaa CR")
+
+        # Build a patch command to enable guaranteed bucket logs
+        bucket_notifs_dict = {"connections": [], "enabled": True}
+
+        # Add the bucketLoggingPVC field if provided
+        if notifs_pvc:
+            bucket_notifs_dict["pvc"] = notifs_pvc
+
+        patch_params = [
+            {
+                "op": "add",
+                "path": NOTIFS_YAML_PATH_NB_CR,
+                "value": bucket_notifs_dict,
+            }
+        ]
+
+        # Try patching via add, and if it fails - replace instead
+        try:
+            self.nb_config_resource.patch(
+                params=json.dumps(patch_params),
+                format_type="json",
+            )
+        except CommandFailed as e:
+            if "already exists" in str(e).lower():
+                patch_params[0]["op"] = "replace"
+                self.nb_config_resource.patch(
+                    params=json.dumps(patch_params),
+                    format_type="json",
+                )
+            else:
+                logger.error(f"Failed to enable bucket notifications: {e}")
+                raise e
+
+        self.cur_logs_pvc = (
+            notifs_pvc if notifs_pvc else constants.DEFAULT_MCG_BUCKET_NOTIFS_PVC
+        )
+
+        wait_for_pods_to_be_running(
+            pod_names=[constants.NOOBAA_CORE_POD],
+            timeout=60,
+            sleep=10,
+        )
+
+        logger.info("Guaranteed bucket logs have been enabled")
+
+    def disable_bucket_logging_on_cr(self):
+        """
+        Unset the bucket notifications feature on the NooBaa CR
+        """
+        logger.info("Disabling bucket notifications on the NooBaa CR")
+
+        try:
+            patch_params = [
+                {
+                    "op": "replace",
+                    "path": NOTIFS_YAML_PATH_NB_CR,
+                    "value": None,
+                },
+            ]
+            self.nb_config_resource.patch(
+                params=json.dumps(patch_params),
+                format_type="json",
+            )
+
+        except CommandFailed as e:
+            if "not found" in str(e):
+                logger.info("The bucketNotifications field was not found")
+            else:
+                logger.error(f"Failed to disable bucket notifications: {e}")
+                raise e
+
+        wait_for_pods_to_be_running(
+            pod_names=[constants.NOOBAA_CORE_POD],
+            timeout=60,
+            sleep=10,
+        )
+
+        logger.info("Bucket notifications have been disabled")
+
+    def add_new_notif_conn(self, name=""):
+        """
+        1. Create a Kafka topic
+        2. Create a secret with the Kafka connection details
+        3. Add the connection to the NooBaa CR
+        """
+        topic_name = name or create_unique_resource_name(
+            resource_description="nb-notif", resource_type="kafka-topic"
+        )
+        topic_obj = self.amq.create_kafka_topic(topic_name)
+        self.kafka_topics.append(topic_obj)
+        secret, conn_file_name = self.create_kafka_connection_secret(topic_name)
+        self.add_notif_conn_to_noobaa_cr(secret)
+
+        return topic_name, conn_file_name
+
+    def create_kafka_connection_secret(self, topic):
+        """
+        TODO
+        """
+        namespace = config.ENV_DATA["cluster_namespace"]
+        conn_name = create_unique_resource_name(
+            resource_description="nb-notif", resource_type="kafka-conn"
+        )
+        secret_name = conn_name + "-secret"
+        file_name = ""
+
+        kafka_conn_config = {
+            "metadata.broker.list": "my-cluster-kafka-bootstrap.myproject.svc.cluster.local:9092",
+            "notification_protocol": "kafka",
+            "topic": topic,
+            "name": conn_name,
+        }
+
+        with tempfile.NamedTemporaryFile(
+            mode="w+", prefix="kafka_conn_", suffix=".json", delete=True
+        ) as conn_file:
+            file_name = os.path.basename(conn_file.name)
+            conn_file.write(json.dumps(kafka_conn_config))
+            conn_file.flush()  # Ensure that the data is written
+
+            OCP().exec_oc_cmd(
+                f"create secret generic {secret_name} --from-file={conn_file.name} -n {namespace}"
+            )
+
+        secret_ocp_obj = OCP(
+            kind="secret",
+            namespace=namespace,
+            resource_name=secret_name,
+        )
+        self.conn_secrets.append(secret_ocp_obj)
+
+        return secret_ocp_obj, file_name
+
+    def add_notif_conn_to_noobaa_cr(self, secret):
+        """
+        TODO
+        """
+        nb_ocp_obj = OCP(
+            kind="noobaa",
+            namespace=config.ENV_DATA["cluster_namespace"],
+            resource_name="noobaa",
+        )
+        conn_data = {
+            "name": secret.resource_name,
+            "namespace": secret.namespace,
+        }
+        patch_path = "/spec/bucketNotifications/connections"
+        add_op = [{"op": "add", "path": f"{patch_path}/-", "value": conn_data}]
+        nb_ocp_obj.patch(
+            resource_name=constants.NOOBAA_RESOURCE_NAME,
+            params=json.dumps(add_op),
+            format_type="json",
+        )
+
+        # Wait for noobaa to process the change
+        nb_pods = [pod.name for pod in get_noobaa_endpoint_pods()]
+        nb_pods += [constants.NOOBAA_CORE_POD]
+        wait_for_pods_to_be_running(
+            namespace=config.ENV_DATA["cluster_namespace"],
+            pod_names=nb_pods,
+            timeout=60,
+            sleep=10,
+        )
+        CephCluster().wait_for_noobaa_health_ok()
+
+    def put_bucket_notification(self, awscli_pod, mcg_obj, bucket, events, conn_file):
+        """
+        TODO
+        """
+        rand_id = create_unique_resource_name(
+            resource_description="notif", resource_type="id"
+        )
+
+        notif_config = {
+            "TopicConfiguration": {
+                "Id": rand_id,
+                "Events": events,
+                "Topic": conn_file,  # See https://issues.redhat.com/browse/DFBUGS-947
+            }
+        }
+        # Serialize JSON with escaped quotes
+        notif_config_json = json.dumps(notif_config).replace('"', '\\"')
+        awscli_pod.exec_cmd_on_pod(
+            command=craft_s3_command(
+                f"put-bucket-notification --bucket {bucket} --notification-configuration '{notif_config_json}'",
+                mcg_obj=mcg_obj,
+                api=True,
+            )
+        )
+        logger.info("Waiting for put-bucket-notification to propogate")
+        time.sleep(60)
+
+    def get_bucket_notification(self, awscli_pod, mcg_obj, bucket):
+        """
+        TODO
+        """
+        return awscli_pod.exec_cmd_on_pod(
+            command=craft_s3_command(
+                f"get-bucket-notification --bucket {bucket}",
+                mcg_obj=mcg_obj,
+                api=True,
+            )
+        )
+
+    def get_events(self, topic):
+        """
+        TODO
+        """
+        # TODO: use constants and add variability where possible
+        kafka_pod = Pod(
+            **get_pods_having_label(
+                namespace=constants.AMQ_NAMESPACE, label="strimzi.io/pool-name=kafka"
+            )[0]
+        )
+        cmd = (
+            f"bin/kafka-console-consumer.sh --bootstrap-server {constants.KAFKA_ENDPOINT} "
+            f"--topic {topic} --from-beginning --timeout-ms 5000"
+        )
+        raw_resp = kafka_pod.exec_cmd_on_pod(command=cmd, out_yaml_format=False)
+
+        # Parse the raw response into a list of event dictionaries
+        events = []
+        for line in raw_resp.split("\n"):
+            if line:
+                full_event_dict = json.loads(line)
+
+                # Every event is nested in a single-element list
+                event_dict = full_event_dict["Records"][0]
+
+                events.append(event_dict)
+
+        return events
+
+    def cleanup(self):
+        """
+        TODO
+        """
+        self.disable_bucket_logging_on_cr()
+        for secret in self.conn_secrets:
+            secret.delete(resource_name=secret.resource_name)
+        self.cleanup_kafka()