1 files changed, 123 insertions, 0 deletions
diff --git a/tests/cli_tests/test_pager.py b/tests/cli_tests/test_pager.py
new file mode 100644
index 0000000..e8106a4
--- /dev/null
+++ b/tests/cli_tests/test_pager.py
@@ -0,0 +1,123 @@
+# noqa: F541
+from contextlib import contextmanager
+import os
+import pathlib
+import sys
+from textwrap import dedent
+from packaging.version import parse as version_parse
+
+import pexpect
+
+
+TEST_IREDISRC = "/tmp/.iredisrc.test"
+TEST_PAGER_BOUNDARY = "---boundary---"
+TEST_PAGER_BOUNDARY_NUMBER = "---88938347271---"
+
+env_pager = "{} {} {}".format(
+    sys.executable,
+    os.path.join(pathlib.Path(__file__).parent, "wrappager.py"),
+    TEST_PAGER_BOUNDARY,
+)
+env_pager_numbers = "{} {} {}".format(
+    sys.executable,
+    os.path.join(pathlib.Path(__file__).parent, "wrappager.py"),
+    TEST_PAGER_BOUNDARY_NUMBER,
+)
+
+long_list_type = "quicklist"
+if version_parse(os.environ["REDIS_VERSION"]) >= version_parse("7"):
+    long_list_type = "listpack"
+
+
+@contextmanager
+def pager_enabled_cli():
+    env = os.environ
+    env["PAGER"] = env_pager
+    child = pexpect.spawn("iredis -n 15", timeout=3, env=env)
+    child.logfile_read = open("cli_test.log", "ab")
+    child.expect("127.0.0.1")
+    try:
+        yield child
+    finally:
+        child.close()
+
+
+def test_using_pager_when_rows_too_high(clean_redis):
+    for index in range(100):
+        clean_redis.lpush("long-list", f"value-{index}")
+    with pager_enabled_cli() as child:
+        child.sendline("lrange long-list 0 -1")
+        child.expect(TEST_PAGER_BOUNDARY)
+        child.expect("value-1")
+        child.expect(TEST_PAGER_BOUNDARY)
+
+
+def test_using_pager_works_for_help():
+    with pager_enabled_cli() as child:
+        child.sendline("help set")
+        child.expect(TEST_PAGER_BOUNDARY)
+        child.expect("Set the string value of a key")
+        child.expect(TEST_PAGER_BOUNDARY)
+
+
+def test_pager_works_for_peek(clean_redis):
+    for index in range(100):
+        clean_redis.lpush("long-list", f"value-{index}")
+    with pager_enabled_cli() as child:
+        child.sendline("peek long-list")
+        child.expect(TEST_PAGER_BOUNDARY)
+        child.expect(f"({long_list_type})")
+        child.expect("value-1")
+        child.expect(TEST_PAGER_BOUNDARY)
+
+
+def test_using_pager_from_config(clean_redis):
+    config_content = dedent(
+        f"""
+        [main]
+        log_location = /tmp/iredis1.log
+        pager = {env_pager_numbers}
+        """
+    )
+
+    with open(TEST_IREDISRC, "w+") as test_iredisrc:
+        test_iredisrc.write(config_content)
+
+    child = pexpect.spawn(f"iredis -n 15 --iredisrc {TEST_IREDISRC}", timeout=3)
+    child.logfile_read = open("cli_test.log", "ab")
+    child.expect("127.0.0.1")
+    for index in range(100):
+        clean_redis.lpush("long-list", f"value-{index}")
+    child.sendline("lrange long-list 0 -1")
+    child.expect(TEST_PAGER_BOUNDARY_NUMBER)
+    child.expect("value-1")
+    child.expect(TEST_PAGER_BOUNDARY_NUMBER)
+    child.close()
+
+
+def test_using_pager_from_config_when_env_config_both_set(clean_redis):
+    config_content = dedent(
+        f"""
+        [main]
+        log_location = /tmp/iredis1.log
+        pager = {env_pager_numbers}
+        """
+    )
+
+    with open(TEST_IREDISRC, "w+") as test_iredisrc:
+        test_iredisrc.write(config_content)
+
+    env = os.environ
+    env["PAGER"] = env_pager
+    child = pexpect.spawn(
+        f"iredis -n 15 --iredisrc {TEST_IREDISRC}", timeout=3, env=env
+    )
+    child.logfile_read = open("cli_test.log", "ab")
+    child.expect("127.0.0.1")
+    for index in range(100):
+        clean_redis.lpush("long-list", f"value-{index}")
+    child.sendline("lrange long-list 0 -1")
+    child.expect(TEST_PAGER_BOUNDARY_NUMBER)
+    child.expect("value-1")
+    child.expect(TEST_PAGER_BOUNDARY_NUMBER)
+    child.close()