Merge pull request #933 from yozel/master

Add support for UNLINK command

2 files changed, 19 insertions, 0 deletions

diff --git a/redis/client.py b/redis/client.py
index f46c7cf..f7b4ca1 100755
--- a/redis/client.py
+++ b/redis/client.py
@@ -1358,6 +1358,10 @@ class StrictRedis(object):
         warnings.warn(
             DeprecationWarning('Call UNWATCH from a Pipeline object'))
 
+    def unlink(self, *names):
+        "Unlink one or more keys specified by ``names``"
+        return self.execute_command('UNLINK', *names)
+
     # LIST COMMANDS
     def blpop(self, keys, timeout=0):
         """
diff --git a/tests/test_commands.py b/tests/test_commands.py
index 1d0e42e..bba29b7 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -303,6 +303,21 @@ class TestRedisCommands(object):
         del r['a']
         assert r.get('a') is None
 
+    @skip_if_server_version_lt('4.0.0')
+    def test_unlink(self, r):
+        assert r.unlink('a') == 0
+        r['a'] = 'foo'
+        assert r.unlink('a') == 1
+        assert r.get('a') is None
+
+    @skip_if_server_version_lt('4.0.0')
+    def test_unlink_with_multiple_keys(self, r):
+        r['a'] = 'foo'
+        r['b'] = 'bar'
+        assert r.unlink('a', 'b') == 2
+        assert r.get('a') is None
+        assert r.get('b') is None
+
     @skip_if_server_version_lt('2.6.0')
     def test_dump_and_restore(self, r):
         r['a'] = 'foo'