1 files changed, 91 insertions, 0 deletions
diff --git a/tests/cli_tests/test_cli_start.py b/tests/cli_tests/test_cli_start.py
new file mode 100644
index 0000000..8045947
--- /dev/null
+++ b/tests/cli_tests/test_cli_start.py
@@ -0,0 +1,91 @@
+import pexpect
+import pytest
+from textwrap import dedent
+
+
+def test_start_on_connection_error():
+    cli = pexpect.spawn("iredis -p 12345", timeout=1)
+    cli.logfile_read = open("cli_test.log", "ab")
+    cli.expect(r"Error \d+ connecting to 127.0.0.1:12345. Connection refused.")
+    cli.close()
+
+
+def test_start_with_client_name():
+    cli = pexpect.spawn("iredis --client_name custom_name", timeout=2)
+    cli.expect("iredis")
+    cli.sendline("CLIENT GETNAME")
+    cli.expect("custom_name")
+    cli.close()
+
+
+def test_short_help_option(config):
+    c = pexpect.spawn("iredis -h", timeout=2)
+
+    c.expect("Show this message and exit.")
+
+    c = pexpect.spawn("iredis -h 127.0.0.1")
+    c.expect("127.0.0.1:6379>")
+
+    c.close()
+
+
+@pytest.mark.skipif("int(os.environ['REDIS_VERSION']) != 5")
+def test_server_version_in_starting_on5():
+    c = pexpect.spawn("iredis", timeout=2)
+    c.expect("redis-server  5")
+    c.close()
+
+
+@pytest.mark.skipif("int(os.environ['REDIS_VERSION']) != 6")
+def test_server_version_in_starting_on6():
+    c = pexpect.spawn("iredis", timeout=2)
+    c.expect("redis-server  6")
+    c.close()
+
+
+def test_connection_using_url(clean_redis):
+    c = pexpect.spawn("iredis --url redis://localhost:6379/7", timeout=2)
+    c.logfile_read = open("cli_test.log", "ab")
+    c.expect(["iredis", "127.0.0.1:6379[7]>"])
+    c.sendline("set current-db 7")
+    c.expect("OK")
+    c.close()
+
+
+def test_connection_using_url_from_env(clean_redis, monkeypatch):
+    monkeypatch.setenv("IREDIS_URL", "redis://localhost:6379/7")
+    c = pexpect.spawn("iredis", timeout=2)
+    c.logfile_read = open("cli_test.log", "ab")
+    c.expect(["iredis", "localhost:6379[7]>"])
+    c.sendline("set current-db 7")
+    c.expect("OK")
+    c.close()
+
+
+@pytest.mark.xfail(reason="current test in github action, socket not supported.")
+# https://github.community/t5/GitHub-Actions/Job-service-command/td-p/33901#
+# https://help.github.com/en/actions/reference/workflow-syntax-for-github-actions#jobsjob_idservices
+def test_connect_via_socket(fake_redis_socket):
+    config_content = dedent(
+        """
+        [main]
+        log_location = /tmp/iredis1.log
+        no_info=True
+        """
+    )
+    with open("/tmp/iredisrc", "w+") as etc_config:
+        etc_config.write(config_content)
+
+    c = pexpect.spawn("iredis --iredisrc /tmp/iredisrc -s /tmp/test.sock", timeout=2)
+    c.logfile_read = open("cli_test.log", "ab")
+    c.expect("redis /tmp/test.sock")
+
+    c.close()
+
+
+def test_iredis_start_with_prompt():
+    cli = pexpect.spawn("iredis --prompt '{host}abc{port}def{client_name}'", timeout=2)
+    cli.logfile_read = open("cli_test.log", "ab")
+    cli.expect("iredis")
+    cli.expect("127.0.0.1abc6379defNone")
+    cli.close()