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
|
# coding=utf-8
"""
Cmd2 testing for argument parsing
"""
import re
import argparse
import pytest
import cmd2
from conftest import run_cmd, StdOut
class ArgparseApp(cmd2.Cmd):
def __init__(self):
self.maxrepeats = 3
cmd2.Cmd.__init__(self)
say_parser = argparse.ArgumentParser()
say_parser.add_argument('-p', '--piglatin', action='store_true', help='atinLay')
say_parser.add_argument('-s', '--shout', action='store_true', help='N00B EMULATION MODE')
say_parser.add_argument('-r', '--repeat', type=int, help='output [n] times')
say_parser.add_argument('words', nargs='+', help='words to say')
@cmd2.with_argument_parser(say_parser)
def do_say(self, arglist, args=None):
"""Repeat what you tell me to."""
words = []
for word in args.words:
if word is None:
word = ''
if args.piglatin:
word = '%s%say' % (word[1:], word[0])
if args.shout:
word = word.upper()
words.append(word)
repetitions = args.repeat or 1
for i in range(min(repetitions, self.maxrepeats)):
self.stdout.write(' '.join(words))
self.stdout.write('\n')
tag_parser = argparse.ArgumentParser(description='create a html tag')
tag_parser.add_argument('tag', nargs=1, help='tag')
tag_parser.add_argument('content', nargs='+', help='content to surround with tag')
@cmd2.with_argument_parser(tag_parser)
def do_tag(self, arglist, args=None):
self.stdout.write('<{0}>{1}</{0}>'.format(args.tag[0], ' '.join(args.content)))
self.stdout.write('\n')
compare_parser = argparse.ArgumentParser()
compare_parser.add_argument('args', nargs='*')
@cmd2.with_argument_parser(compare_parser)
def do_compare(self, arglist, args=None):
cmdline_str = re.sub('\s+', ' ', ' '.join(arglist))
args_str = re.sub('\s+', ' ', ' '.join(args.args))
if cmdline_str == args_str:
self.stdout.write('True')
else:
self.stdout.write('False')
argpasre_arglist_parser = argparse.ArgumentParser()
argpasre_arglist_parser.add_argument('args', nargs='*')
@cmd2.with_argument_parser(argpasre_arglist_parser)
def do_argparse_arglist(self, arglist, args=None):
if isinstance(arglist, list):
self.stdout.write('True')
else:
self.stdout.write('False')
arglist_and_argparser_parser = argparse.ArgumentParser()
arglist_and_argparser_parser.add_argument('args', nargs='*')
@cmd2.with_argument_list
@cmd2.with_argument_parser(arglist_and_argparser_parser)
def do_arglistandargparser(self, arglist, args=None):
if isinstance(arglist, list):
self.stdout.write(' '.join(arglist))
else:
self.stdout.write('False')
@cmd2.with_argument_list
def do_arglist(self, arglist):
if isinstance(arglist, list):
self.stdout.write('True')
else:
self.stdout.write('False')
@cmd2.with_argument_list
@cmd2.with_argument_list
def do_arglisttwice(self, arglist):
if isinstance(arglist, list):
self.stdout.write(' '.join(arglist))
else:
self.stdout.write('False')
@pytest.fixture
def argparse_app():
app = ArgparseApp()
app.stdout = StdOut()
return app
def test_argparse_basic_command(argparse_app):
out = run_cmd(argparse_app, 'say hello')
assert out == ['hello']
def test_argparse_quoted_arguments(argparse_app):
argparse_app.POSIX = False
argparse_app.STRIP_QUOTES_FOR_NON_POSIX = True
out = run_cmd(argparse_app, 'say "hello there"')
assert out == ['hello there']
def test_argparse_quoted_arguments_multiple(argparse_app):
argparse_app.POSIX = False
argparse_app.STRIP_QUOTES_FOR_NON_POSIX = True
out = run_cmd(argparse_app, 'say "hello there" "rick & morty"')
assert out == ['hello there rick & morty']
def test_argparse_quoted_arguments_posix(argparse_app):
argparse_app.POSIX = True
out = run_cmd(argparse_app, 'tag strong this should be loud')
assert out == ['<strong>this should be loud</strong>']
def test_argparse_quoted_arguments_posix_multiple(argparse_app):
argparse_app.POSIX = True
out = run_cmd(argparse_app, 'tag strong this "should be" loud')
assert out == ['<strong>this should be loud</strong>']
def test_argparse_help_docstring(argparse_app):
out = run_cmd(argparse_app, 'help say')
assert out[0] == 'Repeat what you tell me to.'
def test_argparse_help_description(argparse_app):
out = run_cmd(argparse_app, 'help tag')
assert out[2] == 'create a html tag'
def test_argparse_prog(argparse_app):
out = run_cmd(argparse_app, 'help tag')
progname = out[0].split(' ')[1]
assert progname == 'tag'
def test_argparse_cmdline(argparse_app):
out = run_cmd(argparse_app, 'compare this is a test')
assert out[0] == 'True'
def test_argparse_arglist(argparse_app):
out = run_cmd(argparse_app, 'argparse_arglist "some arguments" and some more')
assert out[0] == 'True'
def test_arglist(argparse_app):
out = run_cmd(argparse_app, 'arglist "we should" get these')
assert out[0] == 'True'
def test_arglist_decorator_twice(argparse_app):
out = run_cmd(argparse_app, 'arglisttwice "we should" get these')
assert out[0] == 'we should get these'
def test_arglist_and_argparser(argparse_app):
out = run_cmd(argparse_app, 'arglistandargparser some "quoted words"')
assert out[0] == 'some quoted words'
|
