openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/statistics/StatisticsManagerImplTest.java

   1 /*
   2  * Copyright (c) 2015 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 package org.opendaylight.openflowplugin.impl.statistics;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.mockito.ArgumentMatchers.any;
  14 import static org.mockito.Mockito.verify;
  15 import static org.mockito.Mockito.when;
  16
  17 import com.google.common.util.concurrent.ListenableFuture;
  18 import com.google.common.util.concurrent.MoreExecutors;
  19 import java.util.List;
  20 import java.util.concurrent.ExecutionException;
  21 import java.util.concurrent.Future;
  22 import org.junit.Before;
  23 import org.junit.Test;
  24 import org.junit.runner.RunWith;
  25 import org.mockito.Mock;
  26 import org.mockito.junit.MockitoJUnitRunner;
  27 import org.opendaylight.mdsal.binding.api.DataBroker;
  28 import org.opendaylight.mdsal.binding.api.RpcProviderService;
  29 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
  30 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueue;
  31 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
  32 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  33 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  34 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
  35 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
  36 import org.opendaylight.openflowplugin.api.openflow.device.handlers.MultiMsgCollector;
  37 import org.opendaylight.openflowplugin.api.openflow.lifecycle.ReconciliationFrameworkRegistrar;
  38 import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsContext;
  39 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageSpy;
  40 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManager;
  41 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManagerFactory;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.NonZeroUint32Type;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfigBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.ChangeStatisticsWorkModeInputBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.ChangeStatisticsWorkModeOutput;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutput;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsWorkMode;
  54 import org.opendaylight.yangtools.concepts.Registration;
  55 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
  56 import org.opendaylight.yangtools.yang.common.RpcResult;
  57 import org.opendaylight.yangtools.yang.common.Uint32;
  58
  59 @RunWith(MockitoJUnitRunner.class)
  60 public class StatisticsManagerImplTest {
  61     public static final NodeId NODE_ID = new NodeId("ofp-unit-dummy-node-id");
  62
  63     @Mock
  64     private ConnectionContext mockedPrimConnectionContext;
  65     @Mock
  66     private FeaturesReply mockedFeatures;
  67     @Mock
  68     private ConnectionAdapter mockedConnectionAdapter;
  69     @Mock
  70     private MessageSpy mockedMessagSpy;
  71     @Mock
  72     private DeviceContext mockedDeviceContext;
  73     @Mock
  74     private DeviceState mockedDeviceState;
  75     @Mock
  76     private DeviceInfo mockedDeviceInfo;
  77     @Mock
  78     private RpcProviderService rpcProviderRegistry;
  79     @Mock
  80     private OutboundQueue outboundQueue;
  81     @Mock
  82     private MultiMsgCollector multiMagCollector;
  83     @Mock
  84     private Registration serviceControlRegistration;
  85     @Mock
  86     private DeviceInfo deviceInfo;
  87     @Mock
  88     private DataBroker dataBroker;
  89     @Mock
  90     private ReconciliationFrameworkRegistrar reconciliationFrameworkRegistrar;
  91     @Mock
  92     private StatisticsContext statisticContext;
  93
  94     private RequestContext<List<MultipartReply>> currentRequestContext;
  95     private StatisticsManagerImpl statisticsManager;
  96
  97     @Before
  98     public void initialization() {
  99         final KeyedInstanceIdentifier<Node, NodeKey> nodePath = KeyedInstanceIdentifier
 100                 .create(Nodes.class)
 101                 .child(Node.class, new NodeKey(new NodeId("openflow:10")));
 102
 103         when(rpcProviderRegistry.registerRpcImplementations(any())).thenReturn(serviceControlRegistration);
 104
 105         final ConvertorManager convertorManager = ConvertorManagerFactory.createDefaultManager();
 106
 107         statisticsManager = new StatisticsManagerImpl(
 108                 new OpenflowProviderConfigBuilder()
 109                         .setBasicTimerDelay(new NonZeroUint32Type(Uint32.valueOf(3000)))
 110                         .setMaximumTimerDelay(new NonZeroUint32Type(Uint32.valueOf(900000)))
 111                         .setIsStatisticsPollingOn(false)
 112                         .build(), rpcProviderRegistry,
 113                 convertorManager,
 114                 MoreExecutors.directExecutor());
 115     }
 116
 117     @Test
 118     public void testGetStatisticsWorkMode() throws Exception {
 119         final Future<RpcResult<GetStatisticsWorkModeOutput>> workMode = statisticsManager.getStatisticsWorkMode(null);
 120         assertTrue(workMode.isDone());
 121         assertTrue(workMode.get().isSuccessful());
 122         assertNotNull(workMode.get().getResult());
 123         assertEquals(StatisticsWorkMode.COLLECTALL, workMode.get().getResult().getMode());
 124     }
 125
 126     /**
 127      * switching to {@link StatisticsWorkMode#FULLYDISABLED}; no pollTimeout and no lifecycleRegistry.
 128      *
 129      */
 130     @Test
 131     public void testChangeStatisticsWorkMode1() throws Exception {
 132         statisticsManager.contexts.put(deviceInfo, statisticContext);
 133
 134         final ChangeStatisticsWorkModeInputBuilder changeStatisticsWorkModeInputBld =
 135                 new ChangeStatisticsWorkModeInputBuilder()
 136                         .setMode(StatisticsWorkMode.FULLYDISABLED);
 137
 138         final ListenableFuture<RpcResult<ChangeStatisticsWorkModeOutput>> workMode = statisticsManager
 139                 .changeStatisticsWorkMode(changeStatisticsWorkModeInputBld.build());
 140
 141         checkWorkModeChangeOutcome(workMode);
 142         verify(statisticContext).disableGathering();
 143     }
 144
 145     private static void checkWorkModeChangeOutcome(
 146             final ListenableFuture<RpcResult<ChangeStatisticsWorkModeOutput>> workMode)
 147             throws InterruptedException, ExecutionException {
 148         assertTrue(workMode.isDone());
 149         assertTrue(workMode.get().isSuccessful());
 150     }
 151
 152
 153     /**
 154      * Switching to {@link StatisticsWorkMode#FULLYDISABLED}; with pollTimeout and lifecycleRegistry.
 155      *
 156      */
 157     @Test
 158     public void testChangeStatisticsWorkMode2() throws Exception {
 159         statisticsManager.contexts.put(deviceInfo, statisticContext);
 160
 161         final ChangeStatisticsWorkModeInputBuilder changeStatisticsWorkModeInputBld =
 162                 new ChangeStatisticsWorkModeInputBuilder()
 163                         .setMode(StatisticsWorkMode.FULLYDISABLED);
 164
 165         ListenableFuture<RpcResult<ChangeStatisticsWorkModeOutput>> workMode = statisticsManager
 166             .changeStatisticsWorkMode(changeStatisticsWorkModeInputBld.build());
 167         checkWorkModeChangeOutcome(workMode);
 168
 169         verify(statisticContext).disableGathering();
 170     }
 171
 172     /**
 173      * Tests switching to {@link StatisticsWorkMode#FULLYDISABLED} and back
 174      * to {@link StatisticsWorkMode#COLLECTALL}; with lifecycleRegistry and pollTimeout.
 175      *
 176      */
 177     @Test
 178     public void testChangeStatisticsWorkMode3() throws Exception {
 179         statisticsManager.contexts.put(deviceInfo, statisticContext);
 180
 181         final ChangeStatisticsWorkModeInputBuilder changeStatisticsWorkModeInputBld =
 182                 new ChangeStatisticsWorkModeInputBuilder()
 183                         .setMode(StatisticsWorkMode.FULLYDISABLED);
 184
 185         ListenableFuture<RpcResult<ChangeStatisticsWorkModeOutput>> workMode;
 186         workMode = statisticsManager.changeStatisticsWorkMode(
 187                 changeStatisticsWorkModeInputBld.build());
 188         checkWorkModeChangeOutcome(workMode);
 189
 190         verify(statisticContext).disableGathering();
 191
 192         changeStatisticsWorkModeInputBld.setMode(StatisticsWorkMode.COLLECTALL);
 193         workMode = statisticsManager.changeStatisticsWorkMode(
 194                 changeStatisticsWorkModeInputBld.build());
 195         checkWorkModeChangeOutcome(workMode);
 196
 197         verify(statisticContext).enableGathering();
 198     }
 199
 200     @Test
 201     public void testClose() {
 202         statisticsManager.close();
 203         verify(serviceControlRegistration).close();
 204     }
 205 }