ReadHandler implements RequestHandler { @Override public Result handleRequest(Request request) { Query query = request.getResource().getQuery(); query.setPageRequest(request.getPageRequest()); query.setSortRequest(request.getSortRequest()); query.setRenderer(request.getRenderer()); RequestBody body = request.getBody(); if(body != null) { query.setRequestInfoProps(body.getRequestInfoProperties()); } try { addFieldsToQuery(request, query); } catch (IllegalArgumentException e) { return new ResultImpl(new ResultStatus(ResultStatus.STATUS.BAD_REQUEST, e.getMessage())); } Result result; Predicate p = null; try { p = request.getQueryPredicate(); query.setUserPredicate(p); result = query.execute(); result.setResultStatus(new ResultStatus(ResultStatus.STATUS.OK)); } catch (AuthorizationException e) { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.FORBIDDEN, e.getMessage())); } catch (SystemException e) { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.SERVER_ERROR, e)); } catch (NoSuchParentResourceException e) { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.NOT_FOUND, e.getMessage())); } catch (UnsupportedPropertyException e) { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.BAD_REQUEST, e.getMessage())); } catch (NoSuchResourceException e) { if (p == null) { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.NOT_FOUND, e.getMessage())); } else { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.OK, e)); result.getResultTree().setProperty("isCollection", "true"); } } catch (IllegalArgumentException e) { result = new ResultImpl(new ResultStatus(ResultStatus.STATUS.BAD_REQUEST, "Invalid Request: " + e.getMessage())); LOG.error("Bad request: ", e); } catch (RuntimeException e) { if (LOG.isErrorEnabled()) { LOG.error("Caught a runtime exception executing a query", e); } throw e; } return result; }  @Override Result handleRequest(Request request);  }
@Test public void testHandleRequest__InvalidField() { Request request = createNiceMock(Request.class); ResourceInstance resource = createNiceMock(ResourceInstance.class); Query query = createStrictMock(Query.class); Map<String, TemporalInfo> mapPartialResponseFields = new HashMap<String, TemporalInfo>(); mapPartialResponseFields.put("foo/bar", null); Renderer renderer = new DefaultRenderer(); expect(request.getResource()).andReturn(resource); expect(request.getFields()).andReturn(mapPartialResponseFields); expect(request.getRenderer()).andReturn(renderer); expect(resource.getQuery()).andReturn(query); query.setPageRequest(null); query.setSortRequest(null); query.setRenderer(renderer); query.addProperty("foo/bar", null); expectLastCall().andThrow(new IllegalArgumentException("testMsg")); replay(request, resource, query); ReadHandler handler = new ReadHandler(); Result result = handler.handleRequest(request); assertEquals(ResultStatus.STATUS.BAD_REQUEST, result.getStatus().getStatus()); assertEquals("testMsg", result.getStatus().getMessage()); verify(request, resource, query); }
@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); RequestBody body = createStrictMock(RequestBody.class); Renderer renderer = new DefaultRenderer(); Capture<ResultStatus> resultStatusCapture = EasyMock.newCapture(); Map<String, String> requestInfoProperties = Collections.singletonMap("directive", "value"); 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); 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.getBody()).andReturn(body); expect(request.getFields()).andReturn(mapPartialResponseFields); expect(body.getRequestInfoProperties()).andReturn(requestInfoProperties); query.setRequestInfoProps(requestInfoProperties); 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, body, query, predicate, result); ReadHandler handler = new ReadHandler(); assertSame(result, handler.handleRequest(request)); assertEquals(ResultStatus.STATUS.OK, resultStatusCapture.getValue().getStatus()); verify(request, resource, body, query, predicate, result); }
@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.getBody()).andReturn(null); 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); 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__NoSuchParentResourceException() throws Exception { Request request = createStrictMock(Request.class); ResourceInstance resource = createStrictMock(ResourceInstance.class); Query query = createMock(Query.class); Predicate predicate = createMock(Predicate.class); NoSuchParentResourceException exception = new NoSuchParentResourceException("exceptionMsg", 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.getBody()).andReturn(null); 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); ReadHandler handler = new ReadHandler(); Result result = handler.handleRequest(request); assertEquals(ResultStatus.STATUS.NOT_FOUND, result.getStatus().getStatus()); assertEquals("exceptionMsg", result.getStatus().getMessage()); verify(request, resource, query, predicate); }
@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.getBody()).andReturn(null); 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); 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_OK() throws Exception { Request request = createStrictMock(Request.class); ResourceInstance resource = createStrictMock(ResourceInstance.class); Query query = createMock(Query.class); Predicate predicate = createMock(Predicate.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.getBody()).andReturn(null); expect(request.getFields()).andReturn(Collections.<String, TemporalInfo>emptyMap()); expect(request.getQueryPredicate()).andReturn(predicate).anyTimes(); query.setUserPredicate(predicate); query.setPageRequest(null); query.setSortRequest(null); query.setRenderer(renderer); expect(query.execute()).andThrow(exception); replay(request, resource, query, predicate); ReadHandler handler = new ReadHandler(); Result result = handler.handleRequest(request); assertEquals(ResultStatus.STATUS.OK, result.getStatus().getStatus()); 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.getBody()).andReturn(null); 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); ReadHandler handler = new ReadHandler(); Result result = handler.handleRequest(request); assertEquals(ResultStatus.STATUS.NOT_FOUND, result.getStatus().getStatus()); assertEquals(exception.getMessage(), result.getStatus().getMessage()); verify(request, resource, query); }
@Test public void testHandleRequest__AuthorizationException() 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.getBody()).andReturn(null); 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); AuthorizationException authorizationException = new AuthorizationException("testMsg"); expect(query.execute()).andThrow(authorizationException); replay(request, resource, query, predicate); ReadHandler handler = new ReadHandler(); Result result = handler.handleRequest(request); assertEquals(ResultStatus.STATUS.FORBIDDEN, result.getStatus().getStatus()); assertEquals(authorizationException.getMessage(), result.getStatus().getMessage()); verify(request, resource, query, predicate); }