com.ning.http.client.FluentCaseInsensitiveStringsMap Java Examples

public HttpRequest build() {
  final byte[] body;
  MultipartRequestEntity requestEntity = new TerminatingMultipartRequestEntity(parts, new FluentCaseInsensitiveStringsMap());
  try (ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
    requestEntity.writeRequest(baos);
    baos.flush();
    body = baos.toByteArray();
  } catch (IOException e) {
    throw new RuntimeException("Error generating body bytes", e);
  }

  builder.addHeader("Content-Type", requestEntity.getContentType());
  builder.setBody(body);

  return builder.build();
}
 

@BeforeTest
public void setupCommon(){
    req = mock(Request.class);
    when(req.getUri()).thenReturn(Uri.create("http://test.yahoo.com"));
    when(req.getMethod()).thenReturn("POST");
    when(req.getHeaders()).thenReturn(new FluentCaseInsensitiveStringsMap().add(ParsecClientDefine.HEADER_HOST,
            Arrays.asList("dummyhost")));

    resp = mock(Response.class);
    when(resp.getHeader(anyString())).thenReturn("mock resp header");
    when(resp.getHeader("content-length")).thenReturn("100");
    when(resp.getStatusCode()).thenReturn(200);



    reqStatus = "test_status";
    progress = new ParsecAsyncProgress();

}
 

public static void calculateSignature(Signer signer, Request request, RequestBuilderBase<?> requestBuilder) {
    RequestContent.Builder sigBuilder = new RequestContent.Builder();

    sigBuilder.setRequestTarget(request.getMethod(), getRequestPath(request));
    for (FluentCaseInsensitiveStringsMap.Entry<String, List<String>> entry : request.getHeaders().entrySet()) {
        for (String value : entry.getValue()) {
            sigBuilder.addHeader(entry.getKey(), value);
        }
    }

    if (sigBuilder.build().getDate() == null) {
        sigBuilder.addDateNow();
        requestBuilder.addHeader(Constants.HEADER_DATE, sigBuilder.build().getDate());
    }

    Authorization authz = signer.sign(sigBuilder.build());
    if (authz != null) {
        requestBuilder.setHeader(Constants.AUTHORIZATION, authz.getHeaderValue());
    }
}
 

@Override
public void setHeader(Request request, String name, String value) {
    final FluentCaseInsensitiveStringsMap httpRequestHeaders = request.getHeaders();
    final List<String> valueList = new ArrayList<String>();
    valueList.add(value);
    httpRequestHeaders.put(name, valueList);
    if (isDebug) {
        logger.debug("Set header {}={}", name, value);
    }
}
 

@Override
public Collection<HttpHeader> getHeaders() {
	FluentCaseInsensitiveStringsMap headers = response.getHeaders();

	List<HttpHeader> results = new ArrayList<>(headers.size());
	for (Map.Entry<String, List<String>> entry : headers.entrySet()) {
		results.add(HttpHeader.header(entry.getKey(), entry.getValue()));
	}

	return results;
}
 

public TerminatingMultipartRequestEntity(List<Part> parts, FluentCaseInsensitiveStringsMap requestHeaders) {
  super(parts, requestHeaders);
}
 

private void fillNode(FluentCaseInsensitiveStringsMap headers, ObjectNode node){
    headers.keySet().forEach(
            k -> node.put(k, headers.getJoinedValue(k, ","))
    );
}
 

@Test
public void testGet() throws Exception {
    FluentCaseInsensitiveStringsMap responseHeaders = new FluentCaseInsensitiveStringsMap();

    responseHeaders.add("header1", "header1_value1");
    responseHeaders.add("header2", "header2_value1");

    List<Cookie> ningCookies = new ArrayList<>();

    ningCookies.add(new Cookie(
        "cookie1_name",
        "cookie1_value",
        false,
        null,
        "cookie1_path",
        1,
        true,
        true
    ));

    ningCookies.add(new Cookie(
        "cookie2_name",
        "cookie2_value",
        false,
        null,
        "cookie2_path",
        2,
        false,
        false
    ));

    when(mockNingResponse.getHeaders()).thenReturn(responseHeaders);
    when(mockNingResponse.getContentType()).thenReturn(MediaType.APPLICATION_JSON);
    when(mockNingResponse.getCookies()).thenReturn(ningCookies);
    when(mockNingResponse.getStatusCode()).thenReturn(200);
    when(mockNingResponse.hasResponseHeaders()).thenReturn(true);

    when(mockNingFuture.get()).thenReturn(mockNingResponse);
    when(mockNingFuture.get(anyLong(), any(TimeUnit.class))).thenReturn(mockNingResponse);

    ParsecCompletableFuture<Response> future = new ParsecCompletableFuture<>(mockNingFuture);
    Response response = future.get();

    assertEquals(response.getContentType(), MediaType.APPLICATION_JSON);
    assertEquals(response.getHeaders().size(), 2);

    response = future.get(3, TimeUnit.SECONDS);

    assertEquals(response.getContentType(), MediaType.APPLICATION_JSON);
    assertEquals(response.getHeaders().size(), 2);
}
 

private NingHttpHeaders(FluentCaseInsensitiveStringsMap map) {
    this.map = map;
}
 

@Override
public FluentCaseInsensitiveStringsMap getHeaders() {
    return map;
}
 

public NingHttpHeaders build() {
    return new NingHttpHeaders(
            new FluentCaseInsensitiveStringsMap(map));
}