/** * This test just make sure the hack used to catch disconnected channel under win7 doesn't throw any exception. The onComplete method must be only called once. * * @throws Exception if something wrong happens. */ @Test public void win7DisconnectTest() throws Exception { final AtomicInteger count = new AtomicInteger(0); try (AsyncHttpClient client = asyncHttpClient()) { AsyncCompletionHandler<Response> handler = new AsyncCompletionHandlerAdapter() { @Override public Response onCompleted(Response response) throws Exception { count.incrementAndGet(); StackTraceElement e = new StackTraceElement("sun.nio.ch.SocketDispatcher", "read0", null, -1); IOException t = new IOException(); t.setStackTrace(new StackTraceElement[]{e}); throw t; } }; try { client.prepareGet(getTargetUrl()).execute(handler).get(); fail("Must have received an exception"); } catch (ExecutionException ex) { assertNotNull(ex); assertNotNull(ex.getCause()); assertEquals(ex.getCause().getClass(), IOException.class); assertEquals(count.get(), 1); } } }
final CountDownLatch latch = new CountDownLatch(16); for (int i = 0; i < 16; i++) { client.prepareGet(getTargetUrl()).execute(new AsyncCompletionHandlerBase() { @Override public Response onCompleted(Response response) throws Exception { client.prepareGet(getTargetUrl()).execute(new AsyncCompletionHandlerBase() { @Override public void onThrowable(Throwable t) {
client.prepareGet(getTargetUrl()).execute(handler).get(); server.stop(); server.start(); client.prepareGet(getTargetUrl()).execute(handler);
@Test(expectedExceptions = TooManyConnectionsException.class) public void testMaxTotalConnectionsException() throws Throwable { try (AsyncHttpClient client = asyncHttpClient(config().setKeepAlive(true).setMaxConnections(1))) { String url = getTargetUrl(); List<ListenableFuture<Response>> futures = new ArrayList<>(); for (int i = 0; i < 5; i++) { logger.info("{} requesting url [{}]...", i, url); futures.add(client.prepareGet(url).execute()); } Exception exception = null; for (ListenableFuture<Response> future : futures) { try { future.get(); } catch (Exception ex) { exception = ex; break; } } assertNotNull(exception); throw exception.getCause(); } }
@Test public void testMaxTotalConnections() throws Exception { try (AsyncHttpClient client = asyncHttpClient(config().setKeepAlive(true).setMaxConnections(1))) { String url = getTargetUrl(); int i; Exception exception = null; for (i = 0; i < 3; i++) { try { logger.info("{} requesting url [{}]...", i, url); Response response = client.prepareGet(url).execute().get(); logger.info("{} response [{}].", i, response); } catch (Exception ex) { exception = ex; } } assertNull(exception); } }
@Test public void multipleMaxConnectionOpenTestWithQuery() throws Exception { try (AsyncHttpClient c = asyncHttpClient(config().setKeepAlive(true).setConnectTimeout(5000).setMaxConnections(1))) { String body = "hello there"; // once Response response = c.preparePost(getTargetUrl() + "?foo=bar").setBody(body).execute().get(TIMEOUT, TimeUnit.SECONDS); assertEquals(response.getResponseBody(), "foo_" + body); // twice Exception exception = null; try { response = c.preparePost(getTargetUrl()).setBody(body).execute().get(TIMEOUT, TimeUnit.SECONDS); } catch (Exception ex) { ex.printStackTrace(); exception = ex; } assertNull(exception); assertNotNull(response); assertEquals(response.getStatusCode(), 200); } }
@Test(expectedExceptions = TooManyConnectionsException.class) public void multipleMaxConnectionOpenTest() throws Throwable { try (AsyncHttpClient c = asyncHttpClient(config().setKeepAlive(true).setConnectTimeout(5000).setMaxConnections(1))) { String body = "hello there"; // once Response response = c.preparePost(getTargetUrl()).setBody(body).execute().get(TIMEOUT, TimeUnit.SECONDS); assertEquals(response.getResponseBody(), body); // twice Exception exception = null; try { c.preparePost(String.format("http://localhost:%d/foo/test", port2)).setBody(body).execute().get(TIMEOUT, TimeUnit.SECONDS); fail("Should throw exception. Too many connections issued."); } catch (Exception ex) { ex.printStackTrace(); exception = ex; } assertNotNull(exception); throw exception.getCause(); } }
@Test public void nonPoolableConnectionReleaseSemaphoresTest() throws Throwable { RequestBuilder request = get(getTargetUrl()).setHeader("Connection", "close"); try (AsyncHttpClient client = asyncHttpClient(config().setMaxConnections(6).setMaxConnectionsPerHost(3))) { client.executeRequest(request).get(); Thread.sleep(1000); client.executeRequest(request).get(); Thread.sleep(1000); client.executeRequest(request).get(); Thread.sleep(1000); client.executeRequest(request).get(); } }