openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/device/DeviceManagerImplTest.java

   1 /*
   2  *
   3  *  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   4  *  *
   5  *  * This program and the accompanying materials are made available under the
   6  *  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   7  *  * and is available at http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  *
  10  */
  11
  12 package org.opendaylight.openflowplugin.impl.device;
  13
  14 import static org.mockito.Matchers.any;
  15 import static org.mockito.Mockito.doThrow;
  16 import static org.mockito.Mockito.eq;
  17 import static org.mockito.Mockito.inOrder;
  18 import static org.mockito.Mockito.mock;
  19 import static org.mockito.Mockito.times;
  20 import static org.mockito.Mockito.verify;
  21 import static org.mockito.Mockito.when;
  22
  23 import com.google.common.collect.Lists;
  24 import com.google.common.util.concurrent.CheckedFuture;
  25 import com.google.common.util.concurrent.FutureCallback;
  26 import com.google.common.util.concurrent.Futures;
  27 import com.google.common.util.concurrent.ListenableFuture;
  28 import java.lang.reflect.Field;
  29 import java.math.BigInteger;
  30 import java.util.ArrayList;
  31 import java.util.Collection;
  32 import java.util.Collections;
  33 import java.util.List;
  34 import java.util.Set;
  35 import org.junit.Assert;
  36 import org.junit.Before;
  37 import org.junit.Test;
  38 import org.junit.runner.RunWith;
  39 import org.mockito.InOrder;
  40 import org.mockito.Matchers;
  41 import org.mockito.Mock;
  42 import org.mockito.Mockito;
  43 import org.mockito.invocation.InvocationOnMock;
  44 import org.mockito.runners.MockitoJUnitRunner;
  45 import org.mockito.stubbing.Answer;
  46 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
  47 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  48 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  49 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  50 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
  51 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  52 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
  53 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueue;
  54 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueueHandler;
  55 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueueHandlerRegistration;
  56 import org.opendaylight.openflowplugin.api.OFConstants;
  57 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
  58 import org.opendaylight.openflowplugin.api.openflow.connection.OutboundQueueProvider;
  59 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  60 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
  61 import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
  62 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
  63 import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceInitializationPhaseHandler;
  64 import org.opendaylight.openflowplugin.api.openflow.md.core.TranslatorKey;
  65 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageIntelligenceAgency;
  66 import org.opendaylight.openflowplugin.openflow.md.util.OpenflowPortsUtil;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.statistics.rev131111.NodeGroupFeatures;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterFeatures;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ActionType;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.Capabilities;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.CapabilitiesV10;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupCapabilities;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupTypes;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterBandTypeBitmap;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterFlags;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessage;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessageBuilder;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInput;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.features.reply.PhyPortBuilder;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyDescCaseBuilder;
  94 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupFeaturesCaseBuilder;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterFeaturesCaseBuilder;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyPortDescCaseBuilder;
  97 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyTableFeaturesCaseBuilder;
  98 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.desc._case.MultipartReplyDesc;
  99 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.desc._case.MultipartReplyDescBuilder;
 100 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.features._case.MultipartReplyGroupFeatures;
 101 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.features._case.MultipartReplyGroupFeaturesBuilder;
 102 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.features._case.MultipartReplyMeterFeaturesBuilder;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.desc._case.MultipartReplyPortDescBuilder;
 104 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.desc._case.multipart.reply.port.desc.PortsBuilder;
 105 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.MultipartReplyTableFeatures;
 106 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.MultipartReplyTableFeaturesBuilder;
 107 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.multipart.reply.table.features.TableFeatures;
 108 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.multipart.reply.table.features.TableFeaturesBuilder;
 109 import org.opendaylight.yang.gen.v1.urn.opendaylight.role.service.rev150727.OfpRole;
 110 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 111 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
 112 import org.opendaylight.yangtools.yang.common.RpcError;
 113 import org.opendaylight.yangtools.yang.common.RpcResult;
 114 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
 115
 116 @RunWith(MockitoJUnitRunner.class)
 117 public class DeviceManagerImplTest {
 118
 119     private static final boolean TEST_VALUE_SWITCH_FEATURE_MANDATORY = true;
 120     private static final long TEST_VALUE_GLOBAL_NOTIFICATION_QUOTA = 2000l;
 121     private static final KeyedInstanceIdentifier<Node, NodeKey> DUMMY_NODE_II = InstanceIdentifier.create(Nodes.class)
 122             .child(Node.class, new NodeKey(new NodeId("dummyNodeId")));
 123     private static final Short DUMMY_TABLE_ID = 1;
 124     private static final Long DUMMY_MAX_METER = 544L;
 125     private static final String DUMMY_DATAPATH_ID = "44";
 126     private static final Long DUMMY_PORT_NUMBER = 21L;
 127
 128     @Mock
 129     CheckedFuture<Void, TransactionCommitFailedException> mockedFuture;
 130     @Mock
 131     private FeaturesReply mockFeatures;
 132     @Mock
 133     private OutboundQueue outboundQueueProvider;
 134     @Mock
 135     private DeviceInitializationPhaseHandler deviceInitPhaseHandler;
 136     @Mock
 137     private TranslatorLibrary translatorLibrary;
 138     @Mock
 139     private ConnectionContext mockConnectionContext;
 140     @Mock
 141     private ConnectionAdapter mockedConnectionAdapter;
 142     @Mock
 143     private DeviceContextImpl mockedDeviceContext;
 144
 145     @Before
 146     public void setUp() throws Exception {
 147         OpenflowPortsUtil.init();
 148
 149         when(mockConnectionContext.getNodeId()).thenReturn(new NodeId("dummyNodeId"));
 150         when(mockConnectionContext.getFeatures()).thenReturn(mockFeatures);
 151         when(mockConnectionContext.getConnectionAdapter()).thenReturn(mockedConnectionAdapter);
 152         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockConnectionContext);
 153
 154         final Capabilities capabilitiesV13 = Mockito.mock(Capabilities.class);
 155         final CapabilitiesV10 capabilitiesV10 = Mockito.mock(CapabilitiesV10.class);
 156         when(mockFeatures.getCapabilities()).thenReturn(capabilitiesV13);
 157         when(mockFeatures.getCapabilitiesV10()).thenReturn(capabilitiesV10);
 158         when(mockFeatures.getDatapathId()).thenReturn(BigInteger.valueOf(21L));
 159     }
 160
 161     @Test
 162     public void onDeviceContextLevelUpFailTest() {
 163         onDeviceContextLevelUp(true);
 164     }
 165
 166     @Test
 167     public void onDeviceContextLevelUpSuccessTest() {
 168         onDeviceContextLevelUp(false);
 169     }
 170
 171     private DeviceManagerImpl prepareDeviceManager() {
 172         return prepareDeviceManager(false);
 173     }
 174
 175     private DeviceManagerImpl prepareDeviceManager(boolean withException) {
 176         DataBroker mockedDataBroker = mock(DataBroker.class);
 177         WriteTransaction mockedWriteTransaction = mock(WriteTransaction.class);
 178
 179         BindingTransactionChain mockedTxChain = mock(BindingTransactionChain.class);
 180         WriteTransaction mockedWTx = mock(WriteTransaction.class);
 181         when(mockedTxChain.newWriteOnlyTransaction()).thenReturn(mockedWTx);
 182         when(mockedDataBroker.createTransactionChain(any(TransactionChainListener.class))).thenReturn
 183                 (mockedTxChain);
 184         when(mockedDataBroker.newWriteOnlyTransaction()).thenReturn(mockedWriteTransaction);
 185
 186         when(mockedWriteTransaction.submit()).thenReturn(mockedFuture);
 187
 188         MessageIntelligenceAgency mockedMessageIntelligenceAgency = mock(MessageIntelligenceAgency.class);
 189         DeviceManagerImpl deviceManager = new DeviceManagerImpl(mockedDataBroker, mockedMessageIntelligenceAgency, TEST_VALUE_SWITCH_FEATURE_MANDATORY,
 190                 TEST_VALUE_GLOBAL_NOTIFICATION_QUOTA);
 191         deviceManager.setDeviceInitializationPhaseHandler(deviceInitPhaseHandler);
 192
 193         return deviceManager;
 194     }
 195
 196     public void onDeviceContextLevelUp(boolean withException) {
 197         DeviceManagerImpl deviceManager = prepareDeviceManager(withException);
 198         DeviceState mockedDeviceState = mock(DeviceState.class);
 199         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 200         when(mockedDeviceState.getRole()).thenReturn(OfpRole.BECOMEMASTER);
 201
 202         if (withException) {
 203             doThrow(new IllegalStateException("dummy")).when(mockedDeviceContext).initialSubmitTransaction();
 204         }
 205
 206         deviceManager.onDeviceContextLevelUp(mockedDeviceContext);
 207         if (withException) {
 208             verify(mockedDeviceContext).close();
 209         } else {
 210             verify(mockedDeviceContext).initialSubmitTransaction();
 211             verify(mockedDeviceContext).onPublished();
 212         }
 213     }
 214
 215     @Test
 216     public void deviceConnectedTest() {
 217         DeviceManagerImpl deviceManager = prepareDeviceManager();
 218         injectMockTranslatorLibrary(deviceManager);
 219         ConnectionContext mockConnectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_3);
 220
 221         deviceManager.deviceConnected(mockConnectionContext);
 222
 223         InOrder order = inOrder(mockConnectionContext);
 224         order.verify(mockConnectionContext).getFeatures();
 225         order.verify(mockConnectionContext).setOutboundQueueProvider(any(OutboundQueueProvider.class));
 226         order.verify(mockConnectionContext).setOutboundQueueHandleRegistration(
 227                 Mockito.<OutboundQueueHandlerRegistration<OutboundQueueProvider>>any());
 228         order.verify(mockConnectionContext).getNodeId();
 229         order.verify(mockConnectionContext).setDeviceDisconnectedHandler(any(DeviceContext.class));
 230
 231         Mockito.verify(deviceInitPhaseHandler).onDeviceContextLevelUp(Matchers.<DeviceContext>any());
 232     }
 233
 234     @Test
 235     public void deviceConnectedV10Test() {
 236         DeviceManagerImpl deviceManager = prepareDeviceManager();
 237         injectMockTranslatorLibrary(deviceManager);
 238         ConnectionContext mockConnectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_0);
 239
 240         PhyPortBuilder phyPort = new PhyPortBuilder()
 241                 .setPortNo(41L);
 242         when(mockFeatures.getPhyPort()).thenReturn(Collections.singletonList(phyPort.build()));
 243         MessageTranslator<Object, Object> mockedTranslator = Mockito.mock(MessageTranslator.class);
 244         when(mockedTranslator.translate(Matchers.<Object>any(), Matchers.<DeviceContext>any(), Matchers.any()))
 245                 .thenReturn(null);
 246         when(translatorLibrary.lookupTranslator(Matchers.<TranslatorKey>any())).thenReturn(mockedTranslator);
 247
 248         deviceManager.deviceConnected(mockConnectionContext);
 249
 250         InOrder order = inOrder(mockConnectionContext);
 251         order.verify(mockConnectionContext).getFeatures();
 252         order.verify(mockConnectionContext).setOutboundQueueProvider(any(OutboundQueueProvider.class));
 253         order.verify(mockConnectionContext).setOutboundQueueHandleRegistration(
 254                 Mockito.<OutboundQueueHandlerRegistration<OutboundQueueProvider>>any());
 255         order.verify(mockConnectionContext).getNodeId();
 256         order.verify(mockConnectionContext).setDeviceDisconnectedHandler(any(DeviceContext.class));
 257
 258         Mockito.verify(deviceInitPhaseHandler).onDeviceContextLevelUp(Matchers.<DeviceContext>any());
 259     }
 260
 261     protected ConnectionContext buildMockConnectionContext(short ofpVersion) {
 262         when(mockFeatures.getVersion()).thenReturn(ofpVersion);
 263         when(outboundQueueProvider.reserveEntry()).thenReturn(43L);
 264         Mockito.doAnswer(new Answer<Void>() {
 265             @Override
 266             public Void answer(InvocationOnMock invocation) throws Throwable {
 267                 final FutureCallback<OfHeader> callBack = (FutureCallback<OfHeader>) invocation.getArguments()[2];
 268                 callBack.onSuccess(null);
 269                 return null;
 270             }
 271         })
 272                 .when(outboundQueueProvider)
 273                 .commitEntry(Matchers.anyLong(), Matchers.<MultipartRequestInput>any(), Matchers.<FutureCallback<OfHeader>>any());
 274
 275         when(mockedConnectionAdapter.registerOutboundQueueHandler(Matchers.<OutboundQueueHandler>any(), Matchers.anyInt(), Matchers.anyLong()))
 276                 .thenAnswer(new Answer<OutboundQueueHandlerRegistration<OutboundQueueHandler>>() {
 277                     @Override
 278                     public OutboundQueueHandlerRegistration<OutboundQueueHandler> answer(InvocationOnMock invocation) throws Throwable {
 279                         OutboundQueueHandler handler = (OutboundQueueHandler) invocation.getArguments()[0];
 280                         handler.onConnectionQueueChanged(outboundQueueProvider);
 281                         return null;
 282                     }
 283                 });
 284
 285         when(mockConnectionContext.getOutboundQueueProvider()).thenReturn(outboundQueueProvider);
 286         return mockConnectionContext;
 287     }
 288
 289     private void injectMockTranslatorLibrary(DeviceManagerImpl deviceManager) {
 290         deviceManager.setTranslatorLibrary(translatorLibrary);
 291     }
 292
 293     @Test
 294     public void chainTableTrunkWriteOF10Test() {
 295         DeviceState mockedDeviceState = mock(DeviceState.class);
 296
 297         GetFeaturesOutput mockedFeatures = mock(GetFeaturesOutput.class);
 298         when(mockedFeatures.getTables()).thenReturn((short) 2);
 299         when(mockedDeviceState.getFeatures()).thenReturn(mockedFeatures);
 300
 301         when(mockedDeviceState.getNodeInstanceIdentifier()).thenReturn(DUMMY_NODE_II);
 302         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 303
 304         RpcResult<List<MultipartReply>> mockedRpcResult = mock(RpcResult.class);
 305         when(mockedRpcResult.isSuccessful()).thenReturn(true);
 306         List<RpcResult<List<MultipartReply>>> data = new ArrayList<RpcResult<List<MultipartReply>>>();
 307         data.add(mockedRpcResult);
 308         data.add(mockedRpcResult);
 309
 310         DeviceManagerImpl.chainTableTrunkWriteOF10(mockedDeviceContext, Futures.immediateFuture(data));
 311         verify(mockedDeviceContext, times(3))
 312                 .writeToTransaction(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(FlowCapableNode.class));
 313     }
 314
 315     @Test
 316     public void testTranslateAndWriteReplyTypeDesc() {
 317         final ConnectionContext connectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_3);
 318         Mockito.when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(connectionContext);
 319         DeviceState deviceState = Mockito.mock(DeviceState.class);
 320         Mockito.when(mockedDeviceContext.getDeviceState()).thenReturn(deviceState);
 321
 322         Collection<MultipartReply> multipartReplyMessages = prepareDataforTypeDesc(mockedDeviceContext);
 323
 324         DeviceManagerImpl.translateAndWriteReply(MultipartType.OFPMPDESC, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 325         verify(mockedDeviceContext)
 326                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class)), any(FlowCapableNode.class));
 327     }
 328
 329     private Collection<MultipartReply> prepareDataforTypeDesc(final DeviceContext mockedDeviceContext) {
 330         MultipartReplyDesc multipartReplyDesc = new MultipartReplyDescBuilder().build();
 331
 332         MultipartReplyDescCaseBuilder multipartReplyDescCaseBuilder = new MultipartReplyDescCaseBuilder();
 333         multipartReplyDescCaseBuilder.setMultipartReplyDesc(multipartReplyDesc);
 334
 335         MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyDescCaseBuilder.build()).build();
 336         return Collections.<MultipartReply>singleton(multipartReplyMessage);
 337
 338     }
 339
 340     @Test
 341     public void translateAndWriteReplyTypeTableFeatures() {
 342         TableFeaturesBuilder tableFeature = new TableFeaturesBuilder();
 343         tableFeature.setTableId(DUMMY_TABLE_ID);
 344         List<TableFeatures> tableFeatures = new ArrayList<>();
 345         tableFeatures.add(tableFeature.build());
 346
 347         MultipartReplyTableFeatures multipartReplyTableFeatures = new MultipartReplyTableFeaturesBuilder().setTableFeatures(tableFeatures).build();
 348         MultipartReplyTableFeaturesCaseBuilder multipartReplyTableFeaturesCaseBuilder = new MultipartReplyTableFeaturesCaseBuilder();
 349         multipartReplyTableFeaturesCaseBuilder.setMultipartReplyTableFeatures(multipartReplyTableFeatures);
 350
 351         MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyTableFeaturesCaseBuilder.build()).build();
 352         Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 353         DeviceManagerImpl.translateAndWriteReply(MultipartType.OFPMPTABLEFEATURES, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 354         verify(mockedDeviceContext)
 355                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 356                         eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class).child(Table.class, new TableKey(DUMMY_TABLE_ID))), any(Table.class));
 357
 358     }
 359
 360     @Test
 361     public void translateAndWriteReplyTypeMeterFeatures() {
 362         DeviceState mockedDeviceState = mock(DeviceState.class);
 363         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 364
 365         MultipartReplyMeterFeaturesBuilder multipartReplyMeterFeaturesBuilder = new MultipartReplyMeterFeaturesBuilder();
 366         multipartReplyMeterFeaturesBuilder.setBandTypes(new MeterBandTypeBitmap(true, true));
 367         multipartReplyMeterFeaturesBuilder.setCapabilities(new MeterFlags(true, true, true, true));
 368         multipartReplyMeterFeaturesBuilder.setMaxMeter(DUMMY_MAX_METER);
 369
 370         MultipartReplyMeterFeaturesCaseBuilder multipartReplyMeterFeaturesCaseBuilder = new MultipartReplyMeterFeaturesCaseBuilder();
 371         multipartReplyMeterFeaturesCaseBuilder.setMultipartReplyMeterFeatures(multipartReplyMeterFeaturesBuilder.build());
 372
 373         MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyMeterFeaturesCaseBuilder.build()).build();
 374         Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 375         DeviceManagerImpl.translateAndWriteReply(MultipartType.OFPMPMETERFEATURES, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 376         verify(mockedDeviceContext)
 377                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(NodeMeterFeatures.class)), any(NodeMeterFeatures.class));
 378         verify(mockedDeviceState).setMeterAvailable(eq(true));
 379     }
 380
 381     @Test
 382     public void translateAndWriteReplyTypeGroupFeatures() {
 383         MultipartReplyGroupFeaturesBuilder multipartReplyGroupFeaturesBuilder = new MultipartReplyGroupFeaturesBuilder();
 384         multipartReplyGroupFeaturesBuilder.setTypes(new GroupTypes(true, true, true, true));
 385         multipartReplyGroupFeaturesBuilder.setCapabilities(new GroupCapabilities(true, true, true, true));
 386         ActionType actionType = new ActionType(true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true);
 387         multipartReplyGroupFeaturesBuilder.setActionsBitmap(Lists.newArrayList(actionType));
 388
 389         MultipartReplyGroupFeatures multipartReplyGroupFeatures = multipartReplyGroupFeaturesBuilder.build();
 390
 391         MultipartReplyGroupFeaturesCaseBuilder multipartReplyGroupFeaturesCaseBuilder = new MultipartReplyGroupFeaturesCaseBuilder();
 392         multipartReplyGroupFeaturesCaseBuilder.setMultipartReplyGroupFeatures(multipartReplyGroupFeatures);
 393
 394         MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyGroupFeaturesCaseBuilder.build()).build();
 395         Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 396
 397         DeviceManagerImpl.translateAndWriteReply(MultipartType.OFPMPGROUPFEATURES, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 398         verify(mockedDeviceContext)
 399                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(NodeGroupFeatures.class)), any(NodeGroupFeatures.class));
 400     }
 401
 402
 403     @Test
 404     public void translateAndWriteReplyTypePortDesc() {
 405         ConnectionContext mockedPrimaryConnectionContext = mock(ConnectionContext.class);
 406         FeaturesReply mockedFeatures = mock(FeaturesReply.class);
 407         when(mockedFeatures.getDatapathId()).thenReturn(new BigInteger(DUMMY_DATAPATH_ID));
 408         when(mockedPrimaryConnectionContext.getFeatures()).thenReturn(mockedFeatures);
 409         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockedPrimaryConnectionContext);
 410         DeviceState mockedDeviceState = mock(DeviceState.class);
 411         when(mockedDeviceState.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_0);
 412         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 413         MessageTranslator mockedTranslator = mock(MessageTranslator.class);
 414         when(translatorLibrary.lookupTranslator(any(TranslatorKey.class))).thenReturn(mockedTranslator);
 415         when(mockedDeviceContext.oook()).thenReturn(translatorLibrary);
 416
 417         MultipartReplyPortDescBuilder multipartReplyPortDescBuilder = new MultipartReplyPortDescBuilder();
 418
 419         PortsBuilder portsBuilder = new PortsBuilder();
 420         portsBuilder.setPortNo(DUMMY_PORT_NUMBER);
 421
 422         multipartReplyPortDescBuilder.setPorts(Lists.newArrayList(portsBuilder.build()));
 423
 424         MultipartReplyPortDescCaseBuilder multipartReplyPortDescCaseBuilder = new MultipartReplyPortDescCaseBuilder();
 425         multipartReplyPortDescCaseBuilder.setMultipartReplyPortDesc(multipartReplyPortDescBuilder.build());
 426
 427         MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyPortDescCaseBuilder.build()).build();
 428         Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 429
 430         OpenflowPortsUtil.init();
 431         DeviceManagerImpl.translateAndWriteReply(MultipartType.OFPMPPORTDESC, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 432         verify(mockedDeviceContext)
 433                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), any(InstanceIdentifier.class), any(NodeConnector.class));
 434     }
 435
 436     @Test
 437     public void createSuccessProcessingCallbackTest() {
 438         DeviceState mockedDeviceState = mock(DeviceState.class);
 439         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 440
 441         final ConnectionContext connectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_3);
 442
 443         List<MultipartReply> multipartReplies = new ArrayList<>(prepareDataforTypeDesc(mockedDeviceContext));
 444         RpcResult<List<MultipartReply>> result = RpcResultBuilder.<List<MultipartReply>>success(multipartReplies).build();
 445         ListenableFuture<RpcResult<List<MultipartReply>>> mockedRequestContextFuture = Futures.immediateFuture(result);
 446
 447         DeviceManagerImpl.createSuccessProcessingCallback(MultipartType.OFPMPDESC, mockedDeviceContext, DUMMY_NODE_II, mockedRequestContextFuture);
 448         verify(mockedDeviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class)), any(FlowCapableNode.class));
 449
 450         RpcResult<List<MultipartReply>> rpcResult = RpcResultBuilder.<List<MultipartReply>>failed().withError(RpcError.ErrorType.PROTOCOL, "dummy error").build();
 451         mockedRequestContextFuture = Futures.immediateFuture(rpcResult);
 452         DeviceManagerImpl.createSuccessProcessingCallback(MultipartType.OFPMPDESC, mockedDeviceContext, DUMMY_NODE_II, mockedRequestContextFuture);
 453         verify(mockedDeviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class)), any(FlowCapableNode.class));
 454     }
 455
 456     @Test
 457     public void testClose() throws Exception {
 458         DeviceContext deviceContext = Mockito.mock(DeviceContext.class);
 459         final DeviceManagerImpl deviceManager = prepareDeviceManager();
 460         final Set<DeviceContext> deviceContexts = getContextsCollection(deviceManager);
 461         deviceContexts.add(deviceContext);
 462         Assert.assertEquals(1, deviceContexts.size());
 463
 464         deviceManager.close();
 465
 466         Mockito.verify(deviceContext).close();
 467     }
 468
 469     private static Set<DeviceContext> getContextsCollection(DeviceManagerImpl deviceManager) throws NoSuchFieldException, IllegalAccessException {
 470         // HACK: contexts collection for testing shall be accessed in some more civilized way
 471         final Field contextsField = DeviceManagerImpl.class.getDeclaredField("deviceContexts");
 472         Assert.assertNotNull(contextsField);
 473         contextsField.setAccessible(true);
 474         return (Set<DeviceContext>) contextsField.get(deviceManager);
 475     }
 476
 477 }