api-camoufox/tests/test_wrapper.py

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