This commit is contained in:
@@ -580,9 +580,9 @@ def redis_client():
|
||||
|
||||
|
||||
# Mock для Redis если он недоступен
|
||||
@pytest.fixture(autouse=True)
|
||||
@pytest.fixture
|
||||
def mock_redis_if_unavailable():
|
||||
"""Автоматически мокает Redis если он недоступен"""
|
||||
"""Мокает Redis если он недоступен - для тестов которые нуждаются в Redis"""
|
||||
try:
|
||||
import fakeredis.aioredis
|
||||
# Используем fakeredis для тестов
|
||||
@@ -609,6 +609,16 @@ def mock_redis_if_unavailable():
|
||||
mock_redis.ping = fake_redis.ping
|
||||
mock_redis.is_connected = True
|
||||
|
||||
# Добавляем async методы для connect/disconnect
|
||||
async def mock_connect():
|
||||
return True
|
||||
|
||||
async def mock_disconnect():
|
||||
pass
|
||||
|
||||
mock_redis.connect = mock_connect
|
||||
mock_redis.disconnect = mock_disconnect
|
||||
|
||||
yield
|
||||
except ImportError:
|
||||
# fakeredis не установлен, используем базовый mock
|
||||
@@ -622,6 +632,16 @@ def mock_redis_if_unavailable():
|
||||
mock_redis.ping.return_value = True
|
||||
mock_redis.is_connected = False
|
||||
|
||||
# Добавляем async методы для connect/disconnect
|
||||
async def mock_connect():
|
||||
return True
|
||||
|
||||
async def mock_disconnect():
|
||||
pass
|
||||
|
||||
mock_redis.connect = mock_connect
|
||||
mock_redis.disconnect = mock_disconnect
|
||||
|
||||
yield
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user