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
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
|
"""
Test user added container commands
"""
import sys
import lldb
from lldbsuite.test.decorators import *
from lldbsuite.test.lldbtest import *
class TestCmdContainer(TestBase):
NO_DEBUG_INFO_TESTCASE = True
def test_container_add(self):
self.container_add()
def check_command_tree_exists(self):
"""This makes sure we can still run the command tree we added."""
self.runCmd("test-multi")
self.runCmd("test-multi test-multi-sub")
self.runCmd("test-multi test-multi-sub welcome")
def container_add(self):
# Make sure we can't overwrite built-in commands:
self.expect(
"command container add process",
"Can't replace builtin container command",
substrs=["can't replace builtin command"],
error=True,
)
self.expect(
"command container add process non_such_subcommand",
"Can't add to built-in subcommand",
substrs=["Path component: 'process' is not a user command"],
error=True,
)
self.expect(
"command container add process launch",
"Can't replace builtin subcommand",
substrs=["Path component: 'process' is not a user command"],
error=True,
)
# Now lets make a container command:
self.runCmd("command container add -h 'A test container command' test-multi")
# Make sure the help works:
self.expect(
"help test-multi",
"Help works for top-level multi",
substrs=["A test container command"],
)
# Add a subcommand:
self.runCmd(
"command container add -h 'A test container sub-command' test-multi test-multi-sub"
)
# Make sure the help works:
self.expect(
"help test-multi",
"Help shows sub-multi",
substrs=[
"A test container command",
"test-multi-sub -- A test container sub-command",
],
)
self.expect(
"help test-multi test-multi-sub",
"Help shows sub-multi",
substrs=["A test container sub-command"],
)
# Now add a script based command to the container command:
self.runCmd("command script import welcome.py")
self.runCmd(
"command script add -c welcome.WelcomeCommand test-multi test-multi-sub welcome"
)
# Make sure the help still works:
self.expect(
"help test-multi test-multi-sub",
"Listing subcommands works",
substrs=[
"A test container sub-command",
"welcome -- Just a docstring for Welcome",
],
)
self.expect(
"help test-multi test-multi-sub welcome",
"Subcommand help works",
substrs=["Just a docstring for Welcome"],
)
# And make sure it actually works:
self.expect(
"test-multi test-multi-sub welcome friend",
"Test command works",
substrs=["Hello friend, welcome to LLDB"],
)
# Make sure we can make an alias to this:
self.runCmd(
"command alias my-welcome test-multi test-multi-sub welcome",
"We can make an alias to multi-word",
)
self.expect(
"my-welcome friend",
"Test command works",
substrs=["Hello friend, welcome to LLDB"],
)
self.runCmd("command unalias my-welcome")
# Make sure overwriting works on the leaf command. First using the
# explicit option so we should not be able to remove extant commands by default:
self.expect(
"command script add -c welcome.WelcomeCommand2 test-multi test-multi-sub welcome",
"overwrite command w/o -o",
substrs=["cannot add command: sub-command already exists"],
error=True,
)
# But we can with the -o option:
self.runCmd(
"command script add -c welcome.WelcomeCommand2 -o test-multi test-multi-sub welcome"
)
# Make sure we really did overwrite:
self.expect(
"test-multi test-multi-sub welcome friend",
"Used the new command class",
substrs=["Hello friend, welcome again to LLDB"],
)
self.expect(
"apropos welcome",
"welcome should show up in apropos",
substrs=["A docstring for the second Welcome"],
)
self.expect(
"help test-multi test-multi-sub welcome",
"welcome should show up in help",
substrs=["A docstring for the second Welcome"],
)
self.expect("help", "test-multi should show up in help", substrs=["test-multi"])
# Now switch the default and make sure we can now delete w/o the overwrite option:
self.runCmd("settings set interpreter.require-overwrite 0")
self.runCmd(
"command script add -c welcome.WelcomeCommand test-multi test-multi-sub welcome"
)
# Make sure we really did overwrite:
self.expect(
"test-multi test-multi-sub welcome friend",
"Used the new command class",
substrs=["Hello friend, welcome to LLDB"],
)
# Make sure we give good errors when the input is wrong:
self.expect(
"command script delete test-mult test-multi-sub welcome",
"Delete script command - wrong first path component",
substrs=["'test-mult' not found"],
error=True,
)
self.expect(
"command script delete test-multi test-multi-su welcome",
"Delete script command - wrong second path component",
substrs=["'test-multi-su' not found"],
error=True,
)
self.check_command_tree_exists()
self.expect(
"command script delete test-multi test-multi-sub welcom",
"Delete script command - wrong leaf component",
substrs=["'welcom' not found"],
error=True,
)
self.check_command_tree_exists()
self.expect(
"command script delete test-multi test-multi-sub",
"Delete script command - no leaf component",
substrs=["subcommand 'test-multi-sub' is not a user command"],
error=True,
)
self.check_command_tree_exists()
# You can't use command script delete to delete container commands:
self.expect(
"command script delete test-multi",
"Delete script - can't delete container",
substrs=["command 'test-multi' is a multi-word command."],
error=True,
)
self.expect(
"command script delete test-multi test-multi-sub",
"Delete script - can't delete container",
substrs=["subcommand 'test-multi-sub' is not a user command"],
error=True,
)
# You can't use command container delete to delete scripted commands:
self.expect(
"command container delete test-multi test-multi-sub welcome",
"command container can't delete user commands",
substrs=["subcommand 'welcome' is not a container command"],
error=True,
)
# Also make sure you can't alias on top of container commands:
self.expect(
"command alias test-multi process launch",
"Tried to alias on top of a container command",
substrs=[
"'test-multi' is a user container command and cannot be overwritten."
],
error=True,
)
self.check_command_tree_exists()
# Also assert that we can't delete builtin commands:
self.expect(
"command script delete process launch",
"Delete builtin command fails",
substrs=["command 'process' is not a user command"],
error=True,
)
# Now let's do the version that works
self.expect(
"command script delete test-multi test-multi-sub welcome",
"Delete script command by path",
substrs=["Deleted command: test-multi test-multi-sub welcome"],
)
# Now overwrite the sub-command, it should end up empty:
self.runCmd(
"command container add -h 'A different help string' -o test-multi test-multi-sub"
)
# welcome should be gone:
self.expect(
"test-multi test-multi-sub welcome friend",
"did remove subcommand",
substrs=["'test-multi-sub' does not have any subcommands."],
error=True,
)
# We should have the new help:
self.expect(
"help test-multi test-multi-sub",
"help changed",
substrs=["A different help string"],
)
# Now try deleting commands.
self.runCmd("command container delete test-multi test-multi-sub")
self.expect(
"test-multi test-multi-sub",
"Command is not active",
error=True,
substrs=["'test-multi' does not have any subcommands"],
)
self.expect("help test-multi", matching=False, substrs=["test-multi-sub"])
# Next the root command:
self.runCmd("command container delete test-multi")
self.expect(
"test-multi",
"Root command gone",
substrs=["'test-multi' is not a valid command."],
error=True,
)
|
