Add uninstall workflow to local CLI

2026-09-05 07:27:15 +08:00 · 2026-04-20 13:38:06 +08:00
parent 4ba8d42272
commit 93a19b467b
5 changed files with 600 additions and 4 deletions
@@ -0,0 +1,63 @@
+from __future__ import annotations
+
+import importlib.util
+import unittest
+import uuid
+from pathlib import Path
+from unittest.mock import patch
+
+
+MODULE_PATH = Path(__file__).resolve().parents[1] / "scripts" / "local_setup.py"
+
+
+def load_local_setup_module():
+    module_name = f"moviepilot_local_setup_config_{uuid.uuid4().hex}"
+    spec = importlib.util.spec_from_file_location(module_name, MODULE_PATH)
+    module = importlib.util.module_from_spec(spec)
+    assert spec and spec.loader
+    spec.loader.exec_module(module)
+    return module
+
+
+class LocalSetupConfigDirTests(unittest.TestCase):
+    def test_setup_prompts_for_config_dir_when_not_provided(self):
+        module = load_local_setup_module()
+        default_dir = Path("/tmp/default-moviepilot-config")
+        custom_dir = Path("/tmp/custom-moviepilot-config")
+
+        with patch.object(module, "_is_interactive", return_value=True), patch.object(
+            module, "resolve_config_dir", return_value=default_dir
+        ), patch.object(
+            module, "_prompt_path", return_value=str(custom_dir)
+        ):
+            result = module._resolve_interactive_config_dir("setup", None)
+
+        self.assertEqual(result, custom_dir)
+
+    def test_setup_keeps_default_config_dir_when_user_accepts_default(self):
+        module = load_local_setup_module()
+        default_dir = Path("/tmp/default-moviepilot-config")
+
+        with patch.object(module, "_is_interactive", return_value=True), patch.object(
+            module, "resolve_config_dir", return_value=default_dir
+        ), patch.object(
+            module, "_prompt_path", return_value=str(default_dir)
+        ):
+            result = module._resolve_interactive_config_dir("init", None)
+
+        self.assertEqual(result, default_dir)
+
+    def test_non_setup_command_does_not_prompt_for_config_dir(self):
+        module = load_local_setup_module()
+
+        with patch.object(module, "_is_interactive", return_value=True), patch.object(
+            module, "_prompt_path"
+        ) as prompt_mock:
+            result = module._resolve_interactive_config_dir("install-deps", None)
+
+        self.assertIsNone(result)
+        prompt_mock.assert_not_called()
+
+
+if __name__ == "__main__":
+    unittest.main()
@@ -0,0 +1,167 @@
+from __future__ import annotations
+
+import importlib.util
+import tempfile
+import unittest
+import uuid
+from contextlib import ExitStack
+from pathlib import Path
+from unittest.mock import patch
+
+
+MODULE_PATH = Path(__file__).resolve().parents[1] / "scripts" / "local_setup.py"
+
+
+def load_local_setup_module():
+    module_name = f"moviepilot_local_setup_{uuid.uuid4().hex}"
+    spec = importlib.util.spec_from_file_location(module_name, MODULE_PATH)
+    module = importlib.util.module_from_spec(spec)
+    assert spec and spec.loader
+    spec.loader.exec_module(module)
+    return module
+
+
+class LocalSetupUninstallTests(unittest.TestCase):
+    def prepare_install_tree(self, *, legacy_config: bool = False):
+        module = load_local_setup_module()
+        temp_dir = tempfile.TemporaryDirectory()
+        self.addCleanup(temp_dir.cleanup)
+
+        temp_path = Path(temp_dir.name)
+        root_dir = temp_path / "MoviePilot"
+        helper_dir = root_dir / "app" / "helper"
+        runtime_dir = root_dir / ".runtime"
+        public_dir = root_dir / "public"
+        venv_dir = root_dir / "venv"
+        install_env_file = root_dir / ".moviepilot.env"
+        config_dir = root_dir / "config" if legacy_config else temp_path / "moviepilot-config"
+        temp_config_dir = config_dir / "temp"
+
+        helper_dir.mkdir(parents=True)
+        runtime_dir.mkdir(parents=True)
+        public_dir.mkdir(parents=True)
+        venv_dir.mkdir(parents=True)
+        temp_config_dir.mkdir(parents=True)
+        install_env_file.write_text("CONFIG_DIR=/tmp/moviepilot-config\n", encoding="utf-8")
+        (root_dir / "moviepilot").write_text("#!/usr/bin/env bash\n", encoding="utf-8")
+        (helper_dir / "sites.py").write_text("generated\n", encoding="utf-8")
+        (helper_dir / "user.sites.v2.bin").write_bytes(b"binary")
+        (temp_config_dir / "moviepilot.runtime.json").write_text("{}", encoding="utf-8")
+        (temp_config_dir / "moviepilot.frontend.runtime.json").write_text(
+            "{}", encoding="utf-8"
+        )
+
+        stack = ExitStack()
+        self.addCleanup(stack.close)
+        stack.enter_context(patch.object(module, "ROOT", root_dir))
+        stack.enter_context(patch.object(module, "HELPER_DIR", helper_dir))
+        stack.enter_context(patch.object(module, "RUNTIME_DIR", runtime_dir))
+        stack.enter_context(patch.object(module, "PUBLIC_DIR", public_dir))
+        stack.enter_context(patch.object(module, "INSTALL_ENV_FILE", install_env_file))
+        stack.enter_context(patch.object(module, "LEGACY_CONFIG_DIR", root_dir / "config"))
+        stack.enter_context(patch.object(module, "CONFIG_DIR", config_dir))
+        stack.enter_context(patch.object(module, "TEMP_DIR", temp_config_dir))
+
+        return module, root_dir, config_dir, venv_dir, install_env_file
+
+    def test_remove_config_data_deletes_legacy_config_directory(self):
+        module, _, config_dir, _, _ = self.prepare_install_tree(legacy_config=True)
+        category_file = config_dir / "category.yaml"
+        category_file.write_text("seed\n", encoding="utf-8")
+        (config_dir / "logs").mkdir(exist_ok=True)
+        (config_dir / "user.db").write_text("db\n", encoding="utf-8")
+
+        removed = module._remove_config_data(config_dir)
+
+        self.assertFalse(config_dir.exists())
+        self.assertFalse(category_file.exists())
+        self.assertIn(
+            str(config_dir.resolve()),
+            {str(path.resolve()) for path in removed},
+        )
+
+    def test_uninstall_keeps_config_by_default(self):
+        module, root_dir, config_dir, venv_dir, install_env_file = self.prepare_install_tree()
+        cli_dir = root_dir.parent / "bin"
+        cli_dir.mkdir()
+        cli_link = cli_dir / "moviepilot"
+        cli_link.symlink_to(root_dir / "moviepilot")
+
+        yes_no_answers = iter([False, True])
+        with patch.object(module, "_is_interactive", return_value=True), patch.object(
+            module,
+            "_prompt_yes_no",
+            side_effect=lambda label, default=False: next(yes_no_answers),
+        ), patch.object(
+            module,
+            "_prompt_text",
+            side_effect=lambda label, default=None, allow_empty=False, secret=False: module.UNINSTALL_CONFIRM_TEXT,
+        ), patch.object(module, "_stop_managed_services", return_value=None):
+            result = module.uninstall_local(
+                venv_dir=venv_dir,
+                config_dir=config_dir,
+                launch_path=str(cli_link),
+            )
+
+        self.assertFalse(result["cancelled"])
+        self.assertTrue(config_dir.exists())
+        self.assertTrue(install_env_file.exists())
+        self.assertFalse(venv_dir.exists())
+        self.assertFalse((root_dir / ".runtime").exists())
+        self.assertFalse((root_dir / "public").exists())
+        self.assertFalse((root_dir / "app" / "helper" / "sites.py").exists())
+        self.assertFalse((root_dir / "app" / "helper" / "user.sites.v2.bin").exists())
+        self.assertFalse(cli_link.exists())
+
+    def test_uninstall_deletes_external_config_when_requested(self):
+        module, _, config_dir, venv_dir, install_env_file = self.prepare_install_tree()
+        yes_no_answers = iter([True, True])
+
+        with patch.object(module, "_is_interactive", return_value=True), patch.object(
+            module,
+            "_prompt_yes_no",
+            side_effect=lambda label, default=False: next(yes_no_answers),
+        ), patch.object(
+            module,
+            "_prompt_text",
+            side_effect=lambda label, default=None, allow_empty=False, secret=False: module.UNINSTALL_CONFIRM_TEXT,
+        ), patch.object(module, "_stop_managed_services", return_value=None):
+            result = module.uninstall_local(
+                venv_dir=venv_dir,
+                config_dir=config_dir,
+            )
+
+        self.assertFalse(result["cancelled"])
+        self.assertTrue(result["config_deleted"])
+        self.assertFalse(config_dir.exists())
+        self.assertFalse(install_env_file.exists())
+
+    def test_uninstall_deletes_legacy_config_when_requested(self):
+        module, _, config_dir, venv_dir, install_env_file = self.prepare_install_tree(
+            legacy_config=True
+        )
+        (config_dir / "category.yaml").write_text("seed\n", encoding="utf-8")
+        yes_no_answers = iter([True, True])
+
+        with patch.object(module, "_is_interactive", return_value=True), patch.object(
+            module,
+            "_prompt_yes_no",
+            side_effect=lambda label, default=False: next(yes_no_answers),
+        ), patch.object(
+            module,
+            "_prompt_text",
+            side_effect=lambda label, default=None, allow_empty=False, secret=False: module.UNINSTALL_CONFIRM_TEXT,
+        ), patch.object(module, "_stop_managed_services", return_value=None):
+            result = module.uninstall_local(
+                venv_dir=venv_dir,
+                config_dir=config_dir,
+            )
+
+        self.assertFalse(result["cancelled"])
+        self.assertTrue(result["config_deleted"])
+        self.assertFalse(config_dir.exists())
+        self.assertFalse(install_env_file.exists())
+
+
+if __name__ == "__main__":
+    unittest.main()