Java tutorial
/* * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved. * * This program and the accompanying materials are made available under the * terms of the Eclipse Public License v1.0 which accompanies this distribution, * and is available at http://www.eclipse.org/legal/epl-v10.html */ package org.opendaylight.controller.netconf.client; import static org.mockito.Matchers.any; import static org.mockito.Matchers.anyObject; import static org.mockito.Matchers.anyString; import static org.mockito.Mockito.doAnswer; import static org.mockito.Mockito.doReturn; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import com.google.common.base.Optional; import io.netty.channel.Channel; import io.netty.channel.ChannelFuture; import io.netty.channel.ChannelHandler; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelInboundHandlerAdapter; import io.netty.channel.ChannelPipeline; import io.netty.channel.ChannelProgressivePromise; import io.netty.handler.ssl.SslHandler; import io.netty.util.HashedWheelTimer; import io.netty.util.Timer; import io.netty.util.concurrent.GenericFutureListener; import io.netty.util.concurrent.Promise; import java.util.Set; import org.junit.Before; import org.junit.Test; import org.mockito.internal.util.collections.Sets; import org.mockito.invocation.InvocationOnMock; import org.mockito.stubbing.Answer; import org.opendaylight.controller.netconf.api.NetconfClientSessionPreferences; import org.opendaylight.controller.netconf.api.NetconfMessage; import org.opendaylight.controller.netconf.nettyutil.handler.ChunkedFramingMechanismEncoder; import org.opendaylight.controller.netconf.nettyutil.handler.NetconfXMLToHelloMessageDecoder; import org.opendaylight.controller.netconf.nettyutil.handler.NetconfXMLToMessageDecoder; import org.opendaylight.controller.netconf.nettyutil.handler.exi.NetconfStartExiMessage; import org.opendaylight.controller.netconf.util.messages.NetconfHelloMessage; import org.opendaylight.controller.netconf.util.messages.NetconfHelloMessageAdditionalHeader; import org.opendaylight.controller.netconf.util.test.XmlFileLoader; import org.openexi.proc.common.EXIOptions; import org.w3c.dom.Document; public class NetconfClientSessionNegotiatorTest { private NetconfHelloMessage helloMessage; private ChannelPipeline pipeline; private ChannelFuture future; private Channel channel; private ChannelInboundHandlerAdapter channelInboundHandlerAdapter; @Before public void setUp() throws Exception { helloMessage = NetconfHelloMessage.createClientHello(Sets.newSet("exi:1.0"), Optional.<NetconfHelloMessageAdditionalHeader>absent()); pipeline = mockChannelPipeline(); future = mockChannelFuture(); channel = mockChannel(); } private ChannelHandler mockChannelHandler() { ChannelHandler handler = mock(ChannelHandler.class); return handler; } private Channel mockChannel() { Channel channel = mock(Channel.class); ChannelHandler channelHandler = mockChannelHandler(); doReturn("").when(channel).toString(); doReturn(future).when(channel).close(); doReturn(future).when(channel).writeAndFlush(anyObject()); doReturn(true).when(channel).isOpen(); doReturn(pipeline).when(channel).pipeline(); doReturn("").when(pipeline).toString(); doReturn(pipeline).when(pipeline).remove(any(ChannelHandler.class)); doReturn(channelHandler).when(pipeline).remove(anyString()); return channel; } private ChannelFuture mockChannelFuture() { ChannelFuture future = mock(ChannelFuture.class); doReturn(future).when(future).addListener(any(GenericFutureListener.class)); return future; } private ChannelPipeline mockChannelPipeline() { ChannelPipeline pipeline = mock(ChannelPipeline.class); ChannelHandler handler = mock(ChannelHandler.class); doReturn(pipeline).when(pipeline).addAfter(anyString(), anyString(), any(ChannelHandler.class)); doReturn(null).when(pipeline).get(SslHandler.class); doReturn(pipeline).when(pipeline).addLast(anyString(), any(ChannelHandler.class)); doReturn(handler).when(pipeline).replace(anyString(), anyString(), any(ChunkedFramingMechanismEncoder.class)); NetconfXMLToHelloMessageDecoder messageDecoder = new NetconfXMLToHelloMessageDecoder(); doReturn(messageDecoder).when(pipeline).replace(anyString(), anyString(), any(NetconfXMLToMessageDecoder.class)); doReturn(pipeline).when(pipeline).replace(any(ChannelHandler.class), anyString(), any(NetconfClientSession.class)); return pipeline; } private NetconfClientSessionNegotiator createNetconfClientSessionNegotiator( final Promise<NetconfClientSession> promise, final NetconfMessage startExi) { ChannelProgressivePromise progressivePromise = mock(ChannelProgressivePromise.class); NetconfClientSessionPreferences preferences = new NetconfClientSessionPreferences(helloMessage, startExi); doReturn(progressivePromise).when(promise).setFailure(any(Throwable.class)); long timeout = 10L; NetconfClientSessionListener sessionListener = mock(NetconfClientSessionListener.class); Timer timer = new HashedWheelTimer(); return new NetconfClientSessionNegotiator(preferences, promise, channel, timer, sessionListener, timeout); } @Test public void testNetconfClientSessionNegotiator() throws Exception { Promise promise = mock(Promise.class); doReturn(promise).when(promise).setSuccess(anyObject()); NetconfClientSessionNegotiator negotiator = createNetconfClientSessionNegotiator(promise, null); negotiator.channelActive(null); Set<String> caps = Sets.newSet("a", "b"); NetconfHelloMessage helloServerMessage = NetconfHelloMessage.createServerHello(caps, 10); negotiator.handleMessage(helloServerMessage); verify(promise).setSuccess(anyObject()); } @Test public void testNetconfClientSessionNegotiatorWithEXI() throws Exception { Promise promise = mock(Promise.class); EXIOptions exiOptions = new EXIOptions(); NetconfStartExiMessage exiMessage = NetconfStartExiMessage.create(exiOptions, "msg-id"); doReturn(promise).when(promise).setSuccess(anyObject()); NetconfClientSessionNegotiator negotiator = createNetconfClientSessionNegotiator(promise, exiMessage); negotiator.channelActive(null); Set<String> caps = Sets.newSet("exi:1.0"); NetconfHelloMessage helloMessage = NetconfHelloMessage.createServerHello(caps, 10); doAnswer(new Answer<Object>() { @Override public Object answer(final InvocationOnMock invocationOnMock) throws Throwable { channelInboundHandlerAdapter = ((ChannelInboundHandlerAdapter) invocationOnMock.getArguments()[2]); return null; } }).when(pipeline).addAfter(anyString(), anyString(), any(ChannelHandler.class)); ChannelHandlerContext handlerContext = mock(ChannelHandlerContext.class); doReturn(pipeline).when(handlerContext).pipeline(); negotiator.handleMessage(helloMessage); Document expectedResult = XmlFileLoader.xmlFileToDocument("netconfMessages/rpc-reply_ok.xml"); channelInboundHandlerAdapter.channelRead(handlerContext, new NetconfMessage(expectedResult)); verify(promise).setSuccess(anyObject()); // two calls for exiMessage, 2 for hello message verify(pipeline, times(4)).replace(anyString(), anyString(), any(ChannelHandler.class)); } }