Merge pull request #101 from locationlabs/feature/add-dbsize

Adds dbsize() function.  Closes issue #100

Author: narenst

mockredis/client.py

@@ -255,6 +255,9 @@ def _rename(self, old_key, new_key, nx=False):
             return True
         return False
 
+    def dbsize(self):
+        return len(self.redis.keys())
+
     # String Functions #
 
     def get(self, key):

mockredis/tests/test_redis.py

@@ -246,3 +246,9 @@ def test_renamenx(self):
         eq_(b"bar2", self.redis.get("foo2"))
         eq_(self.redis.renamenx("foo", "foo3"), 1)
         eq_(b"bar", self.redis.get("foo3"))
+
+    def test_dbsize(self):
+        self.redis["foo"] = "bar"
+        eq_(1, self.redis.dbsize())
+        del self.redis["foo"]
+        eq_(0, self.redis.dbsize())