Add class HttpAsyncClientTest

gregellis · Nov 30, 2014 · 4ea3b0f · 4ea3b0f
1 parent 9e7d330
commit 4ea3b0f
Show file tree

Hide file tree

Showing 2 changed files with 184 additions and 0 deletions.
diff --git a/httpclient/pom.xml b/httpclient/pom.xml
@@ -60,6 +60,12 @@
 			<version>1.9</version>
 		</dependency>
 
+		<dependency>
+		  <groupId>org.apache.httpcomponents</groupId>
+		  <artifactId>httpasyncclient</artifactId>
+		  <version>4.1-beta1</version>
+		</dependency>
+
 		<!-- logging -->
 
 		<dependency>

diff --git a/httpclient/src/test/java/org/baeldung/httpclient/HttpAsyncClientTest.java b/httpclient/src/test/java/org/baeldung/httpclient/HttpAsyncClientTest.java
@@ -0,0 +1,178 @@
+package org.baeldung.httpclient;
+
+import static org.hamcrest.Matchers.equalTo;
+import static org.junit.Assert.assertThat;
+
+import java.security.cert.X509Certificate;
+import java.util.concurrent.Future;
+
+import javax.net.ssl.SSLContext;
+
+import org.apache.http.HttpHost;
+import org.apache.http.HttpResponse;
+import org.apache.http.auth.AuthScope;
+import org.apache.http.auth.UsernamePasswordCredentials;
+import org.apache.http.client.CredentialsProvider;
+import org.apache.http.client.config.RequestConfig;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.protocol.HttpClientContext;
+import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
+import org.apache.http.conn.ssl.SSLContexts;
+import org.apache.http.conn.ssl.TrustStrategy;
+import org.apache.http.impl.client.BasicCookieStore;
+import org.apache.http.impl.client.BasicCredentialsProvider;
+import org.apache.http.impl.cookie.BasicClientCookie;
+import org.apache.http.impl.nio.client.CloseableHttpAsyncClient;
+import org.apache.http.impl.nio.client.HttpAsyncClients;
+import org.apache.http.impl.nio.conn.PoolingNHttpClientConnectionManager;
+import org.apache.http.impl.nio.reactor.DefaultConnectingIOReactor;
+import org.apache.http.nio.reactor.ConnectingIOReactor;
+import org.apache.http.protocol.BasicHttpContext;
+import org.apache.http.protocol.HttpContext;
+import org.junit.Test;
+
+public class HttpAsyncClientTest {
+
+    private static final String HOST = "http://www.google.com";
+    private static final String HOST_WITH_SSL = "https://mms.nw.ru/";
+    private static final String HOST_WITH_PROXY = "https://issues.apache.org/";
+    private static final String URL_SECURED_BY_BASIC_AUTHENTICATION = "http://browserspy.dk/password-ok.php";// "http://localhost:8080/spring-security-rest-basic-auth/api/foos/1";
+    private static final String DEFAULT_USER = "test";// "user1";
+    private static final String DEFAULT_PASS = "test";// "user1Pass";
+
+    private static final String HOST_WITH_COOKIE = "http://yuilibrary.com/yui/docs/cookie/cookie-simple-example.html"; // "http://github.com";
+    private static final String COOKIE_DOMAIN = ".yuilibrary.com"; // ".github.com";
+    private static final String COOKIE_NAME = "example"; // "JSESSIONID";
+
+
+    @Test
+    public void whenUseHttpAsyncClient_thenCorrect() throws Exception {
+        final CloseableHttpAsyncClient client = HttpAsyncClients.createDefault();
+        client.start();
+        final HttpGet request = new HttpGet(HOST);
+        final Future<HttpResponse> future = client.execute(request, null);
+        final HttpResponse response = future.get();
+        assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
+        client.close();
+    }
+
+    @Test
+    public void whenUseMultipleHttpAsyncClient_thenCorrect() throws Exception {
+        final ConnectingIOReactor ioReactor = new DefaultConnectingIOReactor();
+        final PoolingNHttpClientConnectionManager cm = new PoolingNHttpClientConnectionManager(ioReactor);
+        final CloseableHttpAsyncClient client = HttpAsyncClients.custom().setConnectionManager(cm).build();
+        client.start();
+        final String[] toGet = { "http://www.google.com/", "http://www.apache.org/", "http://www.bing.com/" };
+
+        final GetThread[] threads = new GetThread[toGet.length];
+        for (int i = 0; i < threads.length; i++) {
+            final HttpGet request = new HttpGet(toGet[i]);
+            threads[i] = new GetThread(client, request);
+        }
+
+        for (final GetThread thread : threads) {
+            thread.start();
+        }
+
+        for (final GetThread thread : threads) {
+            thread.join();
+        }
+
+    }
+
+    @Test
+    public void whenUseProxyWithHttpClient_thenCorrect() throws Exception {
+        final CloseableHttpAsyncClient client = HttpAsyncClients.createDefault();
+        client.start();
+        final HttpHost proxy = new HttpHost("74.50.126.248", 3127);
+        final RequestConfig config = RequestConfig.custom().setProxy(proxy).build();
+        final HttpGet request = new HttpGet(HOST_WITH_PROXY);
+        request.setConfig(config);
+        final Future<HttpResponse> future = client.execute(request, null);
+        final HttpResponse response = future.get();
+        assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
+        client.close();
+    }
+
+    @Test
+    public void whenUseSSLWithHttpAsyncClient_thenCorrect() throws Exception {
+        final TrustStrategy acceptingTrustStrategy = new TrustStrategy() {
+            @Override
+            public final boolean isTrusted(final X509Certificate[] certificate, final String authType) {
+                return true;
+            }
+        };
+        final SSLContext sslContext = SSLContexts.custom().loadTrustMaterial(null, acceptingTrustStrategy).build();
+
+        final CloseableHttpAsyncClient client = HttpAsyncClients.custom().setSSLHostnameVerifier(SSLConnectionSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER).setSSLContext(sslContext).build();
+
+        client.start();
+        final HttpGet request = new HttpGet(HOST_WITH_SSL);
+        final Future<HttpResponse> future = client.execute(request, null);
+        final HttpResponse response = future.get();
+        assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
+        client.close();
+    }
+
+
+    @Test
+    public void whenUseCookiesWithHttpAsyncClient_thenCorrect() throws Exception {
+        final BasicCookieStore cookieStore = new BasicCookieStore();
+        final BasicClientCookie cookie = new BasicClientCookie(COOKIE_NAME, "1234");
+        cookie.setDomain(COOKIE_DOMAIN);
+        cookie.setPath("/");
+        cookieStore.addCookie(cookie);
+        final CloseableHttpAsyncClient client = HttpAsyncClients.custom().build();
+        client.start();
+        final HttpGet request = new HttpGet(HOST_WITH_COOKIE);
+
+        final HttpContext localContext = new BasicHttpContext();
+        localContext.setAttribute(HttpClientContext.COOKIE_STORE, cookieStore);
+
+        final Future<HttpResponse> future = client.execute(request, localContext, null);
+        final HttpResponse response = future.get();
+        assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
+        client.close();
+    }
+
+    @Test
+    public void whenUseAuthenticationWithHttpAsyncClient_thenCorrect() throws Exception {
+        final CredentialsProvider provider = new BasicCredentialsProvider();
+        final UsernamePasswordCredentials credentials = new UsernamePasswordCredentials(DEFAULT_USER, DEFAULT_PASS);
+        provider.setCredentials(AuthScope.ANY, credentials);
+        final CloseableHttpAsyncClient client = HttpAsyncClients.custom().setDefaultCredentialsProvider(provider).build();
+
+        final HttpGet request = new HttpGet(URL_SECURED_BY_BASIC_AUTHENTICATION);
+        client.start();
+        final Future<HttpResponse> future = client.execute(request, null);
+        final HttpResponse response = future.get();
+        assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
+        client.close();
+    }
+
+    static class GetThread extends Thread {
+
+        private final CloseableHttpAsyncClient client;
+        private final HttpContext context;
+        private final HttpGet request;
+
+        public GetThread(final CloseableHttpAsyncClient client, final HttpGet request) {
+            this.client = client;
+            context = HttpClientContext.create();
+            this.request = request;
+        }
+
+        @Override
+        public void run() {
+            try {
+                final Future<HttpResponse> future = client.execute(request, context, null);
+                final HttpResponse response = future.get();
+                assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
+            } catch (final Exception ex) {
+                System.out.println(ex.getLocalizedMessage());
+            }
+        }
+
+    }
+}
+