Add Onap Jakarta support

[it/dep.git] / smo-install / test / pythonsdk / src / orantests / smo / dmaap.py

diff --git a/smo-install/test/pythonsdk/src/orantests/smo/dmaap.py b/smo-install/test/pythonsdk/src/orantests/smo/dmaap.py
new file mode 100644 (file)
index 0000000..b952ec4
--- /dev/null
+++ b/smo-install/test/pythonsdk/src/orantests/smo/dmaap.py
@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+###
+# ============LICENSE_START=======================================================
+# ORAN SMO PACKAGE - PYTHONSDK TESTS
+# ================================================================================
+# Copyright (C) 2021-2022 AT&T Intellectual Property. All rights
+#                             reserved.
+# ================================================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============LICENSE_END============================================
+# ===================================================================
+#
+###
+
+"""Dmaap utils module."""
+import logging
+import logging.config
+from onapsdk.configuration import settings
+from waiting import wait
+from oransdk.dmaap.dmaap import OranDmaap
+
+logging.config.dictConfig(settings.LOG_CONFIG)
+logger = logging.getLogger("DMaap utils")
+dmaap = OranDmaap()
+
+class DmaapUtils():
+    """Can be used to have dmaap utils methods."""
+
+    @classmethod
+    def clean_dmaap(cls, dmaap_group, dmaap_user):
+        """Clean DMAAP useful topics."""
+        dmaap.create_topic(settings.DMAAP_TOPIC_FAULT_JSON)
+        dmaap.create_topic(settings.DMAAP_TOPIC_PNFREG_JSON)
+        # Purge the FAULT TOPIC
+        wait(lambda: (dmaap.get_message_from_topic(settings.DMAAP_TOPIC_FAULT, 5000, dmaap_group, dmaap_user).json() == []), sleep_seconds=10, timeout_seconds=60, waiting_for="DMaap topic SEC_FAULT_OUTPUT to be empty")
+        wait(lambda: (dmaap.get_message_from_topic(settings.DMAAP_TOPIC_PNFREG, 5000, dmaap_group, dmaap_user).json() == []), sleep_seconds=10, timeout_seconds=60, waiting_for="DMaap topic unauthenticated.VES_PNFREG_OUTPUT to be empty")