Handle close for older versions of redis-py

jamesls · Jul 7, 2020 · d25a3a8 · d25a3a8
1 parent 440bc17
commit d25a3a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_fakeredis.py b/test/test_fakeredis.py
@@ -110,7 +110,8 @@ def r(request, create_redis):
     yield r
     if connected:
         r.flushall()
-    r.close()
+    if hasattr(r, 'close'):
+        r.close()     # Older versions of redis-py don't have this method
 
 
 def test_large_command(r):