Add AbstractClientChannelInitializer

[netconf.git] / netconf / netconf-client / src / test / java / org / opendaylight / netconf / client / TcpClientChannelInitializerTest.java

diff --git a/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TcpClientChannelInitializerTest.java b/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TcpClientChannelInitializerTest.java
index 000cfbef3d039a643125ca032f93178c027576b2..b9b3b961ad113a262d9a2f62d3e9d95f1cc42937 100644 (file)
--- a/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TcpClientChannelInitializerTest.java
+++ b/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TcpClientChannelInitializerTest.java
@@ -15,6 +15,7 @@ import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 
 import io.netty.channel.Channel;
+import io.netty.channel.ChannelConfig;
 import io.netty.channel.ChannelHandler;
 import io.netty.channel.ChannelPipeline;
 import io.netty.util.concurrent.Promise;
@@ -37,6 +38,11 @@ public class TcpClientChannelInitializerTest {
         doReturn(pipeline).when(channel).pipeline();
         doReturn("").when(channel).toString();
 
+        ChannelConfig channelConfig = mock(ChannelConfig.class);
+        doReturn(channelConfig).when(channel).config();
+        doReturn(1L).when(factory).getConnectionTimeoutMillis();
+        doReturn(channelConfig).when(channelConfig).setConnectTimeoutMillis(1);
+
         Promise<NetconfClientSession> promise = mock(Promise.class);
         doReturn("").when(promise).toString();