commands: Add str method CommandError
This commit is contained in:
parent
b5360a1360
commit
f5a5f9e9b1
@ -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):
|
||||
|
||||
@ -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')
|
||||
|
||||
Loading…
Reference in New Issue
Block a user