def test_verify_args_missing(self):
c = Command()
def run(a, b, c):
pass
c.run = run
with self.assertRaises(c.UsageError):
c.verify_args((1,))
c.verify_args((1, 2, 3))
def test_verify_args_missing(self):
c = Command()
def run(a, b, c):
pass
c.run = run
with self.assertRaises(c.UsageError):
c.verify_args((1, ))
c.verify_args((1, 2, 3))
def test_run_raises_UsageError(self):
cb = Mock()
c = Command(on_usage_error=cb)
c.verify_args = Mock()
c.run = Mock()
exc = c.run.side_effect = c.UsageError('foo', status=3)
self.assertEqual(c(), exc.status)
cb.assert_called_with(exc)
c.verify_args.assert_called_with(())
def test_run_raises_UsageError(self):
cb = Mock()
c = Command(on_usage_error=cb)
c.verify_args = Mock()
c.run = Mock()
exc = c.run.side_effect = c.UsageError('foo', status=3)
self.assertEqual(c(), exc.status)
cb.assert_called_with(exc)
c.verify_args.assert_called_with(())
