X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=smo-install%2Ftest%2Fpythonsdk%2Funit-tests%2Ftest_policy.py;fp=smo-install%2Ftest%2Fpythonsdk%2Funit-tests%2Ftest_policy.py;h=95534c51f8ec61aa4b8a5b3b36cda894fdf95e8c;hb=7b4b4007db003d28ee347e3d373e8ac605581cb5;hp=acbcda386751404cc7060444773e64cf31df3ea8;hpb=a8414e7e44dea72da6f4f235e14f7fef59b4e266;p=it%2Fdep.git

diff --git a/smo-install/test/pythonsdk/unit-tests/test_policy.py b/smo-install/test/pythonsdk/unit-tests/test_policy.py
index acbcda38..95534c51 100644
--- a/smo-install/test/pythonsdk/unit-tests/test_policy.py
+++ b/smo-install/test/pythonsdk/unit-tests/test_policy.py
@@ -35,12 +35,12 @@ def test_get_policy_status(mock_send_message_json):
                                                    basic_auth=BASIC_AUTH)
 
 
-@mock.patch.object(OranPolicy, 'send_message_json')
-def test_get_policy(mock_send_message_json):
+@mock.patch.object(OranPolicy, 'send_message')
+def test_get_policy(mock_send_message):
     """Test Policy's class method."""
     OranPolicy.get_policy(PolicyType(type="test_type", version="type_version"),
                           "policy_name", "policy_version", BASIC_AUTH)
-    mock_send_message_json.assert_called_once_with('GET',
+    mock_send_message.assert_called_once_with('GET',
                                                    'Get the policy',
                                                    (f"{API_URL}/policy/api/v1/policytypes/test_type/versions/"\
                                                    + "type_version/policies/policy_name/versions/policy_version"),
@@ -69,3 +69,23 @@ def test_deploy_policy(mock_send_message):
                                               data={},
                                               headers=HEADER,
                                               basic_auth=BASIC_AUTH)
+
+@mock.patch.object(OranPolicy, 'send_message')
+def test_undeploy_policy(mock_send_message):
+    """Test Policy's class method."""
+    OranPolicy.undeploy_policy("policy_id","1.0.0", BASIC_AUTH)
+    mock_send_message.assert_called_once_with('DELETE',
+                                              'Undeploy Policy',
+                                              (f"{PAP_URL}/policy/pap/v1/pdps/policies/policy_id/versions/1.0.0"),
+                                              headers=HEADER,
+                                              basic_auth=BASIC_AUTH)
+
+@mock.patch.object(OranPolicy, 'send_message')
+def test_delete_policy(mock_send_message):
+    """Test Policy's class method."""
+    OranPolicy.delete_policy(PolicyType(type="test_type", version="type_version"), "policy_id","1.0.0", BASIC_AUTH)
+    mock_send_message.assert_called_once_with('DELETE',
+                                              'Delete Policy',
+                                              (f"{API_URL}/policy/api/v1/policytypes/test_type/versions/type_version/policies/policy_id/versions/1.0.0"),
+                                              headers=HEADER,
+                                              basic_auth=BASIC_AUTH)