Examples with ReconnectFuture org.opendaylight.netconf.nettyutil.ReconnectFuture used on opensource projects

Example 1 with ReconnectFuture

use of org.opendaylight.netconf.nettyutil.ReconnectFuture in project netconf by opendaylight.

the class NetconfClientDispatcherImplTest method testNetconfClientDispatcherImpl.

@Test
public void testNetconfClientDispatcherImpl() throws Exception {
    EventLoopGroup bossGroup = Mockito.mock(EventLoopGroup.class);
    EventLoopGroup workerGroup = Mockito.mock(EventLoopGroup.class);
    Timer timer = new HashedWheelTimer();
    ChannelFuture chf = Mockito.mock(ChannelFuture.class);
    Channel ch = Mockito.mock(Channel.class);
    doReturn(ch).when(chf).channel();
    Throwable thr = Mockito.mock(Throwable.class);
    doReturn(chf).when(workerGroup).register(any(Channel.class));
    ChannelPromise promise = Mockito.mock(ChannelPromise.class);
    doReturn(promise).when(chf).addListener(any(GenericFutureListener.class));
    doReturn(thr).when(chf).cause();
    doReturn(true).when(chf).isDone();
    doReturn(false).when(chf).isSuccess();
    Long timeout = 200L;
    NetconfHelloMessageAdditionalHeader header = new NetconfHelloMessageAdditionalHeader("a", "host", "port", "trans", "id");
    NetconfClientSessionListener listener = new SimpleNetconfClientSessionListener();
    InetSocketAddress address = InetSocketAddress.createUnresolved("host", 830);
    ReconnectStrategyFactory reconnectStrategyFactory = Mockito.mock(ReconnectStrategyFactory.class);
    AuthenticationHandler handler = Mockito.mock(AuthenticationHandler.class);
    ReconnectStrategy reconnect = Mockito.mock(ReconnectStrategy.class);
    doReturn(5).when(reconnect).getConnectTimeout();
    doReturn("").when(reconnect).toString();
    doReturn("").when(handler).toString();
    doReturn("").when(reconnectStrategyFactory).toString();
    doReturn(reconnect).when(reconnectStrategyFactory).createReconnectStrategy();
    NetconfReconnectingClientConfiguration cfg = NetconfReconnectingClientConfigurationBuilder.create().withProtocol(NetconfClientConfiguration.NetconfClientProtocol.SSH).withAddress(address).withConnectionTimeoutMillis(timeout).withReconnectStrategy(reconnect).withAdditionalHeader(header).withSessionListener(listener).withConnectStrategyFactory(reconnectStrategyFactory).withAuthHandler(handler).build();
    NetconfReconnectingClientConfiguration cfg2 = NetconfReconnectingClientConfigurationBuilder.create().withProtocol(NetconfClientConfiguration.NetconfClientProtocol.TCP).withAddress(address).withConnectionTimeoutMillis(timeout).withReconnectStrategy(reconnect).withAdditionalHeader(header).withSessionListener(listener).withConnectStrategyFactory(reconnectStrategyFactory).withAuthHandler(handler).build();
    NetconfClientDispatcherImpl dispatcher = new NetconfClientDispatcherImpl(bossGroup, workerGroup, timer);
    Future<NetconfClientSession> sshSession = dispatcher.createClient(cfg);
    Future<NetconfClientSession> tcpSession = dispatcher.createClient(cfg2);
    ReconnectFuture sshReconn = dispatcher.createReconnectingClient(cfg);
    final ReconnectFuture tcpReconn = dispatcher.createReconnectingClient(cfg2);
    assertNotNull(sshSession);
    assertNotNull(tcpSession);
    assertNotNull(sshReconn);
    assertNotNull(tcpReconn);
    SslHandlerFactory sslHandlerFactory = Mockito.mock(SslHandlerFactory.class);
    NetconfReconnectingClientConfiguration cfg3 = NetconfReconnectingClientConfigurationBuilder.create().withProtocol(NetconfClientConfiguration.NetconfClientProtocol.TLS).withAddress(address).withConnectionTimeoutMillis(timeout).withReconnectStrategy(reconnect).withAdditionalHeader(header).withSessionListener(listener).withConnectStrategyFactory(reconnectStrategyFactory).withSslHandlerFactory(sslHandlerFactory).build();
    Future<NetconfClientSession> tlsSession = dispatcher.createClient(cfg3);
    ReconnectFuture tlsReconn = dispatcher.createReconnectingClient(cfg3);
    assertNotNull(tlsSession);
    assertNotNull(tlsReconn);
}