DUBBO-70 增加Cluster层传递RpcException的ErrorCode测试用例

git-svn-id: http://code.alibabatech.com/svn/dubbo/trunk@752 1a56cb94-b969-4eaa-88fa-be21384802f2

dubbo-cluster/src/test/java/com/alibaba/dubbo/rpc/cluster/support/FailfastClusterInvokerTest.java

@@ -28,13 +28,17 @@ import org.junit.Before;
 import org.junit.Test;
 
 import com.alibaba.dubbo.common.URL;
+import com.alibaba.dubbo.common.utils.NetUtils;
 import com.alibaba.dubbo.rpc.Invocation;
 import com.alibaba.dubbo.rpc.Invoker;
 import com.alibaba.dubbo.rpc.Result;
 import com.alibaba.dubbo.rpc.RpcContext;
 import com.alibaba.dubbo.rpc.RpcException;
+import com.alibaba.dubbo.rpc.RpcInvocation;
 import com.alibaba.dubbo.rpc.RpcResult;
 import com.alibaba.dubbo.rpc.cluster.Directory;
+import com.alibaba.dubbo.rpc.cluster.directory.StaticDirectory;
+import com.alibaba.dubbo.rpc.cluster.filter.DemoService;
 
 /**
  * FailfastClusterInvokerTest
@@ -130,6 +134,39 @@ public class FailfastClusterInvokerTest {
         } catch (RpcException expected) {
         }
     }
-
+    
+    @Test()
+    public void testTimeoutExceptionCode() {
+        List<Invoker<DemoService>> invokers = new ArrayList<Invoker<DemoService>>();
+        invokers.add(new Invoker<DemoService>() {
+
+            public Class<DemoService> getInterface() {
+                return DemoService.class;
+            }
+
+            public URL getUrl() {
+                return URL.valueOf("dubbo://" + NetUtils.getLocalHost() + ":20880/" + DemoService.class.getName());
+            }
+
+            public boolean isAvailable() {
+                return false;
+            }
+
+            public Result invoke(Invocation invocation) throws RpcException {
+                throw new RpcException(RpcException.TIMEOUT_EXCEPTION, "test timeout");
+            }
+
+            public void destroy() {
+            }
+        });
+        Directory<DemoService> directory = new StaticDirectory<DemoService>(invokers);
+        FailfastClusterInvoker<DemoService> cluster = new FailfastClusterInvoker<DemoService>(directory);
+        try {
+            cluster.invoke(new RpcInvocation("sayHello", new Class<?>[0], new Object[0]));
+            Assert.fail();
+        } catch (RpcException e) {
+            Assert.assertEquals(RpcException.TIMEOUT_EXCEPTION, e.getCode());
+        }
+    }
 
 }
\ No newline at end of file

dubbo-cluster/src/test/java/com/alibaba/dubbo/rpc/cluster/support/FailoverClusterInvokerTest.java

@@ -23,17 +23,23 @@ import static org.junit.Assert.fail;
 import java.util.ArrayList;
 import java.util.List;
 
+import junit.framework.Assert;
+
 import org.easymock.EasyMock;
 import org.junit.Before;
 import org.junit.Test;
 
 import com.alibaba.dubbo.common.URL;
+import com.alibaba.dubbo.common.utils.NetUtils;
 import com.alibaba.dubbo.rpc.Invocation;
 import com.alibaba.dubbo.rpc.Invoker;
 import com.alibaba.dubbo.rpc.Result;
 import com.alibaba.dubbo.rpc.RpcException;
+import com.alibaba.dubbo.rpc.RpcInvocation;
 import com.alibaba.dubbo.rpc.RpcResult;
 import com.alibaba.dubbo.rpc.cluster.Directory;
+import com.alibaba.dubbo.rpc.cluster.directory.StaticDirectory;
+import com.alibaba.dubbo.rpc.cluster.filter.DemoService;
 
 /**
  * FailoverClusterInvokerTest
@@ -171,5 +177,39 @@ public class FailoverClusterInvokerTest {
         } catch (RpcException expected) {
             expected.printStackTrace();
         }
+    }
+    
+    @Test()
+    public void testTimeoutExceptionCode() {
+        List<Invoker<DemoService>> invokers = new ArrayList<Invoker<DemoService>>();
+        invokers.add(new Invoker<DemoService>() {
+
+            public Class<DemoService> getInterface() {
+                return DemoService.class;
+            }
+
+            public URL getUrl() {
+                return URL.valueOf("dubbo://" + NetUtils.getLocalHost() + ":20880/" + DemoService.class.getName());
+            }
+
+            public boolean isAvailable() {
+                return false;
+            }
+
+            public Result invoke(Invocation invocation) throws RpcException {
+                throw new RpcException(RpcException.TIMEOUT_EXCEPTION, "test timeout");
+            }
+
+            public void destroy() {
+            }
+        });
+        Directory<DemoService> directory = new StaticDirectory<DemoService>(invokers);
+        FailoverClusterInvoker<DemoService> cluster = new FailoverClusterInvoker<DemoService>(directory);
+        try {
+            cluster.invoke(new RpcInvocation("sayHello", new Class<?>[0], new Object[0]));
+            Assert.fail();
+        } catch (RpcException e) {
+            Assert.assertEquals(RpcException.TIMEOUT_EXCEPTION, e.getCode());
+        }
     }
 }
\ No newline at end of file

dubbo-cluster/src/test/java/com/alibaba/dubbo/rpc/cluster/support/ForkingClusterInvokerTest.java

@@ -28,12 +28,16 @@ import org.junit.Before;
 import org.junit.Test;
 
 import com.alibaba.dubbo.common.URL;
+import com.alibaba.dubbo.common.utils.NetUtils;
 import com.alibaba.dubbo.rpc.Invocation;
 import com.alibaba.dubbo.rpc.Invoker;
 import com.alibaba.dubbo.rpc.Result;
 import com.alibaba.dubbo.rpc.RpcException;
+import com.alibaba.dubbo.rpc.RpcInvocation;
 import com.alibaba.dubbo.rpc.RpcResult;
 import com.alibaba.dubbo.rpc.cluster.Directory;
+import com.alibaba.dubbo.rpc.cluster.directory.StaticDirectory;
+import com.alibaba.dubbo.rpc.cluster.filter.DemoService;
 
 /**
  * ForkingClusterInvokerTest
@@ -170,4 +174,38 @@ public class ForkingClusterInvokerTest {
             assertTrue(expected.getMessage().contains("No provider"));
         }
     }
+    
+    @Test()
+    public void testTimeoutExceptionCode() {
+        List<Invoker<DemoService>> invokers = new ArrayList<Invoker<DemoService>>();
+        invokers.add(new Invoker<DemoService>() {
+
+            public Class<DemoService> getInterface() {
+                return DemoService.class;
+            }
+
+            public URL getUrl() {
+                return URL.valueOf("dubbo://" + NetUtils.getLocalHost() + ":20880/" + DemoService.class.getName());
+            }
+
+            public boolean isAvailable() {
+                return false;
+            }
+
+            public Result invoke(Invocation invocation) throws RpcException {
+                throw new RpcException(RpcException.TIMEOUT_EXCEPTION, "test timeout");
+            }
+
+            public void destroy() {
+            }
+        });
+        Directory<DemoService> directory = new StaticDirectory<DemoService>(invokers);
+        ForkingClusterInvoker<DemoService> cluster = new ForkingClusterInvoker<DemoService>(directory);
+        try {
+            cluster.invoke(new RpcInvocation("sayHello", new Class<?>[0], new Object[0]));
+            Assert.fail();
+        } catch (RpcException e) {
+            Assert.assertEquals(RpcException.TIMEOUT_EXCEPTION, e.getCode());
+        }
+    }
 }
\ No newline at end of file