2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.openflowplugin.impl.statistics.services;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.ArgumentMatchers.eq;
14 import com.google.common.util.concurrent.FutureCallback;
15 import java.math.BigInteger;
16 import java.util.Collections;
17 import java.util.concurrent.Future;
18 import java.util.concurrent.atomic.AtomicLong;
19 import org.junit.After;
20 import org.junit.Assert;
21 import org.junit.Test;
22 import org.mockito.ArgumentCaptor;
23 import org.mockito.ArgumentMatchers;
24 import org.mockito.Captor;
25 import org.mockito.Mockito;
26 import org.opendaylight.openflowplugin.api.OFConstants;
27 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManager;
28 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManagerFactory;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetAllMeterConfigStatisticsInputBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetAllMeterConfigStatisticsOutput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetAllMeterStatisticsInputBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetAllMeterStatisticsOutput;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetMeterFeaturesInputBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetMeterFeaturesOutput;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetMeterStatisticsInputBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetMeterStatisticsOutput;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterBandType;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterBandTypeBitmap;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterFlags;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessageBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInput;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.MeterBandDropCaseBuilder;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.meter.band.header.meter.band.meter.band.drop._case.MeterBandDropBuilder;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterCaseBuilder;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterConfigCaseBuilder;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterFeaturesCaseBuilder;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter._case.MultipartReplyMeterBuilder;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter._case.multipart.reply.meter.MeterStatsBuilder;
51 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;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.config._case.MultipartReplyMeterConfigBuilder;
53 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;
54 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.BandsBuilder;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.features._case.MultipartReplyMeterFeaturesBuilder;
56 import org.opendaylight.yangtools.yang.binding.Notification;
57 import org.opendaylight.yangtools.yang.common.RpcResult;
58 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
61 * Test for {@link OpendaylightMeterStatisticsServiceImpl}.
63 public class OpendaylightMeterStatisticsServiceImplTest extends AbstractSingleStatsServiceTest {
65 private static final org.opendaylight.yang.gen.v1.urn
66 .opendaylight.openflow.common.types.rev130731.MeterId METER_ID =
67 new org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterId(123L);
69 private ArgumentCaptor<MultipartRequestInput> requestInput;
71 private OpendaylightMeterStatisticsServiceImpl meterStatisticsService;
74 final ConvertorManager convertorManager = ConvertorManagerFactory.createDefaultManager();
75 meterStatisticsService = new OpendaylightMeterStatisticsServiceImpl(rqContextStack, deviceContext,
76 new AtomicLong(), notificationPublishService, convertorManager);
78 Mockito.doAnswer(answerVoidToCallback).when(outboundQueueProvider)
79 .commitEntry(eq(42L), requestInput.capture(), any(FutureCallback.class));
83 public void tearDown() throws Exception {
84 Mockito.verify(notificationPublishService).offerNotification(ArgumentMatchers.<Notification>any());
88 public void testGetAllMeterConfigStatistics() throws Exception {
89 GetAllMeterConfigStatisticsInputBuilder input = new GetAllMeterConfigStatisticsInputBuilder()
90 .setNode(createNodeRef("unitProt:123"));
92 rpcResult = RpcResultBuilder.<Object>success(Collections.singletonList(
93 new MultipartReplyMessageBuilder()
94 .setVersion(OFConstants.OFP_VERSION_1_3)
95 .setMultipartReplyBody(new MultipartReplyMeterConfigCaseBuilder()
96 .setMultipartReplyMeterConfig(new MultipartReplyMeterConfigBuilder()
97 .setMeterConfig(Collections.singletonList(new MeterConfigBuilder()
98 .setFlags(new MeterFlags(true, false, false, false))
100 .setBands(Collections.singletonList(new BandsBuilder()
101 .setMeterBand(new MeterBandDropCaseBuilder()
102 .setMeterBandDrop(new MeterBandDropBuilder()
105 .setType(MeterBandType.OFPMBTDROP)
115 final Future<RpcResult<GetAllMeterConfigStatisticsOutput>> resultFuture
116 = meterStatisticsService.getAllMeterConfigStatistics(input.build());
118 Assert.assertTrue(resultFuture.isDone());
119 final RpcResult<GetAllMeterConfigStatisticsOutput> rpcResult = resultFuture.get();
120 Assert.assertTrue(rpcResult.isSuccessful());
121 Assert.assertEquals(MultipartType.OFPMPMETERCONFIG, requestInput.getValue().getType());
125 public void testGetAllMeterStatistics() throws Exception {
126 GetAllMeterStatisticsInputBuilder input = new GetAllMeterStatisticsInputBuilder()
127 .setNode(createNodeRef("unitProt:123"));
129 rpcResult = buildMeterStatisticsReply();
131 final Future<RpcResult<GetAllMeterStatisticsOutput>> resultFuture
132 = meterStatisticsService.getAllMeterStatistics(input.build());
134 Assert.assertTrue(resultFuture.isDone());
135 final RpcResult<GetAllMeterStatisticsOutput> rpcResult = resultFuture.get();
136 Assert.assertTrue(rpcResult.isSuccessful());
137 Assert.assertEquals(MultipartType.OFPMPMETER, requestInput.getValue().getType());
141 public void testGetMeterFeatures() throws Exception {
142 GetMeterFeaturesInputBuilder input = new GetMeterFeaturesInputBuilder()
143 .setNode(createNodeRef("unitProt:123"));
145 rpcResult = RpcResultBuilder.<Object>success(Collections.singletonList(
146 new MultipartReplyMessageBuilder()
147 .setVersion(OFConstants.OFP_VERSION_1_3)
148 .setMultipartReplyBody(new MultipartReplyMeterFeaturesCaseBuilder()
149 .setMultipartReplyMeterFeatures(new MultipartReplyMeterFeaturesBuilder()
150 .setBandTypes(new MeterBandTypeBitmap(true, false))
151 .setCapabilities(new MeterFlags(true, false, false, false))
152 .setMaxBands((short) 71)
153 .setMaxColor((short) 72)
160 final Future<RpcResult<GetMeterFeaturesOutput>> resultFuture
161 = meterStatisticsService.getMeterFeatures(input.build());
163 Assert.assertTrue(resultFuture.isDone());
164 final RpcResult<GetMeterFeaturesOutput> rpcResult = resultFuture.get();
165 Assert.assertTrue(rpcResult.isSuccessful());
166 Assert.assertEquals(MultipartType.OFPMPMETERFEATURES, requestInput.getValue().getType());
170 public void testGetMeterStatistics() throws Exception {
171 GetMeterStatisticsInputBuilder input = new GetMeterStatisticsInputBuilder()
172 .setNode(createNodeRef("unitProt:123"))
173 .setMeterId(new MeterId(21L));
175 rpcResult = buildMeterStatisticsReply();
177 final Future<RpcResult<GetMeterStatisticsOutput>> resultFuture
178 = meterStatisticsService.getMeterStatistics(input.build());
180 Assert.assertTrue(resultFuture.isDone());
181 final RpcResult<GetMeterStatisticsOutput> rpcResult = resultFuture.get();
182 Assert.assertTrue(rpcResult.isSuccessful());
183 Assert.assertEquals(MultipartType.OFPMPMETER, requestInput.getValue().getType());
186 protected RpcResult<Object> buildMeterStatisticsReply() {
187 return RpcResultBuilder.<Object>success(Collections.singletonList(
188 new MultipartReplyMessageBuilder()
189 .setVersion(OFConstants.OFP_VERSION_1_3)
190 .setMultipartReplyBody(new MultipartReplyMeterCaseBuilder()
191 .setMultipartReplyMeter(new MultipartReplyMeterBuilder()
192 .setMeterStats(Collections.singletonList(new MeterStatsBuilder()
193 .setMeterId(METER_ID)
194 .setByteInCount(BigInteger.valueOf(81L))
196 .setDurationNsec(83L)
198 .setPacketInCount(BigInteger.valueOf(85L))
199 .setMeterBandStats(Collections.singletonList(new MeterBandStatsBuilder()
200 .setByteBandCount(BigInteger.valueOf(86L))
201 .setPacketBandCount(BigInteger.valueOf(87L))