def setUp(self): self.datastore_manager = MagicMock() self.datastore_manager.get_datastore_ids.return_value = [ "datastore1", "datastore2" ] self.host_upgrade = HostUpgrade(self.datastore_manager) self.timeout = self.TIMEOUT
class HostUpgradeTestCase(unittest.TestCase): TIMEOUT = 6000 def setUp(self): self.datastore_manager = MagicMock() self.datastore_manager.get_datastore_ids.return_value = [ "datastore1", "datastore2" ] self.host_upgrade = HostUpgrade(self.datastore_manager) self.timeout = self.TIMEOUT def tearDown(self): self.host_upgrade._task_runner.stop() @patch("host.upgrade.softlink_generator.SoftLinkGenerator.process") def test_lifecycle(self, process): assert_that(not self.host_upgrade.in_progress()) self.host_upgrade.start(self.timeout) self.host_upgrade._task_runner.wait_for_task_end() assert_that(not self.host_upgrade.in_progress()) self.assertEqual(process.call_count, 1) @patch("host.upgrade.softlink_generator.SoftLinkGenerator.process") def test_upgrade_exception(self, process): process.side_effect = Exception("create_symlinks failed.") assert_that(not self.host_upgrade.in_progress()) self.host_upgrade.start(self.timeout) self.host_upgrade._task_runner.wait_for_task_end() assert_that(not self.host_upgrade.in_progress()) self.assertEqual(process.call_count, 1)
class HostUpgradeTestCase(unittest.TestCase): TIMEOUT = 6000 def setUp(self): self.datastore_manager = MagicMock() self.datastore_manager.get_datastore_ids.return_value = ["datastore1", "datastore2"] self.host_upgrade = HostUpgrade(self.datastore_manager) self.timeout = self.TIMEOUT def tearDown(self): self.host_upgrade._task_runner.stop() def test_lifecycle(self): assert_that(not self.host_upgrade.in_progress()) self.host_upgrade.start(self.timeout) self.host_upgrade._task_runner.wait_for_task_end() assert_that(not self.host_upgrade.in_progress())
def init(self): # Load agent config config = common.services.get(ServiceName.AGENT_CONFIG) # Create the hypervisor object hv = hypervisor.Hypervisor(config) # When configuration changes, notify hypervisor config.on_config_change(config.CPU_OVERCOMMIT, hv.set_cpu_overcommit) config.on_config_change(config.MEMORY_OVERCOMMIT, hv.set_memory_overcommit) # Register hypervisor in services common.services.register(ServiceName.HYPERVISOR, hv) # Create the upgrade object upgrade = HostUpgrade(hv.datastore_manager) common.services.register(ServiceName.UPGRADE, upgrade) # Create host handler host_handler = HostHandler(hv) common.services.register(Host.Iface, host_handler) if config.hypervisor == "esx": common.services.register(ServiceName.VIM_CLIENT, hv.hypervisor.vim_client) # Load num_threads num_threads = config.host_service_threads # Define and add thrift service service = common.plugin.ThriftService( name="Host", service=Host, handler=host_handler, num_threads=num_threads, ) self.add_thrift_service(service)
def setUp(self): self.datastore_manager = MagicMock() self.datastore_manager.get_datastore_ids.return_value = ["datastore1", "datastore2"] self.host_upgrade = HostUpgrade(self.datastore_manager) self.timeout = self.TIMEOUT