MSA-11489 - Add method delete MS method

efeubiqube · efeubiqube · commit e61d188593a6 · 2022-06-09T14:50:56.000+01:00
diff --git a/msa_sdk/configuration_object.py b/msa_sdk/configuration_object.py
@@ -0,0 +1,47 @@
+"""Module Configuration Object."""
+
+from msa_sdk.msa_api import MSA_API
+
+
+class ConfigurationObject(MSA_API):
+    """Class Configuration Ojbect."""
+
+    def __init__(self):
+        """
+        Initialize.
+
+        Parameters
+        ----------
+        None
+
+        Returns
+        -------
+        None
+
+        """
+        MSA_API.__init__(self)
+        self.api_path = "/configuration-objects"
+
+    def delete(self, device_id: int, object_name: str = "") -> None:
+        """
+        Delete microservice object instance.
+
+        Parameters
+        ----------
+        device_id: Integer
+                Device id
+
+        object_name: String
+                Object name
+
+        Returns
+        -------
+        None
+        """
+        if object_name:
+            self.path = "{}/{}/{}".format(self.api_path, device_id,
+                                          object_name)
+        else:
+            self.path = "{}/{}".format(self.api_path, device_id)
+
+        self._call_delete()
diff --git a/tests/test_configuration_object.py b/tests/test_configuration_object.py
@@ -0,0 +1,34 @@
+"""
+Test Repository
+"""
+from unittest.mock import patch
+
+from msa_sdk.configuration_object import ConfigurationObject
+
+
+def test_delete_device():
+    """
+    Test Delete device
+    """
+
+    with patch('msa_sdk.msa_api.MSA_API._call_delete') as mock_call_delete:
+        conf_object = ConfigurationObject()
+
+        conf_object.delete(200)
+
+        assert conf_object.path == "/configuration-objects/200"
+        mock_call_delete.assert_called_once_with()
+
+
+def test_delete_device_object():
+    """
+    Tetst Delete Device Object
+    """
+
+    with patch('msa_sdk.msa_api.MSA_API._call_delete') as mock_call_delete:
+        conf_object = ConfigurationObject()
+
+        conf_object.delete(200, 'ObjectName')
+
+        assert conf_object.path == "/configuration-objects/200/ObjectName"
+        mock_call_delete.assert_called_once_with()
