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_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() dummy.run_from_argv.side_effect = Error("foo", status="EX_FAILURE") help.reset() x.execute("dummy", ["dummy"]) 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')
def test_on_usage_error(self): x = CeleryCommand(app=self.app) x.error = Mock() x.on_usage_error(x.UsageError("foo"), command=None) self.assertTrue(x.error.called) x.on_usage_error(x.UsageError("foo"), command="dummy")
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')