Specialize protocol-framework to netconf

[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 da238bf90ddbf3d10cf875ccaa29e56ff0154226..e6a518e4cbcf29977ab69aab052ae9b2477924a0 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
@@ -18,13 +18,13 @@ import io.netty.channel.Channel;
 import io.netty.channel.ChannelHandler;
 import io.netty.channel.ChannelPipeline;
 import io.netty.util.concurrent.Promise;
-import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
 import org.mockito.Mock;
-import org.mockito.MockitoAnnotations;
-import org.opendaylight.protocol.framework.SessionListenerFactory;
-import org.opendaylight.protocol.framework.SessionNegotiator;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.opendaylight.netconf.api.NetconfSessionListenerFactory;
 
+@RunWith(MockitoJUnitRunner.class)
 public class TlsClientChannelInitializerTest {
     @Mock
     private SslHandlerFactory sslHandlerFactory;
@@ -33,17 +33,12 @@ public class TlsClientChannelInitializerTest {
     @Mock
     private NetconfClientSessionListener sessionListener;
 
-    @Before
-    public void setUp() {
-        MockitoAnnotations.initMocks(this);
-    }
-
     @SuppressWarnings("unchecked")
     @Test
     public void testInitialize() throws Exception {
-        SessionNegotiator<?> sessionNegotiator = mock(SessionNegotiator.class);
-        doReturn(sessionNegotiator).when(negotiatorFactory).getSessionNegotiator(any(SessionListenerFactory.class),
-                any(Channel.class), any(Promise.class));
+        NetconfClientSessionNegotiator sessionNegotiator = mock(NetconfClientSessionNegotiator.class);
+        doReturn(sessionNegotiator).when(negotiatorFactory).getSessionNegotiator(
+            any(NetconfSessionListenerFactory.class), any(Channel.class), any(Promise.class));
         ChannelPipeline pipeline = mock(ChannelPipeline.class);
         doReturn(pipeline).when(pipeline).addAfter(anyString(), anyString(), any(ChannelHandler.class));
         Channel channel = mock(Channel.class);