59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
import pexpect
|
|
from textwrap import dedent
|
|
from pathlib import Path
|
|
|
|
|
|
def test_log_location_config():
|
|
config_content = dedent(
|
|
"""
|
|
[main]
|
|
log_location = /tmp/iredis1.log
|
|
"""
|
|
)
|
|
with open("/tmp/iredisrc", "w+") as etc_config:
|
|
etc_config.write(config_content)
|
|
|
|
cli = pexpect.spawn("iredis -n 15 --iredisrc /tmp/iredisrc", timeout=1)
|
|
cli.expect("127.0.0.1")
|
|
cli.close()
|
|
|
|
log = Path("/tmp/iredis1.log")
|
|
assert log.exists()
|
|
with open(log, "r") as logfile:
|
|
content = logfile.read()
|
|
|
|
assert len(content) > 100
|
|
|
|
|
|
def test_load_prompt_from_config(iredis_client, clean_redis):
|
|
config_content = dedent(
|
|
"""
|
|
[main]
|
|
prompt = {host}abc{port}xx{db}
|
|
"""
|
|
)
|
|
with open("/tmp/iredisrc", "w+") as etc_config:
|
|
etc_config.write(config_content)
|
|
|
|
cli = pexpect.spawn("iredis -n 15 --iredisrc /tmp/iredisrc", timeout=1)
|
|
cli.expect("iredis")
|
|
cli.expect("127.0.0.1abc6379xx15")
|
|
cli.close()
|
|
|
|
|
|
def test_prompt_cli_overwrite_config(iredis_client, clean_redis):
|
|
config_content = dedent(
|
|
"""
|
|
[main]
|
|
prompt = {host}abc{port}xx{db}
|
|
"""
|
|
)
|
|
with open("/tmp/iredisrc", "w+") as etc_config:
|
|
etc_config.write(config_content)
|
|
|
|
cli = pexpect.spawn(
|
|
"iredis -n 15 --iredisrc /tmp/iredisrc --prompt='{db}-12345'", timeout=1
|
|
)
|
|
cli.expect("iredis")
|
|
cli.expect("15-12345")
|
|
cli.close()
|