def test_set_enabled_off(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=True)
mock_task = mock.Mock()
cluster.tasks = {'task': mock_task}
cluster.set_enabled(False)
assert not cluster.enabled
assert cluster.runner.stop.call_count == 1
assert cluster.tasks == {}
assert mock_task.exited.call_count == 1
def test_set_enabled_off(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=True)
mock_task = mock.Mock()
cluster.tasks = {'task': mock_task}
cluster.set_enabled(False)
assert not cluster.enabled
assert cluster.runner.stop.call_count == 1
assert cluster.tasks == {}
assert mock_task.exited.call_count == 1
def test_set_enabled_on(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=False)
cluster.set_enabled(True)
assert_equal(cluster.enabled, True)
# Basically the same as regular initialization
assert_equal(self.mock_processor.executor_from_config.call_count, 2)
self.mock_runner_cls.assert_called_once_with(
self.mock_processor.executor_from_config.return_value,
self.mock_queue,
)
assert_equal(cluster.runner, self.mock_runner_cls.return_value)
get_event_deferred = cluster.deferred
assert_equal(get_event_deferred, self.mock_queue.get.return_value)
get_event_deferred.addCallback.assert_has_calls([
mock.call(cluster._process_event),
mock.call(cluster.handle_next_event),
])
def test_set_enabled_on(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=False)
cluster.set_enabled(True)
assert_equal(cluster.enabled, True)
# Basically the same as regular initialization
assert_equal(self.mock_processor.executor_from_config.call_count, 2)
self.mock_runner_cls.assert_called_once_with(
self.mock_processor.executor_from_config.return_value,
self.mock_queue,
)
assert_equal(cluster.runner, self.mock_runner_cls.return_value)
get_event_deferred = cluster.deferred
assert_equal(get_event_deferred, self.mock_queue.get.return_value)
get_event_deferred.addCallback.assert_has_calls([
mock.call(cluster._process_event),
mock.call(cluster.handle_next_event),
])
def test_set_enabled_on_already(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=True)
cluster.set_enabled(True)
assert_equal(cluster.enabled, True)
# Runner should have only be created once
assert_equal(self.mock_runner_cls.call_count, 1)
def test_set_enabled_on_already(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=True)
cluster.set_enabled(True)
assert_equal(cluster.enabled, True)
# Runner should have only be created once
assert_equal(self.mock_runner_cls.call_count, 1)
def test_set_enabled_off(self):
cluster = MesosCluster('mesos-cluster-a.me', enabled=True)
cluster.set_enabled(False)
assert_equal(cluster.enabled, False)
assert_equal(cluster.runner.stop.call_count, 1)
