Add new tests to validate O1

[it/dep.git] / smo-install / test / pythonsdk / src / oransdk / a1sim / a1sim.py

diff --git a/smo-install/test/pythonsdk/src/oransdk/a1sim/a1sim.py b/smo-install/test/pythonsdk/src/oransdk/a1sim/a1sim.py
index 19bbe6b..67fd6af 100644 (file)
--- a/smo-install/test/pythonsdk/src/oransdk/a1sim/a1sim.py
+++ b/smo-install/test/pythonsdk/src/oransdk/a1sim/a1sim.py
@@ -1,23 +1,104 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# SPDX-License-Identifier: Apache-2.0
+###
+# ============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============================================
+# ===================================================================
+#
+###
 """Oran A1 Simulator module."""
 
 from onapsdk.onap_service import OnapService
 
 class A1sim(OnapService):
+    """Oran A1 Simulator library."""
 
     @classmethod
-    def check_version(cls, url) -> str:
+    def check_version(cls, base_url) -> str:
         """
         Return ric version.
 
+        Args:
+           base_url: the base url of the ric
+
         Returns:
             the ric version
 
         """
-        url = f"{url}/counter/interface"
+        url = f"{base_url}/counter/interface"
         version = cls.send_message('GET',
                                    'Get ric version',
-                                    url)
+                                   url)
         return version
+
+    @classmethod
+    def check_status(cls, url) -> str:
+        """
+        Return ric status.
+
+        Args:
+           url: the url of the ric
+
+        Returns:
+            the ric status
+
+        """
+        url = f"{url}"
+        status = cls.send_message('GET',
+                                  'Get ric status',
+                                  url)
+        return status
+
+    @classmethod
+    def get_policy_number(cls, url) -> str:
+        """
+        Policy numbers for ric.
+
+        Args:
+           url: the url of the ric
+
+        Returns:
+            the policy numbers for ric
+
+        """
+        url = f"{url}/counter/num_instances"
+        policy_number = cls.send_message('GET',
+                                         'Get policy numbers for ric',
+                                         url)
+        return policy_number
+
+    @classmethod
+    def create_policy_type(cls,
+                           url,
+                           policy_type_id,
+                           policy_type_data) -> None:
+        """
+        Create topic in Dmaap.
+
+        Args:
+           url: the url of the ric
+           policy_type_num: the policy type id
+           policy_type_data: the policy type data in binary format
+
+        """
+        url = f"{url}/policytype?id={policy_type_id}"
+        cls.send_message('PUT',
+                         'Create Policy Type',
+                         url,
+                         data=policy_type_data,
+                         headers={"Accept":"application/json", "Content-Type":"application/json"})