Fix restore command

iredis/client.py

@@ -4,6 +4,7 @@
 import re
 import os
 import sys
+import codecs
 import logging
 from subprocess import run
 from importlib_resources import read_text
@@ -282,6 +283,7 @@ def execute_by_connection(self, connection, command_name, *args, **options):
                     connection.connect()
                     logger.info(f"New connection created, retry on {connection}.")
                 logger.info(f"send_command: {command_name} , {args}")
+
                 connection.send_command(command_name, *args)
                 response = connection.read_response()
             except AuthenticationError:
@@ -547,6 +549,12 @@ def pre_hook(self, command, command_name, args, completer: IRedisCompleter):
         if command_name.upper() in ["ZSCAN", "ZPOPMAX", "ZPOPMIN"]:
             config.withscores = True
 
+        # TODO should we using escape_decode on all strings??
+        if command_name.upper() == "RESTORE":
+            for i, a in enumerate(args):
+                serialized_value = codecs.escape_decode(a)[0]
+                args[i] = serialized_value
+
         # not a tty
         if not completer:
             logger.warning(

tests/cli_tests/test_command_restore.py

@@ -0,0 +1,6 @@
+def test_restore_command(clean_redis, cli):
+    cli.sendline(r'restore foo1 0 "\x00\x03bar\t\x006L\x18\xac\xba\xe0\x9e\xa6"')
+    cli.expect(["OK", "127.0.0.1"])
+
+    cli.sendline("get foo1")
+    cli.expect('"bar"')

tests/unittests/test_client.py

@@ -351,7 +351,7 @@ def test_peek_set_fetch_part(iredis_client, clean_redis):
     peek_result = list(iredis_client.do_peek("myset"))
 
     assert peek_result[0][0] == ("class:dockey", "key: ")
-    assert peek_result[0][1][1].startswith("set (hashtable)  mem: 2")
+    assert peek_result[0][1][1].startswith("set (hashtable)  mem: ")
 
 
 def test_peek_zset_fetch_all(iredis_client, clean_redis):