openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/statistics/StatisticsGatheringUtilsTest.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.mockito.ArgumentMatchers.any;
  11 import static org.mockito.ArgumentMatchers.eq;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.verify;
  14 import static org.mockito.Mockito.when;
  15
  16 import com.google.common.collect.Lists;
  17 import com.google.common.util.concurrent.Futures;
  18 import com.google.common.util.concurrent.ListenableFuture;
  19 import com.google.common.util.concurrent.MoreExecutors;
  20 import java.math.BigInteger;
  21 import java.util.Collections;
  22 import java.util.List;
  23 import java.util.Optional;
  24 import java.util.concurrent.ExecutionException;
  25 import java.util.concurrent.TimeUnit;
  26 import java.util.concurrent.TimeoutException;
  27 import org.junit.Assert;
  28 import org.junit.Before;
  29 import org.junit.Test;
  30 import org.junit.runner.RunWith;
  31 import org.mockito.ArgumentCaptor;
  32 import org.mockito.ArgumentMatchers;
  33 import org.mockito.InOrder;
  34 import org.mockito.Mock;
  35 import org.mockito.Mockito;
  36 import org.mockito.junit.MockitoJUnitRunner;
  37 import org.opendaylight.mdsal.binding.api.ReadTransaction;
  38 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  39 import org.opendaylight.openflowplugin.api.OFConstants;
  40 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
  41 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  42 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  43 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
  44 import org.opendaylight.openflowplugin.api.openflow.registry.flow.DeviceFlowRegistry;
  45 import org.opendaylight.openflowplugin.api.openflow.registry.flow.FlowDescriptor;
  46 import org.opendaylight.openflowplugin.api.openflow.registry.flow.FlowRegistryKey;
  47 import org.opendaylight.openflowplugin.api.openflow.registry.group.DeviceGroupRegistry;
  48 import org.opendaylight.openflowplugin.api.openflow.registry.meter.DeviceMeterRegistry;
  49 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.EventIdentifier;
  50 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.StatisticsGatherer;
  51 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProvider;
  52 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProviderFactory;
  53 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManagerFactory;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowId;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterKey;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableBuilder;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowBuilder;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowKey;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowsStatisticsUpdate;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowsStatisticsUpdateBuilder;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapListBuilder;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsData;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.statistics.FlowTableStatistics;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.Queue;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.QueueKey;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.queue.rev130925.QueueId;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.statistics.rev131111.NodeGroupStatistics;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.statistics.rev131111.group.statistics.GroupStatistics;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.Group;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.GroupKey;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatistics;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatistics;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.FlowModFlags;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupId;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupType;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterFlags;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterId;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartRequestFlags;
  94 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.PortNumber;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
  97 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
  98 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessage;
  99 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessageBuilder;
 100 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.buckets.grouping.BucketsListBuilder;
 101 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.MultipartReplyBody;
 102 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyFlowCaseBuilder;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupCaseBuilder;
 104 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupDescCaseBuilder;
 105 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterCaseBuilder;
 106 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterConfigCaseBuilder;
 107 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyPortStatsCaseBuilder;
 108 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyQueueCaseBuilder;
 109 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyTableCaseBuilder;
 110 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.flow._case.MultipartReplyFlowBuilder;
 111 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.flow._case.multipart.reply.flow.FlowStatsBuilder;
 112 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.MultipartReplyGroupBuilder;
 113 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.multipart.reply.group.GroupStatsBuilder;
 114 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.multipart.reply.group.group.stats.BucketStats;
 115 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.multipart.reply.group.group.stats.BucketStatsBuilder;
 116 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.desc._case.MultipartReplyGroupDescBuilder;
 117 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.desc._case.multipart.reply.group.desc.GroupDescBuilder;
 118 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter._case.MultipartReplyMeterBuilder;
 119 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter._case.multipart.reply.meter.MeterStatsBuilder;
 120 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter._case.multipart.reply.meter.meter.stats.MeterBandStatsBuilder;
 121 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.config._case.MultipartReplyMeterConfigBuilder;
 122 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.config._case.multipart.reply.meter.config.MeterConfigBuilder;
 123 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.stats._case.MultipartReplyPortStatsBuilder;
 124 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.stats._case.multipart.reply.port.stats.PortStatsBuilder;
 125 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.queue._case.MultipartReplyQueueBuilder;
 126 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.queue._case.multipart.reply.queue.QueueStatsBuilder;
 127 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table._case.MultipartReplyTableBuilder;
 128 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table._case.multipart.reply.table.TableStatsBuilder;
 129 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsData;
 130 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.flow.capable.node.connector.statistics.FlowCapableNodeConnectorStatistics;
 131 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
 132 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 133 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
 134 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
 135
 136 @RunWith(MockitoJUnitRunner.class)
 137 public class StatisticsGatheringUtilsTest {
 138
 139     static final String DUMMY_NODE_ID_VALUE = "1";
 140     static final NodeId DUMMY_NODE_ID = new NodeId(DUMMY_NODE_ID_VALUE);
 141
 142     private final KeyedInstanceIdentifier<Node, NodeKey> dummyNodePath = InstanceIdentifier.create(Nodes.class)
 143             .child(Node.class, new NodeKey(DUMMY_NODE_ID));
 144     @Mock
 145     private DeviceContext deviceContext;
 146     @Mock
 147     private DeviceState deviceState;
 148     @Mock
 149     private DeviceFlowRegistry deviceFlowRegistry;
 150     @Mock
 151     private DeviceGroupRegistry deviceGroupRegistry;
 152     @Mock
 153     private DeviceMeterRegistry deviceMeterRegistry;
 154     @Mock
 155     private FlowDescriptor flowDescriptor;
 156     @Mock
 157     private FlowId flowId;
 158     @Mock
 159     private GetFeaturesOutput features;
 160     @Mock
 161     private ReadTransaction readTx;
 162     @Mock
 163     private ConnectionContext connectionAdapter;
 164     @Mock
 165     private StatisticsGatherer<MultipartReply> statisticsService;
 166     @Mock
 167     private DeviceInfo deviceInfo;
 168
 169     private MultipartWriterProvider provider;
 170
 171     @Before
 172     public void setUp() throws Exception {
 173         when(deviceContext.getDeviceInfo()).thenReturn(deviceInfo);
 174         when(deviceContext.getDeviceFlowRegistry()).thenReturn(deviceFlowRegistry);
 175         when(deviceContext.getDeviceGroupRegistry()).thenReturn(deviceGroupRegistry);
 176         when(deviceContext.getDeviceMeterRegistry()).thenReturn(deviceMeterRegistry);
 177         when(deviceFlowRegistry.retrieveDescriptor(any(FlowRegistryKey.class))).thenReturn(flowDescriptor);
 178         when(deviceContext.getReadTransaction()).thenReturn(readTx);
 179         when(deviceContext.getReadTransaction()).thenReturn(readTx);
 180         when(deviceContext.getPrimaryConnectionContext()).thenReturn(connectionAdapter);
 181         when(deviceContext.isTransactionsEnabled()).thenReturn(Boolean.TRUE);
 182         when(connectionAdapter.getFeatures()).thenReturn(features);
 183         when(features.getDatapathId()).thenReturn(BigInteger.ONE);
 184         when(features.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_3);
 185         when(deviceInfo.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_3);
 186         when(deviceInfo.getDatapathId()).thenReturn(BigInteger.ONE);
 187         when(deviceInfo.getNodeInstanceIdentifier()).thenReturn(dummyNodePath);
 188         when(deviceInfo.getNodeId()).thenReturn(DUMMY_NODE_ID);
 189         provider = MultipartWriterProviderFactory.createDefaultProvider(deviceContext);
 190     }
 191
 192     @Test
 193     public void testWriteFlowStatistics() throws Exception {
 194         final ArgumentCaptor<LogicalDatastoreType> dataStoreType = ArgumentCaptor.forClass(LogicalDatastoreType.class);
 195         final ArgumentCaptor<InstanceIdentifier> flowPath = ArgumentCaptor.forClass(InstanceIdentifier.class);
 196         final ArgumentCaptor<Flow> flow = ArgumentCaptor.forClass(Flow.class);
 197
 198         provider.lookup(MultipartType.OFPMPFLOW).get().write(prepareFlowStatisticsData().iterator().next(), false);
 199
 200         Mockito.verify(deviceContext).writeToTransaction(
 201                 dataStoreType.capture(), flowPath.capture(), flow.capture());
 202         Assert.assertEquals(LogicalDatastoreType.OPERATIONAL, dataStoreType.getValue());
 203         final InstanceIdentifier<FlowCapableNode> flowCapableNodePath = flowPath.getValue();
 204         Assert.assertEquals(DUMMY_NODE_ID, flowCapableNodePath.firstKeyOf(Node.class).getId());
 205         Assert.assertEquals(42, flow.getValue().getTableId().intValue());
 206     }
 207
 208     private Iterable<FlowsStatisticsUpdate> prepareFlowStatisticsData() {
 209         final FlowAndStatisticsMapListBuilder flowAndStatsMapListBld = new FlowAndStatisticsMapListBuilder();
 210         flowAndStatsMapListBld.setTableId((short) 42);
 211         flowAndStatsMapListBld.setMatch(new MatchBuilder().build());
 212
 213         final FlowsStatisticsUpdateBuilder flowStatsUpdateBld1 = new FlowsStatisticsUpdateBuilder();
 214         flowStatsUpdateBld1.setFlowAndStatisticsMapList(Lists.newArrayList(flowAndStatsMapListBld.build()));
 215
 216         return Lists.newArrayList(flowStatsUpdateBld1.build());
 217     }
 218
 219
 220     @Test
 221     public void testGatherStatistics_group() throws Exception {
 222         final MultipartType type = MultipartType.OFPMPGROUP;
 223         final long groupIdValue = 19L;
 224
 225         final GroupStatsBuilder groupStatsBld = new GroupStatsBuilder()
 226                 .setBucketStats(Lists.newArrayList(createBucketStat(21L, 42L)))
 227                 .setByteCount(BigInteger.valueOf(84L))
 228                 .setPacketCount(BigInteger.valueOf(63L))
 229                 .setDurationSec(11L)
 230                 .setDurationNsec(12L)
 231                 .setRefCount(13L)
 232                 .setGroupId(new GroupId(groupIdValue));
 233         final MultipartReplyGroupBuilder mpReplyGroupBld = new MultipartReplyGroupBuilder();
 234         mpReplyGroupBld.setGroupStats(Lists.newArrayList(groupStatsBld.build()));
 235         final MultipartReplyGroupCaseBuilder mpReplyGroupCaseBld = new MultipartReplyGroupCaseBuilder();
 236         mpReplyGroupCaseBld.setMultipartReplyGroup(mpReplyGroupBld.build());
 237
 238         final MultipartReply groupStatsUpdated = assembleMPReplyMessage(type, mpReplyGroupCaseBld.build());
 239         final List<MultipartReply> statsData = Collections.singletonList(groupStatsUpdated);
 240
 241         fireAndCheck(type, statsData);
 242
 243         final InstanceIdentifier<GroupStatistics> groupPath = dummyNodePath.augmentation(FlowCapableNode.class)
 244                 .child(Group.class, new GroupKey(new org.opendaylight.yang.gen.v1.urn
 245                         .opendaylight.group.types.rev131018.GroupId(groupIdValue)))
 246                 .augmentation(NodeGroupStatistics.class)
 247                 .child(GroupStatistics.class);
 248         verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 249                 eq(groupPath), any(GroupStatistics.class));
 250     }
 251
 252     @Test
 253     public void testGatherStatistics_groupDesc() throws Exception {
 254         final MultipartType type = MultipartType.OFPMPGROUPDESC;
 255         final long groupIdValue = 27L;
 256
 257         final BucketsListBuilder bucketsListBld = new BucketsListBuilder()
 258                 .setWatchPort(new PortNumber(5L));
 259         final GroupDescBuilder groupStatsBld = new GroupDescBuilder()
 260                 .setBucketsList(Lists.newArrayList(bucketsListBld.build()))
 261                 .setGroupId(new GroupId(groupIdValue))
 262                 .setType(GroupType.OFPGTALL);
 263         final MultipartReplyGroupDescBuilder mpReplyGroupBld = new MultipartReplyGroupDescBuilder();
 264         mpReplyGroupBld.setGroupDesc(Lists.newArrayList(groupStatsBld.build()));
 265         final MultipartReplyGroupDescCaseBuilder mpReplyGroupCaseBld = new MultipartReplyGroupDescCaseBuilder();
 266         mpReplyGroupCaseBld.setMultipartReplyGroupDesc(mpReplyGroupBld.build());
 267
 268         final MultipartReply groupStatsUpdated = assembleMPReplyMessage(type, mpReplyGroupCaseBld.build());
 269         final List<MultipartReply> statsData = Collections.singletonList(groupStatsUpdated);
 270
 271         fireAndCheck(type, statsData);
 272
 273         final org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId storedGroupId =
 274                 new org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId(groupIdValue);
 275         final KeyedInstanceIdentifier<Group, GroupKey> groupPath =
 276                 dummyNodePath.augmentation(FlowCapableNode.class).child(Group.class, new GroupKey(storedGroupId));
 277
 278         verify(deviceContext, Mockito.never()).addDeleteToTxChain(eq(LogicalDatastoreType.OPERATIONAL),
 279                 ArgumentMatchers.<InstanceIdentifier<?>>any());
 280         verify(deviceGroupRegistry).store(storedGroupId);
 281         verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 282                                                  eq(groupPath), any(Group.class));
 283     }
 284
 285     @Test
 286     public void testGatherStatistics_meter() throws Exception {
 287         final MultipartType type = MultipartType.OFPMPMETER;
 288         final long meterIdValue = 19L;
 289
 290         final MeterBandStatsBuilder meterBandStatsBld = new MeterBandStatsBuilder()
 291                 .setByteBandCount(BigInteger.valueOf(91L))
 292                 .setPacketBandCount(BigInteger.valueOf(92L));
 293         final MeterStatsBuilder meterStatsBld = new MeterStatsBuilder()
 294                 .setMeterId(new MeterId(meterIdValue))
 295                 .setByteInCount(BigInteger.valueOf(111L))
 296                 .setDurationSec(112L)
 297                 .setDurationNsec(113L)
 298                 .setFlowCount(114L)
 299                 .setPacketInCount(BigInteger.valueOf(115L))
 300                 .setMeterBandStats(Lists.newArrayList(meterBandStatsBld.build()));
 301         final MultipartReplyMeterBuilder mpReplyMeterBld = new MultipartReplyMeterBuilder();
 302         mpReplyMeterBld.setMeterStats(Lists.newArrayList(meterStatsBld.build()));
 303         final MultipartReplyMeterCaseBuilder mpReplyMeterCaseBld = new MultipartReplyMeterCaseBuilder();
 304         mpReplyMeterCaseBld.setMultipartReplyMeter(mpReplyMeterBld.build());
 305
 306         final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyMeterCaseBld.build());
 307         final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
 308
 309         fireAndCheck(type, statsData);
 310
 311         final InstanceIdentifier<MeterStatistics> meterPath = dummyNodePath.augmentation(FlowCapableNode.class)
 312                 .child(Meter.class, new MeterKey(new org.opendaylight.yang.gen.v1.urn
 313                         .opendaylight.meter.types.rev130918.MeterId(meterIdValue)))
 314                 .augmentation(NodeMeterStatistics.class)
 315                 .child(MeterStatistics.class);
 316         verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 317                 eq(meterPath), any(MeterStatistics.class));
 318     }
 319
 320     @Test
 321     public void testGatherStatistics_nodeConnector() throws Exception {
 322         final MultipartType type = MultipartType.OFPMPPORTSTATS;
 323
 324         final PortStatsBuilder portStatsBld = new PortStatsBuilder()
 325                 .setPortNo(11L);
 326         final MultipartReplyPortStatsBuilder mpReplyMeterBld = new MultipartReplyPortStatsBuilder();
 327         mpReplyMeterBld.setPortStats(Lists.newArrayList(portStatsBld.build()));
 328         final MultipartReplyPortStatsCaseBuilder mpReplyMeterCaseBld = new MultipartReplyPortStatsCaseBuilder();
 329         mpReplyMeterCaseBld.setMultipartReplyPortStats(mpReplyMeterBld.build());
 330
 331         final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyMeterCaseBld.build());
 332         final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
 333
 334         fireAndCheck(type, statsData);
 335
 336         final InstanceIdentifier<FlowCapableNodeConnectorStatistics> portPath = dummyNodePath
 337                 .child(NodeConnector.class,
 338                         new NodeConnectorKey(new NodeConnectorId("openflow:" + DUMMY_NODE_ID_VALUE + ":11")))
 339                 .augmentation(FlowCapableNodeConnectorStatisticsData.class)
 340                 .child(FlowCapableNodeConnectorStatistics.class);
 341         verify(deviceContext).writeToTransaction(
 342                 eq(LogicalDatastoreType.OPERATIONAL),
 343                 eq(portPath),
 344                 any(FlowCapableNodeConnectorStatistics.class));
 345     }
 346
 347     @Test
 348     public void testGatherStatistics_table() throws Exception {
 349         final MultipartType type = MultipartType.OFPMPTABLE;
 350
 351         final TableStatsBuilder tableStatsBld = new TableStatsBuilder()
 352                 .setActiveCount(33L)
 353                 .setLookupCount(BigInteger.valueOf(34L))
 354                 .setMatchedCount(BigInteger.valueOf(35L))
 355                 .setTableId((short) 0);
 356         final MultipartReplyTableBuilder mpReplyTableBld = new MultipartReplyTableBuilder();
 357         mpReplyTableBld.setTableStats(Lists.newArrayList(tableStatsBld.build()));
 358         final MultipartReplyTableCaseBuilder mpReplyTableCaseBld = new MultipartReplyTableCaseBuilder();
 359         mpReplyTableCaseBld.setMultipartReplyTable(mpReplyTableBld.build());
 360
 361         final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyTableCaseBld.build());
 362         final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
 363
 364         fireAndCheck(type, statsData);
 365
 366         final InstanceIdentifier<FlowTableStatistics> tablePath = dummyNodePath
 367                 .augmentation(FlowCapableNode.class)
 368                 .child(Table.class, new TableKey((short) 0))
 369                 .augmentation(FlowTableStatisticsData.class)
 370                 .child(FlowTableStatistics.class);
 371         verify(deviceContext).writeToTransaction(
 372                 eq(LogicalDatastoreType.OPERATIONAL),
 373                 eq(tablePath),
 374                 any(FlowTableStatistics.class));
 375     }
 376
 377     @Test
 378     public void testGatherStatistics_queue() throws Exception {
 379         final MultipartType type = MultipartType.OFPMPQUEUE;
 380
 381         final long queueIdValue = 4L;
 382         final QueueStatsBuilder queueStatsBld = new QueueStatsBuilder()
 383                 .setPortNo(11L)
 384                 .setTxBytes(BigInteger.valueOf(44L))
 385                 .setTxErrors(BigInteger.valueOf(45L))
 386                 .setTxPackets(BigInteger.valueOf(46L))
 387                 .setDurationSec(47L)
 388                 .setDurationNsec(48L)
 389                 .setQueueId(queueIdValue);
 390
 391         final MultipartReplyQueueBuilder mpReplyQueueBld = new MultipartReplyQueueBuilder();
 392         mpReplyQueueBld.setQueueStats(Lists.newArrayList(queueStatsBld.build()));
 393         final MultipartReplyQueueCaseBuilder mpReplyQueueCaseBld = new MultipartReplyQueueCaseBuilder();
 394         mpReplyQueueCaseBld.setMultipartReplyQueue(mpReplyQueueBld.build());
 395
 396         final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyQueueCaseBld.build());
 397         final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
 398
 399         fireAndCheck(type, statsData);
 400
 401         final KeyedInstanceIdentifier<Queue, QueueKey> queuePath = dummyNodePath
 402                 .child(NodeConnector.class,
 403                         new NodeConnectorKey(new NodeConnectorId("openflow:" + DUMMY_NODE_ID_VALUE + ":11")))
 404                 .augmentation(FlowCapableNodeConnector.class)
 405                 .child(Queue.class, new QueueKey(new QueueId(queueIdValue)));
 406         verify(deviceContext).writeToTransaction(
 407                 eq(LogicalDatastoreType.OPERATIONAL),
 408                 eq(queuePath),
 409                 any(Queue.class));
 410     }
 411
 412     @Test
 413     public void testGatherStatistics_flow() throws Exception {
 414         final short tableId = 0;
 415         final MultipartType type = MultipartType.OFPMPFLOW;
 416
 417         final InstanceIdentifier<FlowCapableNode> nodePath =
 418                 deviceInfo.getNodeInstanceIdentifier().augmentation(FlowCapableNode.class);
 419         final TableBuilder tableDataBld = new TableBuilder();
 420         tableDataBld.setId(tableId);
 421         final FlowCapableNodeBuilder flowNodeBuilder = new FlowCapableNodeBuilder();
 422         flowNodeBuilder.setTable(Collections.singletonList(tableDataBld.build()));
 423         final Optional<FlowCapableNode> flowNodeOpt = Optional.of(flowNodeBuilder.build());
 424         doReturn(FluentFutures.immediateFluentFuture(flowNodeOpt)).when(readTx)
 425             .read(LogicalDatastoreType.OPERATIONAL, nodePath);
 426         when(flowDescriptor.getFlowId()).thenReturn(flowId);
 427
 428         final org.opendaylight.yang.gen.v1.urn
 429                 .opendaylight.openflow.oxm.rev150225.match.grouping.MatchBuilder matchBld =
 430                 new org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.grouping.MatchBuilder()
 431                         .setMatchEntry(Collections.emptyList());
 432         final FlowStatsBuilder flowStatsBld = new FlowStatsBuilder()
 433                 .setByteCount(BigInteger.valueOf(55L))
 434                 .setPacketCount(BigInteger.valueOf(56L))
 435                 .setDurationSec(57L)
 436                 .setDurationNsec(58L)
 437                 .setTableId((short) 0)
 438                 .setMatch(matchBld.build())
 439                 .setFlags(new FlowModFlags(true, false, false, false, true));
 440
 441         final MultipartReplyFlowBuilder mpReplyFlowBld = new MultipartReplyFlowBuilder();
 442         mpReplyFlowBld.setFlowStats(Lists.newArrayList(flowStatsBld.build()));
 443         final MultipartReplyFlowCaseBuilder mpReplyFlowCaseBld = new MultipartReplyFlowCaseBuilder();
 444         mpReplyFlowCaseBld.setMultipartReplyFlow(mpReplyFlowBld.build());
 445
 446         final MultipartReply flowStatsUpdated = assembleMPReplyMessage(type, mpReplyFlowCaseBld.build());
 447         final List<MultipartReply> statsData = Collections.singletonList(flowStatsUpdated);
 448         fireAndCheck(type, statsData);
 449
 450         final FlowBuilder flowBld = new FlowBuilder()
 451                 .setTableId((short) 0)
 452                 .setMatch(new MatchBuilder().build());
 453         final KeyedInstanceIdentifier<Table, TableKey> tablePath = dummyNodePath.augmentation(FlowCapableNode.class)
 454                 .child(Table.class, new TableKey((short) 0));
 455         final KeyedInstanceIdentifier<Flow, FlowKey> flowPath =  tablePath.child(Flow.class, new FlowKey(flowId));
 456
 457         verify(deviceContext, Mockito.never()).addDeleteToTxChain(eq(LogicalDatastoreType.OPERATIONAL),
 458                 ArgumentMatchers.<InstanceIdentifier<?>>any());
 459
 460         final InOrder inOrder = Mockito.inOrder(deviceContext);
 461         inOrder.verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 462                                                          any(),
 463                                                          any());
 464     }
 465
 466     @Test
 467     public void testGatherStatistics_meterConfig() throws Exception {
 468         final MultipartType type = MultipartType.OFPMPMETERCONFIG;
 469         final Long meterIdValue = 55L;
 470
 471         final MeterConfigBuilder meterConfigBld = new MeterConfigBuilder()
 472                 .setMeterId(new MeterId(meterIdValue))
 473                 .setFlags(new MeterFlags(false, true, false, true))
 474                 .setBands(Collections.emptyList());
 475
 476         final MultipartReplyMeterConfigBuilder mpReplyMeterConfigBld = new MultipartReplyMeterConfigBuilder();
 477         mpReplyMeterConfigBld.setMeterConfig(Lists.newArrayList(meterConfigBld.build()));
 478         final MultipartReplyMeterConfigCaseBuilder mpReplyMeterConfigCaseBld =
 479                 new MultipartReplyMeterConfigCaseBuilder();
 480         mpReplyMeterConfigCaseBld.setMultipartReplyMeterConfig(mpReplyMeterConfigBld.build());
 481
 482         final MultipartReply meterConfigUpdated = assembleMPReplyMessage(type, mpReplyMeterConfigCaseBld.build());
 483         final List<MultipartReply> statsData = Collections.singletonList(meterConfigUpdated);
 484
 485         fireAndCheck(type, statsData);
 486
 487         final org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId meterId =
 488                 new org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId(meterIdValue);
 489         final KeyedInstanceIdentifier<Meter, MeterKey> meterPath = dummyNodePath.augmentation(FlowCapableNode.class)
 490                 .child(Meter.class, new MeterKey(meterId));
 491         verify(deviceContext, Mockito.never()).addDeleteToTxChain(eq(LogicalDatastoreType.OPERATIONAL),
 492                 ArgumentMatchers.<InstanceIdentifier<?>>any());
 493         verify(deviceMeterRegistry).store(meterId);
 494         verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL),
 495                                                  eq(meterPath), any(Meter.class));
 496     }
 497
 498     private void fireAndCheck(final MultipartType type, final List<MultipartReply> statsData)
 499             throws InterruptedException, ExecutionException, TimeoutException {
 500         when(statisticsService.getStatisticsOfType(any(EventIdentifier.class),eq(type)))
 501                 .thenReturn(Futures.immediateFuture(RpcResultBuilder.success(statsData).build()));
 502
 503         final ListenableFuture<Boolean> gatherStatisticsResult = StatisticsGatheringUtils.gatherStatistics(
 504             statisticsService,
 505             deviceInfo,
 506             type,
 507             deviceContext,
 508             deviceContext,
 509             ConvertorManagerFactory.createDefaultManager(),
 510             provider,
 511             MoreExecutors.newDirectExecutorService());
 512
 513         Assert.assertTrue(gatherStatisticsResult.get(1, TimeUnit.SECONDS));
 514         verify(deviceContext).submitTransaction();
 515     }
 516
 517     private static MultipartReplyMessage assembleMPReplyMessage(final MultipartType type,
 518                                                                 final MultipartReplyBody mpReplyGroupCaseBld) {
 519         return new MultipartReplyMessageBuilder()
 520                 .setMultipartReplyBody(mpReplyGroupCaseBld)
 521                 .setType(type)
 522                 .setFlags(new MultipartRequestFlags(false))
 523                 .setXid(42L)
 524                 .build();
 525     }
 526
 527     private static BucketStats createBucketStat(final long byteCount, final long packetCount) {
 528         return new BucketStatsBuilder()
 529                 .setByteCount(BigInteger.valueOf(byteCount)).setPacketCount(BigInteger.valueOf(packetCount)).build();
 530     }
 531
 532     @Test
 533     public void testDeleteAllKnownFlows() throws Exception {
 534         final short tableId = 0;
 535         final InstanceIdentifier<FlowCapableNode> nodePath =
 536                 deviceInfo.getNodeInstanceIdentifier().augmentation(FlowCapableNode.class);
 537         final TableBuilder tableDataBld = new TableBuilder();
 538         tableDataBld.setId(tableId);
 539         final FlowCapableNodeBuilder flowNodeBuilder = new FlowCapableNodeBuilder();
 540         flowNodeBuilder.setTable(Collections.singletonList(tableDataBld.build()));
 541         final Optional<FlowCapableNode> flowNodeOpt = Optional.of(flowNodeBuilder.build());
 542         doReturn(FluentFutures.immediateFluentFuture(flowNodeOpt)).when(readTx)
 543             .read(LogicalDatastoreType.OPERATIONAL, nodePath);
 544         StatisticsGatheringUtils.deleteAllKnownFlows(deviceContext, deviceInfo.getNodeInstanceIdentifier()
 545             .augmentation(FlowCapableNode.class), deviceFlowRegistry);
 546
 547         verify(deviceContext).isTransactionsEnabled();
 548         verify(deviceContext).getReadTransaction();
 549         verify(deviceContext).writeToTransaction(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
 550     }
 551 }