Exception handling added for the uninstall_dev_env calls. The error reporting for the Platform Error cleaned up.

Author: janosmurai

dem/cli/command/create_cmd.py

@@ -5,6 +5,7 @@
 from dem.core.dev_env import DevEnv, DevEnv
 from dem.core.tool_images import ToolImages
 from dem.core.platform import Platform
+from dem.core.exceptions import PlatformError
 from dem.cli.console import stdout, stderr
 from dem.cli.tui.panel.tool_type_selector import ToolTypeSelectorPanel
 from dem.cli.tui.panel.tool_image_selector import ToolImageSelectorPanel
@@ -171,7 +172,11 @@ def create_dev_env(platform: Platform, dev_env_name: str) -> None:
         if dev_env_original.is_installed:
             typer.confirm("The Development Environment is installed, so it can't be overwritten. " + \
                           "Uninstall it first?", abort=True)
-            platform.uninstall_dev_env(dev_env_original)
+            try:
+                platform.uninstall_dev_env(dev_env_original)
+            except PlatformError as e:
+                stderr.print(f"[red]{str(e)}[/]")
+                typer.Abort()
 
     tool_image_list = get_tool_image_list(platform.tool_images)
     new_dev_env_descriptor = get_dev_env_descriptor_from_user(dev_env_name, tool_image_list)

dem/cli/command/delete_cmd.py

@@ -20,7 +20,7 @@ def execute(platform: Platform, dev_env_name: str) -> None:
             try:
                 platform.uninstall_dev_env(dev_env_to_delete)
             except PlatformError as e:
-                stderr.print(f"[red]Error: The deletion failed, because the Dev Env can't be uninstalled. {str(e)}[/]")
+                stderr.print(f"[red]{str(e)}[/]")
                 return
 
         stdout.print("Deleting the Development Environment...")

dem/cli/command/init_cmd.py

@@ -36,7 +36,7 @@ def execute(platform: Platform, project_path: str) -> None:
                 try:
                     platform.uninstall_dev_env(local_dev_env)
                 except PlatformError as e:
-                    stderr.print(f"[red]Error: The Dev Env can't be uninstalled. {str(e)}")
+                    stderr.print(f"[red]{str(e)}[/]")
                     return
 
             platform.local_dev_envs.remove(local_dev_env)

dem/cli/command/modify_cmd.py

@@ -5,6 +5,7 @@
 from dem.core.dev_env import DevEnv, DevEnv
 from dem.core.tool_images import ToolImages
 from dem.core.platform import Platform
+from dem.core.exceptions import PlatformError
 from dem.cli.console import stderr, stdout
 from dem.cli.tui.renderable.menu import SelectMenu
 from dem.cli.tui.panel.tool_type_selector import ToolTypeSelectorPanel
@@ -207,7 +208,11 @@ def execute(platform: Platform, dev_env_name: str, tool_type: str, tool_image: s
     elif dev_env.is_installed is True:
         stdout.print("[yellow]The Development Environment is installed, so it can't be modified.[/]")
         typer.confirm("Do you want to uninstall it first?", abort=True)
-        platform.uninstall_dev_env(dev_env)
+        try:
+            platform.uninstall_dev_env(dev_env)
+        except PlatformError as e:
+            stderr.print(f"[red]{str(e)}[/]")
+            return
 
     if (tool_type or tool_image):
         modify_single_tool(platform, dev_env, tool_type, tool_image)

dem/cli/command/uninstall_cmd.py

@@ -23,6 +23,6 @@ def execute(platform: Platform, dev_env_name: str) -> None:
         try:
             platform.uninstall_dev_env(dev_env_to_uninstall)
         except PlatformError as e:
-            stderr.print(f"[red]Error: {str(e)}[/]")
+            stderr.print(f"[red]{str(e)}[/]")
         else:
             stdout.print(f"[green]Successfully uninstalled the {dev_env_name}![/]")

dem/core/container_engine.py

@@ -114,7 +114,7 @@ def remove(self, image: str) -> None:
         except docker.errors.ImageNotFound:
             self.user_output.msg(f"[yellow]The {image} doesn't exist. Unable to remove it.[/]\n")
         except docker.errors.APIError:
-            raise ContainerEngineError(f"The {image} is used by a container. Unable to remove it.[/]\n")
+            raise ContainerEngineError(f"The {image} is used by a container. Unable to remove it.\n")
         else:
             self.user_output.msg(f"[green]Successfully removed the {image}![/]\n")
 

dem/core/platform.py

@@ -166,11 +166,11 @@ def install_dev_env(self, dev_env_to_install: DevEnv) -> None:
     def uninstall_dev_env(self, dev_env_to_uninstall: DevEnv) -> None:
         """ Uninstall the Dev Env by removing the images not required anymore.
 
-            Exceptions:
-                PlatformError -- if the uninstall fails
-        
             Args:
                 dev_env_to_uninstall -- the Development Environment to uninstall
+
+            Exceptions:
+                PlatformError -- if the uninstall fails
         """
         all_required_tool_images = set()
         for dev_env in self.local_dev_envs:
@@ -188,7 +188,7 @@ def uninstall_dev_env(self, dev_env_to_uninstall: DevEnv) -> None:
             try:
                 self.container_engine.remove(tool_image)
             except ContainerEngineError as e:
-                raise PlatformError(f"Dev Env uninstall failed. {str(e)}")
+                raise PlatformError(f"Dev Env uninstall failed. <-caused by- {str(e)}")
 
         dev_env_to_uninstall.is_installed = False
         self.flush_descriptors()

tests/cli/test_create_cmd.py

@@ -124,6 +124,38 @@ def test_create_dev_env_overwrite(mock_confirm, mock_get_tool_image_list,
     mock_get_dev_env_descriptor_from_user.assert_called_once_with(expected_dev_env_name,
                                                                   mock_tool_images)
 
+@patch("dem.cli.command.create_cmd.typer.Abort")
+@patch("dem.cli.command.create_cmd.stderr.print")
+@patch("dem.cli.command.create_cmd.typer.confirm")
+def test_create_dev_env_overwrite_PlatformError(mock_confirm: MagicMock, mock_stderr_print: MagicMock, 
+                                                mock_Abort: MagicMock) -> None:
+    # Test setup
+    mock_platform = MagicMock()
+    mock_dev_env_original = MagicMock()
+    mock_dev_env_original.is_installed = True
+    mock_platform.get_dev_env_by_name.return_value = mock_dev_env_original
+    test_exception_text = "test_exception_text"
+    mock_platform.uninstall_dev_env.side_effect = create_cmd.PlatformError(test_exception_text)
+    mock_Abort.side_effect = Exception("")
+
+    test_dev_env_name = "test_dev_env"
+
+    # Run unit under test
+    with pytest.raises(Exception):
+        create_cmd.create_dev_env(mock_platform, test_dev_env_name)
+
+    # Check expectations
+    mock_platform.get_dev_env_by_name.assert_called_once_with(test_dev_env_name)
+    mock_confirm.assert_has_calls([
+        call("The input name is already used by a Development Environment. Overwrite it?", 
+             abort=True),
+        call("The Development Environment is installed, so it can't be overwritten. " + \
+             "Uninstall it first?", abort=True)
+    ])
+    mock_platform.uninstall_dev_env.assert_called_once_with(mock_dev_env_original)
+    mock_stderr_print.assert_called_once_with(f"[red]Platform error: {test_exception_text}[/]")
+    mock_Abort.assert_called_once()
+
 @patch("dem.cli.command.create_cmd.get_dev_env_descriptor_from_user")
 @patch("dem.cli.command.create_cmd.typer.confirm")
 def test_create_dev_env_abort(mock_confirm, mock_get_dev_env_descriptor_from_user):

tests/cli/test_delete_cmd.py

@@ -66,7 +66,7 @@ def test_delete_uninstall_failed(mock_stderr_print: MagicMock, mock_confirm: Mag
     mock_confirm.assert_called_once_with("The Development Environment is installed. Do you want to uninstall it?", 
                                         abort=True)
     mock_platform.uninstall_dev_env.assert_called_once_with(test_dev_env)
-    mock_stderr_print.assert_called_once_with(f"[red]Error: The deletion failed, because the Dev Env can't be uninstalled. Platform error: {test_exception_text}[/]")
+    mock_stderr_print.assert_called_once_with(f"[red]Platform error: {test_exception_text}[/]")
 
 @patch("dem.cli.command.delete_cmd.stderr.print")
 def test_delete_not_existing(mock_stderr_print: MagicMock) -> None:

tests/cli/test_init_cmd.py

@@ -133,4 +133,4 @@ def test_execute_reinit_installed_uninstall_fails(mock_DevEnv, mock_confirm, moc
     mock_confirm.assert_has_calls([call("Would you like to re-init the Dev Env? All local changes will be lost!", abort=True),
                                    call("The Development Environment is installed, so it can't be deleted. Do you want to uninstall it first?", abort=True)])
     mock_platform.uninstall_dev_env.assert_called_once_with(mock_local_dev_env)
-    mock_stderr_print.assert_called_once_with(f"[red]Error: The Dev Env can't be uninstalled. Platform error: {test_exception_text}")
+    mock_stderr_print.assert_called_once_with(f"[red]Platform error: {test_exception_text}[/]")