本文整理汇总了Python中celery.bin.multi.MultiTool.call_command方法的典型用法代码示例。如果您正苦于以下问题：Python MultiTool.call_command方法的具体用法？Python MultiTool.call_command怎么用？Python MultiTool.call_command使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类celery.bin.multi.MultiTool的用法示例。
在下文中一共展示了MultiTool.call_command方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_MultiTool
# 需要导入模块: from celery.bin.multi import MultiTool [as 别名]
# 或者: from celery.bin.multi.MultiTool import call_command [as 别名]
class test_MultiTool(AppCase):
def setup(self):
self.fh = WhateverIO()
self.env = {}
self.t = MultiTool(env=self.env, fh=self.fh)
self.t.cluster_from_argv = Mock(name='cluster_from_argv')
self.t._cluster_from_argv = Mock(name='cluster_from_argv')
self.t.Cluster = Mock(name='Cluster')
self.t.carp = Mock(name='.carp')
self.t.usage = Mock(name='.usage')
self.t.splash = Mock(name='.splash')
self.t.say = Mock(name='.say')
self.t.ok = Mock(name='.ok')
self.cluster = self.t.Cluster.return_value
def _cluster_from_argv(argv):
p = self.t.OptionParser(argv)
p.parse()
return p, self.cluster
self.t.cluster_from_argv.return_value = self.cluster
self.t._cluster_from_argv.side_effect = _cluster_from_argv
def test_findsig(self):
self.assert_sig_argument(['a', 'b', 'c', '-1'], 1)
self.assert_sig_argument(['--foo=1', '-9'], 9)
self.assert_sig_argument(['-INT'], signal.SIGINT)
self.assert_sig_argument([], signal.SIGTERM)
self.assert_sig_argument(['-s'], signal.SIGTERM)
self.assert_sig_argument(['-log'], signal.SIGTERM)
def assert_sig_argument(self, args, expected):
p = self.t.OptionParser(args)
p.parse()
self.assertEqual(self.t._find_sig_argument(p), expected)
def test_execute_from_commandline(self):
self.t.call_command = Mock(name='call_command')
self.t.execute_from_commandline(
'multi start --verbose 10 --foo'.split(),
cmd='X',
)
self.assertEqual(self.t.cmd, 'X')
self.assertEqual(self.t.prog_name, 'multi')
self.t.call_command.assert_called_with('start', ['10', '--foo'])
def test_execute_from_commandline__arguments(self):
self.assertTrue(self.t.execute_from_commandline('multi'.split()))
self.assertTrue(self.t.execute_from_commandline('multi -bar'.split()))
def test_call_command(self):
cmd = self.t.commands['foo'] = Mock(name='foo')
self.t.retcode = 303
self.assertIs(
self.t.call_command('foo', ['1', '2', '--foo=3']),
cmd.return_value,
)
cmd.assert_called_with('1', '2', '--foo=3')
def test_call_command__error(self):
self.assertEqual(
self.t.call_command('asdqwewqe', ['1', '2']),
1,
)
self.t.carp.assert_called()
def test_handle_reserved_options(self):
self.assertListEqual(
self.t._handle_reserved_options(
['a', '-q', 'b', '--no-color', 'c']),
['a', 'b', 'c'],
)
def test_start(self):
self.cluster.start.return_value = [0, 0, 1, 0]
self.assertTrue(self.t.start('10', '-A', 'proj'))
self.t.splash.assert_called_with()
self.t.cluster_from_argv.assert_called_with(('10', '-A', 'proj'))
self.cluster.start.assert_called_with()
def test_start__exitcodes(self):
self.cluster.start.return_value = [0, 0, 0]
self.assertFalse(self.t.start('foo', 'bar', 'baz'))
self.cluster.start.assert_called_with()
self.cluster.start.return_value = [0, 1, 0]
self.assertTrue(self.t.start('foo', 'bar', 'baz'))
def test_stop(self):
self.t.stop('10', '-A', 'proj', retry=3)
self.t.splash.assert_called_with()
self.t._cluster_from_argv.assert_called_with(('10', '-A', 'proj'))
self.cluster.stop.assert_called_with(retry=3, sig=signal.SIGTERM)
def test_stopwait(self):
self.t.stopwait('10', '-A', 'proj', retry=3)
self.t.splash.assert_called_with()
self.t._cluster_from_argv.assert_called_with(('10', '-A', 'proj'))
self.cluster.stopwait.assert_called_with(retry=3, sig=signal.SIGTERM)
#.........这里部分代码省略.........