Merge pull request #619 from foxx/patch-1

Fixes #613
main
Selwin Ong 9 years ago
commit 961e443611

@ -1,2 +1,2 @@
redis>=2.7 redis>=2.7
click>=3.0.0 click<6.0

@ -22,9 +22,6 @@ PATCHED_METHODS = ['_setex', '_lrem', '_zadd', '_pipeline', '_ttl']
def patch_connection(connection): def patch_connection(connection):
if not isinstance(connection, StrictRedis):
raise ValueError('A StrictRedis or Redis connection is required.')
# Don't patch already patches objects # Don't patch already patches objects
if all([hasattr(connection, attr) for attr in PATCHED_METHODS]): if all([hasattr(connection, attr) for attr in PATCHED_METHODS]):
return connection return connection
@ -38,7 +35,8 @@ def patch_connection(connection):
if hasattr(connection, 'pttl'): if hasattr(connection, 'pttl'):
connection._pttl = fix_return_type(partial(StrictRedis.pttl, connection)) connection._pttl = fix_return_type(partial(StrictRedis.pttl, connection))
elif isinstance(connection, StrictRedis): # add support for mock redis objects
elif hasattr(connection, 'setex'):
connection._setex = connection.setex connection._setex = connection.setex
connection._lrem = connection.lrem connection._lrem = connection.lrem
connection._zadd = connection.zadd connection._zadd = connection.zadd

Loading…
Cancel
Save