flushdb in MockRedis, fix #23

Author: samuelcolvin

arq/testing.py

@@ -79,6 +79,12 @@ def _get(self, key, default=None):
     async def llen(self, list_name):
         return len(self._get(list_name, []))
 
+    async def flushdb(self):
+        self.data = {}
+
+    async def set(self, key, value):
+        self.data[key] = value
+
     async def setex(self, key, expires, value):
         self.data[key] = value
         self._expiry[key] = datetime.now() + timedelta(seconds=expires)

tests/test_utils.py

@@ -41,3 +41,11 @@ async def test_mock_redis_expiry_expired(loop):
     r = MockRedis(loop=loop)
     await r.setex('foo', -10, 'bar')
     assert None is await r.get('foo')
+
+
+async def test_mock_redis_flushdb(loop):
+    r = MockRedis(loop=loop)
+    await r.set('foo', 'bar')
+    assert 'bar' == await r.get('foo')
+    await r.flushdb()
+    assert None is await r.get('foo')