1 files changed, 97 insertions, 0 deletions
diff --git a/tests/unittests/command_parse/test_connection.py b/tests/unittests/command_parse/test_connection.py
new file mode 100644
index 0000000..8952e53
--- /dev/null
+++ b/tests/unittests/command_parse/test_connection.py
@@ -0,0 +1,97 @@
+def test_auth(judge_command):
+    judge_command("auth 123", {"command": "auth", "password": "123"})
+
+
+def test_auth_redis6(judge_command):
+    from iredis.commands import command2syntax
+    from iredis.redis_grammar import get_command_grammar
+
+    get_command_grammar.cache_clear()
+
+    old = command2syntax["AUTH"]
+    command2syntax["AUTH"] = "command_usernamex_password"
+    judge_command(
+        "auth default 123",
+        {"command": "auth", "password": "123", "username": "default"},
+    )
+    judge_command("AUTH 123", {"command": "AUTH", "password": "123"})
+    command2syntax["AUTH"] = old
+
+
+def test_echo(judge_command):
+    judge_command("echo hello", {"command": "echo", "message": "hello"})
+
+
+def test_ping(judge_command):
+    judge_command("ping hello", {"command": "ping", "message": "hello"})
+    judge_command("ping", {"command": "ping", "message": None})
+    judge_command("ping hello world", None)
+
+
+def test_select(judge_command):
+    for index in range(16):
+        judge_command(f"select {index}", {"command": "select", "index": str(index)})
+    for index in range(16, 100):
+        judge_command(f"select {index}", None)
+    judge_command("select acb", None)
+
+
+def test_swapdb(judge_command):
+    for index1 in range(16):
+        for index2 in range(16):
+            judge_command(
+                f"swapdb {index1} {index2}",
+                {"command": "swapdb", "index": [str(index1), str(index2)]},
+            )
+    judge_command("swapdb abc 1", None)
+    judge_command("swapdb 1", None)
+
+
+def test_client_caching(judge_command):
+    judge_command("CLIENT CACHING YES", {"command": "CLIENT CACHING", "yes": "YES"})
+    judge_command("CLIENT CACHING   NO", {"command": "CLIENT CACHING", "yes": "NO"})
+    judge_command("CLIENT CACHING", None)
+    judge_command("CLIENT CACHING abc", None)
+
+
+def test_client_tracking(judge_command):
+    judge_command("CLIENT TRACKING on", {"command": "CLIENT TRACKING", "on_off": "on"})
+    judge_command(
+        "CLIENT TRACKING ON REDIRECT 123",
+        {
+            "command": "CLIENT TRACKING",
+            "on_off": "ON",
+            "redirect_const": "REDIRECT",
+            "clientid": "123",
+        },
+    )
+    judge_command(
+        "CLIENT TRACKING ON PREFIX foo",
+        {
+            "command": "CLIENT TRACKING",
+            "on_off": "ON",
+            "prefix_const": "PREFIX",
+            "prefix": "foo",
+        },
+    )
+    judge_command(
+        "CLIENT TRACKING ON PREFIX foo",
+        {
+            "command": "CLIENT TRACKING",
+            "on_off": "ON",
+            "prefix_const": "PREFIX",
+            "prefix": "foo",
+        },
+    )
+    judge_command(
+        "CLIENT TRACKING ON PREFIX foo BCAST NOLOOP OPTIN",
+        {
+            "command": "CLIENT TRACKING",
+            "on_off": "ON",
+            "prefix_const": "PREFIX",
+            "prefix": "foo",
+            "bcast_const": "BCAST",
+            "noloop_const": "NOLOOP",
+            "optin_const": "OPTIN",
+        },
+    )