gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test fix

Browse source
This commit is contained in:
hajdul88 2025-10-17 10:25:26 +02:00
parent 47cce90112
commit 339de5a0b8
1 changed files with 63 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
cognee/tests/unit/modules/retrieval/conversation_history_test.py
View file

@ -64,7 +64,17 @@ class TestConversationHistoryUtils:
"cognee.infrastructure.databases.cache.get_cache_engine.get_cache_engine", "cognee.infrastructure.databases.cache.get_cache_engine.get_cache_engine",
return_value=mock_cache, return_value=mock_cache,
): ):
from cognee.modules.retrieval.utils.session_cache import get_conversation_history with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
# Enable caching
mock_config = MagicMock()
mock_config.caching = True
MockCacheConfig.return_value = mock_config
from cognee.modules.retrieval.utils.session_cache import (
get_conversation_history,
)
result = await get_conversation_history(session_id="test_session") result = await get_conversation_history(session_id="test_session")
@ -86,7 +96,17 @@ class TestConversationHistoryUtils:
"cognee.infrastructure.databases.cache.get_cache_engine.get_cache_engine", "cognee.infrastructure.databases.cache.get_cache_engine.get_cache_engine",
return_value=mock_cache, return_value=mock_cache,
): ):
from cognee.modules.retrieval.utils.session_cache import save_to_session_cache with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
# Enable caching
mock_config = MagicMock()
mock_config.caching = True
MockCacheConfig.return_value = mock_config
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
)
result = await save_to_session_cache( result = await save_to_session_cache(
query="What is Python?", query="What is Python?",
@ -101,7 +121,9 @@ class TestConversationHistoryUtils:
call_kwargs = mock_cache.add_qa.call_args.kwargs call_kwargs = mock_cache.add_qa.call_args.kwargs
assert call_kwargs["question"] == "What is Python?" assert call_kwargs["question"] == "What is Python?"
assert call_kwargs["context"] == "Python is a programming language" assert call_kwargs["context"] == "Python is a programming language"
assert call_kwargs["answer"] == "Python is a high-level programming language" assert (
call_kwargs["answer"] == "Python is a high-level programming language"
)
assert call_kwargs["session_id"] == "my_session" assert call_kwargs["session_id"] == "my_session"
@pytest.mark.asyncio @pytest.mark.asyncio
@ -116,7 +138,17 @@ class TestConversationHistoryUtils:
"cognee.infrastructure.databases.cache.get_cache_engine.get_cache_engine", "cognee.infrastructure.databases.cache.get_cache_engine.get_cache_engine",
return_value=mock_cache, return_value=mock_cache,
): ):
from cognee.modules.retrieval.utils.session_cache import save_to_session_cache with patch(
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
# Enable caching
mock_config = MagicMock()
mock_config.caching = True
MockCacheConfig.return_value = mock_config
from cognee.modules.retrieval.utils.session_cache import (
save_to_session_cache,
)
result = await save_to_session_cache( result = await save_to_session_cache(
query="Test question", query="Test question",