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)
