mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 23:58:34 +00:00
Fix logger caching test: Reset global state for consistent testing
Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
@@ -139,18 +139,25 @@ def test_concurrent_llm_calls():
|
||||
def test_logger_caching_performance():
|
||||
"""Test that logger instance is cached for performance."""
|
||||
from crewai.llm import suppress_litellm_output
|
||||
import crewai.llm
|
||||
|
||||
with patch('logging.getLogger') as mock_get_logger:
|
||||
mock_logger = MagicMock()
|
||||
mock_get_logger.return_value = mock_logger
|
||||
|
||||
with suppress_litellm_output():
|
||||
pass
|
||||
|
||||
with suppress_litellm_output():
|
||||
pass
|
||||
original_logger = crewai.llm._litellm_logger
|
||||
crewai.llm._litellm_logger = None
|
||||
|
||||
try:
|
||||
with patch('logging.getLogger') as mock_get_logger:
|
||||
mock_logger = MagicMock()
|
||||
mock_get_logger.return_value = mock_logger
|
||||
|
||||
mock_get_logger.assert_called_once_with("litellm")
|
||||
with suppress_litellm_output():
|
||||
pass
|
||||
|
||||
with suppress_litellm_output():
|
||||
pass
|
||||
|
||||
mock_get_logger.assert_called_once_with("litellm")
|
||||
finally:
|
||||
crewai.llm._litellm_logger = original_logger
|
||||
|
||||
|
||||
def test_suppression_error_handling():
|
||||
|
||||
Reference in New Issue
Block a user