apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: unit test

Browse source
This commit is contained in:
莘权 马 2024-01-08 18:02:24 +08:00
parent ef5304961e
commit 360f5dcd1b
1 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/metagpt/utils/test_redis.py
View file

@ -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"