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

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

@ -64,15 +64,25 @@ 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
result = await get_conversation_history(session_id="test_session") from cognee.modules.retrieval.utils.session_cache import (
get_conversation_history,
)
assert "Previous conversation:" in result result = await get_conversation_history(session_id="test_session")
assert "[2024-01-15 10:30:45]" in result
assert "QUESTION: What is AI?" in result assert "Previous conversation:" in result
assert "CONTEXT: AI is artificial intelligence" in result assert "[2024-01-15 10:30:45]" in result
assert "ANSWER: AI stands for Artificial Intelligence" in result assert "QUESTION: What is AI?" in result
assert "CONTEXT: AI is artificial intelligence" in result
assert "ANSWER: AI stands for Artificial Intelligence" in result
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_save_to_session_cache_saves_correctly(self): async def test_save_to_session_cache_saves_correctly(self):
@ -86,23 +96,35 @@ 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
result = await save_to_session_cache( from cognee.modules.retrieval.utils.session_cache import (
query="What is Python?", save_to_session_cache,
context_summary="Python is a programming language", )
answer="Python is a high-level programming language",
session_id="my_session",
)
assert result is True result = await save_to_session_cache(
mock_cache.add_qa.assert_called_once() query="What is Python?",
context_summary="Python is a programming language",
answer="Python is a high-level programming language",
session_id="my_session",
)
call_kwargs = mock_cache.add_qa.call_args.kwargs assert result is True
assert call_kwargs["question"] == "What is Python?" mock_cache.add_qa.assert_called_once()
assert call_kwargs["context"] == "Python is a programming language"
assert call_kwargs["answer"] == "Python is a high-level programming language" call_kwargs = mock_cache.add_qa.call_args.kwargs
assert call_kwargs["session_id"] == "my_session" assert call_kwargs["question"] == "What is Python?"
assert call_kwargs["context"] == "Python is a programming language"
assert (
call_kwargs["answer"] == "Python is a high-level programming language"
)
assert call_kwargs["session_id"] == "my_session"
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_save_to_session_cache_uses_default_session_when_none(self): async def test_save_to_session_cache_uses_default_session_when_none(self):
@ -116,15 +138,25 @@ 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
result = await save_to_session_cache( from cognee.modules.retrieval.utils.session_cache import (
query="Test question", save_to_session_cache,
context_summary="Test context", )
answer="Test answer",
session_id=None,
)
assert result is True result = await save_to_session_cache(
call_kwargs = mock_cache.add_qa.call_args.kwargs query="Test question",
assert call_kwargs["session_id"] == "default_session" context_summary="Test context",
answer="Test answer",
session_id=None,
)
assert result is True
call_kwargs = mock_cache.add_qa.call_args.kwargs
assert call_kwargs["session_id"] == "default_session"