Add tests for connection events

This is groundwork for adding more events in AsyncHttpClient#732.

Sadly I couldn’t put these higher up in the test hierarchy, because the
Grizzly provider doesn’t seem to fire most events.

Author: barnardb

src/test/java/com/ning/http/client/async/EventCollectingHandler.java

@@ -0,0 +1,89 @@
+package com.ning.http.client.async;
+
+import org.testng.Assert;
+
+import com.ning.http.client.AsyncCompletionHandlerBase;
+import com.ning.http.client.AsyncHandlerExtensions;
+import com.ning.http.client.HttpResponseHeaders;
+import com.ning.http.client.HttpResponseStatus;
+import com.ning.http.client.Response;
+
+import java.util.Queue;
+import java.util.concurrent.ConcurrentLinkedQueue;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
+
+public class EventCollectingHandler extends AsyncCompletionHandlerBase implements AsyncHandlerExtensions {
+    public Queue<String> firedEvents = new ConcurrentLinkedQueue<String>();
+    private CountDownLatch completionLatch = new CountDownLatch(1);
+
+    public void waitForCompletion() throws InterruptedException {
+        if (!completionLatch.await(AbstractBasicTest.TIMEOUT, TimeUnit.SECONDS)) {
+            Assert.fail("Timeout out");
+        }
+    }
+
+    @Override
+    public Response onCompleted(Response response) throws Exception {
+        firedEvents.add("Completed");
+        try {
+            return super.onCompleted(response);
+        } finally {
+            completionLatch.countDown();
+        }
+    }
+
+    @Override
+    public STATE onStatusReceived(HttpResponseStatus status) throws Exception {
+        firedEvents.add("StatusReceived");
+        return super.onStatusReceived(status);
+    }
+
+    @Override
+    public STATE onHeadersReceived(HttpResponseHeaders headers) throws Exception {
+        firedEvents.add("HeadersReceived");
+        return super.onHeadersReceived(headers);
+    }
+
+    @Override
+    public STATE onHeaderWriteCompleted() {
+        firedEvents.add("HeaderWriteCompleted");
+        return super.onHeaderWriteCompleted();
+    }
+
+    @Override
+    public STATE onContentWriteCompleted() {
+        firedEvents.add("ContentWriteCompleted");
+        return super.onContentWriteCompleted();
+    }
+
+    @Override
+    public void onOpenConnection() {
+        firedEvents.add("OpenConnection");
+    }
+
+    @Override
+    public void onConnectionOpen() {
+        firedEvents.add("ConnectionOpen");
+    }
+
+    @Override
+    public void onPoolConnection() {
+        firedEvents.add("PoolConnection");
+    }
+
+    @Override
+    public void onConnectionPooled() {
+        firedEvents.add("ConnectionPooled");
+    }
+
+    @Override
+    public void onSendRequest(Object request) {
+        firedEvents.add("SendRequest");
+    }
+
+    @Override
+    public void onRetry() {
+        firedEvents.add("Retry");
+    }
+}

src/test/java/com/ning/http/client/async/netty/NettyAsyncProviderBasicTest.java

@@ -12,15 +12,27 @@
  */
 package com.ning.http.client.async.netty;
 
+import static org.testng.Assert.assertEquals;
+
 import org.testng.annotations.Test;
 
+import com.google.common.base.Joiner;
 import com.ning.http.client.AsyncHttpClient;
 import com.ning.http.client.AsyncHttpClientConfig;
 import com.ning.http.client.AsyncHttpProviderConfig;
+import com.ning.http.client.Request;
+import com.ning.http.client.RequestBuilder;
 import com.ning.http.client.async.AsyncProvidersBasicTest;
+import com.ning.http.client.async.EventCollectingHandler;
 import com.ning.http.client.async.ProviderUtil;
 import com.ning.http.client.providers.netty.NettyAsyncHttpProviderConfig;
 
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
 @Test
 public class NettyAsyncProviderBasicTest extends AsyncProvidersBasicTest {
 
@@ -38,4 +50,29 @@ public AsyncHttpClient getAsyncHttpClient(AsyncHttpClientConfig config) {
     protected String generatedAcceptEncodingHeader() {
         return "gzip,deflate";
     }
+
+    @Test(groups = { "standalone", "default_provider", "async" })
+    public void testNewConnectionEventsFired() throws InterruptedException, TimeoutException, ExecutionException {
+        Request request = new RequestBuilder("GET").setUrl("http://127.0.0.1:" + port1 + "/Test").build();
+
+        try (AsyncHttpClient client = getAsyncHttpClient(null)) {
+            EventCollectingHandler handler = new EventCollectingHandler();
+            client.executeRequest(request, handler).get(3, TimeUnit.SECONDS);
+            handler.waitForCompletion();
+
+            List<String> expectedEvents = Arrays.asList(
+                    "PoolConnection",
+                    "OpenConnection",
+                    "ConnectionOpen",
+                    "SendRequest",
+                    "HeaderWriteCompleted",
+                    "StatusReceived",
+                    "HeadersReceived",
+                    "Completed");
+
+            assertEquals(handler.firedEvents, expectedEvents,
+                    "Got: " + Joiner.on(", ").join(handler.firedEvents));
+        }
+
+    }
 }

src/test/java/com/ning/http/client/async/netty/NettyBasicHttpsTest.java

@@ -12,13 +12,49 @@
  */
 package com.ning.http.client.async.netty;
 
+import static org.testng.Assert.assertEquals;
+
+import org.testng.annotations.Test;
+
+import com.google.common.base.Joiner;
 import com.ning.http.client.AsyncHttpClient;
 import com.ning.http.client.AsyncHttpClientConfig;
 import com.ning.http.client.async.BasicHttpsTest;
+import com.ning.http.client.async.EventCollectingHandler;
 import com.ning.http.client.async.ProviderUtil;
 
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+import java.util.concurrent.atomic.AtomicBoolean;
+
 public class NettyBasicHttpsTest extends BasicHttpsTest {
 
+    @Test
+    public void testNormalEventsFired() throws InterruptedException, TimeoutException, ExecutionException {
+        try (AsyncHttpClient client = getAsyncHttpClient(new AsyncHttpClientConfig.Builder().setSSLContext(createSSLContext(new AtomicBoolean(true))).build())) {
+            EventCollectingHandler handler = new EventCollectingHandler();
+            client.preparePost(getTargetUrl()).setBody("whatever").execute(handler).get(3, TimeUnit.SECONDS);
+            handler.waitForCompletion();
+
+            List<String> expectedEvents = Arrays.asList(
+                    "PoolConnection",
+                    "OpenConnection",
+                    "ConnectionOpen",
+                    "SendRequest",
+                    "HeaderWriteCompleted",
+                    "StatusReceived",
+                    "HeadersReceived",
+                    "Completed");
+
+            assertEquals(handler.firedEvents, expectedEvents,
+                    "Got: " + Joiner.on(", ").join(handler.firedEvents));
+        }
+
+    }
+
     @Override
     public AsyncHttpClient getAsyncHttpClient(AsyncHttpClientConfig config) {
         return ProviderUtil.nettyProvider(config);

src/test/java/com/ning/http/client/async/netty/NettyConnectionPoolTest.java

@@ -21,16 +21,22 @@
 import org.jboss.netty.channel.Channel;
 import org.testng.annotations.Test;
 
+import com.google.common.base.Joiner;
 import com.ning.http.client.AsyncHttpClient;
 import com.ning.http.client.AsyncHttpClientConfig;
+import com.ning.http.client.Request;
+import com.ning.http.client.RequestBuilder;
 import com.ning.http.client.Response;
 import com.ning.http.client.async.ConnectionPoolTest;
+import com.ning.http.client.async.EventCollectingHandler;
 import com.ning.http.client.async.ProviderUtil;
 import com.ning.http.client.providers.netty.NettyAsyncHttpProviderConfig;
 import com.ning.http.client.providers.netty.channel.pool.ChannelPool;
 import com.ning.http.client.providers.netty.channel.pool.NoopChannelPool;
 
 import java.net.ConnectException;
+import java.util.Arrays;
+import java.util.List;
 import java.util.concurrent.TimeUnit;
 
 public class NettyConnectionPoolTest extends ConnectionPoolTest {
@@ -131,5 +137,31 @@ public void testHostNotContactable() {
         }
     }
 
+    @Test
+    public void testPooledEventsFired() throws Exception {
+        Request request = new RequestBuilder("GET").setUrl("http://127.0.0.1:" + port1 + "/Test").build();
+
+        try (AsyncHttpClient client = getAsyncHttpClient(null)) {
+            EventCollectingHandler firstHandler = new EventCollectingHandler();
+            client.executeRequest(request, firstHandler).get(3, TimeUnit.SECONDS);
+            firstHandler.waitForCompletion();
+
+            EventCollectingHandler secondHandler = new EventCollectingHandler();
+            client.executeRequest(request, secondHandler).get(3, TimeUnit.SECONDS);
+            secondHandler.waitForCompletion();
+
+            List<String> expectedEvents = Arrays.asList(
+                    "PoolConnection",
+                    "ConnectionPooled",
+                    "SendRequest",
+                    "HeaderWriteCompleted",
+                    "StatusReceived",
+                    "HeadersReceived",
+                    "Completed");
+
+            assertEquals(secondHandler.firedEvents, expectedEvents,
+                    "Got: " + Joiner.on(", ").join(secondHandler.firedEvents));
+        }
+    }
 
 }