def test_execute(self):
x = CeleryCommand(app=self.app)
Help = x.commands['help'] = Mock()
help = Help.return_value = Mock()
x.execute('fooox', ['a'])
help.run_from_argv.assert_called_with(x.prog_name, [], command='help')
help.reset()
x.execute('help', ['help'])
help.run_from_argv.assert_called_with(x.prog_name, [], command='help')
Dummy = x.commands['dummy'] = Mock()
dummy = Dummy.return_value = Mock()
exc = dummy.run_from_argv.side_effect = Error(
'foo', status='EX_FAILURE',
)
x.on_error = Mock(name='on_error')
help.reset()
x.execute('dummy', ['dummy'])
x.on_error.assert_called_with(exc)
dummy.run_from_argv.assert_called_with(
x.prog_name, [], command='dummy',
)
help.run_from_argv.assert_called_with(
x.prog_name, [], command='help',
)
exc = dummy.run_from_argv.side_effect = x.UsageError('foo')
x.on_usage_error = Mock()
x.execute('dummy', ['dummy'])
x.on_usage_error.assert_called_with(exc)
def test_on_usage_error(self):
x = CeleryCommand(app=self.app)
x.error = Mock()
x.on_usage_error(x.UsageError('foo'), command=None)
x.error.assert_called()
x.on_usage_error(x.UsageError('foo'), command='dummy')