Examples with QueryReturnValue org.ovirt.engine.core.common.queries.QueryReturnValue used on opensource projects

Example 41 with QueryReturnValue

use of org.ovirt.engine.core.common.queries.QueryReturnValue in project ovirt-engine by oVirt.

the class GetVmConfigurationBySnapshotQueryTest method testQuery.

@Test
public void testQuery() throws Exception {
    GetVmConfigurationBySnapshotQuery<IdQueryParameters> query = setupQueryBySnapshotId(existingSnapshotId);
    VM vm = new VM();
    doReturn(vm).when(snapshotVmConfigurationHelper).getVmFromConfiguration(any(), any(), any());
    query.execute();
    QueryReturnValue returnValue = query.getQueryReturnValue();
    assertNotNull("Return value from query cannot be null", returnValue);
    VM returnedVm = returnValue.getReturnValue();
    assertEquals(vm, returnedVm);
}

Example 42 with QueryReturnValue

use of org.ovirt.engine.core.common.queries.QueryReturnValue in project ovirt-engine by oVirt.

the class GetVmsByStorageDomainQueryTest method testExecuteQueryCommandNoDisks.

@Test
public void testExecuteQueryCommandNoDisks() {
    List<VM> vmsOfDomain = new ArrayList<>();
    VM vm1 = mock(VM.class);
    VM vm2 = mock(VM.class);
    vmsOfDomain.add(vm1);
    vmsOfDomain.add(vm2);
    when(vmDao.getAllForStorageDomain(domainId)).thenReturn(vmsOfDomain);
    QueryReturnValue returnValue = new QueryReturnValue();
    returnValue.setSucceeded(true);
    List<DiskImage> diskImages = new ArrayList<>();
    returnValue.setReturnValue(diskImages);
    doReturn(returnValue).when(query).getAllDisksByStorageDomain(domainId);
    query.executeQueryCommand();
    List<VM> vms = query.getQueryReturnValue().getReturnValue();
    assertEquals(2, vms.size());
}

Example 43 with QueryReturnValue

use of org.ovirt.engine.core.common.queries.QueryReturnValue in project ovirt-engine by oVirt.

the class StorageDomainValidatorTest method invalidRunningVmsAndVmLeasesForBackupDomain.

@Test
public void invalidRunningVmsAndVmLeasesForBackupDomain() {
    QueryReturnValue ret = new QueryReturnValue();
    List<VmBase> vmLeases = new ArrayList<>();
    VM vmWithLease = new VM();
    vmWithLease.setName("firstVM");
    vmLeases.add(vmWithLease.getStaticData());
    ret.setReturnValue(vmLeases);
    ret.setSucceeded(true);
    // VM1
    VM vm1 = new VM();
    vm1.setName("firstVM");
    Map<Guid, Disk> attachedDisksForVm1 = new HashMap<>();
    DiskImage diskVm1 = new DiskImage();
    diskVm1.setStorageIds(new ArrayList<>(Collections.singletonList(domain.getId())));
    diskVm1.setPlugged(true);
    attachedDisksForVm1.put(Guid.newGuid(), diskVm1);
    vm1.setDiskMap(attachedDisksForVm1);
    // VM2
    VM vm2 = new VM();
    vm2.setName("secondVM");
    Map<Guid, Disk> attachedDisksForVm2 = new HashMap<>();
    DiskImage diskVm2 = new DiskImage();
    diskVm2.setStorageIds(new ArrayList<>(Collections.singletonList(domain.getId())));
    diskVm2.setPlugged(true);
    attachedDisksForVm2.put(Guid.newGuid(), diskVm2);
    vm2.setDiskMap(attachedDisksForVm2);
    List<VM> runningVMs = new ArrayList<>();
    runningVMs.add(vm1);
    runningVMs.add(vm2);
    when(vmDao.getAllActiveForStorageDomain(any())).thenReturn(runningVMs);
    doReturn(ret).when(validator).getEntitiesWithLeaseIdForStorageDomain(any());
    assertThat(validator.isRunningVmsOrVmLeasesForBackupDomain(vmHandler), failsWith(EngineMessage.ACTION_TYPE_FAILED_RUNNING_VM_OR_VM_LEASES_PRESENT_ON_STORAGE_DOMAIN));
}

Example 44 with QueryReturnValue

use of org.ovirt.engine.core.common.queries.QueryReturnValue in project ovirt-engine by oVirt.

the class StorageDomainValidatorTest method validRunningVmsOrVmLeasesForBackupDomain.

@Test
public void validRunningVmsOrVmLeasesForBackupDomain() {
    when(vmDao.getAllActiveForStorageDomain(any())).thenReturn(Collections.EMPTY_LIST);
    QueryReturnValue ret = new QueryReturnValue();
    ret.setReturnValue(new ArrayList<VmBase>());
    ret.setSucceeded(true);
    doReturn(ret).when(validator).getEntitiesWithLeaseIdForStorageDomain(any());
    assertThat(validator.isRunningVmsOrVmLeasesForBackupDomain(vmHandler), isValid());
}

Example 45 with QueryReturnValue

use of org.ovirt.engine.core.common.queries.QueryReturnValue in project ovirt-engine by oVirt.

the class StorageDomainValidatorTest method invalidVmLeasesQueryForBackupDomain.

@Test
public void invalidVmLeasesQueryForBackupDomain() {
    QueryReturnValue ret = new QueryReturnValue();
    ret.setSucceeded(false);
    doReturn(ret).when(validator).getEntitiesWithLeaseIdForStorageDomain(any());
    assertThat(validator.isRunningVmsOrVmLeasesForBackupDomain(vmHandler), failsWith(EngineMessage.ACTION_TYPE_FAILED_RETRIEVE_VMS_FOR_WITH_LEASES));
}