2 * Copyright (c) 2017 Pantheon Technologies s.r.o. and others. All rights reserved.
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
9 package org.opendaylight.openflowplugin.impl.device.initialization;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.ArgumentMatchers.eq;
13 import static org.mockito.Mockito.times;
14 import static org.mockito.Mockito.verify;
15 import static org.mockito.Mockito.when;
17 import java.util.Collections;
18 import java.util.Optional;
19 import java.util.concurrent.Future;
20 import org.junit.Before;
21 import org.junit.Test;
22 import org.junit.runner.RunWith;
23 import org.mockito.Mock;
24 import org.mockito.runners.MockitoJUnitRunner;
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
26 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
27 import org.opendaylight.openflowplugin.api.openflow.connection.OutboundQueueProvider;
28 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
29 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
30 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
31 import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
32 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
33 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
34 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
35 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageSpy;
36 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProvider;
37 import org.opendaylight.openflowplugin.impl.datastore.multipart.AbstractMultipartWriter;
38 import org.opendaylight.openflowplugin.impl.util.DeviceStateUtil;
39 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.multipart.reply.multipart.reply.body.MultipartReplyDescBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartReplyBuilder;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.Capabilities;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessageBuilder;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyDescCaseBuilder;
50 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
51 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
53 @RunWith(MockitoJUnitRunner.class)
54 public class OF13DeviceInitializerTest {
56 private DeviceContext deviceContext;
58 private MultipartWriterProvider multipartWriterProvider;
60 private AbstractMultipartWriter abstractMultipartWriter;
62 private ConvertorExecutor convertorExecutor;
64 private ConnectionContext connectionContext;
66 private FeaturesReply featuresReply;
68 private Capabilities capabilities;
70 private DeviceInfo deviceInfo;
72 private DeviceState deviceState;
74 private MessageSpy messageSpy;
76 private TranslatorLibrary translatorLibrary;
78 private MessageTranslator messageTranslator;
80 private RequestContext requestContext;
82 private OutboundQueueProvider outboundQueueProvider;
83 private AbstractDeviceInitializer deviceInitializer;
86 public void setUp() throws Exception {
87 final KeyedInstanceIdentifier<Node, NodeKey> nodeInstanceIdentifier = DeviceStateUtil
88 .createNodeInstanceIdentifier(new NodeId("openflow:1"));
90 deviceInitializer = new OF13DeviceInitializer();
91 when(multipartWriterProvider.lookup(any())).thenReturn(Optional
92 .of(abstractMultipartWriter));
93 when(featuresReply.getCapabilities()).thenReturn(capabilities);
94 when(connectionContext.getFeatures()).thenReturn(featuresReply);
95 when(connectionContext.getOutboundQueueProvider()).thenReturn(outboundQueueProvider);
96 when(deviceContext.getDeviceState()).thenReturn(deviceState);
97 when(deviceInfo.getNodeInstanceIdentifier()).thenReturn(nodeInstanceIdentifier);
98 when(deviceContext.getDeviceInfo()).thenReturn(deviceInfo);
99 when(deviceContext.getMessageSpy()).thenReturn(messageSpy);
100 when(deviceContext.oook()).thenReturn(translatorLibrary);
101 when(requestContext.getXid()).thenReturn(new Xid(42L));
102 when(deviceContext.createRequestContext()).thenReturn(requestContext);
103 when(deviceContext.getPrimaryConnectionContext()).thenReturn(connectionContext);
107 public void initializeSingleLayer() throws Exception {
108 when(deviceContext.canUseSingleLayerSerialization()).thenReturn(true);
109 when(requestContext.getFuture()).thenReturn(RpcResultBuilder
110 .success(Collections.singletonList(new MultipartReplyBuilder()
111 .setMultipartReplyBody(new MultipartReplyDescBuilder().build())
115 final Future<Void> initialize = deviceInitializer.initialize(deviceContext,
118 multipartWriterProvider,
122 verify(messageSpy, times(5)).spyMessage(eq(MultipartType.class), any());
123 verify(requestContext, times(5)).getFuture();
124 verify(featuresReply).getCapabilities();
125 verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
129 public void initializeMultiLayer() throws Exception {
130 when(deviceContext.canUseSingleLayerSerialization()).thenReturn(false);
131 when(requestContext.getFuture()).thenReturn(RpcResultBuilder
132 .success(Collections.singletonList(new MultipartReplyMessageBuilder()
133 .setType(MultipartType.OFPMPDESC)
134 .setMultipartReplyBody(new MultipartReplyDescCaseBuilder()
135 .setMultipartReplyDesc(new org.opendaylight.yang.gen.v1.urn.opendaylight.openflow
136 .protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply
137 .desc._case.MultipartReplyDescBuilder()
143 final Future<Void> initialize = deviceInitializer.initialize(deviceContext,
146 multipartWriterProvider,
150 verify(messageSpy, times(5)).spyMessage(eq(MultipartType.class), any());
151 verify(requestContext, times(5)).getFuture();
152 verify(featuresReply).getCapabilities();
153 verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), any(), any());