Update lock methods to match current redis-py

mockredis/client.py

@@ -74,9 +74,14 @@ def ping(self):
 
     # Transactions Functions #
 
-    def lock(self, key, timeout=0, sleep=0):
+    def lock(self, name, timeout=None, sleep=0.1, blocking_timeout=None,
+             lock_class=None, thread_local=True):
         """Emulate lock."""
-        return MockRedisLock(self, key, timeout, sleep)
+        if lock_class is None:
+            lock_class = MockRedisLock
+        return lock_class(self, name, timeout=timeout, sleep=sleep,
+                          blocking_timeout=blocking_timeout,
+                          thread_local=thread_local)
 
     def pipeline(self, transaction=True, shard_hint=None):
         """Emulate a redis-python pipeline."""

mockredis/lock.py

@@ -4,14 +4,18 @@ class MockRedisLock(object):
     to allow testing without a real redis server.
     """
 
-    def __init__(self, redis, name, timeout=None, sleep=0.1):
+    def __init__(self, redis, name, timeout=None, sleep=0.1,
+                 blocking=True, blocking_timeout=None, thread_local=True):
         """Initialize the object."""
 
         self.redis = redis
         self.name = name
         self.acquired_until = None
         self.timeout = timeout
         self.sleep = sleep
+        self.blocking = blocking
+        self.blocking_timeout = blocking_timeout
+        self.thread_local = bool(thread_local)
 
     def acquire(self, blocking=True):  # pylint: disable=R0201,W0613
         """Emulate acquire."""
@@ -23,8 +27,14 @@ def release(self):   # pylint: disable=R0201
 
         return
 
+    def extend(self, additional_time):  # pylint: disable=R0201,W0613
+        """Emulate extend."""
+
+        return
+
     def __enter__(self):
-        return self.acquire()
+        self.acquire(blocking=True)
+        return self
 
     def __exit__(self, exc_type, exc_value, traceback):
         self.release()