Adding UTs for model deployment

Author: kumar-shivam-ranjan

tests/unitary/with_extras/aqua/test_deployment_handler.py

@@ -92,6 +92,26 @@ def test_get_deployment(self, mock_get):
         self.deployment_handler.get(id="mock-model-id")
         mock_get.assert_called()
 
+    @patch("ads.aqua.modeldeployment.AquaDeploymentApp.delete")
+    def test_delete_deployment(self,mock_delete):
+        self.deployment_handler.request.path = "aqua/deployments"
+        self.deployment_handler.delete("mock-model-id")
+        mock_delete.assert_called()
+
+    @patch("ads.aqua.modeldeployment.AquaDeploymentApp.activate")
+    def test_activate_deployment(self,mock_activate):
+        self.deployment_handler.request.path = "aqua/deployments/ocid1.datasciencemodeldeployment.oc1.iad.xxx/activate"
+        mock_activate.return_value={"lifecycle_state":"UPDATING"}
+        self.deployment_handler.put()
+        mock_activate.assert_called()
+
+    @patch("ads.aqua.modeldeployment.AquaDeploymentApp.deactivate")
+    def test_deactivate_deployment(self,mock_deactivate):
+        self.deployment_handler.request.path = "aqua/deployments/ocid1.datasciencemodeldeployment.oc1.iad.xxx/deactivate"
+        mock_deactivate.return_value={"lifecycle_state":"UPDATING"}
+        self.deployment_handler.put()
+        mock_deactivate.assert_called()
+
     @patch("ads.aqua.modeldeployment.AquaDeploymentApp.list")
     def test_list_deployment(self, mock_list):
         """Test get method to return a list of model deployments."""

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -20,6 +20,7 @@
 )
 from ads.aqua.model import AquaModelApp
 from ads.aqua.model.entities import AquaModel, AquaModelSummary, HFModelSummary
+from ads.aqua.ui import AquaContainerConfig
 
 
 class ModelHandlerTestCase(TestCase):
@@ -93,6 +94,32 @@ def test_delete_with_id(self, mock_delete, mock_urlparse):
             mock_urlparse.assert_called()
             mock_delete.assert_called()
 
+    @patch.object(AquaContainerConfig,"from_container_index_json")
+    @patch.object(AquaModelApp,"edit_registered_model")
+    def test_put(self,mock_edit,mock_container_index):
+        mock_edit.return_value={"state":"EDITED"}
+        mock_inference = MagicMock()
+        mock_inference.values.return_value = [
+            MagicMock(family="odsc-vllm-serving"),
+            MagicMock(family="odsc-tgi-serving"),
+            MagicMock(family="odsc-vllm-serving"),
+        ]
+
+        mock_container_index.return_value = MagicMock(inference=mock_inference)
+        self.model_handler.get_json_body = MagicMock(
+            return_value=dict(
+                task="text_generation",
+                enable_finetuning="true",
+                inference_container="odsc-tgi-serving",
+            )
+        )
+        with patch("ads.aqua.extension.base_handler.AquaAPIhandler.finish") as mock_finish:
+            mock_finish.side_effect = lambda x: x
+            result = self.model_handler.put(id="ocid1.datasciencemodel.oc1.iad.xxx")
+            print(f"result: ",result)
+            assert result["state"] is "EDITED"
+            mock_edit.assert_called()
+
     @patch.object(AquaModelApp, "list")
     def test_list(self, mock_list):
         with patch(