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 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_memory_yes(self):
save_memory_string = 'Yes'
self.assertTrue(
SaveSnapshotCommand._get_save_vm_memory_to_snapshot(
save_memory_string))
save_memory_string = 'yEs'
self.assertTrue(
SaveSnapshotCommand._get_save_vm_memory_to_snapshot(
save_memory_string))
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')
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')
# Assert
vm.CreateSnapshot.called_with('new_snapshot', 'Created by CloudShell vCenterShell', False, True)
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)
