Java tutorial
/* * Copyright 2009-2016 Weibo, Inc. * * 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.weibo.api.motan.transport.netty4.http; import static org.junit.Assert.*; import io.netty.buffer.ByteBuf; import io.netty.buffer.PooledByteBufAllocator; import io.netty.channel.ChannelHandlerContext; import io.netty.handler.codec.http.DefaultFullHttpRequest; import io.netty.handler.codec.http.DefaultHttpHeaders; import io.netty.handler.codec.http.FullHttpRequest; import io.netty.handler.codec.http.FullHttpResponse; import io.netty.handler.codec.http.HttpMethod; import io.netty.handler.codec.http.HttpResponseStatus; import io.netty.handler.codec.http.HttpVersion; import org.jmock.Expectations; import org.jmock.api.Invocation; import org.jmock.integration.junit4.JUnit4Mockery; import org.jmock.lib.action.CustomAction; import org.jmock.lib.legacy.ClassImposteriser; import org.junit.After; import org.junit.Before; import org.junit.Test; import com.weibo.api.motan.common.MotanConstants; import com.weibo.api.motan.transport.Channel; import com.weibo.api.motan.transport.MessageHandler; import com.weibo.api.motan.util.MotanSwitcherUtil; /** * * @Description NettyHttpRequestHandlerTest * @author zhanglei * @date 2016727 * */ public class NettyHttpRequestHandlerTest { public static JUnit4Mockery mockery = null; @Before public void setUp() throws Exception { mockery = new JUnit4Mockery() { { setImposteriser(ClassImposteriser.INSTANCE); } }; } @After public void tearDown() throws Exception { } @Test public void testChannelRead0() throws Exception { final MessageHandler messageHandler = mockery.mock(MessageHandler.class); final ChannelHandlerContext ctx = mockery.mock(ChannelHandlerContext.class); final FullHttpResponse response = mockery.mock(FullHttpResponse.class); mockery.checking(new Expectations() { { allowing(ctx).write(with(any(FullHttpResponse.class))); will(new CustomAction("verify") { @Override public Object invoke(Invocation invocation) throws Throwable { FullHttpResponse actualResponse = (FullHttpResponse) invocation.getParameter(0); assertNotNull(actualResponse); assertEquals(response, actualResponse); return null; } }); allowing(ctx).flush(); will(returnValue(null)); allowing(ctx).close(); will(returnValue(null)); atLeast(1).of(messageHandler).handle(with(any(Channel.class)), with(anything())); will(returnValue(response)); allowing(response).headers(); will(returnValue(new DefaultHttpHeaders())); } }); FullHttpRequest httpRequest = buildHttpRequest("anyPath"); NettyHttpRequestHandler handler = new NettyHttpRequestHandler(null, messageHandler); handler.channelRead0(ctx, httpRequest); } @Test public void testServerStatus() throws Exception { final MessageHandler messageHandler = mockery.mock(MessageHandler.class); final ChannelHandlerContext ctx = mockery.mock(ChannelHandlerContext.class); mockery.checking(new Expectations() { { allowing(ctx).write(with(any(FullHttpResponse.class))); will(new CustomAction("verify") { @Override public Object invoke(Invocation invocation) throws Throwable { verifyStatus((FullHttpResponse) invocation.getParameter(0)); return null; } }); allowing(ctx).flush(); will(returnValue(null)); allowing(ctx).close(); will(returnValue(null)); allowing(messageHandler).handle(with(any(Channel.class)), with(anything())); will(returnValue(null)); } }); FullHttpRequest httpRequest = buildHttpRequest(NettyHttpRequestHandler.ROOT_PATH); NettyHttpRequestHandler handler = new NettyHttpRequestHandler(null, messageHandler); // MotanSwitcherUtil.setSwitcherValue(MotanConstants.REGISTRY_HEARTBEAT_SWITCHER, false); handler.channelRead0(ctx, httpRequest); // MotanSwitcherUtil.setSwitcherValue(MotanConstants.REGISTRY_HEARTBEAT_SWITCHER, true); handler.channelRead0(ctx, httpRequest); } private void verifyStatus(FullHttpResponse response) { if (MotanSwitcherUtil.isOpen(MotanConstants.REGISTRY_HEARTBEAT_SWITCHER)) { assertEquals(HttpResponseStatus.OK, response.getStatus()); } else { assertEquals(HttpResponseStatus.SERVICE_UNAVAILABLE, response.getStatus()); } } private FullHttpRequest buildHttpRequest(String requestPath) throws Exception { PooledByteBufAllocator allocator = new PooledByteBufAllocator(); ByteBuf buf = allocator.buffer(0); FullHttpRequest httpReqeust = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.POST, requestPath, buf); return httpReqeust; } }