commands: Add str method CommandError

This commit is contained in:
Thomas Adamcik 2013-11-12 22:58:59 +01:00
parent b5360a1360
commit f5a5f9e9b1
2 changed files with 14 additions and 0 deletions

View File

@ -9,6 +9,9 @@ class CommandError(Exception):
self.message = message
self.usage = usage
def __str__(self):
return '%s\n\nerror: %s\n' % (self.usage, self.message)
class ArgumentParser(argparse.ArgumentParser):
def error(self, message):

View File

@ -423,3 +423,14 @@ class HelpTest(unittest.TestCase):
' some text about this sub-command.\n\n'
' --test TEST the great and wonderful')
self.assertEqual(expected, cmd.format_help('foo').strip())
class CommandErrorTest(unittest.TestCase):
def test_args_get_stored(self):
error = command.CommandError('message', usage='usage: foo')
self.assertEqual(error.message, 'message')
self.assertEqual(error.usage, 'usage: foo')
def test_str_command_error(self):
error = command.CommandError('message', usage='usage: foo')
self.assertEqual(str(error), 'usage: foo\n\nerror: message\n')