def test_patch_completer():
client = Client("127.0.0.1", "6379", None)
completer = IRedisCompleter()
client.pre_hook("MGET foo bar hello world", "MGET", "foo bar hello world",
completer)
assert completer.key_completer.words == ["world", "hello", "bar", "foo"]
assert completer.key_completer.words == ["world", "hello", "bar", "foo"]
client.pre_hook("GET bar", "GET", "bar", completer)
assert completer.key_completer.words == ["bar", "world", "hello", "foo"]
def test_patch_completer(completer):
client = Client("127.0.0.1", "6379", None)
client.pre_hook("MGET foo bar hello world", completer)
assert completer.completers["key"].words == [
"world", "hello", "bar", "foo"
]
assert completer.completers["keys"].words == [
"world", "hello", "bar", "foo"
]
client.pre_hook("GET bar", completer)
assert completer.completers["keys"].words == [
"bar", "world", "hello", "foo"
]
def test_patch_completer():
client = Client("127.0.0.1", "6379", None)
grammar = get_command_grammar("MGET")
completer = GrammarCompleter(grammar, completer_mapping)
client.pre_hook(
"MGET foo bar hello world", "MGET", "foo bar hello world", completer
)
assert completer.completers["key"].words == ["world", "hello", "bar", "foo"]
assert completer.completers["keys"].words == ["world", "hello", "bar", "foo"]
grammar = get_command_grammar("GET")
completer = GrammarCompleter(grammar, completer_mapping)
client.pre_hook("GET bar", "GET", "bar", completer)
assert completer.completers["keys"].words == ["bar", "world", "hello", "foo"]
def test_send_command(_input, command_name, expect_args):
client = Client("127.0.0.1", "6379", None)
client.execute_command_and_read_response = MagicMock()
client.send_command(_input, None)
args, kwargs = client.execute_command_and_read_response.call_args
assert args == (None, command_name, *expect_args)
def iredis_client():
return Client("127.0.0.1", "6379", db=15)
def test_send_command(_input, command_name, expect_args):
client = Client("127.0.0.1", "6379", None)
client.execute = MagicMock()
next(client.send_command(_input, None))
args, kwargs = client.execute.call_args
assert args == (command_name, *expect_args)
def test_can_not_connect_on_startup(capfd):
with pytest.raises(SystemExit):
Client("localhost", "16111", 15)
out, err = capfd.readouterr()
assert "connecting to localhost:16111." in err
def test_stipe_quote_escaple_in_quote(test_input, expected):
client = Client("127.0.0.1", "6379", None)
assert list(client._strip_quote_args(test_input)) == expected