import pytest
import importlib
import sys
from unittest.mock import patch

from crewai.utilities.errors import ChromaDBRequiredError


def test_import_without_chromadb():
    """Test that crewai can be imported without chromadb."""
    with patch.dict(sys.modules, {"chromadb": None, "chromadb.errors": None, "chromadb.api": None, "chromadb.config": None}):
        modules_to_reload = [
            "crewai.memory.storage.rag_storage",
            "crewai.knowledge.storage.knowledge_storage", 
            "crewai.utilities.embedding_configurator"
        ]
        for module in modules_to_reload:
            if module in sys.modules:
                importlib.reload(sys.modules[module])
            
        from crewai import Agent, Task, Crew, Process
        
        agent = Agent(role="Test Agent", goal="Test Goal", backstory="Test Backstory")
        task = Task(description="Test Task", agent=agent)
        _ = Crew(agents=[agent], tasks=[task], process=Process.sequential)


def test_memory_storage_without_chromadb():
    """Test that memory storage raises appropriate error when chromadb is not available."""
    with patch.dict(sys.modules, {"chromadb": None, "chromadb.errors": None, "chromadb.api": None, "chromadb.config": None}):
        if "crewai.memory.storage.rag_storage" in sys.modules:
            importlib.reload(sys.modules["crewai.memory.storage.rag_storage"])
            
        from crewai.memory.storage.rag_storage import RAGStorage, HAS_CHROMADB
        
        assert not HAS_CHROMADB
        
        with pytest.raises(ChromaDBRequiredError) as excinfo:
            storage = RAGStorage("memory", allow_reset=True, crew=None)
            
        assert "ChromaDB is required for memory storage" in str(excinfo.value)


def test_knowledge_storage_without_chromadb():
    """Test that knowledge storage raises appropriate error when chromadb is not available."""
    with patch.dict(sys.modules, {"chromadb": None, "chromadb.errors": None, "chromadb.api": None, "chromadb.config": None}):
        modules_to_reload = [
            "crewai.knowledge.storage.knowledge_storage",
            "crewai.utilities.embedding_configurator"
        ]
        for module in modules_to_reload:
            if module in sys.modules:
                importlib.reload(sys.modules[module])
            
        from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage, HAS_CHROMADB
        
        assert not HAS_CHROMADB
        
        with pytest.raises(ChromaDBRequiredError) as excinfo:
            storage = KnowledgeStorage()
            storage.initialize_knowledge_storage()
            
        assert "ChromaDB is required for knowledge storage" in str(excinfo.value)


def test_embedding_configurator_without_chromadb():
    """Test that embedding configurator raises appropriate error when chromadb is not available."""
    with patch.dict(sys.modules, {"chromadb": None, "chromadb.errors": None, "chromadb.api": None, "chromadb.config": None}):
        if "crewai.utilities.embedding_configurator" in sys.modules:
            importlib.reload(sys.modules["crewai.utilities.embedding_configurator"])
            
        from crewai.utilities.embedding_configurator import EmbeddingConfigurator, HAS_CHROMADB
        
        assert not HAS_CHROMADB
        
        with pytest.raises(ChromaDBRequiredError) as excinfo:
            configurator = EmbeddingConfigurator()
            configurator.configure_embedder()
            
        assert "ChromaDB is required for embedding functionality" in str(excinfo.value)