@Override public BackendInternal getBackend() { BackendInternal backend = mock(BackendInternal.class); when(backend.runInternalQuery( eq(VdcQueryType.GetVmsFromExportDomain), any(VdcQueryParametersBase.class))) .thenReturn(createVmQueryResult()); when(backend.runInternalQuery( eq(VdcQueryType.IsVmWithSameNameExist), any(VdcQueryParametersBase.class))) .thenReturn(createDuplicateResult()); return backend; }
private boolean runQuery(HttpServletResponse response, String sessionID) { boolean returnValue = false; log.debug("Calling ValidateSession query"); VdcQueryReturnValue queryReturnValue = backend.runInternalQuery( VdcQueryType.ValidateSession, new VdcQueryParametersBase(sessionID)); if (queryReturnValue != null) { returnValue = queryReturnValue.getSucceeded(); if (returnValue) { DbUser user = queryReturnValue.getReturnValue(); // We get the user name only in case the validation succeeded, and the user is an // administrator if (user.isAdmin()) { log.debug("Getting user name"); printUPNToResponse(response, getUPN(user)); } else { log.error("User '{}' is not authorized to perform operation", user.getLoginName()); returnValue = false; } } } else { log.error("Got NULL from backend.RunQuery"); } return returnValue; }
private void setupMock() throws Exception { vdsBrokerFrontend = mock(VDSBrokerFrontend.class); clusterUtils = mock(ClusterUtils.class); vdsDaoMock = mock(VdsDao.class); vdsGroupDaoMock = mock(VdsGroupDao.class); dbUtils = mock(GlusterDBUtils.class); backendInternal = mock(BackendInternal.class); doReturn(vdsBrokerFrontend).when(getQuery()).getResourceManager(); doReturn(clusterUtils).when(getQuery()).getClusterUtils(); doReturn(dbUtils).when(getQuery()).getDbUtils(); doReturn(getVds(VDSStatus.Up)).when(clusterUtils).getUpServer(CLUSTER_ID); doReturn(backendInternal).when(getQuery()).getBackend(); VDSReturnValue returnValue = getVDSReturnValue(); when(vdsBrokerFrontend.RunVdsCommand( eq(VDSCommandType.GlusterServersList), any(VDSParametersBase.class))) .thenReturn(returnValue); VdcQueryReturnValue vdcReturnValue = getVdcReturnValue(); when(backendInternal.runInternalQuery( eq(VdcQueryType.GetServerSSHKeyFingerprint), any(VdcQueryParametersBase.class), any(EngineContext.class))) .thenReturn(vdcReturnValue); doReturn(params.getClusterId()).when(getQueryParameters()).getClusterId(); doReturn(true).when(getQueryParameters()).isServerKeyFingerprintRequired(); doReturn(vdsDaoMock).when(clusterUtils).getVdsDao(); doReturn(vdsGroupDaoMock).when(getQuery()).getVdsGroupDao(); doReturn(serversList).when(vdsDaoMock).getAllForVdsGroup(CLUSTER_ID); }