本文整理汇总了Python中app.master.cluster_master.ClusterMaster.get_build方法的典型用法代码示例。如果您正苦于以下问题:Python ClusterMaster.get_build方法的具体用法?Python ClusterMaster.get_build怎么用?Python ClusterMaster.get_build使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类app.master.cluster_master.ClusterMaster
的用法示例。
在下文中一共展示了ClusterMaster.get_build方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_updating_slave_to_setup_completed_state_should_tell_build_to_begin_subjob_execution
# 需要导入模块: from app.master.cluster_master import ClusterMaster [as 别名]
# 或者: from app.master.cluster_master.ClusterMaster import get_build [as 别名]
def test_updating_slave_to_setup_completed_state_should_tell_build_to_begin_subjob_execution(self):
master = ClusterMaster()
fake_build = MagicMock()
master.get_build = MagicMock(return_value=fake_build)
slave_url = 'raphael.turtles.gov'
master.connect_slave(slave_url, 10)
slave = master.get_slave(slave_url=slave_url)
master.handle_slave_state_update(slave, SlaveState.SETUP_COMPLETED)
fake_build.begin_subjob_executions_on_slave.assert_called_once_with(slave)
示例2: test_updating_slave_to_setup_completed_state_should_tell_build_to_begin_subjob_execution
# 需要导入模块: from app.master.cluster_master import ClusterMaster [as 别名]
# 或者: from app.master.cluster_master.ClusterMaster import get_build [as 别名]
def test_updating_slave_to_setup_completed_state_should_tell_build_to_begin_subjob_execution(self):
master = ClusterMaster()
slave_registry = SlaveRegistry.singleton()
fake_build = MagicMock(spec_set=Build)
master.get_build = MagicMock(return_value=fake_build)
slave_url = 'raphael.turtles.gov'
master.connect_slave(slave_url, 10)
slave = slave_registry.get_slave(slave_url=slave_url)
mock_scheduler = self.mock_scheduler_pool.get(fake_build)
scheduler_begin_event = Event()
mock_scheduler.begin_subjob_executions_on_slave.side_effect = lambda **_: scheduler_begin_event.set()
master.handle_slave_state_update(slave, SlaveState.SETUP_COMPLETED)
was_called = scheduler_begin_event.wait(timeout=5)
self.assertTrue(was_called, 'scheduler.begin_subjob_executions_on_slave should be called in response '
'to slave setup completing.')
_, call_kwargs = mock_scheduler.begin_subjob_executions_on_slave.call_args
self.assertEqual(call_kwargs.get('slave'), slave)