real-ladybug 0.0.1.dev1__cp311-cp311-win_amd64.whl

real_ladybug-source/tools/shell/test/test_shell_esc_search.py

@@ -0,0 +1,162 @@
+import pexpect
+import pytest
+
+from test_helper import KEY_ACTION
+
+
+def set_up_search(test) -> None:
+    test.send_finished_statement('RETURN "databases rule" AS a;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )
+    test.send_finished_statement('RETURN "lbug is cool" AS b;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 lbug is cool \u2502", pexpect.EOF])
+        == 0
+    )
+    test.send_finished_statement('RETURN "the shell is fun" AS c;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 the shell is fun \u2502", pexpect.EOF])
+        == 0
+    )
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+
+
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_accept_search(test) -> None:
+    test.send_finished_statement('RETURN "databases" AS a;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases \u2502", pexpect.EOF]) == 0
+    )
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("databases")
+    test.send_statement("\x1b\x1b")
+    test.send_finished_statement(" rule\x1b[F\r")
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )
+
+
+@pytest.mark.parametrize(
+    "esc",
+    [
+        "\x1b[H",
+        "\x1b[1~",
+        "\x1bOH",
+    ],
+)
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_accept_move_home(test, esc) -> None:
+    test.send_finished_statement('"databases rule" AS a;\r')
+    assert (
+        test.shell_process.expect_exact(
+            [
+                "Error: Parser exception: extraneous input '\"databases rule\"'",
+                pexpect.EOF,
+            ],
+        )
+        == 0
+    )
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("databases")
+    test.send_statement(esc)
+    test.send_statement("RETURN \x1b[F")
+    test.send_finished_statement(KEY_ACTION.ENTER.value)
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )
+
+
+@pytest.mark.parametrize(
+    "esc",
+    [
+        "\x1b[4~",
+        "\x1b[8~",
+        "\x1b[F",
+        "\x1bOF",
+    ],
+)
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_accept_move_end(test, esc) -> None:
+    test.send_finished_statement("RETURN ;\r")
+    assert (
+        test.shell_process.expect_exact(
+            [
+                "Error: Parser exception: Invalid input <RETURN ;>: expected rule oC_RegularQuery (line: 1, offset: 7)",
+                pexpect.EOF,
+            ],
+        )
+        == 0
+    )
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("return")
+    test.send_statement(esc)
+    test.send_statement(KEY_ACTION.BACKSPACE.value)  # remove semicolon
+    test.send_finished_statement('"databases rule" AS a;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )
+
+
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_next_history(test) -> None:
+    set_up_search(test)
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("the shell is fun")
+    test.send_statement("\x1b[A" * 2)
+    test.send_finished_statement(KEY_ACTION.ENTER.value)
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )
+
+
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_prev_history(test) -> None:
+    set_up_search(test)
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("databases")
+    test.send_statement("\x1b[B" * 2)
+    test.send_finished_statement(KEY_ACTION.ENTER.value)
+    assert (
+        test.shell_process.expect_exact(["\u2502 the shell is fun \u2502", pexpect.EOF])
+        == 0
+    )
+
+
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_move_left(test) -> None:
+    test.send_finished_statement('RETURN "databasesrule" AS a;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 databasesrule \u2502", pexpect.EOF])
+        == 0
+    )
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("databasesr")
+    test.send_statement("\x1b[D")
+    test.send_finished_statement(" \x1b[F\r")
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )
+
+
+@pytest.mark.parametrize("test", ["multiline", "singleline"], indirect=True)
+def test_move_right(test) -> None:
+    test.send_finished_statement('RETURN "databasesrule" AS a;\r')
+    assert (
+        test.shell_process.expect_exact(["\u2502 databasesrule \u2502", pexpect.EOF])
+        == 0
+    )
+    test.send_control_statement(KEY_ACTION.CTRL_R.value)
+    test.send_statement("database")
+    test.send_statement("\x1b[C")
+    test.send_finished_statement(" \x1b[F\r")
+    assert (
+        test.shell_process.expect_exact(["\u2502 databases rule \u2502", pexpect.EOF])
+        == 0
+    )