private static VDSReturnValue createGetVGInfoReturnValue(List<LUNs> luns) {
VDSReturnValue returnValue = new VDSReturnValue();
returnValue.setSucceeded(true);
returnValue.setReturnValue(luns);
return returnValue;
}
private void expectGetDeviceList() {
VDSReturnValue returnValue = new VDSReturnValue();
returnValue.setSucceeded(true);
returnValue.setReturnValue(setUpLunsFromDeviceList());
when(vdsBrokerFrontendMock.runVdsCommand(
eq(VDSCommandType.GetDeviceList), any(GetDeviceListVDSCommandParameters.class)))
.thenReturn(returnValue);
}
private void mockGetStorageDomainInfoVdsCommand(StorageDomainStatic storageDomain) {
Pair<StorageDomainStatic, Guid> pairResult = new Pair<>(storageDomain, null);
VDSReturnValue returnValueForGetStorageDomainInfo = new VDSReturnValue();
returnValueForGetStorageDomainInfo.setSucceeded(true);
returnValueForGetStorageDomainInfo.setReturnValue(pairResult);
when(vdsBrokerFrontend.runVdsCommand(
eq(VDSCommandType.HSMGetStorageDomainInfo),
any(HSMGetStorageDomainInfoVDSCommandParameters.class)))
.thenReturn(returnValueForGetStorageDomainInfo);
}
private VDSReturnValue createGetStorageDomainInfoReturnValue() {
VDSReturnValue returnValue = new VDSReturnValue();
returnValue.setSucceeded(true);
StorageDomain storageDomain = new StorageDomain();
storageDomain.setId(storageDomainId);
Pair<StorageDomainStatic, Object> pair = new Pair<>(storageDomain.getStorageStaticData(), null);
returnValue.setReturnValue(pair);
return returnValue;
}
private void mockBackend(boolean succeeded, EngineError errorCode) {
doReturn(backend).when(cmd).getBackend();
doNothing().when(cmd).startSubStep();
doReturn(asyncTaskToBeReturned).when(cmd).handleTaskReturn(asyncTaskToBeReturned);
doNothing().when(cmd).updateBricksWithTaskID(asyncTaskToBeReturned);
VDSReturnValue vdsReturnValue = new VDSReturnValue();
vdsReturnValue.setReturnValue(asyncTaskToBeReturned);
vdsReturnValue.setSucceeded(succeeded);
if (!succeeded) {
vdsReturnValue.setVdsError(new VDSError(errorCode, ""));
}
when(vdsBrokerFrontend.runVdsCommand(
eq(VDSCommandType.StartRemoveGlusterVolumeBricks),
argThat(anyGlusterVolumeRemoveBricksVDSParameters())))
.thenReturn(vdsReturnValue);
}
/** Test query execution */
private void internalExecuteQuery() {
// Run 'GetDeviceList' command
VDSReturnValue returnValue = new VDSReturnValue();
returnValue.setSucceeded(true);
returnValue.setReturnValue(lunsInput);
when(vdsBrokerFrontendMock.RunVdsCommand(
eq(VDSCommandType.GetDeviceList), any(GetDeviceListVDSCommandParameters.class)))
.thenReturn(returnValue);
// Return 'lunsFromDb'
when(lunDaoMock.getAll()).thenReturn(lunsFromDb);
// Execute command
getQuery().executeQueryCommand();
// Assert the query's results
List<LUNs> lunsActual = getQuery().getQueryReturnValue().getReturnValue();
assertEquals(lunsExpected, lunsActual);
}
private VDSReturnValue getVDSReturnValue() {
VDSReturnValue returnValue = new VDSReturnValue();
returnValue.setSucceeded(true);
returnValue.setReturnValue(expectedServers);
return returnValue;
}