mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-23 07:26:46 +00:00
0ee9fec1d2
- Replace Playwright-based browser emulation with CloakBrowser as default - Update config to support CloakBrowser options and humanization presets - Refactor browser helper to use CloakBrowser context and remove cf_clearance dependency - Update Dockerfile, entrypoint, and update scripts to install CloakBrowser runtime - Ensure CloakBrowser kernel is pre-installed during local setup and dependency updates - Add tests for CloakBrowser integration and legacy compatibility
94 lines
3.4 KiB
Python
94 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import tempfile
|
|
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()
|
|
|
|
def test_install_deps_installs_browser_runtime(self):
|
|
module = load_local_setup_module()
|
|
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
venv_dir = (Path(temp_dir) / "venv").resolve()
|
|
venv_python = venv_dir / "bin" / "python"
|
|
venv_pip = venv_dir / "bin" / "pip"
|
|
|
|
with patch.object(module, "ensure_supported_python"), \
|
|
patch.object(
|
|
module,
|
|
"configure_venv_pip_compat",
|
|
return_value=venv_pip,
|
|
), \
|
|
patch.object(module, "run") as run_mock, \
|
|
patch.object(module, "install_browser_runtime") as install_browser:
|
|
result = module.install_deps(
|
|
python_bin="python3",
|
|
venv_dir=venv_dir,
|
|
recreate=False,
|
|
)
|
|
|
|
self.assertEqual(result, venv_python)
|
|
run_mock.assert_any_call(["python3", "-m", "venv", str(venv_dir)])
|
|
run_mock.assert_any_call(
|
|
[str(venv_pip), "install", "-r", str(module.ROOT / "requirements.txt")]
|
|
)
|
|
install_browser.assert_called_once_with(venv_python)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|