X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=smo-install%2Ftest%2Fpythonsdk%2Fsrc%2Foransdk%2Fpolicy%2Fpolicy.py;h=304d6987ba047320d5de486d24e328e8b092ad3d;hb=dd88f8e183d4463db48e1db2abb5d101e3dfb02a;hp=34139097770be7c7cddc35fdbfe825edf85ec943;hpb=285d9a5c96b23594b419d95c71c5d6a2cf52052e;p=it%2Fdep.git

diff --git a/smo-install/test/pythonsdk/src/oransdk/policy/policy.py b/smo-install/test/pythonsdk/src/oransdk/policy/policy.py
index 34139097..304d6987 100644
--- a/smo-install/test/pythonsdk/src/oransdk/policy/policy.py
+++ b/smo-install/test/pythonsdk/src/oransdk/policy/policy.py
@@ -1,12 +1,32 @@
 #!/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============================================
+# ===================================================================
+#
+###
 """Onap Policy module."""
 
 from dataclasses import dataclass
 from typing import Dict
-from oransdk.configuration import settings
 from onapsdk.onap_service import OnapService
+from oransdk.configuration import settings
 
 @dataclass
 class PolicyType:
@@ -44,8 +64,7 @@ class OranPolicy(OnapService):
         return status
 
     @classmethod
-    def get_policy_status(cls,
-                          basic_auth: Dict[str, str]) -> Dict:
+    def get_policy_status(cls, basic_auth: Dict[str, str]) -> Dict:
         """
         Get status of all the policies.
 
@@ -61,11 +80,7 @@ class OranPolicy(OnapService):
         return status
 
     @classmethod
-    def get_policy(cls,
-                   policy_type: PolicyType,
-                   policy_name,
-                   policy_version,
-                   basic_auth: Dict[str, str]) -> Dict:
+    def get_policy(cls, policy_type: PolicyType, policy_name, policy_version, basic_auth: Dict[str, str]) -> Dict:
         """
         Get the policy.
 
@@ -76,22 +91,15 @@ class OranPolicy(OnapService):
            basic_auth: (Dict[str, str]) for example:{ 'username': 'bob', 'password': 'secret' }
 
         Returns:
-           the policy
+           the policy reponse
 
         """
-        url = f"{cls.api_url}/policy/api/v1/policytypes/{policy_type.type}/versions/"\
-              + f"{policy_type.version}/policies/{policy_name}/versions/{policy_version}"
-        policy = cls.send_message_json('GET',
-                                       'Get the policy',
-                                       url,
-                                       basic_auth=basic_auth)
-        return policy
+        url = f"{cls.api_url}/policy/api/v1/policytypes/{policy_type.type}/versions/{policy_type.version}/policies/{policy_name}/versions/{policy_version}"
+        policy_response = cls.send_message('GET', 'Get the policy', url, basic_auth=basic_auth)
+        return policy_response
 
     @classmethod
-    def create_policy(cls,
-                      policy_type: PolicyType,
-                      policy_data,
-                      basic_auth: Dict[str, str]) -> None:
+    def create_policy(cls, policy_type: PolicyType, policy_data, basic_auth: Dict[str, str]) -> None:
         """
         Create a policy.
 
@@ -101,19 +109,12 @@ class OranPolicy(OnapService):
            policy_data: the policy to be created, in binary format
 
         """
-        url = f"{cls.api_url}/policy/api/v1/policytypes/{policy_type.type}/"\
-              + f"versions/{policy_type.version}/policies"
-        cls.send_message('POST',
-                         'Create Policy',
-                         url,
-                         data=policy_data,
-                         headers=cls.header,
+        url = f"{cls.api_url}/policy/api/v1/policytypes/{policy_type.type}/versions/{policy_type.version}/policies"
+        cls.send_message('POST', 'Create Policy', url, data=policy_data, headers=cls.header,
                          basic_auth=basic_auth)
 
     @classmethod
-    def deploy_policy(cls,
-                      policy_data,
-                      basic_auth: Dict[str, str]) -> None:
+    def deploy_policy(cls, policy_data, basic_auth: Dict[str, str]) -> None:
         """
         Deploy a policy.
 
@@ -122,9 +123,31 @@ class OranPolicy(OnapService):
 
         """
         url = f"{cls.pap_url}/policy/pap/v1/pdps/policies"
-        cls.send_message('POST',
-                         'Deploy Policy',
-                         url,
-                         data=policy_data,
-                         headers=cls.header,
-                         basic_auth=basic_auth)
+        cls.send_message('POST', 'Deploy Policy', url, data=policy_data, headers=cls.header, basic_auth=basic_auth)
+
+    @classmethod
+    def undeploy_policy(cls, policy_id, policy_version, basic_auth: Dict[str, str]) -> None:
+        """
+        Undeploy a policy.
+
+        Args:
+           policy_id: The policy id as provided during the create
+           policy_version: The policy version as provided during the create
+
+        """
+        url = f"{cls.pap_url}/policy/pap/v1/pdps/policies/{policy_id}/versions/{policy_version}"
+        cls.send_message('DELETE', 'Undeploy Policy', url, headers=cls.header, basic_auth=basic_auth)
+
+    @classmethod
+    def delete_policy(cls, policy_type: PolicyType, policy_id, policy_version, basic_auth: Dict[str, str]) -> None:
+        """
+        Delete a policy.
+
+        Args:
+           policy_type: the policy type
+           policy_id: The policy id as provided during the create
+           policy_version: The policy version as provided during the create
+
+        """
+        url = f"{cls.api_url}/policy/api/v1/policytypes/{policy_type.type}/versions/{policy_type.version}/policies/{policy_id}/versions/{policy_version}"
+        cls.send_message('DELETE', 'Delete Policy', url, headers=cls.header, basic_auth=basic_auth)