Class TestEmbedderFactory

tests/unit/test_all_embedders.py:146–197

Test the embedder factory function.

class TestEmbedderFactory:
    """Test the embedder factory function."""
    
    def test_get_embedder_with_explicit_type(self):
        """Test get_embedder with explicit embedder_type parameter."""
        from api.tools.embedder import get_embedder
        
        # Test Google embedder
        google_embedder = get_embedder(embedder_type='google')
        assert google_embedder is not None, "Google embedder should be created"

        # Test Bedrock embedder (mock boto3 to avoid hitting AWS credential providers)
        with patch("api.bedrock_client.boto3.Session") as mock_session_cls:
            mock_session = MagicMock()
            mock_session.client.return_value = MagicMock()
            mock_session_cls.return_value = mock_session
            bedrock_embedder = get_embedder(embedder_type='bedrock')
            assert bedrock_embedder is not None, "Bedrock embedder should be created"
        
        # Test OpenAI embedder
        openai_embedder = get_embedder(embedder_type='openai')
        assert openai_embedder is not None, "OpenAI embedder should be created"
        
        # Test Ollama embedder (may fail if Ollama not available, but should not crash)
        try:
            ollama_embedder = get_embedder(embedder_type='ollama')
            assert ollama_embedder is not None, "Ollama embedder should be created"
        except Exception as e:
            logger.warning(f"Ollama embedder creation failed (expected if Ollama not available): {e}")

    def test_get_embedder_with_legacy_params(self):
        """Test get_embedder with legacy boolean parameters."""
        from api.tools.embedder import get_embedder
        
        # Test with use_google_embedder=True
        google_embedder = get_embedder(use_google_embedder=True)
        assert google_embedder is not None, "Google embedder should be created with use_google_embedder=True"
        
        # Test with is_local_ollama=True
        try:
            ollama_embedder = get_embedder(is_local_ollama=True)
            assert ollama_embedder is not None, "Ollama embedder should be created with is_local_ollama=True"
        except Exception as e:
            logger.warning(f"Ollama embedder creation failed (expected if Ollama not available): {e}")

    def test_get_embedder_auto_detection(self):
        """Test get_embedder with automatic type detection."""
        from api.tools.embedder import get_embedder
        
        # Test auto-detection (should use current configuration)
        embedder = get_embedder()
        assert embedder is not None, "Auto-detected embedder should be created"


class TestEmbedderClients: