90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
|
import pytest
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
from pr.core.enhanced_assistant import EnhancedAssistant
|
||
|
|
|
||
|
|
def test_enhanced_assistant_init():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assert assistant.base == mock_base
|
||
|
|
assert assistant.current_conversation_id is not None
|
||
|
|
|
||
|
|
def test_enhanced_call_api_with_cache():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
mock_base.model = 'test-model'
|
||
|
|
mock_base.api_url = 'http://test'
|
||
|
|
mock_base.api_key = 'key'
|
||
|
|
mock_base.use_tools = False
|
||
|
|
mock_base.verbose = False
|
||
|
|
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.api_cache = MagicMock()
|
||
|
|
assistant.api_cache.get.return_value = {'cached': True}
|
||
|
|
|
||
|
|
result = assistant.enhanced_call_api([{'role': 'user', 'content': 'test'}])
|
||
|
|
assert result == {'cached': True}
|
||
|
|
assistant.api_cache.get.assert_called_once()
|
||
|
|
|
||
|
|
def test_enhanced_call_api_without_cache():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
mock_base.model = 'test-model'
|
||
|
|
mock_base.api_url = 'http://test'
|
||
|
|
mock_base.api_key = 'key'
|
||
|
|
mock_base.use_tools = False
|
||
|
|
mock_base.verbose = False
|
||
|
|
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.api_cache = None
|
||
|
|
|
||
|
|
# It will try to call API and fail with network error, but that's expected
|
||
|
|
result = assistant.enhanced_call_api([{'role': 'user', 'content': 'test'}])
|
||
|
|
assert 'error' in result
|
||
|
|
|
||
|
|
def test_execute_workflow_not_found():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.workflow_storage = MagicMock()
|
||
|
|
assistant.workflow_storage.load_workflow_by_name.return_value = None
|
||
|
|
|
||
|
|
result = assistant.execute_workflow('nonexistent')
|
||
|
|
assert 'error' in result
|
||
|
|
|
||
|
|
def test_create_agent():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.agent_manager = MagicMock()
|
||
|
|
assistant.agent_manager.create_agent.return_value = 'agent_id'
|
||
|
|
|
||
|
|
result = assistant.create_agent('role')
|
||
|
|
assert result == 'agent_id'
|
||
|
|
|
||
|
|
def test_search_knowledge():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.knowledge_store = MagicMock()
|
||
|
|
assistant.knowledge_store.search_entries.return_value = [{'result': True}]
|
||
|
|
|
||
|
|
result = assistant.search_knowledge('query')
|
||
|
|
assert result == [{'result': True}]
|
||
|
|
|
||
|
|
def test_get_cache_statistics():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.api_cache = MagicMock()
|
||
|
|
assistant.api_cache.get_statistics.return_value = {'hits': 10}
|
||
|
|
assistant.tool_cache = MagicMock()
|
||
|
|
assistant.tool_cache.get_statistics.return_value = {'misses': 5}
|
||
|
|
|
||
|
|
stats = assistant.get_cache_statistics()
|
||
|
|
assert 'api_cache' in stats
|
||
|
|
assert 'tool_cache' in stats
|
||
|
|
|
||
|
|
def test_clear_caches():
|
||
|
|
mock_base = MagicMock()
|
||
|
|
assistant = EnhancedAssistant(mock_base)
|
||
|
|
assistant.api_cache = MagicMock()
|
||
|
|
assistant.tool_cache = MagicMock()
|
||
|
|
|
||
|
|
assistant.clear_caches()
|
||
|
|
assistant.api_cache.clear_all.assert_called_once()
|
||
|
|
assistant.tool_cache.clear_all.assert_called_once()
|