summaryrefslogtreecommitdiff
path: root/tests/unit/test_shell.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/test_shell.py')
-rw-r--r--tests/unit/test_shell.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/unit/test_shell.py b/tests/unit/test_shell.py
index 0bd7fda..daf4cd8 100644
--- a/tests/unit/test_shell.py
+++ b/tests/unit/test_shell.py
@@ -356,3 +356,25 @@ class TestShell(unittest.TestCase):
connection.return_value.get_capabilities.return_value = {'swift': None}
swiftclient.shell.main(argv)
connection.return_value.get_capabilities.assert_called_with(None)
+
+
+class TestSubcommandHelp(unittest.TestCase):
+
+ def test_subcommand_help(self):
+ for command in swiftclient.shell.commands:
+ help_var = 'st_%s_help' % command
+ self.assertTrue(help_var in vars(swiftclient.shell))
+ out = six.StringIO()
+ with mock.patch('sys.stdout', out):
+ argv = ['', command, '--help']
+ self.assertRaises(SystemExit, swiftclient.shell.main, argv)
+ expected = vars(swiftclient.shell)[help_var]
+ self.assertEqual(out.getvalue().strip('\n'), expected)
+
+ def test_no_help(self):
+ out = six.StringIO()
+ with mock.patch('sys.stdout', out):
+ argv = ['', 'bad_command', '--help']
+ self.assertRaises(SystemExit, swiftclient.shell.main, argv)
+ expected = 'no help for bad_command'
+ self.assertEqual(out.getvalue().strip('\n'), expected)