def test_get_server_verison_after_client(config):
Client("127.0.0.1", "6379", None)
assert config.version.startswith("5.")
config.version = "Unknown"
config.no_info = True
Client("127.0.0.1", "6379", None)
assert config.version == "Unknown"
def test_get_server_verison_after_client(config):
Client("127.0.0.1", "6379", None)
assert re.match(r"\d+\..*", config.version)
config.version = "Unknown"
config.no_info = True
Client("127.0.0.1", "6379", None)
assert config.version == "Unknown"
def test_get_server_verison_after_client():
from iredis.config import config
Client("127.0.0.1", "6379", None)
assert config.version.startswith("5.")
config.version = "Unknown"
Client("127.0.0.1", "6379", None, get_info=False)
assert config.version == "Unknown"
def test_socket_keepalive(config):
config.socket_keepalive = True
from iredis.client import Client
newclient = Client("127.0.0.1", "6379", 0)
assert newclient.connection.socket_keepalive
# keepalive off
config.socket_keepalive = False
newclient = Client("127.0.0.1", "6379", 0)
assert not newclient.connection.socket_keepalive
def test_do_help(config):
client = Client("127.0.0.1", "6379", None)
config.version = "5.0.0"
resp = client.do_help("SET")
assert resp[10] == ("", "1.0.0 (Avaiable on your redis-server: 5.0.0)")
config.version = "2.0.0"
resp = client.do_help("cluster", "addslots")
assert resp[10] == ("", "3.0.0 (Not avaiable on your redis-server: 2.0.0)")
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