Skip to content

Commit bbed2f2

Browse files
committed
Enable testing with FakeConnection
1 parent 0e11e7d commit bbed2f2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

fakeredis/aioredis.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import async_timeout
1313
import redis.asyncio as redis_async # aioredis was integrated into redis in version 4.2.0 as redis.asyncio
1414

15-
from . import _fakesocket
15+
from . import _fakesocket, FakeServer
1616
from . import _helpers
1717
from . import _msgs as msgs
1818
from . import _server
@@ -100,7 +100,9 @@ def writelines(self, data):
100100

101101
class FakeConnection(redis_async.Connection):
102102
def __init__(self, *args, **kwargs):
103-
self._server = kwargs.pop('server')
103+
self._server = kwargs.pop('server', None)
104+
if self._server is None:
105+
self._server = FakeServer()
104106
self._sock = None
105107
super().__init__(*args, **kwargs)
106108

0 commit comments

Comments
 (0)