update username completer for acl users command.

iredis/completers.py

@@ -257,6 +257,8 @@ def update_completer_for_response(self, command_name, args, response):
         # only update categoryname completer when `ACL CAT` without args.
         if command_name == "ACL CAT" and not args:
             self.catetoryname_completer.touch_words(response)
+        if command_name == "ACL USERS":
+            self.username_completer.touch_words(response)
 
     def _touch_members(self, items):
         _step = 1

tests/cli_tests/test_completer.py

@@ -1,3 +1,6 @@
+import pytest
+
+
 def test_integer_type_completer(cli):
     cli.expect("127.0.0.1")
     cli.send("BITFIELD meykey GET ")
@@ -32,3 +35,17 @@ def test_command_completion_when_space_command(cli, clean_redis):
 
     cli.send("command in")
     cli.expect("command info")
+
+
+@pytest.mark.skipif("int(os.environ['REDIS_VERSION']) < 6")
+def test_username_completer(cli, iredis_client):
+    iredis_client.execute("acl setuser", "foo1")
+    iredis_client.execute("acl setuser", "bar2")
+
+    cli.expect("127.0.0.1")
+    cli.sendline("acl users")
+    cli.expect("foo1")
+
+    cli.send("acl deluser ")
+    cli.expect("foo1")
+    cli.expect("bar2")

tests/unittests/test_completers.py

@@ -333,3 +333,14 @@ def test_username_completer():
         "antirez",
         "laixintao",
     ]
+
+
+def test_username_touch_for_response():
+    c = IRedisCompleter()
+    c.update_completer_for_response(
+        "acl   users", (), [b"hello", b"world"],
+    )
+    assert sorted(c.username_completer.words) == [
+        "hello",
+        "world",
+    ]