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_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)