@Test
public void testHandleRequest__UnsupportedPropertyException() throws Exception {
Request request = createStrictMock(Request.class);
ResourceInstance resource = createStrictMock(ResourceInstance.class);
Query query = createMock(Query.class);
Predicate predicate = createMock(Predicate.class);
Renderer renderer = new DefaultRenderer();
UnsupportedPropertyException exception =
new UnsupportedPropertyException(Resource.Type.Cluster, Collections.singleton("foo"));
expect(request.getResource()).andReturn(resource);
expect(resource.getQuery()).andReturn(query);
expect(request.getPageRequest()).andReturn(null);
expect(request.getSortRequest()).andReturn(null);
expect(request.getRenderer()).andReturn(renderer);
expect(request.getFields()).andReturn(Collections.<String, TemporalInfo>emptyMap());
expect(request.getQueryPredicate()).andReturn(predicate);
query.setUserPredicate(predicate);
query.setPageRequest(null);
query.setSortRequest(null);
query.setRenderer(renderer);
expect(query.execute()).andThrow(exception);
replay(request, resource, query, predicate);
// test
ReadHandler handler = new ReadHandler();
Result result = handler.handleRequest(request);
assertEquals(ResultStatus.STATUS.BAD_REQUEST, result.getStatus().getStatus());
assertEquals(exception.getMessage(), result.getStatus().getMessage());
verify(request, resource, query, predicate);
}
@Test
public void testHandleRequest__NoSuchResourceException_NOT_FOUND() throws Exception {
Request request = createStrictMock(Request.class);
ResourceInstance resource = createStrictMock(ResourceInstance.class);
Query query = createMock(Query.class);
NoSuchResourceException exception = new NoSuchResourceException("msg", new RuntimeException());
Renderer renderer = new DefaultRenderer();
expect(request.getResource()).andReturn(resource);
expect(resource.getQuery()).andReturn(query);
expect(request.getPageRequest()).andReturn(null);
expect(request.getSortRequest()).andReturn(null);
expect(request.getRenderer()).andReturn(renderer);
expect(request.getFields()).andReturn(Collections.<String, TemporalInfo>emptyMap());
expect(request.getQueryPredicate()).andReturn(null).anyTimes();
query.setUserPredicate(null);
query.setPageRequest(null);
query.setSortRequest(null);
query.setRenderer(renderer);
expect(query.execute()).andThrow(exception);
replay(request, resource, query);
// test
ReadHandler handler = new ReadHandler();
Result result = handler.handleRequest(request);
// not a query, so not found
assertEquals(ResultStatus.STATUS.NOT_FOUND, result.getStatus().getStatus());
assertEquals(exception.getMessage(), result.getStatus().getMessage());
verify(request, resource, query);
}
@Test
public void testHandleRequest__SystemException() throws Exception {
Request request = createStrictMock(Request.class);
ResourceInstance resource = createStrictMock(ResourceInstance.class);
Query query = createMock(Query.class);
Predicate predicate = createMock(Predicate.class);
Renderer renderer = new DefaultRenderer();
expect(request.getResource()).andReturn(resource);
expect(resource.getQuery()).andReturn(query);
expect(request.getPageRequest()).andReturn(null);
expect(request.getSortRequest()).andReturn(null);
expect(request.getRenderer()).andReturn(renderer);
expect(request.getFields()).andReturn(Collections.<String, TemporalInfo>emptyMap());
expect(request.getQueryPredicate()).andReturn(predicate);
query.setUserPredicate(predicate);
query.setPageRequest(null);
query.setSortRequest(null);
query.setRenderer(renderer);
SystemException systemException = new SystemException("testMsg", new RuntimeException());
expect(query.execute()).andThrow(systemException);
replay(request, resource, query, predicate);
// test
ReadHandler handler = new ReadHandler();
Result result = handler.handleRequest(request);
assertEquals(ResultStatus.STATUS.SERVER_ERROR, result.getStatus().getStatus());
assertEquals(systemException.toString(), result.getStatus().getMessage());
verify(request, resource, query, predicate);
}
@Test
public void testHandleRequest__OK() throws Exception {
Request request = createStrictMock(Request.class);
ResourceInstance resource = createStrictMock(ResourceInstance.class);
Query query = createMock(Query.class);
Predicate predicate = createMock(Predicate.class);
Result result = createStrictMock(Result.class);
Renderer renderer = new DefaultRenderer();
Capture<ResultStatus> resultStatusCapture = new Capture<ResultStatus>();
Map<String, TemporalInfo> mapPartialResponseFields = new HashMap<String, TemporalInfo>();
mapPartialResponseFields.put("foo", null);
mapPartialResponseFields.put("bar/c", null);
mapPartialResponseFields.put("bar/d/e", null);
mapPartialResponseFields.put("category/", null);
// expectations
expect(request.getResource()).andReturn(resource);
expect(resource.getQuery()).andReturn(query);
expect(request.getPageRequest()).andReturn(null);
expect(request.getSortRequest()).andReturn(null);
expect(request.getRenderer()).andReturn(renderer);
expect(request.getFields()).andReturn(mapPartialResponseFields);
query.addProperty("foo", null);
query.addProperty("bar/c", null);
query.addProperty("bar/d/e", null);
query.addProperty("category/", null);
expect(request.getQueryPredicate()).andReturn(predicate);
query.setUserPredicate(predicate);
query.setPageRequest(null);
query.setSortRequest(null);
query.setRenderer(renderer);
expect(query.execute()).andReturn(result);
result.setResultStatus(capture(resultStatusCapture));
replay(request, resource, query, predicate, result);
// test
ReadHandler handler = new ReadHandler();
assertSame(result, handler.handleRequest(request));
assertEquals(ResultStatus.STATUS.OK, resultStatusCapture.getValue().getStatus());
verify(request, resource, query, predicate, result);
}