Merge pull request #718 from iorisa/refactor/unittest

refactor: test_redis.py

tests/metagpt/utils/test_redis.py

@@ -5,29 +5,25 @@
 @Author  : mashenquan
 @File    : test_redis.py
 """
+from unittest.mock import AsyncMock
 
-import mock
 import pytest
 
 from metagpt.config import CONFIG
 from metagpt.utils.redis import Redis
 
 
 async def async_mock_from_url(*args, **kwargs):
-    mock_client = mock.AsyncMock()
+    mock_client = AsyncMock()
     mock_client.set.return_value = None
     mock_client.get.side_effect = [b"test", b""]
     return mock_client
 
 
 @pytest.mark.asyncio
-@mock.patch("aioredis.from_url", return_value=async_mock_from_url())
-async def test_redis(mock_from_url):
+async def test_redis(mocker):
     # Mock
-    # mock_client = mock.AsyncMock()
-    # mock_client.set.return_value=None
-    # mock_client.get.side_effect = [b'test', b'']
-    # mock_from_url.return_value = mock_client
+    mocker.patch("aioredis.from_url", return_value=async_mock_from_url())
 
     # Prerequisites
     CONFIG.REDIS_HOST = "MOCK_REDIS_HOST"