def test_save_snapshot_should_succeed_when_snapshot_with_the_same_name_does_not_exists( self): vm = Mock() pyvmomi_service = Mock() pyvmomi_service.find_by_uuid = Mock(return_value=vm) save_snapshot_command = SaveSnapshotCommand(pyvmomi_service, Mock()) si = Mock() # Act with patch(GET_CURRENT_SNAPSHOT_NAME) as get_current_snapshot_name: with patch(GET_VM_SNAPSHOTS) as get_vm_snapshots: get_current_snapshot_name.return_value = 'snapshot1/snapshot2' get_vm_snapshots.return_value = { 'snapshot1/snapshot2': None, 'snapshot1': None } save_snapshot_command.save_snapshot( si=si, logger=Mock(), vm_uuid='machine1', snapshot_name='new_snapshot', save_memory='No') # Assert vm.CreateSnapshot.called_with('new_snapshot', 'Created by CloudShell vCenterShell', False, True)
def test_save_snapshot_should_fail_if_snaphost_exists(self): vm = Mock() pyvmomi_service = Mock() pyvmomi_service.find_by_uuid = Mock(return_value=vm) save_snapshot_command = SaveSnapshotCommand(pyvmomi_service, Mock()) si = Mock() with patch(GET_CURRENT_SNAPSHOT_NAME) as get_current_snapshot_name: with patch(GET_VM_SNAPSHOTS) as get_vm_snapshots: get_current_snapshot_name.return_value = 'snapshot1' get_vm_snapshots.return_value = { 'snapshot1': None, 'snapshot1/snapshot2': None } # Act + Assert with self.assertRaises(SnapshotAlreadyExistsException): save_snapshot_command.save_snapshot( si=si, logger=Mock(), vm_uuid='machine1', snapshot_name='snapshot2', save_memory='No')
def __init__(self): """ Initialize the driver session, this function is called everytime a new instance of the driver is created in here the driver is going to be bootstrapped :param context: models.QualiDriverModels.InitCommandContext """ self.cs_helper = CloudshellDriverHelper() synchronous_task_waiter = SynchronousTaskWaiter() pv_service = pyVmomiService(connect=SmartConnect, disconnect=Disconnect, task_waiter=synchronous_task_waiter) self.resource_model_parser = ResourceModelParser() port_group_name_generator = DvPortGroupNameGenerator() vnic_to_network_mapper = VnicToNetworkMapper( quali_name_generator=port_group_name_generator) resource_remover = CloudshellResourceRemover() ovf_service = OvfImageDeployerService(self.resource_model_parser) self.vm_loader = VMLoader(pv_service) vm_deployer = VirtualMachineDeployer( pv_service=pv_service, name_generator=generate_unique_name, ovf_service=ovf_service, cs_helper=self.cs_helper, resource_model_parser=ResourceModelParser()) dv_port_group_creator = DvPortGroupCreator( pyvmomi_service=pv_service, synchronous_task_waiter=synchronous_task_waiter) virtual_machine_port_group_configurer = \ VirtualMachinePortGroupConfigurer(pyvmomi_service=pv_service, synchronous_task_waiter=synchronous_task_waiter, vnic_to_network_mapper=vnic_to_network_mapper, vnic_service=VNicService(), name_gen=port_group_name_generator) virtual_switch_to_machine_connector = VirtualSwitchToMachineConnector( dv_port_group_creator, virtual_machine_port_group_configurer) # Command Wrapper self.command_wrapper = CommandWrapper( pv_service=pv_service, cloud_shell_helper=self.cs_helper, resource_model_parser=self.resource_model_parser, context_based_logger_factory=ContextBasedLoggerFactory()) # Deploy Command self.deploy_command = DeployCommand(deployer=vm_deployer) # Virtual Switch Revoke self.virtual_switch_disconnect_command = \ VirtualSwitchToMachineDisconnectCommand( pyvmomi_service=pv_service, port_group_configurer=virtual_machine_port_group_configurer, resource_model_parser=self.resource_model_parser) # Virtual Switch Connect virtual_switch_connect_command = \ VirtualSwitchConnectCommand( pv_service=pv_service, virtual_switch_to_machine_connector=virtual_switch_to_machine_connector, dv_port_group_name_generator=DvPortGroupNameGenerator(), vlan_spec_factory=VlanSpecFactory(), vlan_id_range_parser=VLanIdRangeParser()) self.connection_orchestrator = ConnectionCommandOrchestrator( connector=virtual_switch_connect_command, disconnector=self.virtual_switch_disconnect_command, resource_model_parser=self.resource_model_parser) # Destroy VM Command self.destroy_virtual_machine_command = \ DestroyVirtualMachineCommand(pv_service=pv_service, resource_remover=resource_remover, disconnector=self.virtual_switch_disconnect_command) # Power Command self.vm_power_management_command = \ VirtualMachinePowerManagementCommand(pyvmomi_service=pv_service, synchronous_task_waiter=synchronous_task_waiter) ip_manager = VMIPManager() # Refresh IP command self.refresh_ip_command = RefreshIpCommand( pyvmomi_service=pv_service, resource_model_parser=ResourceModelParser(), ip_manager=ip_manager) # Save Snapshot self.snapshot_saver = SaveSnapshotCommand( pyvmomi_service=pv_service, task_waiter=synchronous_task_waiter) # Snapshot Restorer self.snapshot_restorer = SnapshotRestoreCommand( pyvmomi_service=pv_service, task_waiter=synchronous_task_waiter) self.snapshots_retriever = RetrieveSnapshotsCommand( pyvmomi_service=pv_service)