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
|
# 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)
argparser = argparse.ArgumentParser()
argparser.add_argument('-p', '--piglatin', action='store_true', help='atinLay')
argparser.add_argument('-s', '--shout', action='store_true', help='N00B EMULATION MODE')
argparser.add_argument('-r', '--repeat', type=int, help='output [n] times')
argparser.add_argument('words', nargs='+', help='words to say')
@cmd2.with_argument_parser(argparser)
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')
argparser = argparse.ArgumentParser(description='create a html tag')
argparser.add_argument('tag', nargs=1, help='tag')
argparser.add_argument('content', nargs='+', help='content to surround with tag')
@cmd2.with_argument_parser(argparser)
def do_tag(self, arglist, args=None):
self.stdout.write('<{0}>{1}</{0}>'.format(args.tag[0], ' '.join(args.content)))
self.stdout.write('\n')
argparser = argparse.ArgumentParser()
argparser.add_argument('args', nargs='*')
@cmd2.with_argument_parser(argparser)
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')
argparser = argparse.ArgumentParser()
argparser.add_argument('args', nargs='*')
@cmd2.with_argument_parser(argparser)
def do_arglist(self, arglist, args=None):
if isinstance(arglist, list):
self.stdout.write('True')
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, 'arglist "some arguments" and some more')
assert out[0] == 'True'
|
