Add AbstractClientChannelInitializer

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

diff --git a/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TlsClientChannelInitializerTest.java b/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TlsClientChannelInitializerTest.java
index e6a518e4cbcf29977ab69aab052ae9b2477924a0..b9f853aa5d1cfd54409cc21273f39168dd6ecfa1 100644 (file)
--- a/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TlsClientChannelInitializerTest.java
+++ b/netconf/netconf-client/src/test/java/org/opendaylight/netconf/client/TlsClientChannelInitializerTest.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;
@@ -47,6 +48,11 @@ public class TlsClientChannelInitializerTest {
         doReturn(pipeline).when(pipeline).addFirst(anyString(), any(ChannelHandler.class));
         doReturn(pipeline).when(pipeline).addLast(anyString(), any(ChannelHandler.class));
 
+        ChannelConfig channelConfig = mock(ChannelConfig.class);
+        doReturn(channelConfig).when(channel).config();
+        doReturn(1L).when(negotiatorFactory).getConnectionTimeoutMillis();
+        doReturn(channelConfig).when(channelConfig).setConnectTimeoutMillis(1);
+
         Promise<NetconfClientSession> promise = mock(Promise.class);
 
         TlsClientChannelInitializer initializer = new TlsClientChannelInitializer(sslHandlerFactory,