openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/util/DeviceInitializationUtilsTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.openflowplugin.impl.util;
  10
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Matchers.eq;
  13 import static org.mockito.Mockito.atLeastOnce;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.times;
  16 import static org.mockito.Mockito.verify;
  17 import static org.mockito.Mockito.when;
  18
  19 import com.google.common.collect.Lists;
  20 import com.google.common.util.concurrent.CheckedFuture;
  21 import com.google.common.util.concurrent.FutureCallback;
  22 import com.google.common.util.concurrent.Futures;
  23 import com.google.common.util.concurrent.ListenableFuture;
  24 import java.math.BigInteger;
  25 import java.util.ArrayList;
  26 import java.util.Collection;
  27 import java.util.Collections;
  28 import java.util.List;
  29 import java.util.Set;
  30 import org.junit.Before;
  31 import org.junit.Test;
  32 import org.junit.runner.RunWith;
  33 import org.mockito.Matchers;
  34 import org.mockito.Mock;
  35 import org.mockito.Mockito;
  36 import org.mockito.invocation.InvocationOnMock;
  37 import org.mockito.runners.MockitoJUnitRunner;
  38 import org.mockito.stubbing.Answer;
  39 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  40 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  41 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
  42 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueue;
  43 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueueHandler;
  44 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueueHandlerRegistration;
  45 import org.opendaylight.openflowplugin.api.OFConstants;
  46 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
  47 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  48 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  49 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
  50 import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
  51 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
  52 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
  53 import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceInitializationPhaseHandler;
  54 import org.opendaylight.openflowplugin.api.openflow.device.handlers.MultiMsgCollector;
  55 import org.opendaylight.openflowplugin.api.openflow.md.core.TranslatorKey;
  56 import org.opendaylight.openflowplugin.impl.device.DeviceContextImpl;
  57 import org.opendaylight.openflowplugin.openflow.md.util.OpenflowPortsUtil;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.statistics.rev131111.NodeGroupFeatures;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterFeatures;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ActionType;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.Capabilities;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.CapabilitiesV10;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupCapabilities;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupTypes;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterBandTypeBitmap;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterFlags;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessage;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessageBuilder;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInput;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyDescCaseBuilder;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupFeaturesCaseBuilder;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterFeaturesCaseBuilder;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyPortDescCaseBuilder;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyTableFeaturesCaseBuilder;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.desc._case.MultipartReplyDesc;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.desc._case.MultipartReplyDescBuilder;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.features._case.MultipartReplyGroupFeatures;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.features._case.MultipartReplyGroupFeaturesBuilder;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.features._case.MultipartReplyMeterFeaturesBuilder;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.desc._case.MultipartReplyPortDescBuilder;
  94 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;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.MultipartReplyTableFeatures;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.MultipartReplyTableFeaturesBuilder;
  97 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;
  98 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;
  99 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 100 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
 101 import org.opendaylight.yangtools.yang.common.RpcError;
 102 import org.opendaylight.yangtools.yang.common.RpcResult;
 103 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
 104
 105 @RunWith(MockitoJUnitRunner.class)
 106 public class DeviceInitializationUtilsTest {
 107
 108     public static final String DUMMY_NODE_ID = "dummyNodeId";
 109     private static final KeyedInstanceIdentifier<Node, NodeKey> DUMMY_NODE_II = InstanceIdentifier.create(Nodes.class)
 110             .child(Node.class, new NodeKey(new NodeId(DUMMY_NODE_ID)));
 111     private static final Short DUMMY_TABLE_ID = 1;
 112     private static final Long DUMMY_MAX_METER = 544L;
 113     private static final String DUMMY_DATAPATH_ID = "44";
 114     private static final Long DUMMY_PORT_NUMBER = 21L;
 115
 116     @Mock
 117     CheckedFuture<Void, TransactionCommitFailedException> mockedFuture;
 118     @Mock
 119     private FeaturesReply mockFeatures;
 120     @Mock
 121     private OutboundQueue outboundQueueProvider;
 122     @Mock
 123     private DeviceInitializationPhaseHandler deviceInitPhaseHandler;
 124     @Mock
 125     private TranslatorLibrary translatorLibrary;
 126     @Mock
 127     private ConnectionContext mockConnectionContext;
 128     @Mock
 129     private ConnectionAdapter mockedConnectionAdapter;
 130     @Mock
 131     private DeviceContextImpl mockedDeviceContext;
 132     @Mock
 133     private DeviceInfo mockedDeviceInfo;
 134     @Mock
 135     private DeviceInitializationUtils deviceInitializationUtils;
 136     @Mock
 137     private DeviceInfo deviceInfo;
 138
 139     @Before
 140     public void setUp() throws Exception {
 141         OpenflowPortsUtil.init();
 142
 143         when(mockConnectionContext.getNodeId()).thenReturn(new NodeId(DUMMY_NODE_ID));
 144         when(mockConnectionContext.getFeatures()).thenReturn(mockFeatures);
 145         when(mockConnectionContext.getConnectionAdapter()).thenReturn(mockedConnectionAdapter);
 146         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockConnectionContext);
 147         when(mockedDeviceContext.getDeviceInfo()).thenReturn(mockedDeviceInfo);
 148
 149         final Capabilities capabilitiesV13 = mock(Capabilities.class);
 150         final CapabilitiesV10 capabilitiesV10 = mock(CapabilitiesV10.class);
 151         when(mockFeatures.getCapabilities()).thenReturn(capabilitiesV13);
 152         when(mockFeatures.getCapabilitiesV10()).thenReturn(capabilitiesV10);
 153         when(mockFeatures.getDatapathId()).thenReturn(BigInteger.valueOf(21L));
 154     }
 155
 156     @Test
 157     public void initializeNodeInformationTest() throws Exception {
 158         DeviceState mockedDeviceState = mock(DeviceState.class);
 159         MultiMsgCollector msgCollector = mock(MultiMsgCollector.class);
 160         TranslatorLibrary tLibrary = mock(TranslatorLibrary.class);
 161
 162         GetFeaturesOutput mockedFeatures = mock(GetFeaturesOutput.class);
 163         when(mockedFeatures.getTables()).thenReturn((short) 2);
 164         when(mockedDeviceInfo.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_0);
 165
 166         when(mockedDeviceInfo.getNodeInstanceIdentifier()).thenReturn(DUMMY_NODE_II);
 167         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 168         when(mockedDeviceContext.getMultiMsgCollector(Mockito.any(RequestContext.class))).thenReturn(msgCollector);
 169         when(mockedDeviceContext.oook()).thenReturn(tLibrary);
 170
 171         final ConnectionContext connectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_0);
 172         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(connectionContext);
 173
 174         DeviceInitializationUtils.initializeNodeInformation(mockedDeviceContext, true);
 175
 176         verify(mockFeatures, atLeastOnce()).getPhyPort();
 177         verify(tLibrary, atLeastOnce()).lookupTranslator(any(TranslatorKey.class));
 178     }
 179
 180     @Test
 181     public void chainTableTrunkWriteOF10Test() throws Exception {
 182         DeviceState mockedDeviceState = mock(DeviceState.class);
 183
 184         final GetFeaturesOutput mockedFeatures = mock(GetFeaturesOutput.class);
 185         when(mockedFeatures.getTables()).thenReturn((short) 2);
 186         when(mockConnectionContext.getFeatures()).thenReturn(mockedFeatures);
 187
 188         when(mockedDeviceInfo.getNodeInstanceIdentifier()).thenReturn(DUMMY_NODE_II);
 189         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 190
 191         final RpcResult<List<MultipartReply>> mockedRpcResult = mock(RpcResult.class);
 192         when(mockedRpcResult.isSuccessful()).thenReturn(true);
 193         final List<RpcResult<List<MultipartReply>>> data = new ArrayList<RpcResult<List<MultipartReply>>>();
 194         data.add(mockedRpcResult);
 195         data.add(mockedRpcResult);
 196
 197         DeviceInitializationUtils.chainTableTrunkWriteOF10(mockedDeviceContext, Futures.immediateFuture(data));
 198         verify(mockedDeviceContext, times(3)).writeToTransaction(any(LogicalDatastoreType.class),
 199                 Matchers.<InstanceIdentifier<FlowCapableNode>> any(), any(FlowCapableNode.class));
 200     }
 201
 202     @Test
 203     public void testTranslateAndWriteReplyTypeDesc() throws Exception {
 204         final ConnectionContext connectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_3);
 205         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(connectionContext);
 206         final DeviceState deviceState = mock(DeviceState.class);
 207         when(mockedDeviceContext.getDeviceState()).thenReturn(deviceState);
 208
 209         final Collection<MultipartReply> multipartReplyMessages = prepareDataforTypeDesc(mockedDeviceContext);
 210
 211         DeviceInitializationUtils.translateAndWriteReply(MultipartType.OFPMPDESC, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 212         verify(mockedDeviceContext)
 213                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class)), any(FlowCapableNode.class));
 214     }
 215
 216     @Test
 217     public void translateAndWriteReplyTypeTableFeatures() throws Exception {
 218         TableFeaturesBuilder tableFeature = new TableFeaturesBuilder();
 219         tableFeature.setTableId(DUMMY_TABLE_ID);
 220         final List<TableFeatures> tableFeatures = new ArrayList<>();
 221         tableFeatures.add(tableFeature.build());
 222
 223         final MultipartReplyTableFeatures multipartReplyTableFeatures = new MultipartReplyTableFeaturesBuilder().setTableFeatures(tableFeatures).build();
 224         final MultipartReplyTableFeaturesCaseBuilder multipartReplyTableFeaturesCaseBuilder = new MultipartReplyTableFeaturesCaseBuilder();
 225         multipartReplyTableFeaturesCaseBuilder.setMultipartReplyTableFeatures(multipartReplyTableFeatures);
 226
 227         final MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyTableFeaturesCaseBuilder.build()).build();
 228         final Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 229         DeviceInitializationUtils.translateAndWriteReply(MultipartType.OFPMPTABLEFEATURES, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 230         verify(mockedDeviceContext)
 231                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 232                         eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class).child(Table.class, new TableKey(DUMMY_TABLE_ID))), any(Table.class));
 233
 234     }
 235
 236     @Test
 237     public void translateAndWriteReplyTypeMeterFeatures() throws Exception {
 238         DeviceState mockedDeviceState = mock(DeviceState.class);
 239         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 240
 241         final MultipartReplyMeterFeaturesBuilder multipartReplyMeterFeaturesBuilder = new MultipartReplyMeterFeaturesBuilder();
 242         multipartReplyMeterFeaturesBuilder.setBandTypes(new MeterBandTypeBitmap(true, true));
 243         multipartReplyMeterFeaturesBuilder.setCapabilities(new MeterFlags(true, true, true, true));
 244         multipartReplyMeterFeaturesBuilder.setMaxMeter(DUMMY_MAX_METER);
 245
 246         final MultipartReplyMeterFeaturesCaseBuilder multipartReplyMeterFeaturesCaseBuilder = new MultipartReplyMeterFeaturesCaseBuilder();
 247         multipartReplyMeterFeaturesCaseBuilder.setMultipartReplyMeterFeatures(multipartReplyMeterFeaturesBuilder.build());
 248
 249         final MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyMeterFeaturesCaseBuilder.build()).build();
 250         final Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 251         DeviceInitializationUtils.translateAndWriteReply(MultipartType.OFPMPMETERFEATURES, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 252         verify(mockedDeviceContext)
 253                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(NodeMeterFeatures.class)), any(NodeMeterFeatures.class));
 254         verify(mockedDeviceState).setMeterAvailable(eq(true));
 255     }
 256
 257     @Test
 258     public void translateAndWriteReplyTypeGroupFeatures() throws Exception {
 259         MultipartReplyGroupFeaturesBuilder multipartReplyGroupFeaturesBuilder = new MultipartReplyGroupFeaturesBuilder();
 260         multipartReplyGroupFeaturesBuilder.setTypes(new GroupTypes(true, true, true, true));
 261         multipartReplyGroupFeaturesBuilder.setCapabilities(new GroupCapabilities(true, true, true, true));
 262         final ActionType actionType = new ActionType(true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true);
 263         multipartReplyGroupFeaturesBuilder.setActionsBitmap(Lists.newArrayList(actionType));
 264
 265         final MultipartReplyGroupFeatures multipartReplyGroupFeatures = multipartReplyGroupFeaturesBuilder.build();
 266
 267         final MultipartReplyGroupFeaturesCaseBuilder multipartReplyGroupFeaturesCaseBuilder = new MultipartReplyGroupFeaturesCaseBuilder();
 268         multipartReplyGroupFeaturesCaseBuilder.setMultipartReplyGroupFeatures(multipartReplyGroupFeatures);
 269
 270         final MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyGroupFeaturesCaseBuilder.build()).build();
 271         final Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 272
 273         DeviceInitializationUtils.translateAndWriteReply(MultipartType.OFPMPGROUPFEATURES, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 274         verify(mockedDeviceContext)
 275                 .writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(NodeGroupFeatures.class)), any(NodeGroupFeatures.class));
 276     }
 277
 278
 279     @Test
 280     public void translateAndWriteReplyTypePortDesc() throws Exception {
 281         ConnectionContext mockedPrimaryConnectionContext = mock(ConnectionContext.class);
 282         FeaturesReply mockedFeatures = mock(FeaturesReply.class);
 283         when(mockedFeatures.getDatapathId()).thenReturn(new BigInteger(DUMMY_DATAPATH_ID));
 284         when(mockedPrimaryConnectionContext.getFeatures()).thenReturn(mockedFeatures);
 285         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockedPrimaryConnectionContext);
 286         final DeviceState mockedDeviceState = mock(DeviceState.class);
 287         when(mockedDeviceInfo.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_0);
 288         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 289         final MessageTranslator mockedTranslator = mock(MessageTranslator.class);
 290         when(translatorLibrary.lookupTranslator(any(TranslatorKey.class))).thenReturn(mockedTranslator);
 291         when(mockedDeviceContext.oook()).thenReturn(translatorLibrary);
 292
 293         final MultipartReplyPortDescBuilder multipartReplyPortDescBuilder = new MultipartReplyPortDescBuilder();
 294
 295         final PortsBuilder portsBuilder = new PortsBuilder();
 296         portsBuilder.setPortNo(DUMMY_PORT_NUMBER);
 297
 298         multipartReplyPortDescBuilder.setPorts(Lists.newArrayList(portsBuilder.build()));
 299
 300         final MultipartReplyPortDescCaseBuilder multipartReplyPortDescCaseBuilder = new MultipartReplyPortDescCaseBuilder();
 301         multipartReplyPortDescCaseBuilder.setMultipartReplyPortDesc(multipartReplyPortDescBuilder.build());
 302
 303         final MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyPortDescCaseBuilder.build()).build();
 304         final Set<MultipartReply> multipartReplyMessages = Collections.<MultipartReply>singleton(multipartReplyMessage);
 305
 306         OpenflowPortsUtil.init();
 307         DeviceInitializationUtils.translateAndWriteReply(MultipartType.OFPMPPORTDESC, mockedDeviceContext, DUMMY_NODE_II, multipartReplyMessages);
 308         verify(mockedDeviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 309                 Matchers.<InstanceIdentifier<NodeConnector>> any(), any(NodeConnector.class));
 310     }
 311
 312     @Test
 313     public void createSuccessProcessingCallbackTest() throws Exception {
 314         DeviceState mockedDeviceState = mock(DeviceState.class);
 315         when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
 316
 317         final ConnectionContext connectionContext = buildMockConnectionContext(OFConstants.OFP_VERSION_1_3);
 318
 319         final List<MultipartReply> multipartReplies = new ArrayList<>(prepareDataforTypeDesc(mockedDeviceContext));
 320         final RpcResult<List<MultipartReply>> result = RpcResultBuilder.<List<MultipartReply>>success(multipartReplies).build();
 321         ListenableFuture<RpcResult<List<MultipartReply>>> mockedRequestContextFuture = Futures.immediateFuture(result);
 322
 323         DeviceInitializationUtils.createSuccessProcessingCallback(MultipartType.OFPMPDESC, mockedDeviceContext, DUMMY_NODE_II, mockedRequestContextFuture);
 324         verify(mockedDeviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class)), any(FlowCapableNode.class));
 325
 326         final RpcResult<List<MultipartReply>> rpcResult = RpcResultBuilder.<List<MultipartReply>>failed().withError(RpcError.ErrorType.PROTOCOL, "dummy error").build();
 327         mockedRequestContextFuture = Futures.immediateFuture(rpcResult);
 328         DeviceInitializationUtils.createSuccessProcessingCallback(MultipartType.OFPMPDESC, mockedDeviceContext, DUMMY_NODE_II, mockedRequestContextFuture);
 329         verify(mockedDeviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), eq(DUMMY_NODE_II.augmentation(FlowCapableNode.class)), any(FlowCapableNode.class));
 330     }
 331
 332     private Collection<MultipartReply> prepareDataforTypeDesc(final DeviceContext mockedDeviceContext) {
 333         final MultipartReplyDesc multipartReplyDesc = new MultipartReplyDescBuilder().build();
 334
 335         final MultipartReplyDescCaseBuilder multipartReplyDescCaseBuilder = new MultipartReplyDescCaseBuilder();
 336         multipartReplyDescCaseBuilder.setMultipartReplyDesc(multipartReplyDesc);
 337
 338         final MultipartReplyMessage multipartReplyMessage = new MultipartReplyMessageBuilder().setMultipartReplyBody(multipartReplyDescCaseBuilder.build()).build();
 339         return Collections.<MultipartReply>singleton(multipartReplyMessage);
 340
 341     }
 342
 343     protected ConnectionContext buildMockConnectionContext(final short ofpVersion) {
 344         when(mockFeatures.getVersion()).thenReturn(ofpVersion);
 345         when(outboundQueueProvider.reserveEntry()).thenReturn(43L);
 346         Mockito.doAnswer(new Answer<Void>() {
 347             @Override
 348             public Void answer(final InvocationOnMock invocation) throws Throwable {
 349                 final FutureCallback<OfHeader> callBack = (FutureCallback<OfHeader>) invocation.getArguments()[2];
 350                 callBack.onSuccess(null);
 351                 return null;
 352             }
 353         })
 354                 .when(outboundQueueProvider)
 355                 .commitEntry(Matchers.anyLong(), Matchers.<MultipartRequestInput>any(), Matchers.<FutureCallback<OfHeader>>any());
 356
 357         when(mockedConnectionAdapter.registerOutboundQueueHandler(Matchers.<OutboundQueueHandler>any(), Matchers.anyInt(), Matchers.anyLong()))
 358                 .thenAnswer(new Answer<OutboundQueueHandlerRegistration<OutboundQueueHandler>>() {
 359                     @Override
 360                     public OutboundQueueHandlerRegistration<OutboundQueueHandler> answer(final InvocationOnMock invocation) throws Throwable {
 361                         final OutboundQueueHandler handler = (OutboundQueueHandler) invocation.getArguments()[0];
 362                         handler.onConnectionQueueChanged(outboundQueueProvider);
 363                         return null;
 364                     }
 365                 });
 366
 367         when(mockConnectionContext.getOutboundQueueProvider()).thenReturn(outboundQueueProvider);
 368         return mockConnectionContext;
 369     }
 370 }