github/crewAI
1
1
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 22:48:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: properly test OpenMemory import error scenario

Browse Source 
Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-12-07 16:44:08 +00:00
parent 2a62ac0a23
commit 5ec39a70a9
1 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
lib/crewai/tests/storage/test_openmemory_storage.py
View File

@@ -204,12 +204,25 @@ def test_openmemory_storage_different_memory_types():
def test_openmemory_import_error(): def test_openmemory_import_error():
with patch.dict("sys.modules", {"openmemory": None}): import sys
with patch( import importlib
"crewai.memory.storage.openmemory_storage.OpenMemory",
side_effect=ImportError("No module named 'openmemory'"), original_modules = sys.modules.copy()
): sys.modules["openmemory"] = None # type: ignore[assignment]
pass
try:
if "crewai.memory.storage.openmemory_storage" in sys.modules:
del sys.modules["crewai.memory.storage.openmemory_storage"]
with pytest.raises(ImportError, match="OpenMemory is not installed"):
from crewai.memory.storage import openmemory_storage
importlib.reload(openmemory_storage)
openmemory_storage.OpenMemoryStorage(
type="external", config={"path": "/tmp/test.sqlite"}
)
finally:
sys.modules.update(original_modules)
def test_external_memory_openmemory_provider(): def test_external_memory_openmemory_provider():