tests/unit/utils/test_service_loader.py

from unittest.mock import MagicMock, patch

from src.utils.service_loader import (
    get_analyzer_if_available,
    get_embedding_service_if_available,
)


def test_get_embedding_service_success():
    """Test successful loading of embedding service (free tier fallback)."""
    mock_service = MagicMock()

    # Patch settings to disable premium tier, then patch the local service
    with patch("src.utils.service_loader.settings") as mock_settings:
        mock_settings.has_openai_key = False

        with patch("src.services.embeddings.get_embedding_service", return_value=mock_service):
            service = get_embedding_service_if_available()
            assert service is mock_service


def test_get_embedding_service_import_error():
    """Test handling of ImportError when loading embedding service."""
    # Disable premium tier, then make local service fail
    with patch("src.utils.service_loader.settings") as mock_settings:
        mock_settings.has_openai_key = False

        with patch(
            "src.services.embeddings.get_embedding_service",
            side_effect=ImportError("Missing deps"),
        ):
            service = get_embedding_service_if_available()
            assert service is None


def test_get_embedding_service_generic_error():
    """Test handling of generic Exception when loading embedding service."""
    # Disable premium tier, then make local service fail
    with patch("src.utils.service_loader.settings") as mock_settings:
        mock_settings.has_openai_key = False

        with patch(
            "src.services.embeddings.get_embedding_service",
            side_effect=ValueError("Boom"),
        ):
            service = get_embedding_service_if_available()
            assert service is None


def test_get_analyzer_success():
    """Test successful loading of analyzer."""
    with patch("src.services.statistical_analyzer.get_statistical_analyzer") as mock_get:
        mock_analyzer = MagicMock()
        mock_get.return_value = mock_analyzer

        analyzer = get_analyzer_if_available()

        assert analyzer is mock_analyzer
        mock_get.assert_called_once()


def test_get_analyzer_import_error():
    """Test handling of ImportError when loading analyzer."""
    with patch(
        "src.services.statistical_analyzer.get_statistical_analyzer",
        side_effect=ImportError("No Modal"),
    ):
        analyzer = get_analyzer_if_available()
        assert analyzer is None


def test_get_analyzer_generic_error():
    """Test handling of generic Exception when loading analyzer."""
    with patch(
        "src.services.statistical_analyzer.get_statistical_analyzer",
        side_effect=RuntimeError("Fail"),
    ):
        analyzer = get_analyzer_if_available()
        assert analyzer is None