@Before @Override public void setUp() throws Exception { super.setUp(); rewriters = Lists.newArrayList(new CaptureRewriter(), new CaptureRewriter()); contentRewriters = Lists.<ResponseRewriter>newArrayList(rewriters); registry = new DefaultResponseRewriterRegistry(contentRewriters, parser); }
@Test public void testRewriteHttpResponse() throws Exception { String body = "Hello, world"; HttpRequest request = new HttpRequest(SPEC_URL); HttpResponse response = new HttpResponse(body); HttpResponse rewritten = registry.rewriteHttpResponse(request, response, null); assertTrue("First rewriter not invoked.", rewriters.get(0).responseWasRewritten()); assertTrue("Second rewriter not invoked.", rewriters.get(1).responseWasRewritten()); assertEquals(response, rewritten); }
@Override public void rewrite(HttpRequest request, HttpResponseBuilder original, Gadget gadget) { super.rewrite(request, original, gadget); if (!Strings.isNullOrEmpty(contentToRewrite)) { original.setResponse(contentToRewrite.getBytes()); } } }
@Test public void doRewriting() throws Exception { renderer.render(makeGadget(BASIC_HTML_CONTENT)); assertTrue("Rewriting not performed.", captureRewriter.viewWasRewritten()); }
@Test public void doRewriting() throws Exception { renderer.render(makeGadget(BASIC_HTML_CONTENT)); assertTrue("Rewriting not performed.", captureRewriter.viewWasRewritten()); }
@Test public void testRewriteHttpResponse() throws Exception { String body = "Hello, world"; HttpRequest request = new HttpRequest(SPEC_URL); HttpResponse response = new HttpResponse(body); HttpResponse rewritten = registry.rewriteHttpResponse(request, response, null); assertTrue("First rewriter not invoked.", rewriters.get(0).responseWasRewritten()); assertTrue("Second rewriter not invoked.", rewriters.get(1).responseWasRewritten()); assertEquals(response, rewritten); }
@Test public void doRewriting() throws Exception { renderer.render(makeGadget(BASIC_HTML_CONTENT)); assertTrue("Rewriting not performed.", captureRewriter.viewWasRewritten()); }
@Override public void rewrite(HttpRequest request, HttpResponseBuilder original, Gadget gadget) { super.rewrite(request, original, gadget); if (!Strings.isNullOrEmpty(contentToRewrite)) { original.setResponse(contentToRewrite.getBytes()); } } }
@Before @Override public void setUp() throws Exception { super.setUp(); rewriters = Lists.newArrayList(new CaptureRewriter(), new CaptureRewriter()); contentRewriters = Lists.<ResponseRewriter>newArrayList(rewriters); registry = new DefaultResponseRewriterRegistry(contentRewriters, parser); }
@Test public void testRewriteHttpResponse() throws Exception { String body = "Hello, world"; HttpRequest request = new HttpRequest(SPEC_URL); HttpResponse response = new HttpResponse(body); HttpResponse rewritten = registry.rewriteHttpResponse(request, response); assertTrue("First rewriter not invoked.", rewriters.get(0).responseWasRewritten()); assertTrue("Second rewriter not invoked.", rewriters.get(1).responseWasRewritten()); assertEquals(response, rewritten); }
@Override public void rewrite(HttpRequest request, HttpResponseBuilder original) { super.rewrite(request, original); if (!StringUtils.isEmpty(contentToRewrite)) { original.setResponse(contentToRewrite.getBytes()); } } }
@Before @Override public void setUp() throws Exception { super.setUp(); rewriters = Lists.newArrayList(new CaptureRewriter(), new CaptureRewriter()); contentRewriters = Lists.<ResponseRewriter>newArrayList(rewriters); registry = new DefaultResponseRewriterRegistry(contentRewriters, parser); }
@Test public void testDeleteRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("DELETE"); expectDeleteAndReturnBody(""); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testHeadRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("HEAD"); expectHead(); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testInvalidSigningTypeTreatedAsNone() throws Exception { expectGetAndReturnBody(RESPONSE_BODY); expect(request.getParameter(MakeRequestHandler.AUTHZ_PARAM)).andReturn("garbage"); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertEquals(RESPONSE_BODY, results.get("body")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testPostRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("POST"); expectPostAndReturnBody(REQUEST_BODY, RESPONSE_BODY); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertEquals(RESPONSE_BODY, results.get("body")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testPutRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("PUT"); expectPutAndReturnBody(REQUEST_BODY, RESPONSE_BODY); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertEquals(RESPONSE_BODY, results.get("body")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testPatchRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("PATCH"); expectPatchAndReturnBody(""); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testDeleteRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("DELETE"); expectDeleteAndReturnBody(""); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertTrue(rewriter.responseWasRewritten()); }
@Test public void testPatchRequest() throws Exception { expect(request.getParameter(MakeRequestHandler.METHOD_PARAM)).andReturn("PATCH"); expectPatchAndReturnBody(""); replay(); handler.fetch(request, recorder); JSONObject results = extractJsonFromResponse(); assertEquals(HttpResponse.SC_OK, results.getInt("rc")); assertTrue(rewriter.responseWasRewritten()); }