test init args

cunla · Oct 29, 2024 · e692878 · e692878
1 parent e2b2aaa
commit e692878
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/test/test_internals/test_init_args.py b/test/test_internals/test_init_args.py
@@ -28,18 +28,40 @@ def test_host_init_arg(self):
         db.set("foo", "bar")
         assert db.get("foo") == b"bar"
 
+    def test_with_user_password(self):
+        username = "fakeredis-user"
+        password = "fakeredis-password"
+        db = fakeredis.FakeStrictRedis(host="localhost")
+        db.acl_setuser(username, enabled=True, passwords=[f"+{password}"], commands=["+set", "+get"])
+
+        db = fakeredis.FakeStrictRedis(host="localhost", username=username, password=password)
+        db.set("foo", "bar")
+        assert db.get("foo") == b"bar"
+
     def test_from_url(self):
         db = fakeredis.FakeStrictRedis.from_url("redis://localhost:6390/0")
         db.set("foo", "bar")
         assert db.get("foo") == b"bar"
 
     def test_from_url_user(self):
-        db = fakeredis.FakeStrictRedis.from_url("redis://user@localhost:6390/0")
+        username = "fakeredis-user"
+        password = "fakeredis-password"
+        db = fakeredis.FakeStrictRedis(host="localhost", port=6390, db=0)
+        db.acl_setuser(username, enabled=True, nopass=True, commands=["+set", "+get"])
+
+        db = fakeredis.FakeStrictRedis.from_url(f"redis://{username}@localhost:6390/0")
         db.set("foo", "bar")
         assert db.get("foo") == b"bar"
 
     def test_from_url_user_password(self):
-        db = fakeredis.FakeStrictRedis.from_url("redis://user:password@localhost:6390/0")
+        username = "fakeredis-user"
+        password = "fakeredis-password"
+        db = fakeredis.FakeStrictRedis(host="localhost", server=fakeredis.FakeServer())
+        db.acl_setuser(username, enabled=True, passwords=[f"+{password}"], commands=["+set", "+get"])
+
+        db = fakeredis.FakeStrictRedis.from_url(
+            f"redis://{username}:{password}@localhost:6390/0", server=fakeredis.FakeServer()
+        )
         db.set("foo", "bar")
         assert db.get("foo") == b"bar"