LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 00:23:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: tests

Browse Source
This commit is contained in:
Alex
2024-10-15 11:52:33 +01:00
parent 39e94d4a5e
commit cca6297a64
3 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_cache.py
View File

@@ -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()