def test_process_event_control_stop(self):
event = mock.MagicMock(
kind='control',
message='stop',
)
cluster = MesosCluster('mesos-cluster-a.me')
cluster._process_event(event)
assert cluster.runner.stop.call_count == 1
assert cluster.deferred is None
def test_process_event_task_id_invalid(self):
event = mock_task_event('other_task', 'some_platform_type')
cluster = MesosCluster('mesos-cluster-a.me')
mock_task = mock.MagicMock(spec_set=MesosTask)
mock_task.get_mesos_id.return_value = 'this_task'
cluster.tasks['this_task'] = mock_task
cluster._process_event(event)
assert_equal(mock_task.handle_event.call_count, 0)
def test_process_event_task(self):
event = mock_task_event('this_task', 'some_platform_type')
cluster = MesosCluster('mesos-cluster-a.me')
mock_task = mock.MagicMock(spec_set=MesosTask)
mock_task.get_mesos_id.return_value = 'this_task'
cluster.tasks['this_task'] = mock_task
cluster._process_event(event)
mock_task.handle_event.assert_called_once_with(event)
def test_process_event_control_stop(self):
event = mock.MagicMock(
kind='control',
message='stop',
)
cluster = MesosCluster('mesos-cluster-a.me')
cluster._process_event(event)
assert cluster.runner.stop.call_count == 1
assert cluster.deferred is None
def test_process_event_task_id_invalid(self):
event = mock_task_event('other_task', 'some_platform_type')
cluster = MesosCluster('mesos-cluster-a.me')
mock_task = mock.MagicMock(spec_set=MesosTask)
mock_task.get_mesos_id.return_value = 'this_task'
cluster.tasks['this_task'] = mock_task
cluster._process_event(event)
assert_equal(mock_task.handle_event.call_count, 0)
def test_process_event_task(self):
event = mock_task_event('this_task', 'some_platform_type')
cluster = MesosCluster('mesos-cluster-a.me')
mock_task = mock.MagicMock(spec_set=MesosTask)
mock_task.get_mesos_id.return_value = 'this_task'
cluster.tasks['this_task'] = mock_task
cluster._process_event(event)
mock_task.handle_event.assert_called_once_with(event)
