Use ByteBuf.readRetainedSlice()
[openflowplugin.git] / openflowjava / openflow-protocol-impl / src / test / java / org / opendaylight / openflowjava / protocol / impl / core / connection / ConnectionAdapterImplTest.java
index 53f42a6e02108d5886fd250f0f720782b2384882..27c18dbaa4fbe9b22e77a752633cecd2a65aa4b4 100644 (file)
@@ -24,8 +24,11 @@ import java.util.concurrent.TimeUnit;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
 import org.mockito.Mock;
-import org.mockito.MockitoAnnotations;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter.MessageListener;
+import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter.SystemListener;
 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionReadyListener;
 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInput;
@@ -45,7 +48,6 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessage;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessageBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OpenflowProtocolListener;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketInMessage;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketInMessageBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortStatusMessage;
@@ -54,8 +56,8 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.D
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.DisconnectEventBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.SwitchIdleEvent;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.SwitchIdleEventBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.SystemNotificationsListener;
 import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.common.Uint32;
 
 /**
  * Unit test for ConnectionAdapterImpl.
@@ -63,6 +65,7 @@ import org.opendaylight.yangtools.yang.binding.DataObject;
  * @author michal.polkorab
  * @author madamjak
  */
+@RunWith(MockitoJUnitRunner.class)
 public class ConnectionAdapterImplTest {
 
     private static final int RPC_RESPONSE_EXPIRATION = 1;
@@ -70,13 +73,20 @@ public class ConnectionAdapterImplTest {
     private static final RemovalListener<RpcResponseKey, ResponseExpectedRpcListener<?>> REMOVAL_LISTENER =
         notification -> notification.getValue().discard();
 
-    @Mock SocketChannel channel;
-    @Mock ChannelPipeline pipeline;
-    @Mock OpenflowProtocolListener messageListener;
-    @Mock SystemNotificationsListener systemListener;
-    @Mock ConnectionReadyListener readyListener;
-    @Mock Cache<RpcResponseKey, ResponseExpectedRpcListener<?>> mockCache;
-    @Mock ChannelFuture channelFuture;
+    @Mock
+    private SocketChannel channel;
+    @Mock
+    ChannelPipeline pipeline;
+    @Mock
+    private MessageListener messageListener;
+    @Mock
+    private SystemListener systemListener;
+    @Mock
+    private ConnectionReadyListener readyListener;
+    @Mock
+    private Cache<RpcResponseKey, ResponseExpectedRpcListener<?>> mockCache;
+    @Mock
+    private ChannelFuture channelFuture;
 
     private ConnectionAdapterImpl adapter;
     private Cache<RpcResponseKey, ResponseExpectedRpcListener<?>> cache;
@@ -86,7 +96,6 @@ public class ConnectionAdapterImplTest {
      */
     @Before
     public void setUp() {
-        MockitoAnnotations.initMocks(this);
         when(channel.pipeline()).thenReturn(pipeline);
         adapter = new ConnectionAdapterImpl(channel, InetSocketAddress.createUnresolved("10.0.0.1", 6653), true,
                 CHANNEL_OUTBOUND_QUEUE_SIZE);
@@ -106,37 +115,37 @@ public class ConnectionAdapterImplTest {
     public void testConsume() {
         DataObject message = new EchoRequestMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onEchoRequestMessage((EchoRequestMessage) message);
+        verify(messageListener, times(1)).onEchoRequest((EchoRequestMessage) message);
         message = new ErrorMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onErrorMessage((ErrorMessage) message);
+        verify(messageListener, times(1)).onError((ErrorMessage) message);
         message = new ExperimenterMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onExperimenterMessage((ExperimenterMessage) message);
+        verify(messageListener, times(1)).onExperimenter((ExperimenterMessage) message);
         message = new FlowRemovedMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onFlowRemovedMessage((FlowRemovedMessage) message);
+        verify(messageListener, times(1)).onFlowRemoved((FlowRemovedMessage) message);
         message = new HelloMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onHelloMessage((HelloMessage) message);
+        verify(messageListener, times(1)).onHello((HelloMessage) message);
         message = new MultipartReplyMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onMultipartReplyMessage((MultipartReplyMessage) message);
+        verify(messageListener, times(1)).onMultipartReply((MultipartReplyMessage) message);
         message = new PacketInMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onPacketInMessage((PacketInMessage) message);
+        verify(messageListener, times(1)).onPacketIn((PacketInMessage) message);
         message = new PortStatusMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onPortStatusMessage((PortStatusMessage) message);
+        verify(messageListener, times(1)).onPortStatus((PortStatusMessage) message);
         message = new SwitchIdleEventBuilder().build();
         adapter.consume(message);
-        verify(systemListener, times(1)).onSwitchIdleEvent((SwitchIdleEvent) message);
+        verify(systemListener, times(1)).onSwitchIdle((SwitchIdleEvent) message);
         message = new DisconnectEventBuilder().build();
         adapter.consume(message);
-        verify(systemListener, times(1)).onDisconnectEvent((DisconnectEvent) message);
+        verify(systemListener, times(1)).onDisconnect((DisconnectEvent) message);
         message = new EchoRequestMessageBuilder().build();
         adapter.consume(message);
-        verify(messageListener, times(1)).onEchoRequestMessage((EchoRequestMessage) message);
+        verify(messageListener, times(1)).onEchoRequest((EchoRequestMessage) message);
     }
 
     /**
@@ -145,10 +154,7 @@ public class ConnectionAdapterImplTest {
     @Test
     public void testConsume2() {
         adapter.setResponseCache(mockCache);
-        final BarrierOutputBuilder barrierBuilder = new BarrierOutputBuilder();
-        barrierBuilder.setXid(42L);
-        final BarrierOutput barrier = barrierBuilder.build();
-        adapter.consume(barrier);
+        adapter.consume(new BarrierOutputBuilder().setXid(Uint32.valueOf(42)).build());
         verify(mockCache, times(1)).getIfPresent(any(RpcResponseKey.class));
     }
 
@@ -158,8 +164,8 @@ public class ConnectionAdapterImplTest {
     @Test
     public void testConsume3() {
         final BarrierInputBuilder inputBuilder = new BarrierInputBuilder();
-        inputBuilder.setVersion((short) EncodeConstants.OF13_VERSION_ID);
-        inputBuilder.setXid(42L);
+        inputBuilder.setVersion(EncodeConstants.OF_VERSION_1_3);
+        inputBuilder.setXid(Uint32.valueOf(42));
         final BarrierInput barrierInput = inputBuilder.build();
         final RpcResponseKey key = new RpcResponseKey(42L,
                 "org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutput");
@@ -167,7 +173,7 @@ public class ConnectionAdapterImplTest {
                 "failure", mockCache, key);
         cache.put(key, listener);
         final BarrierOutputBuilder barrierBuilder = new BarrierOutputBuilder();
-        barrierBuilder.setXid(42L);
+        barrierBuilder.setXid(Uint32.valueOf(42));
         final BarrierOutput barrierOutput = barrierBuilder.build();
         adapter.consume(barrierOutput);
         final ResponseExpectedRpcListener<?> ifPresent = cache.getIfPresent(key);