def install_three_tasks_test(self, proxy_getter):
"""Install with three tasks."""
observers = []
observer = Mock()
observer.is_service_available = True
observer.service_name = "A"
observer.proxy.InstallWithTasks.return_value = ["/A/1"]
observers.append(observer)
observer = Mock()
observer.is_service_available = True
observer.service_name = "B"
observer.proxy.InstallWithTasks.return_value = ["/B/1", "/B/2"]
observers.append(observer)
task_proxy = Mock()
task_proxy.Steps = 1
proxy_getter.return_value = task_proxy
install_manager = InstallManager()
install_manager.on_module_observers_changed(observers)
proxies = install_manager.collect_install_system_tasks()
self.assertEqual(proxies, [task_proxy, task_proxy, task_proxy])
proxy_getter.assert_has_calls(
[call("A", "/A/1"),
call("B", "/B/1"),
call("B", "/B/2")])
def install_with_no_tasks_test(self):
"""Install with no tasks."""
observer = Mock()
observer.is_service_available = True
observer.service_name = "A"
observer.proxy.InstallWithTasks.return_value = []
install_manager = InstallManager()
install_manager.on_module_observers_changed([observer])
proxies = install_manager.collect_install_system_tasks()
self.assertEqual(proxies, [])
def install_one_task_test(self, proxy_getter):
"""Install with one task."""
observer = Mock()
observer.is_service_available = True
observer.service_name = "A"
observer.proxy.InstallWithTasks.return_value = ["/A/1"]
task_proxy = Mock()
task_proxy.Steps = 1
proxy_getter.return_value = task_proxy
install_manager = InstallManager()
install_manager.on_module_observers_changed([observer])
proxies = install_manager.collect_install_system_tasks()
self.assertEqual(proxies, [task_proxy])
proxy_getter.assert_called_once_with("A", "/A/1")
def install_with_no_modules_test(self):
"""Install with no modules."""
install_manager = InstallManager()
install_manager.on_module_observers_changed([])
proxies = install_manager.collect_install_system_tasks()
self.assertEqual(proxies, [])
class Boss(Service):
"""The Boss service."""
def __init__(self):
super().__init__()
self._module_manager = ModuleManager()
self._kickstart_manager = KickstartManager()
self._install_manager = InstallManager()
self._ui_module = UIModule()
self._module_manager.module_observers_changed.connect(
self._kickstart_manager.on_module_observers_changed)
self._module_manager.module_observers_changed.connect(
self._install_manager.on_module_observers_changed)
def publish(self):
"""Publish the boss."""
TaskContainer.set_namespace(BOSS.namespace)
# Publish submodules.
self._ui_module.publish()
DBus.publish_object(BOSS.object_path, BossInterface(self))
DBus.register_service(BOSS.service_name)
def get_modules(self):
"""Get service names of running modules.
Get a list of all running DBus modules (including addons)
that were discovered and started by the boss.
:return: a list of service names
"""
return self._module_manager.get_service_names()
def start_modules_with_task(self):
"""Start the modules with the task."""
return self._module_manager.start_modules_with_task()
def stop(self):
"""Stop all modules and then stop the boss."""
self._module_manager.stop_modules()
super().stop()
def read_kickstart_file(self, path):
"""Read the specified kickstart file.
:param path: a path to a file
:returns: a kickstart report
"""
log.info("Reading a kickstart file at %s.", path)
return self._kickstart_manager.read_kickstart_file(path)
def generate_kickstart(self):
"""Return a kickstart representation of modules.
:return: a kickstart string
"""
log.info("Generating kickstart data...")
return self._kickstart_manager.generate_kickstart()
def collect_requirements(self):
"""Collect requirements of the modules.
:return: a list of requirements
"""
return self._install_manager.collect_requirements()
def collect_configure_runtime_tasks(self):
"""Collect tasks for configuration of the runtime environment.
FIXME: This method temporarily uses only addons.
:return: a list of task proxies
"""
return self._install_manager.collect_configure_runtime_tasks()
def collect_configure_bootloader_tasks(self, kernel_versions):
"""Collect tasks for configuration of the bootloader.
FIXME: This method temporarily uses only addons.
FIXME: This is a temporary workaround. The method might change.
:param kernel_versions: a list of kernel versions
:return: a list of task proxies
"""
return self._install_manager.collect_configure_bootloader_tasks(
kernel_versions)
def collect_install_system_tasks(self):
"""Collect tasks for installation of the system.
FIXME: This method temporarily uses only addons.
:return: a list of task proxies
"""
return self._install_manager.collect_install_system_tasks()
def set_locale(self, locale):
"""Set locale of boss and all modules.
:param str locale: locale to set
"""
log.info("Setting locale of all modules to %s.", locale)
super().set_locale(locale)
self._module_manager.set_modules_locale(locale)
def finish_installation_with_tasks(self):
"""Finish installation with tasks.
:return: a list of installation tasks
"""
return [
SetContextsTask(conf.target.system_root),
CopyLogsTask(conf.target.system_root)
]
def test_install_with_no_modules(self):
"""Install with no modules."""
install_manager = InstallManager()
install_manager.on_module_observers_changed([])
proxies = install_manager.collect_install_system_tasks()
assert proxies == []