test: align keyspace test with fakeredis fixture requirements

Use the standard  fixture to test against both RESP2 and RESP3 on FakeRedis.

Author: alexbthundiyil-spec

test/test_mixins/test_keyspace_notifications.py

@@ -13,18 +13,13 @@ def wait_for_message(pubsub: PubSub, timeout=0.5, ignore_subscribe_messages=Fals
         now = time.time()
     return None
 
-def test_keyspace_notifications():
-    import fakeredis
-    
-    server = fakeredis.FakeServer()
-    r = fakeredis.FakeRedis(server=server)
-    
+def test_keyspace_notifications(r):
     # Enable keyspace notifications in redis
     r.config_set('notify-keyspace-events', 'KEA')
 
     p = r.pubsub()
-    p.psubscribe('__keyspace@0__:*')
-    p.psubscribe('__keyevent@0__:*')
+    p.psubscribe('__keyspace@*__:*')
+    p.psubscribe('__keyevent@*__:*')
 
     msg1 = wait_for_message(p)
     msg2 = wait_for_message(p)
@@ -41,8 +36,9 @@ def test_keyspace_notifications():
 
     assert len(msgs) >= 2
 
-    keyspace_msg = next((m for m in msgs if m['channel'] == b'__keyspace@0__:foo'), None)
-    keyevent_msg = next((m for m in msgs if m['channel'] == b'__keyevent@0__:set'), None)
+    # We are using db=2 based on conftest.py
+    keyspace_msg = next((m for m in msgs if m['channel'] == b'__keyspace@2__:foo'), None)
+    keyevent_msg = next((m for m in msgs if m['channel'] == b'__keyevent@2__:set'), None)
 
     assert keyspace_msg is not None
     assert keyspace_msg['data'] == b'set'