add more tests

Author: z7ye

tests/unitary/with_extras/opctl/test_opctl_local_model_deployment_backend.py

@@ -4,9 +4,9 @@
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 
-from mock import ANY, patch, MagicMock, PropertyMock
+from mock import ANY, patch, MagicMock
 import pytest
-from ads.opctl.backend.local import LocalModelDeploymentBackend, ModelCustomMetadata, os, create_signer, OCIClientFactory
+from ads.opctl.backend.local import LocalModelDeploymentBackend, ModelCustomMetadata, os, create_signer, OCIClientFactory, is_in_notebook_session, run_command
 
 
 
@@ -155,3 +155,41 @@ def test_predict_download(
                         mock__run_with_conda_pack.assert_called_once_with(
                             {os.path.expanduser('~/.oci'): {'bind': '/home/datascience/.oci'}, os.path.expanduser('~/.ads_ops/models/fake_id'): {'bind': '/opt/ds/model/deployed_model/'}}, "--artifact-directory /opt/ds/model/deployed_model/ --payload 'fake_data' --auth api_key --profile DEFAULT", install=True, conda_uri='fake_path'
                         )
+    @patch("ads.opctl.backend.local.is_in_notebook_session", return_value=True)
+    @patch("ads.opctl.backend.local.os.listdir", return_value=["path"])
+    @patch("ads.opctl.backend.local.os.path.exists", return_value=False)
+    def test_predict_download_in_notebook(
+        self, mock_path_exists, mock_list_dir, mock_is_in_notebook 
+    ): 
+        with patch("ads.opctl.backend.local._download_model") as mock__download:
+            with patch.object(
+                LocalModelDeploymentBackend,
+                "_get_conda_info_from_custom_metadata",
+                return_value=("fake_slug", "fake_path"),
+            ):
+                with patch.object(
+                    LocalModelDeploymentBackend, "_get_conda_info_from_runtime"
+                ):
+                    with patch(
+                        "ads.opctl.backend.local.run_command"
+                    ) as mock_run_command:
+                        backend = LocalModelDeploymentBackend(self.config)
+                        
+                        backend.predict()
+                        mock__download.assert_called_once_with(
+                            auth='api_key',
+                            profile='DEFAULT',
+                            ocid="fake_id",
+                            artifact_directory=os.path.expanduser(
+                                "~/.ads_ops/models/fake_id"
+                            ),
+                            region=None,
+                            bucket_uri="fake_bucket",
+                            timeout=None,
+                            force_overwrite=True,
+                        )
+                        dir_path = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), "../../../../"))
+                        script_path = os.path.join(dir_path, "ads/opctl/backend/../script.py")
+                        mock_run_command.assert_called_once_with(
+                            cmd=f"python {script_path} --artifact-directory /Users/ziye/.ads_ops/models/fake_id --payload 'fake_data' --auth api_key --profile DEFAULT", shell=True
+                        )