def test_power_management(self): # arrange cred = TestCredentials() pv_service = pyVmomiService(SmartConnect, Disconnect) si = pv_service.connect(cred.host, cred.username, cred.password) uuid = pv_service.find_vm_by_name(si, 'QualiSB/Raz', '2').config.uuid power_manager = VirtualMachinePowerManagementCommand(pv_service, SynchronousTaskWaiter()) power_manager.power_on(si, uuid) power_manager.power_off(si, uuid)
def test_power_management(self): # arrange cred = TestCredentials() pv_service = pyVmomiService(SmartConnect, Disconnect) si = pv_service.connect(cred.host, cred.username, cred.password) uuid = pv_service.find_vm_by_name(si, 'QualiSB/Raz', '2').config.uuid power_manager = VirtualMachinePowerManagementCommand( pv_service, SynchronousTaskWaiter()) power_manager.power_on(si, uuid) power_manager.power_off(si, uuid)
def test_power_off_already(self): vm_uuid = 'uuid' si = Mock(spec=vim.ServiceInstance) vm = Mock(spec=vim.VirtualMachine) vm.summary = Mock() vm.summary.runtime = Mock() vm.summary.runtime.powerState = 'poweredOff' session = Mock() pv_service = Mock() pv_service.find_by_uuid = Mock(return_value=vm) power_manager = VirtualMachinePowerManagementCommand( pv_service, Mock()) # act res = power_manager.power_off(si=si, logger=Mock(), session=session, vcenter_data_model=Mock(), vm_uuid=vm_uuid, resource_fullname=None) # assert self.assertTrue(res, 'already powered off') self.assertFalse(vm.PowerOn.called)
def test_power_off_soft(self): # arrange vcenter_name = 'vcenter name' vm_uuid = 'uuid' session = Mock() si = Mock(spec=vim.ServiceInstance) vm = Mock(spec=vim.VirtualMachine) task = Mock() pv_service = Mock() pv_service.find_by_uuid = Mock(return_value=vm) vm.PowerOff = Mock(return_value=task) synchronous_task_waiter = Mock() synchronous_task_waiter.wait_for_task = Mock(return_value=True) power_manager = VirtualMachinePowerManagementCommand( pv_service, synchronous_task_waiter) power_manager._connect_to_vcenter = Mock(return_value=si) power_manager._get_vm = Mock(return_value=vm) vcenter = Mock() vcenter.shutdown_method = 'soft' # act res = power_manager.power_off(si=si, logger=Mock(), session=session, vcenter_data_model=vcenter, vm_uuid=vm_uuid, resource_fullname=None) # assert self.assertTrue(res) self.assertTrue(vm.ShutdownGuest.called) self.assertTrue( power_manager._connect_to_vcenter.called_with(vcenter_name)) self.assertTrue(power_manager._get_vm.called_with(si, vm_uuid)) self.assertTrue( synchronous_task_waiter.wait_for_task.called_with(task))
def test_power_off_hard(self): # arrange vcenter_name = 'vcenter name' vm_uuid = 'uuid' session = Mock() si = Mock(spec=vim.ServiceInstance) vm = Mock(spec=vim.VirtualMachine) task = Mock() pv_service = Mock() pv_service.find_by_uuid = Mock(return_value=vm) vm.PowerOff = Mock(return_value=task) synchronous_task_waiter = Mock() synchronous_task_waiter.wait_for_task = Mock(return_value=True) power_manager = VirtualMachinePowerManagementCommand(pv_service, synchronous_task_waiter) power_manager._connect_to_vcenter = Mock(return_value=si) power_manager._get_vm = Mock(return_value=vm) vcenter = Mock() vcenter.shutdown_method = 'hard' # act res = power_manager.power_off(si=si, logger=Mock(), session=session, vcenter_data_model=vcenter, vm_uuid=vm_uuid, resource_fullname=None) # assert self.assertTrue(res) self.assertTrue(vm.ShutdownGuest.called) self.assertTrue(power_manager._connect_to_vcenter.called_with(vcenter_name)) self.assertTrue(power_manager._get_vm.called_with(si, vm_uuid)) self.assertTrue(synchronous_task_waiter.wait_for_task.called_with(task))
def test_power_off_already(self): vm_uuid = 'uuid' si = Mock(spec=vim.ServiceInstance) vm = Mock(spec=vim.VirtualMachine) vm.summary = Mock() vm.summary.runtime = Mock() vm.summary.runtime.powerState = 'poweredOff' session = Mock() pv_service = Mock() pv_service.find_by_uuid = Mock(return_value=vm) power_manager = VirtualMachinePowerManagementCommand(pv_service, Mock()) # act res = power_manager.power_off(si=si, logger=Mock(), session=session, vcenter_data_model=Mock(), vm_uuid=vm_uuid, resource_fullname=None) # assert self.assertTrue(res, 'already powered off') self.assertFalse(vm.PowerOn.called)