use of org.forgerock.openam.rest.query.QueryResponseHandler in project OpenAM by OpenRock.
the class ResourceTypesResourceTest method makeQueryResponseHandler.
private static QueryResponseHandler makeQueryResponseHandler() {
QueryResponseHandler mock = mock(QueryResponseHandler.class);
given(mock.handleResource(any(ResourceResponse.class))).willReturn(true);
return mock;
}
use of org.forgerock.openam.rest.query.QueryResponseHandler in project OpenAM by OpenRock.
the class ResourceTypesResourceTest method queryShouldHandleQueryException.
@Test
public void queryShouldHandleQueryException() throws Exception {
//given
setupExistingResourceTypeFromJson(jsonResourceType);
QueryRequest queryRequest = mock(QueryRequest.class);
QueryResponseHandler queryHandler = makeQueryResponseHandler();
Throwable t = new QueryException(QueryException.QueryErrorCode.FILTER_BOOLEAN_LITERAL_FALSE);
when(resourceTypeService.getResourceTypesData(any(Subject.class), anyString())).thenThrow(t);
//when
Promise<QueryResponse, ResourceException> promise = resourceTypesResource.queryCollection(mockServerContext, queryRequest, queryHandler);
//then
assertThat(promise).failedWithResourceException().withCode(ResourceException.BAD_REQUEST);
}
use of org.forgerock.openam.rest.query.QueryResponseHandler in project OpenAM by OpenRock.
the class ResourceTypesResourceTest method shouldHandleQueryPageLargerThanResults.
@Test
public void shouldHandleQueryPageLargerThanResults() throws Exception {
//given
QueryRequest queryRequest = makeMockQueryRequest();
QueryResponseHandler queryHandler = makeQueryResponseHandler();
final int lastPageSize = 5;
final int pageOffset = 10;
final int resultSize = 11;
Map<String, Map<String, Set<String>>> resourceTypes = new HashMap<>();
for (int i = 0; i < resultSize; i++) {
resourceTypes.put(UUID.randomUUID().toString(), rawData);
}
when(resourceTypeService.getResourceTypesData(any(Subject.class), anyString())).thenReturn(resourceTypes);
when(resourceTypeService.getResourceType(any(Subject.class), anyString(), anyString())).thenReturn(EntitlementUtils.resourceTypeFromMap(UUID.randomUUID().toString(), rawData));
when(queryRequest.getPagedResultsOffset()).thenReturn(pageOffset);
when(queryRequest.getPageSize()).thenReturn(lastPageSize);
Answer<Boolean> onlyFirstPage = new Answer<Boolean>() {
int count = pageOffset;
@Override
public Boolean answer(InvocationOnMock invocationOnMock) throws Throwable {
return count++ < resultSize;
}
};
when(queryHandler.handleResource(any(ResourceResponse.class))).thenAnswer(onlyFirstPage);
//when
Promise<QueryResponse, ResourceException> promise = resourceTypesResource.queryCollection(mockServerContext, queryRequest, queryHandler);
//then
QueryResponse queryResponse = promise.getOrThrowUninterruptibly();
verify(queryHandler, times(resultSize - pageOffset)).handleResource(any(ResourceResponse.class));
assertThat(queryResponse.getRemainingPagedResults()).isEqualTo(0);
}
use of org.forgerock.openam.rest.query.QueryResponseHandler in project OpenAM by OpenRock.
the class ResourceTypesResourceTest method queryShouldPageResults.
@Test
public void queryShouldPageResults() throws Exception {
//given
QueryRequest queryRequest = makeMockQueryRequest();
QueryResponseHandler queryHandler = makeQueryResponseHandler();
final int firstPageSize = 2;
final int pageOffset = 0;
final int resultSize = 10;
Map<String, Map<String, Set<String>>> resourceTypes = new HashMap<>();
for (int i = 0; i < resultSize; i++) {
resourceTypes.put(UUID.randomUUID().toString(), rawData);
}
when(resourceTypeService.getResourceTypesData(any(Subject.class), anyString())).thenReturn(resourceTypes);
when(resourceTypeService.getResourceType(any(Subject.class), anyString(), anyString())).thenReturn(EntitlementUtils.resourceTypeFromMap(UUID.randomUUID().toString(), rawData));
when(queryRequest.getPagedResultsOffset()).thenReturn(pageOffset);
when(queryRequest.getPageSize()).thenReturn(firstPageSize);
Answer<Boolean> onlyFirstPage = new Answer<Boolean>() {
int count = 0;
@Override
public Boolean answer(InvocationOnMock invocationOnMock) throws Throwable {
return count++ < firstPageSize;
}
};
when(queryHandler.handleResource(any(ResourceResponse.class))).thenAnswer(onlyFirstPage);
//when
Promise<QueryResponse, ResourceException> promise = resourceTypesResource.queryCollection(mockServerContext, queryRequest, queryHandler);
//then
QueryResponse queryResponse = promise.getOrThrowUninterruptibly();
verify(queryHandler, times(firstPageSize)).handleResource(any(ResourceResponse.class));
assertThat(queryResponse.getRemainingPagedResults()).isEqualTo(resultSize - firstPageSize);
}
use of org.forgerock.openam.rest.query.QueryResponseHandler in project OpenAM by OpenRock.
the class ResourceTypesResourceTest method queryShouldHandleAllResults.
@Test
public void queryShouldHandleAllResults() throws Exception {
//given
QueryRequest queryRequest = makeMockQueryRequest();
QueryResponseHandler queryHandler = makeQueryResponseHandler();
Map<String, Map<String, Set<String>>> resourceTypes = new HashMap<>();
final int resultSize = 10;
for (int i = 0; i < resultSize; i++) {
resourceTypes.put(UUID.randomUUID().toString(), rawData);
}
when(resourceTypeService.getResourceTypesData(any(Subject.class), anyString())).thenReturn(resourceTypes);
when(resourceTypeService.getResourceType(any(Subject.class), anyString(), anyString())).thenReturn(EntitlementUtils.resourceTypeFromMap(UUID.randomUUID().toString(), rawData));
//when
Promise<QueryResponse, ResourceException> promise = resourceTypesResource.queryCollection(mockServerContext, queryRequest, queryHandler);
//then
QueryResponse queryResponse = promise.getOrThrowUninterruptibly();
verify(queryHandler, times(resultSize)).handleResource(any(ResourceResponse.class));
assertThat(queryResponse.getRemainingPagedResults()).isEqualTo(0);
}
Aggregations