11from unittest import mock
22
33import pytest
4- from aioredis import Redis
5- from aioredis.exceptions import RedisError
4+ from redis import asyncio as aioredis
65from {{package_name}}.app.utils import RedisClient
76from {{package_name}}.config import redis as redis_conf
87
@@ -18,14 +17,14 @@ mock.MagicMock.__await__ = lambda x: async_magic().__await__()
1817
1918def test_open_redis_client():
2019 RedisClient.open_redis_client()
21- assert isinstance(RedisClient.redis_client, Redis)
20+ assert isinstance(RedisClient.redis_client, aioredis. Redis)
2221 RedisClient.redis_client = None
2322
2423 redis_conf.REDIS_USERNAME = "John"
2524 redis_conf.REDIS_PASSWORD = "Secret"
2625 redis_conf.REDIS_USE_SENTINEL = True
2726 RedisClient.open_redis_client()
28- assert isinstance(RedisClient.redis_client, Redis)
27+ assert isinstance(RedisClient.redis_client, aioredis. Redis)
2928 RedisClient.redis_client = None
3029
3130
@@ -39,7 +38,7 @@ async def test_ping():
3938@pytest.mark.asyncio
4039async def test_ping_exception():
4140 RedisClient.redis_client = mock.MagicMock()
42- RedisClient.redis_client.ping.side_effect = RedisError("Mock error")
41+ RedisClient.redis_client.ping.side_effect = aioredis. RedisError("Mock error")
4342 assert await RedisClient.ping() is False
4443
4544
@@ -53,8 +52,8 @@ async def test_set():
5352@pytest.mark.asyncio
5453async def test_set_exception():
5554 RedisClient.redis_client = mock.MagicMock()
56- RedisClient.redis_client.set.side_effect = RedisError("Mock error")
57- with pytest.raises(RedisError):
55+ RedisClient.redis_client.set.side_effect = aioredis. RedisError("Mock error")
56+ with pytest.raises(aioredis. RedisError):
5857 await RedisClient.set("key", "value")
5958
6059
@@ -68,8 +67,8 @@ async def test_rpush():
6867@pytest.mark.asyncio
6968async def test_rpush_exception():
7069 RedisClient.redis_client = mock.MagicMock()
71- RedisClient.redis_client.rpush.side_effect = RedisError("Mock error")
72- with pytest.raises(RedisError):
70+ RedisClient.redis_client.rpush.side_effect = aioredis. RedisError("Mock error")
71+ with pytest.raises(aioredis. RedisError):
7372 await RedisClient.rpush("key", "value")
7473
7574
@@ -83,8 +82,8 @@ async def test_exists():
8382@pytest.mark.asyncio
8483async def test_exists_exception():
8584 RedisClient.redis_client = mock.MagicMock()
86- RedisClient.redis_client.exists.side_effect = RedisError("Mock error")
87- with pytest.raises(RedisError):
85+ RedisClient.redis_client.exists.side_effect = aioredis. RedisError("Mock error")
86+ with pytest.raises(aioredis. RedisError):
8887 await RedisClient.exists("key")
8988
9089
@@ -98,8 +97,8 @@ async def test_get():
9897@pytest.mark.asyncio
9998async def test_get_exception():
10099 RedisClient.redis_client = mock.MagicMock()
101- RedisClient.redis_client.get.side_effect = RedisError("Mock error")
102- with pytest.raises(RedisError):
100+ RedisClient.redis_client.get.side_effect = aioredis. RedisError("Mock error")
101+ with pytest.raises(aioredis. RedisError):
103102 await RedisClient.get("key")
104103
105104
@@ -113,6 +112,6 @@ async def test_lrange():
113112@pytest.mark.asyncio
114113async def test_lrange_exception():
115114 RedisClient.redis_client = mock.MagicMock()
116- RedisClient.redis_client.lrange.side_effect = RedisError("Mock error")
117- with pytest.raises(RedisError):
115+ RedisClient.redis_client.lrange.side_effect = aioredis. RedisError("Mock error")
116+ with pytest.raises(aioredis. RedisError):
118117 await RedisClient.lrange("key", 1, -1)
0 commit comments