Merge pull request #224 from viljoviitanen/del

Add __delitem__
jamesls · Nov 8, 2018 · da948fd · da948fd
2 parents aa6a0d6 + 49f6f7a
commit da948fd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/fakeredis.py b/fakeredis.py
@@ -980,6 +980,9 @@ def delete(self, *names):
                 continue
         return deleted
 
+    def __delitem__(self, name):
+        return self.delete(name)
+
     @_locked
     def sort(self, name, start=None, num=None, by=None, get=None, desc=False,
              alpha=False, store=None):

diff --git a/test_fakeredis.py b/test_fakeredis.py
@@ -622,6 +622,11 @@ def test_setnx(self):
         self.assertEqual(self.redis.setnx('foo', 'baz'), False)
         self.assertEqual(self.redis.get('foo'), b'bar')
 
+    def test_del_operator(self):
+        self.redis['foo'] = 'bar'
+        del self.redis['foo']
+        self.assertEqual(self.redis.get('foo'), None)
+
     def test_delete(self):
         self.redis['foo'] = 'bar'
         self.assertEqual(self.redis.delete('foo'), True)