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

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