def test_submit_with_clusterman(self): mock_clusterman_metrics = mock.MagicMock() cluster = MesosCluster('mesos-cluster-a.me') mock_task = mock.MagicMock( get_config=mock.Mock(return_value={ 'environment': { 'CLUSTERMAN_RESOURCES': '{"required_cpus|blah=x": 4}', 'EXECUTOR_CLUSTER': 'fake-cluster', 'EXECUTOR_POOL': 'fake-pool', }, }), ) mock_task.get_mesos_id.return_value = 'this_task' with mock.patch( 'tron.mesos.get_clusterman_metrics', return_value=mock_clusterman_metrics, autospec=True, ), staticconf.testing.MockConfiguration( {'clusters': {'fake-cluster': {'aws_region': 'fake-region'}}}, namespace='clusterman', ): cluster.submit(mock_task) assert 'this_task' in cluster.tasks assert cluster.tasks['this_task'] == mock_task cluster.runner.run.assert_called_once_with( mock_task.get_config.return_value, ) assert mock_clusterman_metrics.ClustermanMetricsBotoClient.call_count == 1
def test_submit_disabled(self): cluster = MesosCluster('mesos-cluster-a.me', enabled=False) mock_task = mock.MagicMock() mock_task.get_mesos_id.return_value = 'this_task' cluster.submit(mock_task) assert 'this_task' not in cluster.tasks mock_task.exited.assert_called_once_with(1)
def test_submit(self): cluster = MesosCluster('mesos-cluster-a.me') mock_task = mock.MagicMock() mock_task.get_mesos_id.return_value = 'this_task' cluster.submit(mock_task) assert 'this_task' in cluster.tasks assert cluster.tasks['this_task'] == mock_task cluster.runner.run.assert_called_once_with( mock_task.get_config.return_value, )
def test_submit_disabled(self): cluster = MesosCluster('mesos-cluster-a.me', enabled=False) mock_task = mock.MagicMock() mock_task.get_mesos_id.return_value = 'this_task' with mock.patch( 'tron.mesos.get_clusterman_metrics', return_value=(None, None), autospec=True, ): cluster.submit(mock_task) assert 'this_task' not in cluster.tasks mock_task.exited.assert_called_once_with(1)
def test_submit(self): mock_clusterman_metrics = mock.MagicMock() cluster = MesosCluster('mesos-cluster-a.me') mock_task = mock.MagicMock(get_config=mock.Mock(return_value={'environment': {}})) mock_task.get_mesos_id.return_value = 'this_task' with mock.patch( 'tron.mesos.get_clusterman_metrics', return_value=(mock_clusterman_metrics), autospec=True, ): cluster.submit(mock_task) assert 'this_task' in cluster.tasks assert cluster.tasks['this_task'] == mock_task cluster.runner.run.assert_called_once_with( mock_task.get_config.return_value, ) assert mock_clusterman_metrics.ClustermanMetricsBotoClient.call_count == 0