Update a test to work with redis-py 4.0

bmerry · bmerry · commit bdccdd921fdd · 2021-10-25T14:27:51.000+02:00
diff --git a/test/test_fakeredis.py b/test/test_fakeredis.py
@@ -5008,10 +5008,10 @@ def test_from_url_with_db_arg(self):
         assert db2.get('foo') == b'foo2'
 
     def test_from_url_db_value_error(self):
-        # In ValueError, should default to 0
+        # In case of ValueError, should default to 0, or be absent in redis-py 4.0
         db = fakeredis.FakeStrictRedis.from_url(
             'redis://localhost:6379/a')
-        assert db.connection_pool.connection_kwargs['db'] == 0
+        assert db.connection_pool.connection_kwargs.get('db', 0) == 0
 
     def test_can_pass_through_extra_args(self):
         db = fakeredis.FakeStrictRedis.from_url(
