Example usage for org.apache.http.client.methods HttpUriRequest getFirstHeader

List of usage examples for org.apache.http.client.methods HttpUriRequest getFirstHeader

Introduction

In this page you can find the example usage for org.apache.http.client.methods HttpUriRequest getFirstHeader.

Prototype

Header getFirstHeader(String str);

Source Link

Usage

From source file:com.klinker.android.spotify.loader.OAuthTokenRefresherTest.java

@Test
public void test_addAuthHeader() {
    HttpUriRequest post = new HttpPost("http://www.google.com");
    post = refresher.addAuthHeader(post);

    assertNotNull(post.getFirstHeader("Authorization"));
    assertEquals("application/x-www-form-urlencoded", post.getLastHeader("Content-Type").getValue());
}

From source file:com.github.restdriver.serverdriver.http.HeaderTest.java

public void headerAppliesItselfToRequest() {
    HttpUriRequest request = new HttpGet();
    Header header = new Header("name", "value");
    header.applyTo(new ServerDriverHttpUriRequest(request));
    assertThat(request.getFirstHeader("name").getValue(), is("value"));
}

From source file:org.springframework.cloud.netflix.ribbon.apache.RibbonApacheHttpRequestTests.java

@Test
public void testNullEntity() throws Exception {
    String uri = "http://example.com";
    LinkedMultiValueMap<String, String> headers = new LinkedMultiValueMap<>();
    headers.add("my-header", "my-value");
    headers.add(HttpEncoding.CONTENT_LENGTH, "5192");
    LinkedMultiValueMap<String, String> params = new LinkedMultiValueMap<>();
    params.add("myparam", "myparamval");
    RibbonApacheHttpRequest httpRequest = new RibbonApacheHttpRequest(new RibbonCommandContext("example", "GET",
            uri, false, headers, params, null, new ArrayList<RibbonRequestCustomizer>()));

    HttpUriRequest request = httpRequest.toRequest(RequestConfig.custom().build());

    assertThat("request is wrong type", request, is(not(instanceOf(HttpEntityEnclosingRequest.class))));
    assertThat("uri is wrong", request.getURI().toString(), startsWith(uri));
    assertThat("my-header is missing", request.getFirstHeader("my-header"), is(notNullValue()));
    assertThat("my-header is wrong", request.getFirstHeader("my-header").getValue(), is(equalTo("my-value")));
    assertThat("Content-Length is wrong", request.getFirstHeader(HttpEncoding.CONTENT_LENGTH).getValue(),
            is(equalTo("5192")));
    assertThat("myparam is missing", request.getURI().getQuery(), is(equalTo("myparam=myparamval")));

}

From source file:com.microsoft.live.unittest.ApiTest.java

License:asdf

/**
 * Changes the mockClient to one that checks if the incoming requests contains the
 * Live Library header and then reverts to the original mockClient.
 *///from  www  . j  a v  a2 s .  c  o m
protected void loadLiveLibraryHeaderChecker() {
    final MockHttpClient currentMockClient = this.mockClient;
    MockHttpClient liveLibraryHeaderCheckerClient = new MockHttpClient() {
        @Override
        public HttpResponse execute(HttpUriRequest request) throws IOException, ClientProtocolException {
            Header header = request.getFirstHeader("X-HTTP-Live-Library");

            assertEquals("android/" + Build.VERSION.RELEASE + "_5.0", header.getValue());

            // load the old mock client back after we check.
            mockClient = currentMockClient;
            return currentMockClient.execute(request);
        }

        @Override
        public HttpResponse getHttpResponse() {
            return currentMockClient.getHttpResponse();
        }

        @Override
        public void setHttpResponse(HttpResponse httpResponse) {
            currentMockClient.setHttpResponse(httpResponse);
        }

        @Override
        public void addHttpResponse(HttpResponse httpResponse) {
            currentMockClient.addHttpResponse(httpResponse);
        }

        @Override
        public void clearHttpResponseQueue() {
            currentMockClient.clearHttpResponseQueue();
        }
    };

    this.mockClient = liveLibraryHeaderCheckerClient;
}

From source file:com.betfair.cougar.client.HttpClientCougarRequestFactoryTest.java

@Test
public void shouldAddNewHeadersWithoutDeleteTheExistingHeaders() {

    HttpUriRequest request = new HttpGet();
    request.setHeader("X-UUID", "1111-111-111-111");

    List<Header> headers = new ArrayList<>(1);
    headers.add(new BasicHeader("NEW-HEADER", "value"));
    factory.addHeaders(request, headers);

    assertEquals(2, request.getAllHeaders().length);
    assertEquals("1111-111-111-111", request.getFirstHeader("X-UUID").getValue());
    assertEquals("value", request.getFirstHeader("NEW-HEADER").getValue());
}

From source file:org.flowable.rest.dmn.service.api.BaseSpringDmnRestTestCase.java

protected CloseableHttpResponse internalExecuteRequest(HttpUriRequest request, int expectedStatusCode,
        boolean addJsonContentType) {
    CloseableHttpResponse response = null;
    try {//w w  w.  j av a  2 s.  c o  m
        if (addJsonContentType && request.getFirstHeader(HttpHeaders.CONTENT_TYPE) == null) {
            // Revert to default content-type
            request.addHeader(new BasicHeader(HttpHeaders.CONTENT_TYPE, "application/json"));
        }
        response = client.execute(request);
        Assert.assertNotNull(response.getStatusLine());

        int responseStatusCode = response.getStatusLine().getStatusCode();
        if (expectedStatusCode != responseStatusCode) {
            log.info("Wrong status code : {}, but should be {}", responseStatusCode, expectedStatusCode);
            log.info("Response body: {}", IOUtils.toString(response.getEntity().getContent()));
        }

        Assert.assertEquals(expectedStatusCode, responseStatusCode);
        httpResponses.add(response);
        return response;

    } catch (ClientProtocolException e) {
        Assert.fail(e.getMessage());
    } catch (IOException e) {
        Assert.fail(e.getMessage());
    }
    return null;
}

From source file:org.springframework.cloud.netflix.ribbon.apache.RibbonApacheHttpRequestTests.java

void testEntity(String entityValue, ByteArrayInputStream requestEntity, boolean addContentLengthHeader,
        String method) throws IOException {
    String lengthString = String.valueOf(entityValue.length());
    Long length = null;//from   w  ww  . j  av a2s  .  c  o m
    URI uri = URI.create("http://example.com");
    LinkedMultiValueMap<String, String> headers = new LinkedMultiValueMap<>();
    if (addContentLengthHeader) {
        headers.add("Content-Length", lengthString);
        length = (long) entityValue.length();
    }

    RibbonRequestCustomizer requestCustomizer = new RibbonRequestCustomizer<RequestBuilder>() {
        @Override
        public boolean accepts(Class builderClass) {
            return builderClass == RequestBuilder.class;
        }

        @Override
        public void customize(RequestBuilder builder) {
            builder.addHeader("from-customizer", "foo");
        }
    };
    RibbonCommandContext context = new RibbonCommandContext("example", method, uri.toString(), false, headers,
            new LinkedMultiValueMap<String, String>(), requestEntity,
            Collections.singletonList(requestCustomizer));
    context.setContentLength(length);
    RibbonApacheHttpRequest httpRequest = new RibbonApacheHttpRequest(context);

    HttpUriRequest request = httpRequest.toRequest(RequestConfig.custom().build());

    assertThat("request is wrong type", request, is(instanceOf(HttpEntityEnclosingRequest.class)));
    assertThat("uri is wrong", request.getURI().toString(), startsWith(uri.toString()));
    if (addContentLengthHeader) {
        assertThat("Content-Length is missing", request.getFirstHeader("Content-Length"), is(notNullValue()));
        assertThat("Content-Length is wrong", request.getFirstHeader("Content-Length").getValue(),
                is(equalTo(lengthString)));
    }
    assertThat("from-customizer is missing", request.getFirstHeader("from-customizer"), is(notNullValue()));
    assertThat("from-customizer is wrong", request.getFirstHeader("from-customizer").getValue(),
            is(equalTo("foo")));

    HttpEntityEnclosingRequest entityRequest = (HttpEntityEnclosingRequest) request;
    assertThat("entity is missing", entityRequest.getEntity(), is(notNullValue()));
    HttpEntity entity = entityRequest.getEntity();
    assertThat("contentLength is wrong", entity.getContentLength(), is(equalTo((long) entityValue.length())));
    assertThat("content is missing", entity.getContent(), is(notNullValue()));
    String string = StreamUtils.copyToString(entity.getContent(), Charset.forName("UTF-8"));
    assertThat("content is wrong", string, is(equalTo(entityValue)));
}

From source file:org.activiti.rest.content.service.api.BaseSpringContentRestTestCase.java

protected CloseableHttpResponse internalExecuteRequest(HttpUriRequest request, int expectedStatusCode,
        boolean addJsonContentType) {
    CloseableHttpResponse response = null;
    try {//www. java 2s  .  c om
        if (addJsonContentType && request.getFirstHeader(HttpHeaders.CONTENT_TYPE) == null) {
            // Revert to default content-type
            request.addHeader(new BasicHeader(HttpHeaders.CONTENT_TYPE, "application/json"));
        }
        response = client.execute(request);
        Assert.assertNotNull(response.getStatusLine());

        int responseStatusCode = response.getStatusLine().getStatusCode();
        if (expectedStatusCode != responseStatusCode) {
            log.info("Wrong status code : " + responseStatusCode + ", but should be " + expectedStatusCode);
            log.info("Response body: " + IOUtils.toString(response.getEntity().getContent()));
        }

        Assert.assertEquals(expectedStatusCode, responseStatusCode);
        httpResponses.add(response);
        return response;

    } catch (ClientProtocolException e) {
        Assert.fail(e.getMessage());
    } catch (IOException e) {
        Assert.fail(e.getMessage());
    }
    return null;
}

From source file:org.fcrepo.integration.connector.file.FileConnectorIT.java

private CloseableDataset getDataset(final CloseableHttpClient client, final HttpUriRequest req)
        throws IOException {
    if (!req.containsHeader("Accept")) {
        req.addHeader("Accept", "application/n-triples");
    }//from  w  w w. j  a  v  a 2 s . c om
    logger.debug("Retrieving RDF using mimeType: {}", req.getFirstHeader("Accept"));

    try (final CloseableHttpResponse response = client.execute(req)) {
        assertEquals(OK.getStatusCode(), response.getStatusLine().getStatusCode());
        final CloseableDataset result = parseTriples(response.getEntity());
        logger.trace("Retrieved RDF: {}", result);
        return result;
    }
}

From source file:com.seleritycorp.common.base.http.client.HttpRequestTest.java

@Test
public void testSetUserAgentPlain() throws Exception {
    replayAll();//from   w  ww.j  a  v a  2 s.c  om

    HttpRequest request = createHttpRequest("foo");
    HttpRequest requestAfterSetting = request.setUserAgent("foo");
    HttpResponse response = request.execute();

    verifyAll();

    assertThat(request).isSameAs(requestAfterSetting);
    assertThat(response).isEqualTo(httpResponse);

    HttpUriRequest backendRequest = backendRequestCapture.getValue();
    assertThat(backendRequest.getMethod()).isEqualTo("GET");
    assertThat(backendRequest.getURI().toString()).isEqualTo("foo");
    assertThat(backendRequest.getHeaders("User-Agent")).hasSize(1);
    assertThat(backendRequest.getFirstHeader("User-Agent").getValue()).isEqualTo("foo");
}