| """ |
| Tests autosuggestion using pexpect. |
| """ |
| |
| import lldb |
| from lldbsuite.test.decorators import * |
| from lldbsuite.test.lldbtest import * |
| from lldbsuite.test.lldbpexpect import PExpectTest |
| |
| |
| def cursor_horizontal_abs(s): |
| return "\x1b[" + str(len(s) + 1) + "G" |
| |
| |
| class TestCase(PExpectTest): |
| ANSI_FAINT = "\x1b[2m" |
| ANSI_RESET = "\x1b[0m" |
| ANSI_RED = "\x1b[31m" |
| ANSI_CYAN = "\x1b[36m" |
| |
| # PExpect uses many timeouts internally and doesn't play well |
| # under ASAN on a loaded machine.. |
| @skipIfAsan |
| @skipIfEditlineSupportMissing |
| def test_autosuggestion_add_spaces(self): |
| self.launch( |
| use_colors=True, |
| extra_args=[ |
| "-o", |
| "settings set show-autosuggestion true", |
| "-o", |
| "settings set use-color true", |
| ], |
| ) |
| |
| # Check if spaces are added to hide the previous gray characters. |
| self.expect("help frame var") |
| self.expect("help frame info") |
| self.child.send("help frame v") |
| self.child.expect_exact( |
| cursor_horizontal_abs("(lldb) help frame ") |
| + "v" |
| + self.ANSI_FAINT |
| + "ar" |
| + self.ANSI_RESET |
| + " " |
| ) |
| |
| self.quit() |
| |
| @skipIfAsan |
| @skipIfEditlineSupportMissing |
| def test_autosuggestion(self): |
| self.launch( |
| use_colors=True, |
| extra_args=[ |
| "-o", |
| "settings set show-autosuggestion true", |
| "-o", |
| "settings set use-color true", |
| ], |
| ) |
| |
| # Common input codes. |
| ctrl_f = "\x06" |
| delete = chr(127) |
| |
| frame_output_needle = "Syntax: frame <subcommand>" |
| # Run 'help frame' once to put it into the command history. |
| self.expect("help frame", substrs=[frame_output_needle]) |
| |
| # Check that LLDB shows the autosuggestion in gray behind the text. |
| self.child.send("hel") |
| self.child.expect_exact( |
| cursor_horizontal_abs("(lldb) he") |
| + "l" |
| + self.ANSI_FAINT |
| + "p frame" |
| + self.ANSI_RESET |
| ) |
| |
| # Apply the autosuggestion and press enter. This should print the |
| # 'help frame' output if everything went correctly. |
| self.child.send(ctrl_f + "\n") |
| self.child.expect_exact(frame_output_needle) |
| |
| # Check that pressing Ctrl+F directly after Ctrl+F again does nothing. |
| self.child.send("hel" + ctrl_f + ctrl_f + "\n") |
| self.child.expect_exact(frame_output_needle) |
| |
| # Try autosuggestion using tab and ^f. |
| # \t makes "help" and ^f makes "help frame". If everything went |
| # correct we should see the 'help frame' output again. |
| self.child.send("hel\t" + ctrl_f + "\n") |
| self.child.expect_exact(frame_output_needle) |
| |
| # Check that autosuggestion works after delete. |
| self.child.send("a1234" + 5 * delete + "hel" + ctrl_f + "\n") |
| self.child.expect_exact(frame_output_needle) |
| |
| # Check that autosuggestion works after delete. |
| self.child.send("help x" + delete + ctrl_f + "\n") |
| self.child.expect_exact(frame_output_needle) |
| |
| # Check that autosuggestion complete to the most recent one. |
| self.child.send("help frame variable\n") |
| self.child.send("help fr") |
| self.child.expect_exact(self.ANSI_FAINT + "ame variable" + self.ANSI_RESET) |
| self.child.send("\n") |
| |
| # Try another command. |
| apropos_output_needle = "Syntax: apropos <search-word>" |
| # Run 'help frame' once to put it into the command history. |
| self.expect("help apropos", substrs=[apropos_output_needle]) |
| |
| # Check that 'hel' should have an autosuggestion for 'help apropos' now. |
| self.child.send("hel") |
| self.child.expect_exact( |
| cursor_horizontal_abs("(lldb) he") |
| + "l" |
| + self.ANSI_FAINT |
| + "p apropos" |
| + self.ANSI_RESET |
| ) |
| |
| # Run the command and expect the 'help apropos' output. |
| self.child.send(ctrl_f + "\n") |
| self.child.expect_exact(apropos_output_needle) |
| |
| # Check that pressing Ctrl+F in an empty prompt does nothing. |
| breakpoint_output_needle = "Syntax: breakpoint <subcommand>" |
| self.child.send(ctrl_f + "help breakpoint" + "\n") |
| self.child.expect_exact(breakpoint_output_needle) |
| |
| self.quit() |
| |
| @skipIfAsan |
| @skipIfEditlineSupportMissing |
| def test_autosuggestion_custom_ansi_prefix_suffix(self): |
| self.launch( |
| use_colors=True, |
| extra_args=[ |
| "-o", |
| "settings set show-autosuggestion true", |
| "-o", |
| "settings set use-color true", |
| "-o", |
| "settings set show-autosuggestion-ansi-prefix ${ansi.fg.red}", |
| "-o", |
| "setting set show-autosuggestion-ansi-suffix ${ansi.fg.cyan}", |
| ], |
| ) |
| |
| self.child.send("help frame variable\n") |
| self.child.send("help fr") |
| self.child.expect_exact(self.ANSI_RED + "ame variable" + self.ANSI_CYAN) |
| self.child.send("\n") |
| |
| self.quit() |