1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
"""
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()
|
