summaryrefslogtreecommitdiffstats
path: root/tests/unittests/command_parse/test_generic_parse.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unittests/command_parse/test_generic_parse.py')
-rw-r--r--tests/unittests/command_parse/test_generic_parse.py218
1 files changed, 218 insertions, 0 deletions
diff --git a/tests/unittests/command_parse/test_generic_parse.py b/tests/unittests/command_parse/test_generic_parse.py
new file mode 100644
index 0000000..6c4b0f3
--- /dev/null
+++ b/tests/unittests/command_parse/test_generic_parse.py
@@ -0,0 +1,218 @@
+def test_del(judge_command):
+ judge_command("DEL abc", {"command": "DEL", "keys": "abc"})
+ judge_command("DEL bc1", {"command": "DEL", "keys": "bc1"})
+ judge_command("DEL 1", {"command": "DEL", "keys": "1"})
+ judge_command('DEL "hello world"', {"command": "DEL", "keys": '"hello world"'})
+ judge_command(
+ r'DEL "hell\"o world"', {"command": "DEL", "keys": r'"hell\"o world"'}
+ )
+ judge_command("DEL abc def", {"command": "DEL", "keys": "abc def"})
+ judge_command("DEL 1 2", {"command": "DEL", "keys": "1 2"})
+ judge_command("DEL 'he \"llo'", {"command": "DEL", "keys": "'he \"llo'"})
+ judge_command(
+ """DEL 'he "llo' "abc" 'def' """,
+ {"command": "DEL", "keys": "'he \"llo' \"abc\" 'def'"},
+ )
+
+
+def test_exists(judge_command):
+ judge_command("EXISTS foo bar", {"command": "EXISTS", "keys": "foo bar"})
+ judge_command("EXISTS foo", {"command": "EXISTS", "keys": "foo"})
+ judge_command("EXISTS 1", {"command": "EXISTS", "keys": "1"})
+ judge_command('EXISTS "foo bar"', {"command": "EXISTS", "keys": '"foo bar"'})
+ judge_command(r'EXISTS "\""', {"command": "EXISTS", "keys": r'"\""'})
+
+
+def test_expire(judge_command):
+ judge_command("EXPIRE key 12", {"command": "EXPIRE", "key": "key", "second": "12"})
+ judge_command("EXPIRE key a12", None)
+ judge_command("EXPIRE 12 12", {"command": "EXPIRE", "key": "12", "second": "12"})
+ judge_command("EXPIRE 12", None)
+
+
+def test_expireat(judge_command):
+ judge_command(
+ "EXPIRE key 1565787643",
+ {"command": "EXPIRE", "key": "key", "second": "1565787643"},
+ )
+ judge_command("EXPIRE key a12", None)
+
+
+def test_keys(judge_command):
+ judge_command("KEYS *", {"command": "KEYS", "pattern": "*"})
+ judge_command("KEYS *abc", {"command": "KEYS", "pattern": "*abc"})
+ judge_command("keys abc*", {"command": "keys", "pattern": "abc*"})
+
+
+def test_move(judge_command):
+ judge_command("MOVE key 14", {"command": "MOVE", "key": "key", "index": "14"})
+
+
+def test_pexpire(judge_command):
+ judge_command(
+ "PEXPIRE key 12", {"command": "PEXPIRE", "key": "key", "millisecond": "12"}
+ )
+ judge_command("PEXPIRE key a12", None)
+ judge_command(
+ "PEXPIRE 12 12", {"command": "PEXPIRE", "key": "12", "millisecond": "12"}
+ )
+ judge_command("PEXPIRE 12", None)
+
+
+def test_pexpireat(judge_command):
+ judge_command(
+ "PEXPIREAT key 1565787643",
+ {"command": "PEXPIREAT", "key": "key", "timestampms": "1565787643"},
+ )
+ judge_command("PEXPIREAT key a12", None)
+
+
+def test_rename(judge_command):
+ judge_command(
+ "rename key newkey", {"command": "rename", "key": "key", "newkey": "newkey"}
+ )
+ judge_command(
+ "rename 123 newkey", {"command": "rename", "key": "123", "newkey": "newkey"}
+ )
+ judge_command("rename 123 ", None)
+
+
+def test_scan(judge_command):
+ judge_command(
+ "SCAN 0 MATCH task* COUNT 15 TYPE string",
+ {
+ "command": "SCAN",
+ "cursor": "0",
+ "match": "MATCH",
+ "pattern": "task*",
+ "count_const": "COUNT",
+ "count": "15",
+ "type_const": "TYPE",
+ "type": "string",
+ },
+ )
+ judge_command("SCAN 0", {"command": "SCAN", "cursor": "0"})
+ judge_command(
+ "SCAN 0 MATCH task*",
+ {"command": "SCAN", "cursor": "0", "match": "MATCH", "pattern": "task*"},
+ )
+ judge_command(
+ "SCAN 0 COUNT 15 TYPE string",
+ {
+ "command": "SCAN",
+ "cursor": "0",
+ "count_const": "COUNT",
+ "count": "15",
+ "type_const": "TYPE",
+ "type": "string",
+ },
+ )
+
+
+def test_migrate(judge_command):
+ judge_command(
+ 'MIGRATE 192.168.1.34 6379 " " 0 5000 KEYS key1 key2 key3',
+ {
+ "command": "MIGRATE",
+ "host": "192.168.1.34",
+ "port": "6379",
+ "key": '" "',
+ "index": "0",
+ "timeout": "5000",
+ "const_keys": "KEYS",
+ "keys": "key1 key2 key3",
+ },
+ )
+ judge_command(
+ "MIGRATE 192.168.1.34 6379 foo 0 5000 auth password1 KEYS key1 key2 key3",
+ {
+ "command": "MIGRATE",
+ "host": "192.168.1.34",
+ "port": "6379",
+ "key": "foo",
+ "index": "0",
+ "timeout": "5000",
+ "const_keys": "KEYS",
+ "keys": "key1 key2 key3",
+ "auth": "auth",
+ "password": "password1",
+ },
+ )
+ judge_command(
+ "MIGRATE 192.168.1.34 6379 foo 0 5000 auth username1 password1 KEYS key1 key2 key3",
+ {
+ "command": "MIGRATE",
+ "host": "192.168.1.34",
+ "port": "6379",
+ "key": "foo",
+ "index": "0",
+ "timeout": "5000",
+ "const_keys": "KEYS",
+ "keys": "key1 key2 key3",
+ "auth": "auth",
+ "password": "password1",
+ "username": "username1",
+ },
+ )
+
+
+def test_object(judge_command):
+ judge_command(
+ "object refcount mylist",
+ {"command": "object", "object": "refcount", "key": "mylist"},
+ )
+
+
+def test_wait(judge_command):
+ judge_command("WAIT 3 100", {"command": "WAIT", "count": "3", "timeout": "100"})
+
+
+def test_restore(judge_command):
+ judge_command(
+ 'RESTORE mykey 0 "\n\x17\x17\x00\x00\x00\x12\x00\x00\x00\x03\x00\x00\xc0\x01\x00\x04\xc0\x02\x00\x04\xc0\x03\x00\xff\x04\x00u#<\xc0;.\xe9\xdd"', # noqa
+ {
+ "command": "RESTORE",
+ "key": "mykey",
+ "timeout": "0",
+ "value": '"\n\x17\x17\x00\x00\x00\x12\x00\x00\x00\x03\x00\x00\xc0\x01\x00\x04\xc0\x02\x00\x04\xc0\x03\x00\xff\x04\x00u#<\xc0;.\xe9\xdd"', # noqa
+ },
+ )
+
+
+def test_copy(judge_command):
+ judge_command(
+ "COPY foo bar DB 3 REPLACE",
+ {
+ "command": "COPY",
+ "key": ["foo", "bar"],
+ "db_const": "DB",
+ "index": "3",
+ "replace_const": "REPLACE",
+ },
+ )
+ judge_command(
+ "COPY foo bar REPLACE",
+ {"command": "COPY", "key": ["foo", "bar"], "replace_const": "REPLACE"},
+ )
+ judge_command("COPY foo bar", {"command": "COPY", "key": ["foo", "bar"]})
+
+
+def test_getex(judge_command):
+ judge_command("GETEX foo", {"command": "GETEX", "key": "foo"})
+ judge_command(
+ "GETEX bar ex 5",
+ {"command": "GETEX", "key": "bar", "expiration": "ex", "millisecond": "5"},
+ )
+ judge_command(
+ "GETEX bar px 5",
+ {"command": "GETEX", "key": "bar", "expiration": "px", "millisecond": "5"},
+ )
+ judge_command(
+ "GETEX bar pxat 5",
+ {"command": "GETEX", "key": "bar", "pxat_const": "pxat", "timestampms": "5"},
+ )
+ judge_command(
+ "GETEX bar exat 5",
+ {"command": "GETEX", "key": "bar", "exat_const": "exat", "timestamp": "5"},
+ )
+ judge_command("GETEX bar ex 5 exat 5", None)
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-26 01:03:28 +0000