andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-25 16:18:13 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

refactor: consolidate tests as per PR feedback

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-14 12:50:55 +00:00
parent e705da90e8
commit e37bad4e45
1 changed files with 6 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
tests/storage/test_mem0_storage.py
View File

@@ -29,7 +29,7 @@ def mem0_storage_with_mocked_config(mock_mem0_memory):
"""Fixture to create a Mem0Storage instance with mocked dependencies""" """Fixture to create a Mem0Storage instance with mocked dependencies"""
# Patch the Memory class to return our mock # Patch the Memory class to return our mock
with patch("mem0.memory.main.Memory.from_config", return_value=mock_mem0_memory): with patch("mem0.memory.main.Memory.from_config", return_value=mock_mem0_memory) as mock_from_config:
config = { config = {
"vector_store": { "vector_store": {
"provider": "mock_vector_store", "provider": "mock_vector_store",
@@ -66,13 +66,15 @@ def mem0_storage_with_mocked_config(mock_mem0_memory):
) )
mem0_storage = Mem0Storage(type="short_term", crew=crew) mem0_storage = Mem0Storage(type="short_term", crew=crew)
return mem0_storage return mem0_storage, mock_from_config, config
def test_mem0_storage_initialization(mem0_storage_with_mocked_config, mock_mem0_memory): def test_mem0_storage_initialization(mem0_storage_with_mocked_config, mock_mem0_memory):
"""Test that Mem0Storage initializes correctly with the mocked config""" """Test that Mem0Storage initializes correctly with the mocked config"""
assert mem0_storage_with_mocked_config.memory_type == "short_term" mem0_storage, mock_from_config, config = mem0_storage_with_mocked_config
assert mem0_storage_with_mocked_config.memory is mock_mem0_memory assert mem0_storage.memory_type == "short_term"
assert mem0_storage.memory is mock_mem0_memory
mock_from_config.assert_called_once_with(config)
@pytest.fixture @pytest.fixture
@@ -153,31 +155,3 @@ def test_mem0_storage_with_explict_config(
mem0_storage_with_memory_client_using_explictly_config.memory_config mem0_storage_with_memory_client_using_explictly_config.memory_config
== expected_config == expected_config
) )
@pytest.fixture
def mem0_storage_with_local_config(mock_mem0_memory):
"""Fixture to create a Mem0Storage instance with local mem0 config"""
# Patch the Memory class to return our mock
with patch("mem0.memory.main.Memory.from_config", return_value=mock_mem0_memory) as mock_from_config:
local_config = {
"vector_store": {"provider": "mock_vector_store"},
"llm": {"provider": "mock_llm"},
"embedder": {"provider": "mock_embedder"},
}
crew = MockCrew(
memory_config={
"provider": "mem0",
"config": {"user_id": "test_user", "local_mem0_config": local_config},
}
)
mem0_storage = Mem0Storage(type="short_term", crew=crew)
return mem0_storage, mock_from_config, local_config
def test_mem0_storage_uses_local_config(mem0_storage_with_local_config):
"""Test that Mem0Storage correctly uses local_mem0_config when initializing Memory"""
_, mock_from_config, local_config = mem0_storage_with_local_config
mock_from_config.assert_called_once_with(local_config)