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
|
# -*- coding: utf-8 -*-
import unittest
from wilderness import Application
from wilderness import Command
from wilderness.help import HelpCommand
from wilderness.help import have_man_command
from wilderness.tester import Tester
class TestApp(Application):
__test__ = False # for Pytest users
def __init__(self):
super().__init__(
"testapp",
version="0.1.0",
title="testapp title",
author="John Doe",
description="long description",
extra_sections={"test": "test section"},
)
def register(self):
self.add_argument(
"-q",
"--quiet",
help="be quiet",
action="store_true",
description="Run in quiet mode",
)
class FooCommand(Command):
def __init__(self):
super().__init__(
"foo",
title="the foo command",
description="Description of the foo command",
)
def handle(self) -> int:
return 0
class ApplicationTestCase(unittest.TestCase):
maxDiff = None
def setUp(self):
self._app = TestApp()
self._helptext = "usage: testapp [-h] [-q] command ...\n"
def tearDown(self):
self._app = None
def test_application_base(self):
app = self._app
self.assertEqual(app.name, "testapp")
self.assertEqual(app.author, "John Doe")
self.assertEqual(app.version, "0.1.0")
self.assertEqual(app.description, "long description")
self.assertEqual(len(app.commands), 1)
self.assertTrue(isinstance(app.commands[0], HelpCommand))
def test_application_help_1(self):
self.assertEqual(self._app.format_help(), self._helptext)
def test_application_help_2(self):
# Test that we get help without any arguments
tester = Tester(self._app)
tester.test_application()
self.assertEqual(tester.get_return_code(), 1)
self.assertEqual(tester.get_stdout(), self._helptext)
def test_application_help_3(self):
# Test that we get help with only -h flag
tester = Tester(self._app)
tester.test_application(["-h"])
self.assertEqual(tester.get_return_code(), 1)
self.assertEqual(tester.get_stdout(), self._helptext)
def test_application_help_4(self):
# Test that we get help with only --help
tester = Tester(self._app)
tester.test_application(["--help"])
self.assertEqual(tester.get_return_code(), 1)
self.assertEqual(tester.get_stdout(), self._helptext)
def test_application_help_5(self):
# Test that we get help with --help and other arguments
tester = Tester(self._app)
tester.test_application(["-q", "--help"])
self.assertEqual(tester.get_return_code(), 1)
self.assertEqual(tester.get_stdout(), self._helptext)
def test_application_help_6(self):
# Test that we get help when help command supplied
tester = Tester(self._app)
tester.test_application(["help"])
self.assertEqual(tester.get_return_code(), 1)
self.assertEqual(tester.get_stdout(), self._helptext)
@unittest.skipUnless(have_man_command(), "no man command")
def test_application_help_7(self):
app = self._app
app.add(FooCommand())
tester = Tester(self._app)
tester.test_application(["help"])
self.assertEqual(tester.get_return_code(), 1)
helptext = "\n".join(
[
self._helptext,
"Available commands:",
" help Display help information",
" foo the foo command",
"",
]
)
self.assertEqual(tester.get_stdout(), helptext)
if __name__ == "__main__":
unittest.main()
|
