From cca6297a64543d31d6cefba7c313515ff59e4880 Mon Sep 17 00:00:00 2001 From: Alex Date: Tue, 15 Oct 2024 11:52:33 +0100 Subject: [PATCH] fix: tests --- tests/llm/test_anthropic.py | 4 ++-- tests/llm/test_sagemaker.py | 4 ++-- tests/test_cache.py | 8 ++++---- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/llm/test_anthropic.py b/tests/llm/test_anthropic.py index f4c84afa..689013c0 100644 --- a/tests/llm/test_anthropic.py +++ b/tests/llm/test_anthropic.py @@ -22,7 +22,7 @@ class TestAnthropicLLM(unittest.TestCase): mock_response = Mock() mock_response.completion = "test completion" - with patch("application.cache.make_redis") as mock_make_redis: + with patch("application.cache.get_redis_instance") as mock_make_redis: mock_redis_instance = mock_make_redis.return_value mock_redis_instance.get.return_value = None mock_redis_instance.set = Mock() @@ -47,7 +47,7 @@ class TestAnthropicLLM(unittest.TestCase): ] mock_responses = [Mock(completion="response_1"), Mock(completion="response_2")] - with patch("application.cache.make_redis") as mock_make_redis: + with patch("application.cache.get_redis_instance") as mock_make_redis: mock_redis_instance = mock_make_redis.return_value mock_redis_instance.get.return_value = None mock_redis_instance.set = Mock() diff --git a/tests/llm/test_sagemaker.py b/tests/llm/test_sagemaker.py index e45d4c0d..d659d498 100644 --- a/tests/llm/test_sagemaker.py +++ b/tests/llm/test_sagemaker.py @@ -52,7 +52,7 @@ class TestSagemakerAPILLM(unittest.TestCase): self.response['Body'].read.return_value.decode.return_value = json.dumps(self.result) def test_gen(self): - with patch('application.cache.make_redis') as mock_make_redis: + with patch('application.cache.get_redis_instance') as mock_make_redis: mock_redis_instance = mock_make_redis.return_value mock_redis_instance.get.return_value = None @@ -70,7 +70,7 @@ class TestSagemakerAPILLM(unittest.TestCase): mock_redis_instance.set.assert_called_once() def test_gen_stream(self): - with patch('application.cache.make_redis') as mock_make_redis: + with patch('application.cache.get_redis_instance') as mock_make_redis: mock_redis_instance = mock_make_redis.return_value mock_redis_instance.get.return_value = None diff --git a/tests/test_cache.py b/tests/test_cache.py index d38801bd..4270a181 100644 --- a/tests/test_cache.py +++ b/tests/test_cache.py @@ -27,7 +27,7 @@ def test_gen_cache_key_invalid_message_format(): assert str(context.exception) == "All messages must be dictionaries." # Test for gen_cache decorator -@patch('application.cache.make_redis') # Mock the Redis client +@patch('application.cache.get_redis_instance') # Mock the Redis client def test_gen_cache_hit(mock_make_redis): # Arrange mock_redis_instance = MagicMock() @@ -50,7 +50,7 @@ def test_gen_cache_hit(mock_make_redis): mock_redis_instance.set.assert_not_called() # Ensure the function result is not cached again -@patch('application.cache.make_redis') # Mock the Redis client +@patch('application.cache.get_redis_instance') # Mock the Redis client def test_gen_cache_miss(mock_make_redis): # Arrange mock_redis_instance = MagicMock() @@ -73,7 +73,7 @@ def test_gen_cache_miss(mock_make_redis): assert result == "new_result" mock_redis_instance.get.assert_called_once() -@patch('application.cache.make_redis') +@patch('application.cache.get_redis_instance') def test_stream_cache_hit(mock_make_redis): # Arrange mock_redis_instance = MagicMock() @@ -98,7 +98,7 @@ def test_stream_cache_hit(mock_make_redis): mock_redis_instance.set.assert_not_called() -@patch('application.cache.make_redis') +@patch('application.cache.get_redis_instance') def test_stream_cache_miss(mock_make_redis): # Arrange mock_redis_instance = MagicMock()