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