from unittest.mock import MagicMock, patch from app.use_cases.camoufox.wrapper import ( CamoufoxWrapper, _detect_headless_mode, _resolve_headless_mode, ) class TestDetectHeadlessMode: @patch("sys.platform", "darwin") def test_macos_returns_true(self): result = _detect_headless_mode() assert result is True @patch("sys.platform", "win32") def test_windows_returns_true(self): result = _detect_headless_mode() assert result is True @patch("shutil.which", return_value=None) @patch("sys.platform", "linux") def test_linux_without_xvfb_returns_true(self, mock_which): result = _detect_headless_mode() assert result is True @patch("shutil.which", return_value="/usr/bin/Xvfb") @patch("sys.platform", "linux") def test_linux_with_xvfb_returns_virtual(self, mock_which): result = _detect_headless_mode() assert result == "virtual" class TestResolveHeadlessMode: @patch("app.use_cases.camoufox.wrapper.config") def test_env_virtual_returns_virtual(self, mock_config): mock_config.HEADLESS_MODE = "virtual" result = _resolve_headless_mode() assert result == "virtual" @patch("app.use_cases.camoufox.wrapper.config") def test_env_true_returns_true(self, mock_config): for val in ["true", "True", "1"]: mock_config.HEADLESS_MODE = val result = _resolve_headless_mode() assert result is True @patch("app.use_cases.camoufox.wrapper.config") def test_env_false_returns_false(self, mock_config): for val in ["false", "False", "0"]: mock_config.HEADLESS_MODE = val result = _resolve_headless_mode() assert result is False @patch("app.use_cases.camoufox.wrapper.config") @patch("app.use_cases.camoufox.wrapper._detect_headless_mode") def test_env_auto_uses_detection(self, mock_detect, mock_config): mock_config.HEADLESS_MODE = "auto" mock_detect.return_value = True result = _resolve_headless_mode() mock_detect.assert_called_once() assert result is True class TestCamoufoxWrapper: def test_singleton(self): wrapper1 = CamoufoxWrapper() wrapper2 = CamoufoxWrapper() assert wrapper1 is wrapper2 def test_build_proxy_config_simple(self): wrapper = CamoufoxWrapper() result = wrapper._build_proxy_config("host:8080") assert result["server"] == "http://host:8080" def test_build_proxy_config_with_scheme(self): wrapper = CamoufoxWrapper() result = wrapper._build_proxy_config("http://host:8080") assert result["server"] == "http://host:8080" def test_build_proxy_config_with_auth(self): wrapper = CamoufoxWrapper() result = wrapper._build_proxy_config( "host:8080", {"username": "user", "password": "pass"} ) assert result["server"] == "http://host:8080" assert result["username"] == "user" assert result["password"] == "pass" def test_build_proxy_config_with_auth_no_password(self): wrapper = CamoufoxWrapper() result = wrapper._build_proxy_config( "host:8080", {"username": "user"} ) assert result["server"] == "http://host:8080" assert result["username"] == "user" assert "password" not in result