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)