def test_start_upgrade_no_executors(self, upgrade_util):
controller = WorkController()
controller.app = Mock(rapid_config="testing")
controller._start_upgrade("12345", [])
upgrade_util.upgrade_version.assert_called_with("12345", "testing")
def test_can_work_on_with_work_existing_action_instance(self, os):
controller = WorkController()
controller.app = Mock()
controller.app.rapid_config.executor_count = 1
os.listdir.return_value = ['rapid-11-11111']
eq_(False, controller.can_work_on(Mock(action_instance_id=11)))
def test_can_work_on_with_work(self, os):
controller = WorkController()
controller.app = Mock()
controller.app.rapid_config.executor_count = 1
os.listdir.return_value = []
eq_(True, controller.can_work_on(Mock(action_instance_id=1111)))
def test_perform_upgrade(self, store_service, upgrade_util, mock_logger):
controller = WorkController()
controller.app = Mock(rapid_config="testing")
store_service.get_executors.return_value = []
controller._perform_upgrade("12345")
upgrade_util.upgrade_version.assert_called_with("12345", "testing")
def test_start_upgrade_with_executors(self, store_service, mock_logger):
controller = WorkController()
mock_app = Mock()
controller.app = mock_app
controller._start_upgrade("12345", [1])
store_service.set_updating.assert_called_with(mock_app, False)
def test_check_version_different_versions_is_updating(self, store_service):
controller = WorkController()
controller.app = Mock()
store_service.is_updating.return_value = True
eq_(
False,
controller.check_version(
Mock(headers={version.Version.HEADER: "1"})))
def test_check_version_different_versions_is_not_updating(
self, store_service, threading):
controller = WorkController()
controller.app = Mock()
store_service.is_updating.return_value = False
eq_(
False,
controller.check_version(
Mock(headers={version.Version.HEADER: "1"})))
eq_(1, threading.Thread.call_count)
def test_can_work_on_equal_executors(self, store_service):
controller = WorkController()
controller.app = Mock()
controller.app.rapid_config.executor_count = 1
store_service.get_executors.return_value = [1]
eq_(False, controller.can_work_on())