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
|
# Copyright 2007 Google Inc.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
"""Unit tests for nss_cache/app.py."""
__author__ = "vasilios@google.com (Vasilios Hoffman)"
import logging
import io
import os
import sys
import unittest
from nss_cache import app
class TestNssCacheApp(unittest.TestCase):
"""Unit tests for NssCacheApp class."""
def setUp(self):
dev_null = io.StringIO()
self.stdout = sys.stdout
sys.stdout = dev_null
self.srcdir = os.path.normpath(os.path.join(os.path.dirname(__file__), ".."))
self.conf_filename = os.path.join(self.srcdir, "nsscache.conf")
def tearDown(self):
sys.stdout = self.stdout
def testRun(self):
return_code = app.NssCacheApp().Run([], {})
self.assertEqual(os.EX_USAGE, return_code)
def testParseGlobalOptions(self):
a = app.NssCacheApp()
(options, args) = a.parser.parse_args(["-d", "-v", "command"])
self.assertNotEqual(None, options.debug)
self.assertNotEqual(None, options.verbose)
self.assertEqual(["command"], args)
def testParseCommandLineDebug(self):
a = app.NssCacheApp()
(options, args) = a.parser.parse_args(["-d"])
self.assertNotEqual(None, options.debug)
(options, args) = a.parser.parse_args(["--debug"])
self.assertNotEqual(None, options.debug)
a.Run(["-d"], {})
self.assertEqual(logging.DEBUG, a.log.getEffectiveLevel())
def testParseCommandLineVerbose(self):
a = app.NssCacheApp()
(options, args) = a.parser.parse_args(["-v"])
self.assertNotEqual(None, options.verbose)
self.assertEqual([], args)
(options, args) = a.parser.parse_args(["--verbose"])
self.assertNotEqual(None, options.verbose)
self.assertEqual([], args)
a.Run(["-v"], {})
self.assertEqual(logging.INFO, a.log.getEffectiveLevel())
def testParseCommandLineVerboseDebug(self):
a = app.NssCacheApp()
a.Run(["-v", "-d"], {})
self.assertEqual(logging.DEBUG, a.log.getEffectiveLevel())
def testParseCommandLineConfigFile(self):
a = app.NssCacheApp()
(options, args) = a.parser.parse_args(["-c", "file"])
self.assertNotEqual(None, options.config_file)
self.assertEqual([], args)
(options, args) = a.parser.parse_args(["--config-file", "file"])
self.assertNotEqual(None, options.config_file)
self.assertEqual([], args)
def testBadOptionsCauseNoExit(self):
a = app.NssCacheApp()
stderr_buffer = io.StringIO()
old_stderr = sys.stderr
sys.stderr = stderr_buffer
self.assertEqual(2, a.Run(["--invalid"], {}))
sys.stderr = old_stderr
def testHelpOptionPrintsGlobalHelp(self):
stdout_buffer = io.StringIO()
a = app.NssCacheApp()
old_stdout = sys.stdout
sys.stdout = stdout_buffer
self.assertEqual(0, a.Run(["--help"], {}))
sys.stdout = old_stdout
self.assertNotEqual(0, stdout_buffer.tell())
(prelude, usage, commands, options) = stdout_buffer.getvalue().split("\n\n")
self.assertTrue(prelude.startswith("nsscache synchronises"))
expected_str = "Usage: nsscache [global options] command [command options]"
self.assertEqual(expected_str, usage)
self.assertTrue(commands.startswith("commands:"))
self.assertTrue(options.startswith("Options:"))
self.assertTrue(options.find("show this help message and exit") >= 0)
def testHelpCommandOutput(self):
# trap stdout into a StringIO
stdout_buffer = io.StringIO()
a = app.NssCacheApp()
old_stdout = sys.stdout
sys.stdout = stdout_buffer
self.assertEqual(0, a.Run(["help"], {}))
sys.stdout = old_stdout
self.assertNotEqual(0, stdout_buffer.tell())
self.assertTrue(stdout_buffer.getvalue().find("nsscache synchronises") >= 0)
def testRunBadArgsPrintsGlobalHelp(self):
# trap stdout into a StringIO
stdout_buffer = io.StringIO()
old_stdout = sys.stdout
sys.stdout = stdout_buffer
# verify bad arguments calls help
return_code = app.NssCacheApp().Run(
["blarg"], {"NSSCACHE_CONFIG": self.conf_filename}
)
sys.stdout = old_stdout
assert return_code == 70 # EX_SOFTWARE
assert stdout_buffer.getvalue().find("enable debugging") >= 0
if __name__ == "__main__":
unittest.main()
|
