1 files changed, 47 insertions, 0 deletions
diff --git a/tests/lock.py b/tests/lock.py
new file mode 100644
index 0000000..a6ec9e7
--- /dev/null
+++ b/tests/lock.py
@@ -0,0 +1,47 @@
+import redis
+import time
+import unittest
+from redis.client import Lock
+
+class LockTestCase(unittest.TestCase):
+    def setUp(self):
+        self.client = redis.Redis(host='localhost', port=6379, db=9)
+        self.client.flushdb()
+
+    def tearDown(self):
+        self.client.flushdb()
+
+    def test_lock(self):
+        lock = self.client.lock('foo')
+        self.assert_(lock.acquire())
+        self.assertEquals(self.client['foo'], str(Lock.LOCK_FOREVER))
+        lock.release()
+        self.assertEquals(self.client['foo'], None)
+
+    def test_competing_locks(self):
+        lock1 = self.client.lock('foo')
+        lock2 = self.client.lock('foo')
+        self.assert_(lock1.acquire())
+        self.assertFalse(lock2.acquire(blocking=False))
+        lock1.release()
+        self.assert_(lock2.acquire())
+        self.assertFalse(lock1.acquire(blocking=False))
+        lock2.release()
+
+    def test_timeouts(self):
+        lock1 = self.client.lock('foo')
+        lock2 = self.client.lock('foo')
+        self.assert_(lock1.acquire(timeout=1))
+        self.assertEquals(lock1.acquired_until, long(time.time()) + 1)
+        self.assertEquals(lock1.acquired_until, long(self.client['foo']))
+        self.assertFalse(lock2.acquire(blocking=False))
+        time.sleep(2) # need to wait up to 2 seconds for lock to timeout
+        self.assert_(lock2.acquire(blocking=False))
+        lock2.release()
+
+    def test_non_blocking(self):
+        lock1 = self.client.lock('foo')
+        self.assert_(lock1.acquire(blocking=False))
+        self.assert_(lock1.acquired_until)
+        lock1.release()
+        self.assert_(lock1.acquired_until is None)