mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
5e90b75c98
Added a new flag 'checkBeforeCleanup' to StopCommand based on which check is done to see if VM is running in HV host. If VM is running then in this case it is not stopped and the operation bails out. Also modified the MS code to call the StopCommand with appropriate value for the flag based on the context. Currently it is only set to 'true' when called from the new vmsync logic based on powerstate of VM. For rest it is set to 'false' meaning no change in behaviour.
508 lines
21 KiB
Java
508 lines
21 KiB
Java
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
package com.cloud.vm;
|
|
|
|
import static org.mockito.Matchers.any;
|
|
import static org.mockito.Matchers.anyLong;
|
|
import static org.mockito.Matchers.isA;
|
|
import static org.mockito.Mockito.doNothing;
|
|
import static org.mockito.Mockito.doReturn;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.when;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import junit.framework.Assert;
|
|
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.mockito.Matchers;
|
|
import org.mockito.Mock;
|
|
import org.mockito.MockitoAnnotations;
|
|
import org.mockito.Spy;
|
|
|
|
import org.apache.cloudstack.api.command.user.vm.RestoreVMCmd;
|
|
import org.apache.cloudstack.engine.orchestration.service.NetworkOrchestrationService;
|
|
import org.apache.cloudstack.engine.orchestration.service.VolumeOrchestrationService;
|
|
import org.apache.cloudstack.framework.config.ConfigDepot;
|
|
import org.apache.cloudstack.framework.config.dao.ConfigurationDao;
|
|
import org.apache.cloudstack.storage.datastore.db.PrimaryDataStoreDao;
|
|
import org.apache.cloudstack.storage.datastore.db.StoragePoolVO;
|
|
|
|
import com.cloud.agent.AgentManager;
|
|
import com.cloud.agent.api.CheckVirtualMachineAnswer;
|
|
import com.cloud.agent.api.CheckVirtualMachineCommand;
|
|
import com.cloud.agent.api.Command;
|
|
import com.cloud.agent.api.MigrateWithStorageAnswer;
|
|
import com.cloud.agent.api.MigrateWithStorageCommand;
|
|
import com.cloud.agent.api.MigrateWithStorageCompleteAnswer;
|
|
import com.cloud.agent.api.MigrateWithStorageCompleteCommand;
|
|
import com.cloud.agent.api.MigrateWithStorageReceiveAnswer;
|
|
import com.cloud.agent.api.MigrateWithStorageReceiveCommand;
|
|
import com.cloud.agent.api.MigrateWithStorageSendAnswer;
|
|
import com.cloud.agent.api.MigrateWithStorageSendCommand;
|
|
import com.cloud.agent.api.PrepareForMigrationAnswer;
|
|
import com.cloud.agent.api.PrepareForMigrationCommand;
|
|
import com.cloud.agent.api.ScaleVmAnswer;
|
|
import com.cloud.agent.api.ScaleVmCommand;
|
|
import com.cloud.agent.api.StopAnswer;
|
|
import com.cloud.agent.api.StopCommand;
|
|
import com.cloud.capacity.CapacityManager;
|
|
import com.cloud.dc.dao.ClusterDao;
|
|
import com.cloud.dc.dao.DataCenterDao;
|
|
import com.cloud.dc.dao.HostPodDao;
|
|
import com.cloud.deploy.DeployDestination;
|
|
import com.cloud.deploy.DeploymentPlanner;
|
|
import com.cloud.exception.ConcurrentOperationException;
|
|
import com.cloud.exception.ManagementServerException;
|
|
import com.cloud.exception.OperationTimedoutException;
|
|
import com.cloud.exception.ResourceUnavailableException;
|
|
import com.cloud.exception.VirtualMachineMigrationException;
|
|
import com.cloud.host.HostVO;
|
|
import com.cloud.host.dao.HostDao;
|
|
import com.cloud.hypervisor.Hypervisor.HypervisorType;
|
|
import com.cloud.hypervisor.HypervisorGuru;
|
|
import com.cloud.hypervisor.HypervisorGuruManager;
|
|
import com.cloud.offering.ServiceOffering;
|
|
import com.cloud.service.ServiceOfferingVO;
|
|
import com.cloud.storage.dao.DiskOfferingDao;
|
|
import com.cloud.storage.dao.StoragePoolHostDao;
|
|
import com.cloud.storage.dao.VMTemplateDao;
|
|
import com.cloud.storage.dao.VolumeDao;
|
|
import com.cloud.storage.Storage.ProvisioningType;
|
|
import com.cloud.storage.DiskOfferingVO;
|
|
import com.cloud.storage.StoragePool;
|
|
import com.cloud.storage.StoragePoolHostVO;
|
|
import com.cloud.storage.Volume;
|
|
import com.cloud.storage.VolumeVO;
|
|
import com.cloud.storage.VMTemplateVO;
|
|
import com.cloud.user.Account;
|
|
import com.cloud.user.AccountVO;
|
|
import com.cloud.user.UserVO;
|
|
import com.cloud.user.dao.AccountDao;
|
|
import com.cloud.user.dao.UserDao;
|
|
import com.cloud.utils.Pair;
|
|
import com.cloud.utils.db.EntityManager;
|
|
import com.cloud.utils.exception.CloudRuntimeException;
|
|
import com.cloud.vm.VirtualMachine.Event;
|
|
import com.cloud.vm.VirtualMachine.State;
|
|
import com.cloud.vm.dao.UserVmDao;
|
|
import com.cloud.vm.dao.UserVmDetailsDao;
|
|
import com.cloud.vm.dao.VMInstanceDao;
|
|
import com.cloud.vm.snapshot.VMSnapshotManager;
|
|
|
|
public class VirtualMachineManagerImplTest {
|
|
|
|
@Spy
|
|
VirtualMachineManagerImpl _vmMgr = new VirtualMachineManagerImpl();
|
|
@Mock
|
|
VolumeOrchestrationService _storageMgr;
|
|
@Mock
|
|
Account _account;
|
|
@Mock
|
|
CapacityManager _capacityMgr;
|
|
@Mock
|
|
AgentManager _agentMgr;
|
|
@Mock
|
|
AccountDao _accountDao;
|
|
@Mock
|
|
ConfigurationDao _configDao;
|
|
@Mock
|
|
HostDao _hostDao;
|
|
@Mock
|
|
UserDao _userDao;
|
|
@Mock
|
|
UserVmDao _vmDao;
|
|
@Mock
|
|
ItWorkDao _workDao;
|
|
@Mock
|
|
VMInstanceDao _vmInstanceDao;
|
|
@Mock
|
|
VMTemplateDao _templateDao;
|
|
@Mock
|
|
VolumeDao _volsDao;
|
|
@Mock
|
|
RestoreVMCmd _restoreVMCmd;
|
|
@Mock
|
|
AccountVO _accountMock;
|
|
@Mock
|
|
UserVO _userMock;
|
|
@Mock
|
|
UserVmVO _vmMock;
|
|
@Mock
|
|
VMInstanceVO _vmInstance;
|
|
@Mock
|
|
HostVO _host;
|
|
@Mock
|
|
VMTemplateVO _templateMock;
|
|
@Mock
|
|
VolumeVO _volumeMock;
|
|
@Mock
|
|
List<VolumeVO> _rootVols;
|
|
@Mock
|
|
ItWorkVO _work;
|
|
@Mock
|
|
HostVO hostVO;
|
|
@Mock
|
|
UserVmDetailVO _vmDetailVO;
|
|
|
|
@Mock
|
|
ClusterDao _clusterDao;
|
|
@Mock
|
|
HostPodDao _podDao;
|
|
@Mock
|
|
DataCenterDao _dcDao;
|
|
@Mock
|
|
DiskOfferingDao _diskOfferingDao;
|
|
@Mock
|
|
PrimaryDataStoreDao _storagePoolDao;
|
|
@Mock
|
|
UserVmDetailsDao _vmDetailsDao;
|
|
@Mock
|
|
StoragePoolHostDao _poolHostDao;
|
|
@Mock
|
|
NetworkOrchestrationService _networkMgr;
|
|
@Mock
|
|
HypervisorGuruManager _hvGuruMgr;
|
|
@Mock
|
|
VMSnapshotManager _vmSnapshotMgr;
|
|
|
|
// Mock objects for vm migration with storage test.
|
|
@Mock
|
|
DiskOfferingVO _diskOfferingMock;
|
|
@Mock
|
|
StoragePoolVO _srcStoragePoolMock;
|
|
@Mock
|
|
StoragePoolVO _destStoragePoolMock;
|
|
@Mock
|
|
HostVO _srcHostMock;
|
|
@Mock
|
|
HostVO _destHostMock;
|
|
@Mock
|
|
Map<Volume, StoragePool> _volumeToPoolMock;
|
|
@Mock
|
|
EntityManager _entityMgr;
|
|
@Mock
|
|
ConfigDepot _configDepot;
|
|
|
|
@Before
|
|
public void setup() {
|
|
MockitoAnnotations.initMocks(this);
|
|
|
|
_vmMgr._templateDao = _templateDao;
|
|
_vmMgr._volsDao = _volsDao;
|
|
_vmMgr.volumeMgr = _storageMgr;
|
|
_vmMgr._capacityMgr = _capacityMgr;
|
|
_vmMgr._hostDao = _hostDao;
|
|
_vmMgr._nodeId = 1L;
|
|
_vmMgr._workDao = _workDao;
|
|
_vmMgr._agentMgr = _agentMgr;
|
|
_vmMgr._podDao = _podDao;
|
|
_vmMgr._clusterDao = _clusterDao;
|
|
_vmMgr._dcDao = _dcDao;
|
|
_vmMgr._diskOfferingDao = _diskOfferingDao;
|
|
_vmMgr._storagePoolDao = _storagePoolDao;
|
|
_vmMgr._poolHostDao = _poolHostDao;
|
|
_vmMgr._networkMgr = _networkMgr;
|
|
_vmMgr._hvGuruMgr = _hvGuruMgr;
|
|
_vmMgr._vmSnapshotMgr = _vmSnapshotMgr;
|
|
_vmMgr._vmDao = _vmInstanceDao;
|
|
_vmMgr._uservmDetailsDao = _vmDetailsDao;
|
|
_vmMgr._entityMgr = _entityMgr;
|
|
_vmMgr._configDepot = _configDepot;
|
|
|
|
when(_vmMock.getId()).thenReturn(314l);
|
|
when(_vmInstance.getId()).thenReturn(1L);
|
|
when(_vmInstance.getServiceOfferingId()).thenReturn(2L);
|
|
when(_vmInstance.getInstanceName()).thenReturn("myVm");
|
|
when(_vmInstance.getHostId()).thenReturn(2L);
|
|
when(_vmInstance.getType()).thenReturn(VirtualMachine.Type.User);
|
|
when(_host.getId()).thenReturn(1L);
|
|
when(_hostDao.findById(anyLong())).thenReturn(null);
|
|
when(_entityMgr.findById(Matchers.eq(ServiceOffering.class), anyLong())).thenReturn(getSvcoffering(512));
|
|
when(_workDao.persist(_work)).thenReturn(_work);
|
|
when(_workDao.update("1", _work)).thenReturn(true);
|
|
when(_work.getId()).thenReturn("1");
|
|
doNothing().when(_work).setStep(ItWorkVO.Step.Done);
|
|
when(_vmInstanceDao.findByUuid(any(String.class))).thenReturn(_vmMock);
|
|
//doNothing().when(_volsDao).detachVolume(anyLong());
|
|
//when(_work.setStep(ItWorkVO.Step.Done)).thenReturn("1");
|
|
|
|
}
|
|
|
|
@Test(expected = CloudRuntimeException.class)
|
|
public void testScaleVM1() throws Exception {
|
|
|
|
DeployDestination dest = new DeployDestination(null, null, null, _host);
|
|
long l = 1L;
|
|
|
|
when(_vmInstanceDao.findById(anyLong())).thenReturn(_vmInstance);
|
|
_vmMgr.migrateForScale(_vmInstance.getUuid(), l, dest, l);
|
|
|
|
}
|
|
|
|
@Test(expected = CloudRuntimeException.class)
|
|
public void testScaleVM2() throws Exception {
|
|
|
|
new DeployDestination(null, null, null, _host);
|
|
doReturn(3L).when(_vmInstance).getId();
|
|
when(_vmInstanceDao.findById(anyLong())).thenReturn(_vmInstance);
|
|
ServiceOfferingVO newServiceOffering = getSvcoffering(512);
|
|
doReturn(1L).when(_vmInstance).getHostId();
|
|
doReturn(hostVO).when(_hostDao).findById(1L);
|
|
doReturn(1L).when(_vmInstance).getDataCenterId();
|
|
doReturn(1L).when(hostVO).getClusterId();
|
|
when(CapacityManager.CpuOverprovisioningFactor.valueIn(1L)).thenReturn(1.0f);
|
|
ScaleVmCommand reconfigureCmd =
|
|
new ScaleVmCommand("myVmName", newServiceOffering.getCpu(), newServiceOffering.getSpeed(), newServiceOffering.getSpeed(), newServiceOffering.getRamSize(),
|
|
newServiceOffering.getRamSize(), newServiceOffering.getLimitCpuUse());
|
|
new ScaleVmAnswer(reconfigureCmd, true, "details");
|
|
when(_agentMgr.send(2l, reconfigureCmd)).thenReturn(null);
|
|
_vmMgr.reConfigureVm(_vmInstance.getUuid(), getSvcoffering(256), false);
|
|
|
|
}
|
|
|
|
@Test(expected = CloudRuntimeException.class)
|
|
public void testScaleVM3() throws Exception {
|
|
|
|
/*VirtualMachineProfile profile = new VirtualMachineProfileImpl(vm);
|
|
|
|
Long srcHostId = vm.getHostId();
|
|
Long oldSvcOfferingId = vm.getServiceOfferingId();
|
|
if (srcHostId == null) {
|
|
throw new CloudRuntimeException("Unable to scale the vm because it doesn't have a host id");
|
|
}*/
|
|
|
|
when(_vmInstance.getHostId()).thenReturn(null);
|
|
when(_vmInstanceDao.findById(anyLong())).thenReturn(_vmInstance);
|
|
when(_vmInstanceDao.findByUuid(any(String.class))).thenReturn(_vmInstance);
|
|
DeploymentPlanner.ExcludeList excludeHostList = new DeploymentPlanner.ExcludeList();
|
|
_vmMgr.findHostAndMigrate(_vmInstance.getUuid(), 2l, excludeHostList);
|
|
|
|
}
|
|
|
|
private ServiceOfferingVO getSvcoffering(int ramSize) {
|
|
|
|
String name = "name";
|
|
String displayText = "displayText";
|
|
int cpu = 1;
|
|
//int ramSize = 256;
|
|
int speed = 128;
|
|
|
|
boolean ha = false;
|
|
boolean useLocalStorage = false;
|
|
|
|
ServiceOfferingVO serviceOffering =
|
|
new ServiceOfferingVO(name, cpu, ramSize, speed, null, null, ha, displayText, ProvisioningType.THIN, useLocalStorage, false, null, false, null, false);
|
|
return serviceOffering;
|
|
}
|
|
|
|
private void initializeMockConfigForMigratingVmWithVolumes() throws OperationTimedoutException, ResourceUnavailableException {
|
|
|
|
// Mock the source and destination hosts.
|
|
when(_srcHostMock.getId()).thenReturn(5L);
|
|
when(_destHostMock.getId()).thenReturn(6L);
|
|
when(_hostDao.findById(5L)).thenReturn(_srcHostMock);
|
|
when(_hostDao.findById(6L)).thenReturn(_destHostMock);
|
|
|
|
// Mock the vm being migrated.
|
|
when(_vmMock.getId()).thenReturn(1L);
|
|
when(_vmMock.getHypervisorType()).thenReturn(HypervisorType.XenServer);
|
|
when(_vmMock.getState()).thenReturn(State.Running).thenReturn(State.Running).thenReturn(State.Migrating).thenReturn(State.Migrating);
|
|
when(_vmMock.getHostId()).thenReturn(5L);
|
|
when(_vmInstance.getId()).thenReturn(1L);
|
|
when(_vmInstance.getServiceOfferingId()).thenReturn(2L);
|
|
when(_vmInstance.getInstanceName()).thenReturn("myVm");
|
|
when(_vmInstance.getHostId()).thenReturn(5L);
|
|
when(_vmInstance.getType()).thenReturn(VirtualMachine.Type.User);
|
|
when(_vmInstance.getState()).thenReturn(State.Running).thenReturn(State.Running).thenReturn(State.Migrating).thenReturn(State.Migrating);
|
|
|
|
// Mock the work item.
|
|
when(_workDao.persist(any(ItWorkVO.class))).thenReturn(_work);
|
|
when(_workDao.update("1", _work)).thenReturn(true);
|
|
when(_work.getId()).thenReturn("1");
|
|
doNothing().when(_work).setStep(ItWorkVO.Step.Done);
|
|
|
|
// Mock the vm guru and the user vm object that gets returned.
|
|
_vmMgr._vmGurus = new HashMap<VirtualMachine.Type, VirtualMachineGuru>();
|
|
// UserVmManagerImpl userVmManager = mock(UserVmManagerImpl.class);
|
|
// _vmMgr.registerGuru(VirtualMachine.Type.User, userVmManager);
|
|
|
|
// Mock the iteration over all the volumes of an instance.
|
|
Iterator<VolumeVO> volumeIterator = mock(Iterator.class);
|
|
when(_volsDao.findUsableVolumesForInstance(anyLong())).thenReturn(_rootVols);
|
|
when(_rootVols.iterator()).thenReturn(volumeIterator);
|
|
when(volumeIterator.hasNext()).thenReturn(true, false);
|
|
when(volumeIterator.next()).thenReturn(_volumeMock);
|
|
|
|
// Mock the disk offering and pool objects for a volume.
|
|
when(_volumeMock.getDiskOfferingId()).thenReturn(5L);
|
|
when(_volumeMock.getPoolId()).thenReturn(200L);
|
|
when(_diskOfferingDao.findById(anyLong())).thenReturn(_diskOfferingMock);
|
|
when(_storagePoolDao.findById(anyLong())).thenReturn(_srcStoragePoolMock);
|
|
|
|
// Mock the volume to pool mapping.
|
|
when(_volumeToPoolMock.get(_volumeMock)).thenReturn(_destStoragePoolMock);
|
|
when(_destStoragePoolMock.getId()).thenReturn(201L);
|
|
when(_srcStoragePoolMock.getId()).thenReturn(200L);
|
|
when(_destStoragePoolMock.isLocal()).thenReturn(false);
|
|
when(_diskOfferingMock.getUseLocalStorage()).thenReturn(false);
|
|
when(_poolHostDao.findByPoolHost(anyLong(), anyLong())).thenReturn(mock(StoragePoolHostVO.class));
|
|
|
|
// Mock hypervisor guru.
|
|
HypervisorGuru guruMock = mock(HypervisorGuru.class);
|
|
when(_hvGuruMgr.getGuru(HypervisorType.XenServer)).thenReturn(guruMock);
|
|
|
|
when(_srcHostMock.getClusterId()).thenReturn(3L);
|
|
when(_destHostMock.getClusterId()).thenReturn(3L);
|
|
|
|
// Mock the commands and answers to the agent.
|
|
PrepareForMigrationAnswer prepAnswerMock = mock(PrepareForMigrationAnswer.class);
|
|
when(prepAnswerMock.getResult()).thenReturn(true);
|
|
when(_agentMgr.send(anyLong(), isA(PrepareForMigrationCommand.class))).thenReturn(prepAnswerMock);
|
|
|
|
MigrateWithStorageAnswer migAnswerMock = mock(MigrateWithStorageAnswer.class);
|
|
when(migAnswerMock.getResult()).thenReturn(true);
|
|
when(_agentMgr.send(anyLong(), isA(MigrateWithStorageCommand.class))).thenReturn(migAnswerMock);
|
|
|
|
MigrateWithStorageReceiveAnswer migRecAnswerMock = mock(MigrateWithStorageReceiveAnswer.class);
|
|
when(migRecAnswerMock.getResult()).thenReturn(true);
|
|
when(_agentMgr.send(anyLong(), isA(MigrateWithStorageReceiveCommand.class))).thenReturn(migRecAnswerMock);
|
|
|
|
MigrateWithStorageSendAnswer migSendAnswerMock = mock(MigrateWithStorageSendAnswer.class);
|
|
when(migSendAnswerMock.getResult()).thenReturn(true);
|
|
when(_agentMgr.send(anyLong(), isA(MigrateWithStorageSendCommand.class))).thenReturn(migSendAnswerMock);
|
|
|
|
MigrateWithStorageCompleteAnswer migCompleteAnswerMock = mock(MigrateWithStorageCompleteAnswer.class);
|
|
when(migCompleteAnswerMock.getResult()).thenReturn(true);
|
|
when(_agentMgr.send(anyLong(), isA(MigrateWithStorageCompleteCommand.class))).thenReturn(migCompleteAnswerMock);
|
|
|
|
CheckVirtualMachineAnswer checkVmAnswerMock = mock(CheckVirtualMachineAnswer.class);
|
|
when(checkVmAnswerMock.getResult()).thenReturn(true);
|
|
when(checkVmAnswerMock.getState()).thenReturn(State.Running);
|
|
when(_agentMgr.send(anyLong(), isA(CheckVirtualMachineCommand.class))).thenReturn(checkVmAnswerMock);
|
|
|
|
// Mock the state transitions of vm.
|
|
Pair<Long, Long> opaqueMock = new Pair<Long, Long>(_vmMock.getHostId(), _destHostMock.getId());
|
|
when(_vmSnapshotMgr.hasActiveVMSnapshotTasks(anyLong())).thenReturn(false);
|
|
when(_vmInstanceDao.updateState(State.Running, Event.MigrationRequested, State.Migrating, _vmMock, opaqueMock)).thenReturn(true);
|
|
when(_vmInstanceDao.updateState(State.Migrating, Event.OperationSucceeded, State.Running, _vmMock, opaqueMock)).thenReturn(true);
|
|
}
|
|
|
|
// Check migration of a vm with its volumes within a cluster.
|
|
@Test
|
|
public void testMigrateWithVolumeWithinCluster() throws ResourceUnavailableException, ConcurrentOperationException, ManagementServerException,
|
|
VirtualMachineMigrationException, OperationTimedoutException {
|
|
|
|
initializeMockConfigForMigratingVmWithVolumes();
|
|
when(_srcHostMock.getClusterId()).thenReturn(3L);
|
|
when(_destHostMock.getClusterId()).thenReturn(3L);
|
|
|
|
_vmMgr.migrateWithStorage(_vmInstance.getUuid(), _srcHostMock.getId(), _destHostMock.getId(), _volumeToPoolMock);
|
|
}
|
|
|
|
// Check migration of a vm with its volumes across a cluster.
|
|
@Test
|
|
public void testMigrateWithVolumeAcrossCluster() throws ResourceUnavailableException, ConcurrentOperationException, ManagementServerException,
|
|
VirtualMachineMigrationException, OperationTimedoutException {
|
|
|
|
initializeMockConfigForMigratingVmWithVolumes();
|
|
when(_srcHostMock.getClusterId()).thenReturn(3L);
|
|
when(_destHostMock.getClusterId()).thenReturn(4L);
|
|
|
|
_vmMgr.migrateWithStorage(_vmInstance.getUuid(), _srcHostMock.getId(), _destHostMock.getId(), _volumeToPoolMock);
|
|
}
|
|
|
|
// Check migration of a vm fails when src and destination pool are not of same type; that is, one is shared and
|
|
// other is local.
|
|
@Test(expected = CloudRuntimeException.class)
|
|
public void testMigrateWithVolumeFail1() throws ResourceUnavailableException, ConcurrentOperationException, ManagementServerException,
|
|
VirtualMachineMigrationException, OperationTimedoutException {
|
|
|
|
initializeMockConfigForMigratingVmWithVolumes();
|
|
when(_srcHostMock.getClusterId()).thenReturn(3L);
|
|
when(_destHostMock.getClusterId()).thenReturn(3L);
|
|
|
|
when(_destStoragePoolMock.isLocal()).thenReturn(true);
|
|
when(_diskOfferingMock.getUseLocalStorage()).thenReturn(false);
|
|
|
|
_vmMgr.migrateWithStorage(_vmInstance.getUuid(), _srcHostMock.getId(), _destHostMock.getId(), _volumeToPoolMock);
|
|
}
|
|
|
|
// Check migration of a vm fails when vm is not in Running state.
|
|
@Test(expected = ConcurrentOperationException.class)
|
|
public void testMigrateWithVolumeFail2() throws ResourceUnavailableException, ConcurrentOperationException, ManagementServerException,
|
|
VirtualMachineMigrationException, OperationTimedoutException {
|
|
|
|
initializeMockConfigForMigratingVmWithVolumes();
|
|
when(_srcHostMock.getClusterId()).thenReturn(3L);
|
|
when(_destHostMock.getClusterId()).thenReturn(3L);
|
|
|
|
when(_vmMock.getState()).thenReturn(State.Stopped);
|
|
|
|
_vmMgr.migrateWithStorage(_vmInstance.getUuid(), _srcHostMock.getId(), _destHostMock.getId(), _volumeToPoolMock);
|
|
}
|
|
|
|
@Test
|
|
public void testSendStopWithOkAnswer() throws Exception {
|
|
VirtualMachineGuru guru = mock(VirtualMachineGuru.class);
|
|
VirtualMachine vm = mock(VirtualMachine.class);
|
|
VirtualMachineProfile profile = mock(VirtualMachineProfile.class);
|
|
StopAnswer answer = new StopAnswer(new StopCommand(vm, false, false), "ok", true);
|
|
when(profile.getVirtualMachine()).thenReturn(vm);
|
|
when(vm.getHostId()).thenReturn(1L);
|
|
when(_agentMgr.send(anyLong(), (Command)any())).thenReturn(answer);
|
|
|
|
boolean actual = _vmMgr.sendStop(guru, profile, false, false);
|
|
|
|
Assert.assertTrue(actual);
|
|
}
|
|
|
|
@Test
|
|
public void testSendStopWithFailAnswer() throws Exception {
|
|
VirtualMachineGuru guru = mock(VirtualMachineGuru.class);
|
|
VirtualMachine vm = mock(VirtualMachine.class);
|
|
VirtualMachineProfile profile = mock(VirtualMachineProfile.class);
|
|
StopAnswer answer = new StopAnswer(new StopCommand(vm, false, false), "fail", false);
|
|
when(profile.getVirtualMachine()).thenReturn(vm);
|
|
when(vm.getHostId()).thenReturn(1L);
|
|
when(_agentMgr.send(anyLong(), (Command)any())).thenReturn(answer);
|
|
|
|
boolean actual = _vmMgr.sendStop(guru, profile, false, false);
|
|
|
|
Assert.assertFalse(actual);
|
|
}
|
|
|
|
@Test
|
|
public void testSendStopWithNullAnswer() throws Exception {
|
|
VirtualMachineGuru guru = mock(VirtualMachineGuru.class);
|
|
VirtualMachine vm = mock(VirtualMachine.class);
|
|
VirtualMachineProfile profile = mock(VirtualMachineProfile.class);
|
|
when(profile.getVirtualMachine()).thenReturn(vm);
|
|
when(vm.getHostId()).thenReturn(1L);
|
|
when(_agentMgr.send(anyLong(), (Command)any())).thenReturn(null);
|
|
|
|
boolean actual = _vmMgr.sendStop(guru, profile, false, false);
|
|
|
|
Assert.assertFalse(actual);
|
|
}
|
|
}
|