/* * Copyright 1999-2011 Alibaba Group. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.alibaba.dubbo.remoting.handler; import static org.junit.Assert.*; import java.lang.reflect.Field; import junit.framework.Assert; import org.junit.Before; import org.junit.Test; import com.alibaba.dubbo.common.URL; import com.alibaba.dubbo.remoting.Channel; import com.alibaba.dubbo.remoting.RemotingException; import com.alibaba.dubbo.remoting.transport.dispatcher.WrappedChannelHandler; public class WrappedChannelHandlerTest { WrappedChannelHandler handler ; URL url = URL.valueOf("test://10.20.30.40:1234"); @Before public void setUp() throws Exception { handler = new WrappedChannelHandler(new BizChannelHander(true), url); } @Test public void test_Execute_Error() throws RemotingException{ } protected Object getField(Object obj, String fieldName, int parentdepth){ try{ Class<?> clazz = obj.getClass(); Field field = null; for(int i=0;i <= parentdepth && field == null ;i++){ Field[] fields = clazz.getDeclaredFields(); for(Field f : fields){ if (f.getName().equals(fieldName)){ field = f; break; } } clazz = clazz.getSuperclass(); } if (field != null){ field.setAccessible(true); return field.get(obj); }else { throw new NoSuchFieldException(); } }catch (Exception e) { throw new IllegalStateException(e); } } protected void sleep(int ms){ try { Thread.sleep(ms); } catch (InterruptedException e) { Thread.currentThread().interrupt(); } } class BizChannelHander extends MockedChannelHandler{ private boolean invokeWithBizError; public BizChannelHander(boolean invokeWithBizError) { super(); this.invokeWithBizError = invokeWithBizError; } public BizChannelHander() { super(); } @Override public void connected(Channel channel) throws RemotingException { if (invokeWithBizError){ throw new RemotingException(channel, "test connect biz error"); } sleep(20); } @Override public void disconnected(Channel channel) throws RemotingException { if (invokeWithBizError){ throw new RemotingException(channel, "test disconnect biz error"); } sleep(20); } @Override public void received(Channel channel, Object message) throws RemotingException { if (invokeWithBizError){ throw new RemotingException(channel, "test received biz error"); } sleep(20); } }; class BizException extends RuntimeException{ private static final long serialVersionUID = -7541893754900723624L; } @Test(expected = RemotingException.class) public void test_Connect_Biz_Error() throws RemotingException{ handler.connected(new MockedChannel()); } @Test(expected = RemotingException.class) public void test_Disconnect_Biz_Error() throws RemotingException{ handler.disconnected(new MockedChannel()); } @Test(expected = RemotingException.class) public void test_MessageReceived_Biz_Error() throws RemotingException{ handler.received(new MockedChannel(),""); } @Test public void test_Caught_Biz_Error() throws RemotingException{ try{ handler.caught(new MockedChannel(), new BizException()); fail(); }catch (Exception e) { Assert.assertEquals(BizException.class, e.getCause().getClass()); } } }