else: sys_id = self.remote_.new_system(self.token_) logging.debug('create new system %s', sys_id) return (hostname, sys_id) def _save_system(self, sys_id): """save system config update.""" self.remote_.save_system(sys_id, self.token_) def _update_modify_system(self, sys_id, system_config): """update modify system""" for key, value in system_config.items(): self.remote_.modify_system( sys_id, key, value, self.token_) def update_host_config(self, hostid, config, **kwargs): """update host config.""" profile = self._get_profile(**kwargs) hostname, sys_id = self._get_system(config) system_config = self._get_modify_system( hostname, profile, config, **kwargs) logging.debug('%s system config to update: %s', hostid, system_config) self._update_modify_system(sys_id, system_config) self._save_system(sys_id) os_installer.register(Installer)
def test_multi_registered_installer(self): os_installer.register(DummyInstaller) self.assertRaises(KeyError, os_installer.register, Dummy2Installer)
self.remote_.modify_system(sys_id, key, value, self.token_) def _netboot_enabled(self, sys_id): """enable netboot""" self.remote_.modify_system(sys_id, 'netboot_enabled', True, self.token_) def clean_host_config(self, hostid, config, **kwargs): """clean host config.""" self._clean_system(config) def reinstall_host(self, hostid, config, **kwargs): """reinstall host.""" sys_id = self._get_system(config, False) if sys_id: self._netboot_enabled(sys_id) def update_host_config(self, hostid, config, **kwargs): """update host config.""" self.clean_host_config(hostid, config, **kwargs) profile = self._get_profile(**kwargs) sys_id = self._get_system(config) system_config = self._get_modify_system(profile, config, **kwargs) logging.debug('%s system config to update: %s', hostid, system_config) self._update_modify_system(sys_id, system_config) self._save_system(sys_id) os_installer.register(Installer)
def test_found_installer(self): os_installer.register(DummyInstaller) intaller = os_installer.get_installer_by_name(DummyInstaller.NAME) self.assertIsInstance(intaller, DummyInstaller)
def test_multi_registered_installer(self): """test register multi installers with the same name.""" os_installer.register(DummyInstaller) self.assertRaises(KeyError, os_installer.register, Dummy2Installer)