From 360f5dcd1b69b5cd16e7d6dd6839bdf9b63f6a24 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Mon, 8 Jan 2024 18:02:24 +0800 Subject: [PATCH] refactor: unit test --- tests/metagpt/utils/test_redis.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/tests/metagpt/utils/test_redis.py b/tests/metagpt/utils/test_redis.py index d499418ac..140c04f6b 100644 --- a/tests/metagpt/utils/test_redis.py +++ b/tests/metagpt/utils/test_redis.py @@ -5,8 +5,8 @@ @Author : mashenquan @File : test_redis.py """ +from unittest.mock import AsyncMock -import mock import pytest from metagpt.config import CONFIG @@ -14,20 +14,16 @@ 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"