3 * * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
12 package org.opendaylight.openflowplugin.impl.statistics;
14 import static org.mockito.Mockito.verify;
15 import static org.mockito.Mockito.when;
16 import com.google.common.base.Optional;
17 import com.google.common.collect.Lists;
18 import com.google.common.util.concurrent.CheckedFuture;
19 import com.google.common.util.concurrent.Futures;
20 import com.google.common.util.concurrent.ListenableFuture;
21 import java.math.BigInteger;
22 import java.util.Collections;
23 import java.util.List;
24 import java.util.concurrent.ExecutionException;
25 import java.util.concurrent.TimeUnit;
26 import java.util.concurrent.TimeoutException;
27 import org.junit.After;
28 import org.junit.Assert;
29 import org.junit.Before;
30 import org.junit.Test;
31 import org.junit.runner.RunWith;
32 import org.mockito.ArgumentCaptor;
33 import org.mockito.Matchers;
34 import org.mockito.Mock;
35 import org.mockito.Mockito;
36 import org.mockito.runners.MockitoJUnitRunner;
37 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
38 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
39 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
40 import org.opendaylight.openflowplugin.api.OFConstants;
41 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
42 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
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.FlowRegistryKey;
46 import org.opendaylight.openflowplugin.api.openflow.registry.group.DeviceGroupRegistry;
47 import org.opendaylight.openflowplugin.api.openflow.registry.meter.DeviceMeterRegistry;
48 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.EventIdentifier;
49 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.StatisticsGatherer;
50 import org.opendaylight.openflowplugin.impl.registry.flow.FlowRegistryKeyFactory;
51 import org.opendaylight.openflowplugin.openflow.md.util.OpenflowPortsUtil;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeBuilder;
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.yangtools.yang.binding.InstanceIdentifier;
132 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
133 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
136 @RunWith(MockitoJUnitRunner.class)
137 public class StatisticsGatheringUtilsTest {
139 public static final String DUMMY_NODE_ID_VALUE = "1";
140 public static final NodeId DUMMY_NODE_ID = new NodeId(DUMMY_NODE_ID_VALUE);
142 private final KeyedInstanceIdentifier<Node, NodeKey> dummyNodePath = InstanceIdentifier.create(Nodes.class)
143 .child(Node.class, new NodeKey(DUMMY_NODE_ID));
145 private DeviceContext deviceContext;
147 private DeviceState deviceState;
149 private DeviceFlowRegistry deviceFlowRegistry;
151 private DeviceGroupRegistry deviceGroupRegistry;
153 private DeviceMeterRegistry deviceMeterRegistry;
155 private GetFeaturesOutput features;
157 private ReadOnlyTransaction readTx;
159 private ConnectionContext connectionAdapter;
161 private StatisticsGatherer statisticsService;
163 public StatisticsGatheringUtilsTest() {
164 OpenflowPortsUtil.init();
168 public void setUp() throws Exception {
169 when(deviceContext.getDeviceState()).thenReturn(deviceState);
170 when(deviceContext.getDeviceFlowRegistry()).thenReturn(deviceFlowRegistry);
171 when(deviceContext.getDeviceGroupRegistry()).thenReturn(deviceGroupRegistry);
172 when(deviceContext.getDeviceMeterRegistry()).thenReturn(deviceMeterRegistry);
173 when(deviceContext.getReadTransaction()).thenReturn(readTx);
174 when(deviceContext.getPrimaryConnectionContext()).thenReturn(connectionAdapter);
175 when(connectionAdapter.getNodeId()).thenReturn(DUMMY_NODE_ID);
176 when(deviceState.getFeatures()).thenReturn(features);
177 when(connectionAdapter.getFeatures()).thenReturn(features);
179 when(features.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_3);
180 when(features.getDatapathId()).thenReturn(BigInteger.ONE);
182 when(deviceState.getNodeInstanceIdentifier()).thenReturn(dummyNodePath);
186 public void tearDown() throws Exception {
191 public void testWriteFlowStatistics() {
192 final ArgumentCaptor<LogicalDatastoreType> dataStoreType = ArgumentCaptor.forClass(LogicalDatastoreType.class);
193 final ArgumentCaptor<InstanceIdentifier> flowPath = ArgumentCaptor.forClass(InstanceIdentifier.class);
194 final ArgumentCaptor<Flow> flow = ArgumentCaptor.forClass(Flow.class);
196 StatisticsGatheringUtils.writeFlowStatistics(prepareFlowStatisticsData(), deviceContext);
198 Mockito.verify(deviceContext).writeToTransaction(
199 dataStoreType.capture(), flowPath.capture(), flow.capture());
200 Assert.assertEquals(LogicalDatastoreType.OPERATIONAL, dataStoreType.getValue());
201 final InstanceIdentifier<FlowCapableNode> flowCapableNodePath = flowPath.getValue();
202 Assert.assertEquals(DUMMY_NODE_ID, flowCapableNodePath.firstKeyOf(Node.class, NodeKey.class).getId());
203 Assert.assertEquals(42, flow.getValue().getTableId().intValue());
206 private Iterable<FlowsStatisticsUpdate> prepareFlowStatisticsData() {
207 final FlowAndStatisticsMapListBuilder flowAndStatsMapListBld = new FlowAndStatisticsMapListBuilder();
208 flowAndStatsMapListBld.setTableId((short) 42);
209 flowAndStatsMapListBld.setMatch(new MatchBuilder().build());
211 final FlowsStatisticsUpdateBuilder flowStatsUpdateBld1 = new FlowsStatisticsUpdateBuilder();
212 flowStatsUpdateBld1.setFlowAndStatisticsMapList(Lists.newArrayList(flowAndStatsMapListBld.build()));
214 return Lists.newArrayList(flowStatsUpdateBld1.build());
219 public void testGatherStatistics_group() throws Exception {
220 final MultipartType type = MultipartType.OFPMPGROUP;
221 final long groupIdValue = 19L;
223 final GroupStatsBuilder groupStatsBld = new GroupStatsBuilder()
224 .setBucketStats(Lists.newArrayList(createBucketStat(21L, 42L)))
225 .setByteCount(BigInteger.valueOf(84L))
226 .setPacketCount(BigInteger.valueOf(63L))
228 .setDurationNsec(12L)
230 .setGroupId(new GroupId(groupIdValue));
231 final MultipartReplyGroupBuilder mpReplyGroupBld = new MultipartReplyGroupBuilder();
232 mpReplyGroupBld.setGroupStats(Lists.newArrayList(groupStatsBld.build()));
233 final MultipartReplyGroupCaseBuilder mpReplyGroupCaseBld = new MultipartReplyGroupCaseBuilder();
234 mpReplyGroupCaseBld.setMultipartReplyGroup(mpReplyGroupBld.build());
236 final MultipartReply groupStatsUpdated = assembleMPReplyMessage(type, mpReplyGroupCaseBld.build());
237 final List<MultipartReply> statsData = Collections.singletonList(groupStatsUpdated);
239 fireAndCheck(type, statsData);
241 final InstanceIdentifier<GroupStatistics> groupPath = dummyNodePath.augmentation(FlowCapableNode.class)
242 .child(Group.class, new GroupKey(new org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId(groupIdValue)))
243 .augmentation(NodeGroupStatistics.class)
244 .child(GroupStatistics.class);
245 verify(deviceContext).writeToTransaction(Matchers.eq(LogicalDatastoreType.OPERATIONAL),
246 Matchers.eq(groupPath), Matchers.any(GroupStatistics.class));
250 public void testGatherStatistics_groupDesc() throws Exception {
251 final MultipartType type = MultipartType.OFPMPGROUPDESC;
252 final long groupIdValue = 27L;
254 final BucketsListBuilder bucketsListBld = new BucketsListBuilder()
255 .setWatchPort(new PortNumber(5L));
256 final GroupDescBuilder groupStatsBld = new GroupDescBuilder()
257 .setBucketsList(Lists.newArrayList(bucketsListBld.build()))
258 .setGroupId(new GroupId(groupIdValue))
259 .setType(GroupType.OFPGTALL);
260 final MultipartReplyGroupDescBuilder mpReplyGroupBld = new MultipartReplyGroupDescBuilder();
261 mpReplyGroupBld.setGroupDesc(Lists.newArrayList(groupStatsBld.build()));
262 final MultipartReplyGroupDescCaseBuilder mpReplyGroupCaseBld = new MultipartReplyGroupDescCaseBuilder();
263 mpReplyGroupCaseBld.setMultipartReplyGroupDesc(mpReplyGroupBld.build());
265 final MultipartReply groupStatsUpdated = assembleMPReplyMessage(type, mpReplyGroupCaseBld.build());
266 final List<MultipartReply> statsData = Collections.singletonList(groupStatsUpdated);
268 fireAndCheck(type, statsData);
270 final 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);
271 final KeyedInstanceIdentifier<Group, GroupKey> groupPath = dummyNodePath.augmentation(FlowCapableNode.class).child(Group.class, new GroupKey(storedGroupId));
273 verify(deviceContext, Mockito.never()).addDeleteToTxChain(Matchers.eq(LogicalDatastoreType.OPERATIONAL), Matchers.any(InstanceIdentifier.class));
274 verify(deviceGroupRegistry).removeMarked();
275 verify(deviceGroupRegistry).store(storedGroupId);
276 verify(deviceContext).writeToTransaction(
277 Matchers.eq(LogicalDatastoreType.OPERATIONAL), Matchers.eq(groupPath), Matchers.any(Group.class));
281 public void testGatherStatistics_meter() throws Exception {
282 final MultipartType type = MultipartType.OFPMPMETER;
283 final long meterIdValue = 19L;
285 final MeterBandStatsBuilder meterBandStatsBld = new MeterBandStatsBuilder()
286 .setByteBandCount(BigInteger.valueOf(91L))
287 .setPacketBandCount(BigInteger.valueOf(92L));
288 final MeterStatsBuilder meterStatsBld = new MeterStatsBuilder()
289 .setMeterId(new MeterId(meterIdValue))
290 .setByteInCount(BigInteger.valueOf(111L))
291 .setDurationSec(112L)
292 .setDurationNsec(113L)
294 .setPacketInCount(BigInteger.valueOf(115L))
295 .setMeterBandStats(Lists.newArrayList(meterBandStatsBld.build()));
296 final MultipartReplyMeterBuilder mpReplyMeterBld = new MultipartReplyMeterBuilder();
297 mpReplyMeterBld.setMeterStats(Lists.newArrayList(meterStatsBld.build()));
298 final MultipartReplyMeterCaseBuilder mpReplyMeterCaseBld = new MultipartReplyMeterCaseBuilder();
299 mpReplyMeterCaseBld.setMultipartReplyMeter(mpReplyMeterBld.build());
301 final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyMeterCaseBld.build());
302 final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
304 fireAndCheck(type, statsData);
306 final InstanceIdentifier<MeterStatistics> meterPath = dummyNodePath.augmentation(FlowCapableNode.class)
307 .child(Meter.class, new MeterKey(new org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId(meterIdValue)))
308 .augmentation(NodeMeterStatistics.class)
309 .child(MeterStatistics.class);
310 verify(deviceContext).writeToTransaction(Matchers.eq(LogicalDatastoreType.OPERATIONAL),
311 Matchers.eq(meterPath), Matchers.any(MeterStatistics.class));
315 public void testGatherStatistics_nodeConnector() throws Exception {
316 final MultipartType type = MultipartType.OFPMPPORTSTATS;
318 final PortStatsBuilder portStatsBld = new PortStatsBuilder()
320 final MultipartReplyPortStatsBuilder mpReplyMeterBld = new MultipartReplyPortStatsBuilder();
321 mpReplyMeterBld.setPortStats(Lists.newArrayList(portStatsBld.build()));
322 final MultipartReplyPortStatsCaseBuilder mpReplyMeterCaseBld = new MultipartReplyPortStatsCaseBuilder();
323 mpReplyMeterCaseBld.setMultipartReplyPortStats(mpReplyMeterBld.build());
325 final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyMeterCaseBld.build());
326 final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
328 fireAndCheck(type, statsData);
330 final InstanceIdentifier<FlowCapableNodeConnectorStatistics> portPath = dummyNodePath
331 .child(NodeConnector.class, new NodeConnectorKey(new NodeConnectorId("openflow:" + DUMMY_NODE_ID_VALUE + ":11")))
332 .augmentation(FlowCapableNodeConnectorStatisticsData.class)
333 .child(FlowCapableNodeConnectorStatistics.class);
334 verify(deviceContext).writeToTransaction(
335 Matchers.eq(LogicalDatastoreType.OPERATIONAL),
336 Matchers.eq(portPath),
337 Matchers.any(FlowCapableNodeConnectorStatistics.class));
341 public void testGatherStatistics_table() throws Exception {
342 final MultipartType type = MultipartType.OFPMPTABLE;
344 final TableStatsBuilder tableStatsBld = new TableStatsBuilder()
346 .setLookupCount(BigInteger.valueOf(34L))
347 .setMatchedCount(BigInteger.valueOf(35L))
348 .setTableId((short) 0);
349 final MultipartReplyTableBuilder mpReplyTableBld = new MultipartReplyTableBuilder();
350 mpReplyTableBld.setTableStats(Lists.newArrayList(tableStatsBld.build()));
351 final MultipartReplyTableCaseBuilder mpReplyTableCaseBld = new MultipartReplyTableCaseBuilder();
352 mpReplyTableCaseBld.setMultipartReplyTable(mpReplyTableBld.build());
354 final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyTableCaseBld.build());
355 final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
357 fireAndCheck(type, statsData);
359 final InstanceIdentifier<FlowTableStatistics> tablePath = dummyNodePath
360 .augmentation(FlowCapableNode.class)
361 .child(Table.class, new TableKey((short) 0))
362 .augmentation(FlowTableStatisticsData.class)
363 .child(FlowTableStatistics.class);
364 verify(deviceContext).writeToTransaction(
365 Matchers.eq(LogicalDatastoreType.OPERATIONAL),
366 Matchers.eq(tablePath),
367 Matchers.any(FlowTableStatistics.class));
371 public void testGatherStatistics_queue() throws Exception {
372 final MultipartType type = MultipartType.OFPMPQUEUE;
374 final long queueIdValue = 4L;
375 final QueueStatsBuilder queueStatsBld = new QueueStatsBuilder()
377 .setTxBytes(BigInteger.valueOf(44L))
378 .setTxErrors(BigInteger.valueOf(45L))
379 .setTxPackets(BigInteger.valueOf(46L))
381 .setDurationNsec(48L)
382 .setQueueId(queueIdValue);
384 final MultipartReplyQueueBuilder mpReplyQueueBld = new MultipartReplyQueueBuilder();
385 mpReplyQueueBld.setQueueStats(Lists.newArrayList(queueStatsBld.build()));
386 final MultipartReplyQueueCaseBuilder mpReplyQueueCaseBld = new MultipartReplyQueueCaseBuilder();
387 mpReplyQueueCaseBld.setMultipartReplyQueue(mpReplyQueueBld.build());
389 final MultipartReply meterStatsUpdated = assembleMPReplyMessage(type, mpReplyQueueCaseBld.build());
390 final List<MultipartReply> statsData = Collections.singletonList(meterStatsUpdated);
392 fireAndCheck(type, statsData);
394 final KeyedInstanceIdentifier<Queue, QueueKey> queuePath = dummyNodePath
395 .child(NodeConnector.class, new NodeConnectorKey(new NodeConnectorId("openflow:" + DUMMY_NODE_ID_VALUE + ":11")))
396 .augmentation(FlowCapableNodeConnector.class)
397 .child(Queue.class, new QueueKey(new QueueId(queueIdValue)));
398 verify(deviceContext).writeToTransaction(
399 Matchers.eq(LogicalDatastoreType.OPERATIONAL),
400 Matchers.eq(queuePath),
401 Matchers.any(Queue.class));
405 public void testGatherStatistics_flow() throws Exception {
406 final MultipartType type = MultipartType.OFPMPFLOW;
407 when(deviceFlowRegistry.storeIfNecessary(Matchers.any(FlowRegistryKey.class), Matchers.anyShort()))
408 .thenReturn(new FlowId("openflow:21"));
410 final org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.grouping.MatchBuilder matchBld =
411 new org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev150225.match.grouping.MatchBuilder()
412 .setMatchEntry(Collections.<MatchEntry>emptyList());
413 final FlowStatsBuilder flowStatsBld = new FlowStatsBuilder()
414 .setByteCount(BigInteger.valueOf(55L))
415 .setPacketCount(BigInteger.valueOf(56L))
417 .setDurationNsec(58L)
418 .setTableId((short) 0)
419 .setMatch(matchBld.build())
420 .setFlags(new FlowModFlags(true, false, false, false, true));
422 final MultipartReplyFlowBuilder mpReplyFlowBld = new MultipartReplyFlowBuilder();
423 mpReplyFlowBld.setFlowStats(Lists.newArrayList(flowStatsBld.build()));
424 final MultipartReplyFlowCaseBuilder mpReplyFlowCaseBld = new MultipartReplyFlowCaseBuilder();
425 mpReplyFlowCaseBld.setMultipartReplyFlow(mpReplyFlowBld.build());
427 final MultipartReply flowStatsUpdated = assembleMPReplyMessage(type, mpReplyFlowCaseBld.build());
428 final List<MultipartReply> statsData = Collections.singletonList(flowStatsUpdated);
429 fireAndCheck(type, statsData);
431 final FlowBuilder flowBld = new FlowBuilder()
432 .setTableId((short) 0)
433 .setMatch(new MatchBuilder().build());
434 final KeyedInstanceIdentifier<Flow, FlowKey> flowPath = dummyNodePath.augmentation(FlowCapableNode.class)
435 .child(Table.class, new TableKey((short) 0))
436 .child(Flow.class, new FlowKey(new FlowId("openflow:21")));
437 verify(deviceContext, Mockito.never()).addDeleteToTxChain(Matchers.eq(LogicalDatastoreType.OPERATIONAL), Matchers.any(InstanceIdentifier.class));
438 verify(deviceFlowRegistry).storeIfNecessary(FlowRegistryKeyFactory.create(flowBld.build()), (short) 0);
439 verify(deviceContext).writeToTransaction(Matchers.eq(LogicalDatastoreType.OPERATIONAL), Matchers.eq(flowPath), Matchers.any(Flow.class));
443 public void testGatherStatistics_meterConfig() throws Exception {
444 final MultipartType type = MultipartType.OFPMPMETERCONFIG;
445 final Long meterIdValue = 55L;
447 final MeterConfigBuilder meterConfigBld = new MeterConfigBuilder()
448 .setMeterId(new MeterId(meterIdValue))
449 .setFlags(new MeterFlags(false, true, false, true))
450 .setBands(Collections.<Bands>emptyList());
452 final MultipartReplyMeterConfigBuilder mpReplyMeterConfigBld = new MultipartReplyMeterConfigBuilder();
453 mpReplyMeterConfigBld.setMeterConfig(Lists.newArrayList(meterConfigBld.build()));
454 final MultipartReplyMeterConfigCaseBuilder mpReplyMeterConfigCaseBld = new MultipartReplyMeterConfigCaseBuilder();
455 mpReplyMeterConfigCaseBld.setMultipartReplyMeterConfig(mpReplyMeterConfigBld.build());
457 final MultipartReply meterConfigUpdated = assembleMPReplyMessage(type, mpReplyMeterConfigCaseBld.build());
458 final List<MultipartReply> statsData = Collections.singletonList(meterConfigUpdated);
460 fireAndCheck(type, statsData);
462 final org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId meterId =
463 new org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId(meterIdValue);
464 final KeyedInstanceIdentifier<Meter, MeterKey> meterPath = dummyNodePath.augmentation(FlowCapableNode.class)
465 .child(Meter.class, new MeterKey(meterId));
466 verify(deviceContext, Mockito.never()).addDeleteToTxChain(Matchers.eq(LogicalDatastoreType.OPERATIONAL), Matchers.any(InstanceIdentifier.class));
467 verify(deviceMeterRegistry).store(meterId);
468 verify(deviceContext).writeToTransaction(Matchers.eq(LogicalDatastoreType.OPERATIONAL), Matchers.eq(meterPath), Matchers.any(Meter.class));
471 private void fireAndCheck(final MultipartType type, final List<MultipartReply> statsData) throws InterruptedException, ExecutionException, TimeoutException {
472 when(statisticsService.getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.eq(type)))
473 .thenReturn(Futures.immediateFuture(RpcResultBuilder.success(statsData).build()));
475 final ListenableFuture<Boolean> gatherStatisticsResult = StatisticsGatheringUtils.gatherStatistics(statisticsService, deviceContext, type);
476 Assert.assertTrue(gatherStatisticsResult.get(1, TimeUnit.SECONDS).booleanValue());
477 verify(deviceContext).submitTransaction();
480 private static MultipartReplyMessage assembleMPReplyMessage(final MultipartType type, final MultipartReplyBody mpReplyGroupCaseBld) {
481 return new MultipartReplyMessageBuilder()
482 .setMultipartReplyBody(mpReplyGroupCaseBld)
484 .setFlags(new MultipartRequestFlags(false))
489 private static BucketStats createBucketStat(final long byteCount, final long packetCount) {
490 return new BucketStatsBuilder().setByteCount(BigInteger.valueOf(byteCount)).setPacketCount(BigInteger.valueOf(packetCount)).build();
494 public void testDeleteAllKnownFlowsNotSync() throws Exception {
495 when(deviceState.deviceSynchronized()).thenReturn(false);
496 StatisticsGatheringUtils.deleteAllKnownFlows(deviceContext);
497 Mockito.verifyNoMoreInteractions(deviceFlowRegistry);
501 public void testDeleteAllKnownFlows() throws Exception {
502 final short tableId = 0;
503 when(deviceState.deviceSynchronized()).thenReturn(true);
504 final InstanceIdentifier<FlowCapableNode> nodePath = deviceState.getNodeInstanceIdentifier().augmentation(FlowCapableNode.class);
505 final TableBuilder tableDataBld = new TableBuilder();
506 tableDataBld.setId(tableId);
507 final FlowCapableNodeBuilder flowNodeBuilder = new FlowCapableNodeBuilder();
508 flowNodeBuilder.setTable(Collections.singletonList(tableDataBld.build()));
509 final Optional<FlowCapableNode> flowNodeOpt = Optional.of(flowNodeBuilder.build());
510 final CheckedFuture<Optional<FlowCapableNode>, ReadFailedException> flowNodeFuture = Futures.immediateCheckedFuture(flowNodeOpt);
511 when(readTx.read(LogicalDatastoreType.OPERATIONAL, nodePath)).thenReturn(flowNodeFuture);
512 final KeyedInstanceIdentifier<Table, TableKey> tablePath = deviceState.getNodeInstanceIdentifier()
513 .augmentation(FlowCapableNode.class).child(Table.class, new TableKey(tableId));
515 StatisticsGatheringUtils.deleteAllKnownFlows(deviceContext);
517 verify(deviceContext).writeToTransaction(
518 LogicalDatastoreType.OPERATIONAL,
520 tableDataBld.setFlow(Collections.<Flow>emptyList()).build());