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_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)