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