2 * Copyright (c) 2015 Cisco Systems, Inc. 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;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertTrue;
13 import static org.mockito.Matchers.any;
14 import static org.mockito.Matchers.eq;
15 import static org.mockito.Mockito.mock;
16 import static org.mockito.Mockito.verify;
17 import static org.mockito.Mockito.when;
19 import com.google.common.util.concurrent.CheckedFuture;
20 import io.netty.util.HashedWheelTimer;
21 import java.lang.reflect.Field;
22 import java.math.BigInteger;
23 import java.util.concurrent.ConcurrentHashMap;
24 import org.junit.Assert;
25 import org.junit.Before;
26 import org.junit.Test;
27 import org.junit.runner.RunWith;
28 import org.mockito.Mock;
29 import org.mockito.runners.MockitoJUnitRunner;
30 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
31 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
32 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
33 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
34 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
35 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
36 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
37 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
38 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
39 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
40 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
41 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
42 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageIntelligenceAgency;
43 import org.opendaylight.openflowplugin.impl.device.initialization.DeviceInitializerProviderFactory;
44 import org.opendaylight.openflowplugin.impl.util.DeviceStateUtil;
45 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRemovedBuilder;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeUpdatedBuilder;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.Capabilities;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.CapabilitiesV10;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.NonZeroUint16Type;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.NonZeroUint32Type;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfigBuilder;
58 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
60 @RunWith(MockitoJUnitRunner.class)
61 public class DeviceManagerImplTest {
63 private static final long TEST_VALUE_GLOBAL_NOTIFICATION_QUOTA = 2000L;
64 private static final int barrierCountLimit = 25600;
65 private static final long barrierIntervalNanos = 500;
66 private static final NodeId DUMMY_NODE_ID = new NodeId("dummyNodeId");
67 private static final KeyedInstanceIdentifier<Node, NodeKey> DUMMY_IDENTIFIER = DeviceStateUtil
68 .createNodeInstanceIdentifier(DUMMY_NODE_ID);
71 private CheckedFuture<Void, TransactionCommitFailedException> mockedFuture;
73 private FeaturesReply mockFeatures;
75 private ConnectionContext mockConnectionContext;
77 private ConnectionAdapter mockedConnectionAdapter;
79 private DeviceContextImpl mockedDeviceContext;
81 private MessageIntelligenceAgency messageIntelligenceAgency;
83 private DeviceInfo deviceInfo;
85 private ConvertorExecutor convertorExecutor;
87 private DataBroker dataBroker;
89 private WriteTransaction writeTransaction;
91 private BindingTransactionChain transactionChain;
93 private Capabilities capabilities;
95 private CapabilitiesV10 capabilitiesV10;
97 private NotificationPublishService notificationPublishService;
99 private TranslatorLibrary translatorLibrary;
101 private DeviceManagerImpl deviceManager;
104 public void setUp() throws Exception {
105 when(mockConnectionContext.getNodeId()).thenReturn(DUMMY_NODE_ID);
106 when(mockConnectionContext.getFeatures()).thenReturn(mockFeatures);
107 when(mockConnectionContext.getConnectionAdapter()).thenReturn(mockedConnectionAdapter);
108 when(mockConnectionContext.getDeviceInfo()).thenReturn(deviceInfo);
109 when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockConnectionContext);
110 when(deviceInfo.getNodeInstanceIdentifier()).thenReturn(DUMMY_IDENTIFIER);
111 when(deviceInfo.getNodeId()).thenReturn(DUMMY_NODE_ID);
113 when(mockFeatures.getCapabilities()).thenReturn(capabilities);
114 when(mockFeatures.getCapabilitiesV10()).thenReturn(capabilitiesV10);
115 when(mockFeatures.getDatapathId()).thenReturn(BigInteger.valueOf(21L));
117 when(mockedFuture.isDone()).thenReturn(true);
118 when(writeTransaction.submit()).thenReturn(mockedFuture);
119 when(transactionChain.newWriteOnlyTransaction()).thenReturn(writeTransaction);
120 when(dataBroker.createTransactionChain(any(TransactionChainListener.class))).thenReturn(transactionChain);
121 when(dataBroker.newWriteOnlyTransaction()).thenReturn(writeTransaction);
123 deviceManager = new DeviceManagerImpl(
124 new OpenflowProviderConfigBuilder()
125 .setBarrierCountLimit(new NonZeroUint16Type(barrierCountLimit))
126 .setBarrierIntervalTimeoutLimit(new NonZeroUint32Type(barrierIntervalNanos))
127 .setGlobalNotificationQuota(TEST_VALUE_GLOBAL_NOTIFICATION_QUOTA)
128 .setSwitchFeaturesMandatory(false)
129 .setEnableFlowRemovedNotification(true)
130 .setSkipTableFeatures(false)
131 .setUseSingleLayerSerialization(true)
134 messageIntelligenceAgency,
135 notificationPublishService,
136 new HashedWheelTimer(),
138 DeviceInitializerProviderFactory.createDefaultProvider());
140 deviceManager.setTranslatorLibrary(translatorLibrary);
141 verify(dataBroker).newWriteOnlyTransaction();
142 verify(writeTransaction).merge(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
143 verify(writeTransaction).submit();
147 public void createContext() throws Exception {
148 final DeviceContext context = deviceManager.createContext(mockConnectionContext);
149 assertEquals(deviceInfo, context.getDeviceInfo());
154 public void removeDeviceFromOperationalDS() throws Exception {
155 final CheckedFuture<Void, TransactionCommitFailedException> future = deviceManager
156 .removeDeviceFromOperationalDS(DUMMY_IDENTIFIER);
159 assertTrue(future.isDone());
160 verify(writeTransaction).delete(LogicalDatastoreType.OPERATIONAL, DUMMY_IDENTIFIER);
164 public void sendNodeAddedNotification() throws Exception {
165 deviceManager.sendNodeAddedNotification(DUMMY_IDENTIFIER);
166 deviceManager.sendNodeAddedNotification(DUMMY_IDENTIFIER);
167 verify(notificationPublishService).offerNotification(new NodeUpdatedBuilder()
168 .setId(DUMMY_NODE_ID)
169 .setNodeRef(new NodeRef(DUMMY_IDENTIFIER))
174 public void sendNodeRemovedNotification() throws Exception {
175 deviceManager.sendNodeAddedNotification(DUMMY_IDENTIFIER);
176 deviceManager.sendNodeRemovedNotification(DUMMY_IDENTIFIER);
177 deviceManager.sendNodeRemovedNotification(DUMMY_IDENTIFIER);
178 verify(notificationPublishService).offerNotification(new NodeRemovedBuilder()
179 .setNodeRef(new NodeRef(DUMMY_IDENTIFIER))
184 public void close() throws Exception {
185 final DeviceContext deviceContext = mock(DeviceContext.class);
186 final ConcurrentHashMap<DeviceInfo, DeviceContext> deviceContexts = getContextsCollection(deviceManager);
187 deviceContexts.put(deviceInfo, deviceContext);
188 Assert.assertEquals(1, deviceContexts.size());
189 deviceManager.close();
190 verify(deviceContext).close();
193 @SuppressWarnings("unchecked")
194 private static ConcurrentHashMap<DeviceInfo, DeviceContext> getContextsCollection(final DeviceManagerImpl deviceManager) throws NoSuchFieldException, IllegalAccessException {
195 // HACK: contexts collection for testing shall be accessed in some more civilized way
196 final Field contextsField = DeviceManagerImpl.class.getDeclaredField("deviceContexts");
197 Assert.assertNotNull(contextsField);
198 contextsField.setAccessible(true);
199 return (ConcurrentHashMap<DeviceInfo, DeviceContext>) contextsField.get(deviceManager);