From b34e5299de3449ce9176ae4ee1f3b93cf5ec1558 Mon Sep 17 00:00:00 2001
From: Jaskey <linjunjie1103@gmail.com>
Date: Wed, 28 Dec 2016 11:43:51 +0800
Subject: [PATCH] ROCKET_14 unit test updated

---
 .../remoting/NettyConnectionTest.java         | 53 +++++++++++++++----
 1 file changed, 43 insertions(+), 10 deletions(-)

diff --git a/rocketmq-remoting/src/test/java/com/alibaba/rocketmq/remoting/NettyConnectionTest.java b/rocketmq-remoting/src/test/java/com/alibaba/rocketmq/remoting/NettyConnectionTest.java
index 755d3322605..362897b21db 100644
--- a/rocketmq-remoting/src/test/java/com/alibaba/rocketmq/remoting/NettyConnectionTest.java
+++ b/rocketmq-remoting/src/test/java/com/alibaba/rocketmq/remoting/NettyConnectionTest.java
@@ -20,14 +20,14 @@
 import com.alibaba.rocketmq.remoting.exception.RemotingConnectException;
 import com.alibaba.rocketmq.remoting.exception.RemotingSendRequestException;
 import com.alibaba.rocketmq.remoting.exception.RemotingTimeoutException;
-import com.alibaba.rocketmq.remoting.netty.NettyClientConfig;
-import com.alibaba.rocketmq.remoting.netty.NettyRemotingClient;
-import com.alibaba.rocketmq.remoting.netty.ResponseFuture;
+import com.alibaba.rocketmq.remoting.netty.*;
 import com.alibaba.rocketmq.remoting.protocol.RemotingCommand;
+import io.netty.channel.ChannelHandlerContext;
 import org.junit.Assert;
 import org.junit.Test;
 
 import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.Executors;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
 
@@ -57,34 +57,44 @@ public void test_connect_timeout() throws InterruptedException, RemotingConnectE
         System.out.println("-----------------------------------------------------------------");
     }
 
-
+    /**
+     * test of https://github.com/apache/incubator-rocketmq/pull/2#issuecomment-269290436
+     * by linjunjie1103@gmail.com
+     *
+     * @throws InterruptedException
+     * @throws RemotingConnectException
+     * @throws RemotingSendRequestException
+     * @throws RemotingTimeoutException
+     */
     @Test
     public void test_async_timeout() throws InterruptedException, RemotingConnectException,
             RemotingSendRequestException, RemotingTimeoutException {
+
+        RemotingServer server = createRemotingServer();//mock server,start internally
         RemotingClient client = createRemotingClient();
         final AtomicInteger ai = new AtomicInteger(0);
         final CountDownLatch latch = new CountDownLatch(100);
         for(int i=0;i<100;i++) {
             try {
-                RemotingCommand request = RemotingCommand.createRequestCommand(0, null);
-                client.invokeAsync("localhost:8888", request, 5, new InvokeCallback() {//very easy to timeout
+                RemotingCommand request = RemotingCommand.createRequestCommand(0, null);//request code 0,
+                client.invokeAsync("localhost:8888", request, 5, new InvokeCallback() {//make it very easy to timeout,connect to name server,
                     @Override
                     public void operationComplete(ResponseFuture responseFuture) {
                         if (responseFuture.isTimeout()) {
                             if(ai.getAndIncrement()==4) {
                                 try {
-                                    System.out.println("First try timeout,  blocking 10s" + Thread.currentThread().getName());
+                                    System.out.println("A long timeout callback,  blocking 10s. Current Thread:" + Thread.currentThread().getName());
                                     Thread.sleep(10 * 1000);
                                 } catch (InterruptedException e) {
                                     e.printStackTrace();
                                 }
                             }
                             else{
-                                System.out.println("Timeout callback execute,very short."+Thread.currentThread().getName());
+                                System.out.println("Short timeout callback execute, Current Thread:"+Thread.currentThread().getName());
                             }
                         }
                         else{
-                            System.out.println("Success."+Thread.currentThread().getName());
+                            System.out.println("Success callback.Current Thread:"+Thread.currentThread().getName());
                         }
                         latch.countDown();
 
@@ -97,9 +107,10 @@ public void operationComplete(ResponseFuture responseFuture) {
 
 
 
-        latch.await(1000, TimeUnit.MILLISECONDS);
+        latch.await(3100, TimeUnit.MILLISECONDS);//scan response table will be triggered after 3000ms, so we wait a litter more than 3000ms to let it work
         Assert.assertEquals(1, latch.getCount());//only one should be blocked
         client.shutdown();
+        server.shutdown();
         System.out.println("-----------------------------------------------------------------");
     }
 
@@ -110,4 +121,26 @@ public static RemotingClient createRemotingClient() {
         client.start();
         return client;
     }
+
+    public static RemotingServer createRemotingServer() throws InterruptedException {
+        NettyServerConfig config = new NettyServerConfig();
+        RemotingServer server = new NettyRemotingServer(config);
+        server.registerProcessor(0, new NettyRequestProcessor() {
+            private AtomicInteger i = new AtomicInteger();
+
+            @Override
+            public RemotingCommand processRequest(ChannelHandlerContext ctx, RemotingCommand request) {
+                System.out.println("processRequest=" + request + " " + (i.incrementAndGet()));
+                request.setRemark("hello, I am response " + ctx.channel().remoteAddress());
+                return request;
+            }
+
+            @Override
+            public boolean rejectRequest() {
+                return false;
+            }
+        }, Executors.newCachedThreadPool());
+        server.start();
+        return server;
+    }
 }