@Override public Request filterRequest(Request request) { requireNonNull(request, "request is null"); String newToken = createToken(sequence.getAndIncrement()); return fromRequest(request) .setHeader(TRACETOKEN_HEADER, newToken) .build(); }
@Override public Request filterRequest(Request request) { requireNonNull(request, "request is null"); String newToken = createToken(sequence.getAndIncrement()); return fromRequest(request) .setHeader(TRACETOKEN_HEADER, newToken) .build(); }
@Override public Request filterRequest(Request request) { requireNonNull(request, "request is null"); String newToken = createToken(sequence.getAndIncrement()); return fromRequest(request) .setHeader(TRACETOKEN_HEADER, newToken) .build(); }
@Test public void testRedirectRequestHeaders() throws Exception { String basic = "Basic dGVzdDphYmM="; String bearer = "Bearer testxyz"; Request request = prepareGet() .setUri(URI.create(baseURI.toASCIIString() + "/?redirect=/redirect")) .addHeader("X-Test", "xtest1") .addHeader("X-Test", "xtest2") .setHeader(USER_AGENT, "testagent") .addHeader(AUTHORIZATION, basic) .addHeader(AUTHORIZATION, bearer) .build(); StatusResponse response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 200); assertEquals(servlet.getRequestUri(), URI.create(baseURI.toASCIIString() + "/redirect")); assertThat(servlet.getRequestHeaders("X-Test")).containsExactly("xtest1", "xtest2"); assertThat(servlet.getRequestHeaders(USER_AGENT)).containsExactly("testagent"); assertThat(servlet.getRequestHeaders(AUTHORIZATION)).isEmpty(); request = Request.Builder.fromRequest(request) .setPreserveAuthorizationOnRedirect(true) .build(); response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 200); assertEquals(servlet.getRequestUri(), URI.create(baseURI.toASCIIString() + "/redirect")); assertThat(servlet.getRequestHeaders("X-Test")).containsExactly("xtest1", "xtest2"); assertThat(servlet.getRequestHeaders(USER_AGENT)).containsExactly("testagent"); assertThat(servlet.getRequestHeaders(AUTHORIZATION)).containsExactly(basic, bearer); }
@Test public void testRedirectRequestHeaders() throws Exception { String basic = "Basic dGVzdDphYmM="; String bearer = "Bearer testxyz"; Request request = prepareGet() .setUri(URI.create(baseURI.toASCIIString() + "/?redirect=/redirect")) .addHeader("X-Test", "xtest1") .addHeader("X-Test", "xtest2") .setHeader(USER_AGENT, "testagent") .addHeader(AUTHORIZATION, basic) .addHeader(AUTHORIZATION, bearer) .build(); StatusResponse response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 200); assertEquals(servlet.getRequestUri(), URI.create(baseURI.toASCIIString() + "/redirect")); assertThat(servlet.getRequestHeaders("X-Test")).containsExactly("xtest1", "xtest2"); assertThat(servlet.getRequestHeaders(USER_AGENT)).containsExactly("testagent"); assertThat(servlet.getRequestHeaders(AUTHORIZATION)).isEmpty(); request = Request.Builder.fromRequest(request) .setPreserveAuthorizationOnRedirect(true) .build(); response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 200); assertEquals(servlet.getRequestUri(), URI.create(baseURI.toASCIIString() + "/redirect")); assertThat(servlet.getRequestHeaders("X-Test")).containsExactly("xtest1", "xtest2"); assertThat(servlet.getRequestHeaders(USER_AGENT)).containsExactly("testagent"); assertThat(servlet.getRequestHeaders(AUTHORIZATION)).containsExactly(basic, bearer); }
@Test public void testFollowRedirects() throws Exception { Request request = prepareGet() .setUri(URI.create(baseURI.toASCIIString() + "/test?redirect=/redirect")) .build(); StatusResponse response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 200); assertNull(response.getHeader(LOCATION)); assertEquals(servlet.getRequestUri(), URI.create(baseURI.toASCIIString() + "/redirect")); request = Request.Builder.fromRequest(request) .setFollowRedirects(false) .build(); response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 302); assertEquals(response.getHeader(LOCATION), baseURI.toASCIIString() + "/redirect"); assertEquals(servlet.getRequestUri(), request.getUri()); }
@Test public void testFollowRedirects() throws Exception { Request request = prepareGet() .setUri(URI.create(baseURI.toASCIIString() + "/test?redirect=/redirect")) .build(); StatusResponse response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 200); assertNull(response.getHeader(LOCATION)); assertEquals(servlet.getRequestUri(), URI.create(baseURI.toASCIIString() + "/redirect")); request = Request.Builder.fromRequest(request) .setFollowRedirects(false) .build(); response = executeRequest(request, createStatusResponseHandler()); assertEquals(response.getStatusCode(), 302); assertEquals(response.getHeader(LOCATION), baseURI.toASCIIString() + "/redirect"); assertEquals(servlet.getRequestUri(), request.getUri()); }
@Override public Request filterRequest(Request request) { return fromRequest(request) .addHeader("x-custom-filter", "custom value") .build(); } }
@Test public void testBuilderFromRequest() { Request request = createRequest(); assertEquals(fromRequest(request).build(), request); }
@Test public void testBuilderFromRequest() { Request request = createRequest(); assertEquals(fromRequest(request).build(), request); }
@Override public Request filterRequest(Request request) { if (!requestMatcher.test(request)) { return request; } return fromRequest(request) .addHeader(AUTHORIZATION, authenticationHeader) .build(); } }
@Test public void testBuilderFromRequest() { Request request = createRequest(); assertEquals(fromRequest(request).build(), request); }
@Override public Request filterRequest(Request request) { requireNonNull(request, "request is null"); String token = traceTokenManager.getCurrentRequestToken(); if (token == null) { return request; } return fromRequest(request) .addHeader(TRACETOKEN_HEADER, token) .build(); }
@Override public Request filterRequest(Request request) { return fromRequest(request) .addHeader("x-custom-filter", "customvalue") .build(); } }
@Test public void testBuilderFromRequest() { Request request = createRequest(); assertEquals(fromRequest(request).build(), request); }
@Override public Request filterRequest(Request request) { requireNonNull(request, "request is null"); return fromRequest(request) .addHeader(USER_AGENT, nodeId) .build(); }
@Override public Request filterRequest(Request request) { return fromRequest(request) .addHeader("x-custom-filter", "custom value") .build(); } }
@Override public Request filterRequest(Request request) { return fromRequest(request) .addHeader("x-custom-filter", "custom value") .build(); } }