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
|
# coding=utf-8
"""
Cmd2 testing for argument parsing
"""
import pytest
import cmd2
from conftest import run_cmd, StdOut, normalize
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()
@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()
return app
@pytest.fixture
def secondlevel_app():
app = SecondLevel()
app.stdout = StdOut()
return app
def run_submenu_cmd(app, cmd):
""" Clear StdOut buffers, run the command, extract the buffer contents."""
app.stdout.clear()
second_level_cmd.stdout.clear()
app.onecmd_plus_hooks(cmd)
out1 = app.stdout.buffer
out2 = second_level_cmd.stdout.buffer
app.stdout.clear()
second_level_cmd.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, '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 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, '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, '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, '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 get_top_level_attr')
assert out2 == [str(submenu_app.top_level_attr)]
|
