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
|
# coding=utf-8
"""
Cmd2 testing for argument parsing
"""
import pytest
from cmd2 import cmd2
from .conftest import run_cmd, StdOut, normalize
class SecondLevelB(cmd2.Cmd):
"""To be used as a second level command class. """
def __init__(self, *args, **kwargs):
cmd2.Cmd.__init__(self, *args, **kwargs)
self.prompt = '2ndLevel B '
def do_get_top_level_attr(self, line):
self.poutput(str(self.top_level_attr))
def do_set_top_level_attr(self, line):
self.top_level_attr = 987654321
class SecondLevel(cmd2.Cmd):
"""To be used as a second level command class. """
def __init__(self, *args, **kwargs):
cmd2.Cmd.__init__(self, *args, **kwargs)
self.prompt = '2ndLevel '
self.top_level_attr = None
def do_say(self, line):
self.poutput("You called a command in SecondLevel with '%s'. " % line)
def help_say(self):
self.poutput("This is a second level menu. Options are qwe, asd, zxc")
def complete_say(self, text, line, begidx, endidx):
return [s for s in ['qwe', 'asd', 'zxc'] if s.startswith(text)]
def do_get_top_level_attr(self, line):
self.poutput(str(self.top_level_attr))
def do_get_prompt(self, line):
self.poutput(self.prompt)
second_level_cmd = SecondLevel()
second_level_b_cmd = SecondLevelB()
@cmd2.AddSubmenu(SecondLevelB(),
command='should_work_with_default_kwargs')
@cmd2.AddSubmenu(second_level_b_cmd,
command='secondb',
shared_attributes=dict(top_level_attr='top_level_attr'),
require_predefined_shares=False,
preserve_shares=True
)
@cmd2.AddSubmenu(second_level_cmd,
command='second',
aliases=('second_alias',),
shared_attributes=dict(top_level_attr='top_level_attr'))
class SubmenuApp(cmd2.Cmd):
"""To be used as the main / top level command class that will contain other submenus."""
def __init__(self, *args, **kwargs):
cmd2.Cmd.__init__(self, *args, **kwargs)
self.prompt = 'TopLevel '
self.top_level_attr = 123456789
def do_say(self, line):
self.poutput("You called a command in TopLevel with '%s'. " % line)
def help_say(self):
self.poutput("This is a top level submenu. Options are qwe, asd, zxc")
def complete_say(self, text, line, begidx, endidx):
return [s for s in ['qwe', 'asd', 'zxc'] if s.startswith(text)]
@pytest.fixture
def submenu_app():
app = SubmenuApp()
app.stdout = StdOut()
second_level_cmd.stdout = StdOut()
second_level_b_cmd.stdout = StdOut()
return app
@pytest.fixture
def secondlevel_app():
app = SecondLevel()
app.stdout = StdOut()
return app
@pytest.fixture
def secondlevel_app_b():
app = SecondLevelB()
app.stdout = StdOut()
return app
def run_submenu_cmd(app, second_level_app, cmd):
""" Clear StdOut buffers, run the command, extract the buffer contents."""
app.stdout.clear()
second_level_app.stdout.clear()
app.onecmd_plus_hooks(cmd)
out1 = app.stdout.buffer
out2 = second_level_app.stdout.buffer
app.stdout.clear()
second_level_app.stdout.clear()
return normalize(out1), normalize(out2)
def test_submenu_say_from_top_level(submenu_app):
line = 'testing'
out1, out2 = run_submenu_cmd(submenu_app, second_level_cmd, 'say ' + line)
assert len(out1) == 1
assert len(out2) == 0
assert out1[0] == "You called a command in TopLevel with {!r}.".format(line)
def test_submenu_second_say_from_top_level(submenu_app):
line = 'testing'
out1, out2 = run_submenu_cmd(submenu_app, second_level_cmd, 'second say ' + line)
# No output expected from the top level
assert out1 == []
# Output expected from the second level
assert len(out2) == 1
assert out2[0] == "You called a command in SecondLevel with {!r}.".format(line)
def test_submenu_say_from_second_level(secondlevel_app):
line = 'testing'
out = run_cmd(secondlevel_app, 'say ' + line)
assert out == ["You called a command in SecondLevel with '%s'." % line]
def test_submenu_help_second_say_from_top_level(submenu_app):
out1, out2 = run_submenu_cmd(submenu_app, second_level_cmd, 'help second say')
# No output expected from the top level
assert out1 == []
# Output expected from the second level
assert out2 == ["This is a second level menu. Options are qwe, asd, zxc"]
def test_submenu_help_say_from_second_level(secondlevel_app):
out = run_cmd(secondlevel_app, 'help say')
assert out == ["This is a second level menu. Options are qwe, asd, zxc"]
def test_submenu_help_second(submenu_app):
out1, out2 = run_submenu_cmd(submenu_app, second_level_cmd, 'help second')
out3 = run_cmd(second_level_cmd, 'help')
assert out2 == out3
def test_submenu_from_top_help_second_say(submenu_app):
out1, out2 = run_submenu_cmd(submenu_app, second_level_cmd, 'help second say')
out3 = run_cmd(second_level_cmd, 'help say')
assert out2 == out3
def test_submenu_shared_attribute(submenu_app):
out1, out2 = run_submenu_cmd(submenu_app, second_level_cmd, 'second get_top_level_attr')
assert out2 == [str(submenu_app.top_level_attr)]
def test_submenu_shared_attribute_preserve(submenu_app):
out1, out2 = run_submenu_cmd(submenu_app, second_level_b_cmd, 'secondb get_top_level_attr')
assert out2 == [str(submenu_app.top_level_attr)]
out1, out2 = run_submenu_cmd(submenu_app, second_level_b_cmd, 'secondb set_top_level_attr')
assert submenu_app.top_level_attr == 987654321
out1, out2 = run_submenu_cmd(submenu_app, second_level_b_cmd, 'secondb get_top_level_attr')
assert out2 == [str(987654321)]
|
