def test_power_off(self): # arrange vcenter_name = 'vcenter name' vm_uuid = 'uuid' 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) # act res = power_manager.power_off(si, vm_uuid, None) # assert self.assertTrue(res) 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)) self.assertTrue(vm.PowerOff.called)
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, task_waiter.SynchronousTaskWaiter()) power_manager.power_on(si, uuid) power_manager.power_off(si, uuid)
def __init__(self): py_vmomi_service = pyVmomiService(SmartConnect, Disconnect) cloudshell_data_retriever_service = CloudshellDataRetrieverService() resource_connection_details_retriever = ResourceConnectionDetailsRetriever( helpers, cloudshell_data_retriever_service) resource_remover = CloudshellResourceRemover(helpers) command_wrapper = CommandWrapper(getLogger, py_vmomi_service) template_deployer = VirtualMachineDeployer(py_vmomi_service, generate_unique_name) deploy_from_template_command = DeployFromTemplateCommand( template_deployer) resource_model_parser = ResourceModelParser() vc_model_retriever = VCenterDataModelRetriever(helpers, resource_model_parser) vc_data_model = vc_model_retriever.get_vcenter_data_model() #vnic_to_network_mapper = VnicToNetworkMapper(name_generator, vc_data_model.default_network) port_group_name_generator = DvPortGroupNameGenerator() vnic_to_network_mapper = VnicToNetworkMapper(port_group_name_generator) # Virtual Switch Connect synchronous_task_waiter = SynchronousTaskWaiter() dv_port_group_creator = DvPortGroupCreator(py_vmomi_service, synchronous_task_waiter) virtual_machine_port_group_configurer = VirtualMachinePortGroupConfigurer( py_vmomi_service, synchronous_task_waiter, vnic_to_network_mapper, VNicService()) virtual_switch_to_machine_connector = VirtualSwitchToMachineConnector( dv_port_group_creator, virtual_machine_port_group_configurer) virtual_switch_connect_command = VirtualSwitchConnectCommand( py_vmomi_service, virtual_switch_to_machine_connector, DvPortGroupNameGenerator(), VlanSpecFactory(), VLanIdRangeParser(), getLogger('VirtualSwitchConnectCommand')) # Virtual Switch Revoke virtual_switch_disconnect_command = \ VirtualSwitchToMachineDisconnectCommand(py_vmomi_service, virtual_machine_port_group_configurer, vc_data_model.default_network) destroy_virtual_machine_command = DestroyVirtualMachineCommand( py_vmomi_service, resource_remover, virtual_switch_disconnect_command) # Power Command vm_power_management_command = VirtualMachinePowerManagementCommand( py_vmomi_service, synchronous_task_waiter) # Refresh IP command refresh_ip_command = RefreshIpCommand( py_vmomi_service, cloudshell_data_retriever_service, helpers, resource_model_parser) self.commandExecuterService = CommandExecuterService( jsonpickle, helpers, command_wrapper, resource_connection_details_retriever, vc_data_model, destroy_virtual_machine_command, deploy_from_template_command, virtual_switch_connect_command, virtual_switch_disconnect_command, vm_power_management_command, refresh_ip_command)
def test_power_on(self): # arrange vm_uuid = 'uuid' si = Mock(spec=vim.ServiceInstance) vm = Mock(spec=vim.VirtualMachine) pv_service = Mock() pv_service.find_by_uuid = Mock(return_value=vm) task = Mock() vm.PowerOn = 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) # act res = power_manager.power_on(si, vm_uuid, None) # assert self.assertTrue(res) self.assertTrue(synchronous_task_waiter.wait_for_task.called_with(task)) self.assertTrue(vm.PowerOn.called)
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, task_waiter.SynchronousTaskWaiter()) power_manager.power_on(si, uuid) power_manager.power_off(si, uuid)