Java Code Examples for org.apache.http.HttpRequest#setHeader()

@Test
public void testHeaderIsModified() {
    HttpRequest req = EasyMock.createMock(HttpRequest.class);
    Header header = EasyMock.createMock(Header.class);
    EasyMock.expect(header.getValue()).andReturn("BaseAgent").once();
    EasyMock.expect(req.getFirstHeader(HttpHeaders.USER_AGENT))
        .andReturn(header).once();
    req.setHeader(EasyMock.matches("^User-Agent$"),
        EasyMock.matches("^gerrit-rest-java-client/.* using BaseAgent$"));
    EasyMock.replay(header, req);
    HttpContext ctx = EasyMock.createMock(HttpContext.class);
    UserAgentHttpRequestInterceptor interceptor = new UserAgentHttpRequestInterceptor();

    interceptor.process(req, ctx);

    EasyMock.verify(header, req);
}
 

@Test
public void givenZosmfToken_whenApplyToRequest_thenTestJwtToken() {
    Calendar calendar = Calendar.getInstance();
    QueryResponse queryResponse = new QueryResponse("domain", "username", calendar.getTime(), calendar.getTime(), QueryResponse.Source.ZOSMF);
    AuthConfigurationProperties.CookieProperties cookieProperties = mock(AuthConfigurationProperties.CookieProperties.class);
    when(cookieProperties.getCookieName()).thenReturn("apimlAuthenticationToken");
    when(authConfigurationProperties.getCookieProperties()).thenReturn(cookieProperties);
    RequestContext requestContext = spy(new RequestContext());
    RequestContext.testSetCurrentContext(requestContext);

    HttpRequest httpRequest = new HttpGet("/test/request");
    httpRequest.setHeader(COOKIE_HEADER, "cookie1=1");
    Authentication authentication = new Authentication(AuthenticationScheme.ZOSMF, null);
    HttpServletRequest request = new MockHttpServletRequest();
    requestContext.setRequest(request);

    when(authenticationService.getJwtTokenFromRequest(request)).thenReturn(Optional.of("jwtToken2"));
    when(authenticationService.parseJwtToken("jwtToken2")).thenReturn(queryResponse);
    when(authConfigurationProperties.getCookieProperties().getCookieName()).thenReturn("apimlAuthenticationToken");

    zosmfScheme.createCommand(authentication, () -> queryResponse).applyToRequest(httpRequest);

    assertEquals("cookie1=1;jwtToken=jwtToken2", httpRequest.getFirstHeader("cookie").getValue());
}
 

@Test
void givenJwtInCookie_whenApplyToRequest_thenJwtIsRemoved() {
    AuthenticationCommand command = getPassTicketCommand();
    HttpRequest httpRequest = new HttpGet();
    httpRequest.setHeader("cookie",
        authConfigurationProperties.getCookieProperties().getCookieName() + "=jwt;" +
        "abc=def"
    );

    command.applyToRequest(httpRequest);

    Header[] headers = httpRequest.getHeaders("cookie");
    assertNotNull(headers);
    assertEquals(1, headers.length);
    assertEquals("abc=def", headers[0].getValue());
}
 

private void setXForwardedForHeader(HttpServletRequest servletRequest, HttpRequest proxyRequest) {
    String headerName = "X-Forwarded-For";
    if (doForwardIP) {
        String newHeader = servletRequest.getRemoteAddr();
        String existingHeader = servletRequest.getHeader(headerName);
        if (existingHeader != null) {
            newHeader = existingHeader + ", " + newHeader;
        }
        proxyRequest.setHeader(headerName, newHeader);
    }
}
 

@Test
public void givenRequestWithSetCookie_whenApplyToRequest_thenAppendSetCookie() {
    HttpRequest httpRequest = new HttpGet("/test/request");
    httpRequest.setHeader(COOKIE_HEADER, "cookie1=1");
    prepareAuthenticationService("ltpa2");

    zosmfScheme.createCommand(authentication, () -> queryResponse).applyToRequest(httpRequest);

    assertEquals("cookie1=1;LtpaToken2=ltpa2", httpRequest.getFirstHeader("cookie").getValue());
}
 

protected void setHeaders(HttpRequest request, long accountId, String platformType) {
	TokenObject token = TokenObject.getByPlatform(accountId, platformType);
	long currentTime = System.currentTimeMillis();
	request.setHeader("user_id", String.valueOf(accountId));
	request.setHeader("public_token", token.getPublicToken());
	request.setHeader("disposable_token", token.getDisposableToken(currentTime));
	request.setHeader("timestamp", String.valueOf(currentTime));
}
 

@Override
protected boolean interceptInternodeRequest(HttpRequest httpRequest, HttpContext httpContext) {
  if (forwardCredentials) {
    if (httpContext instanceof HttpClientContext) {
      HttpClientContext httpClientContext = (HttpClientContext) httpContext;
      if (httpClientContext.getUserToken() instanceof BasicAuthUserPrincipal) {
        BasicAuthUserPrincipal principal = (BasicAuthUserPrincipal) httpClientContext.getUserToken();
        String userPassBase64 = Base64.encodeBase64String((principal.getName() + ":" + principal.getPassword()).getBytes(StandardCharsets.UTF_8));
        httpRequest.setHeader(HttpHeaders.AUTHORIZATION, "Basic " + userPassBase64);
        return true;
      }
    }
  }
  return false;
}
 

private void setCrsfToken(HttpRequest request) throws IOException {
    if ((request == null) || !isProtectionRequired(request)) {
        return;
    }

    initializeToken(false);
    if (csrfToken != null) {
        request.setHeader(CSRF_TOKEN_HEADER_NAME, csrfToken);
    }
}
 

private void setAuth(HttpRequest httpget) {
  if (password != null) {
    try {
      byte[] b = Base64.encodeBase64((username+":"+password).getBytes("ASCII"));
      String b64 = new String(b, StandardCharsets.US_ASCII);
      httpget.setHeader("Authorization", "Basic " + b64);
    } catch (UnsupportedEncodingException e) {
    }
  }
}
 

@Override
protected boolean interceptInternodeRequest(HttpRequest httpRequest, HttpContext httpContext) {
  if (httpContext instanceof HttpClientContext) {
    HttpClientContext httpClientContext = (HttpClientContext) httpContext;
    if (httpClientContext.getUserToken() instanceof JWTPrincipal) {
      JWTPrincipal jwtPrincipal = (JWTPrincipal) httpClientContext.getUserToken();
      httpRequest.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + jwtPrincipal.getToken());
      return true;
    }
  }
  return false;
}
 

public void requestInit(final HttpRequest request, final HttpContext context)throws HttpException, IOException {
	request.setHeader(C.ACCEPT_KEY, C.ACCEPT_VAL);
	request.setHeader(C.ACCEPT_ENCODING_KEY, C.ACCEPT_ENCODING_VAL);
	request.setHeader(C.ACCEPT_LANGUAGE_KEY, C.ACCEPT_LANGUAGE_VAL);
	request.setHeader(C.CONNECTION_KEY, C.CONNECTION_VAL);
	request.setHeader(C.CONTENT_TYPE_KEY, C.CONTENT_TYPE_VAL);
	request.setHeader(C.USER_AGENT_KEY, C.USER_AGENT_VAL);
}
 

private void setAuth(HttpRequest httpget) {
  if (password != null) {
    try {
      byte[] b = Base64.encodeBase64((username+":"+password).getBytes("ASCII"));
      String b64 = new String(b, StandardCharsets.US_ASCII);
      httpget.setHeader("Authorization", "Basic " + b64);
    } catch (UnsupportedEncodingException e) {
    }
  }
}
 

@Override
public void process(final HttpRequest request, final HttpContext context) throws HttpException, IOException {
    request.setHeader(HttpHeader.USER_AGENT, value_);
}
 

@Override
public void process(final HttpRequest request, final HttpContext context) throws HttpException, IOException {
    request.setHeader(HttpHeader.ACCEPT, value_);
}
 

private void setHttpRequestHeaders(HttpRequest request) {
    for (Entry<String, String> httpRequestHeaderEntry : httpRequestHeaders.entrySet()) {
        request.setHeader(httpRequestHeaderEntry.getKey(), httpRequestHeaderEntry.getValue());
    }
}
 

@Override
public void process(final HttpRequest request, final HttpContext context) throws HttpException, IOException {
    request.setHeader(HttpHeader.UPGRADE_INSECURE_REQUESTS, value_);
}
 

@Override
public void process(final HttpRequest request, final HttpContext context) throws HttpException, IOException {
    request.setHeader("Accept-Encoding", value_);
}
 

@Override
protected void setAuthorization(HttpRequest request, String signature) {
    request.setHeader("Authorization", signature);
}
 

@Override
public void process(final HttpRequest request, final HttpContext context) throws HttpException, IOException {
    request.setHeader(HttpHeader.SEC_FETCH_SITE, value_);
}
 

@Override
public void process(final HttpRequest request, final HttpContext context) throws HttpException, IOException {
    request.setHeader(HttpHeader.SEC_FETCH_DEST, value_);
}