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