Merge "BUG-4118: Li:backward compatibility - rpcs - stats services"
[openflowplugin.git] / openflowplugin-impl / src / test / java / org / opendaylight / openflowplugin / impl / device / listener / OpenflowProtocolListenerFullImplTest.java
index fd8311f8b6e152f6837c904d11950345035ffe16..b60c618e5f5cc20eee3124c46b2d77297b6e5b15 100644 (file)
@@ -8,6 +8,7 @@
 
 package org.opendaylight.openflowplugin.impl.device.listener;
 
+import java.net.InetSocketAddress;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
@@ -63,6 +64,7 @@ public class OpenflowProtocolListenerFullImplTest {
         // place for mocking method's general behavior for HandshakeContext and ConnectionContext
         ofProtocolListener = new OpenflowProtocolListenerFullImpl(connectionAdapter, deviceReplyProcessor);
         connectionAdapter.setMessageListener(ofProtocolListener);
+        Mockito.when(connectionAdapter.getRemoteAddress()).thenReturn(InetSocketAddress.createUnresolved("ofp-junit.example.org", 6663));
         Mockito.verify(connectionAdapter).setMessageListener(Matchers.any(OpenflowProtocolListener.class));
     }
 
@@ -127,7 +129,9 @@ public class OpenflowProtocolListenerFullImplTest {
         HelloMessage helloMessage = new HelloMessageBuilder()
                 .setVersion(OFConstants.OFP_VERSION_1_3).setXid(xid).build();
         ofProtocolListener.onHelloMessage(helloMessage);
-        // NOOP
+
+        Mockito.verify(connectionAdapter).getRemoteAddress();
+        Mockito.verify(connectionAdapter).disconnect();
     }
 
     /**