2 * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.direct.singlelayer;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertTrue;
13 import static org.mockito.ArgumentMatchers.any;
14 import static org.mockito.ArgumentMatchers.eq;
15 import static org.mockito.Mockito.mock;
16 import static org.mockito.Mockito.verify;
17 import static org.mockito.Mockito.when;
19 import java.math.BigInteger;
20 import java.util.Collections;
21 import java.util.List;
22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
23 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
24 import org.opendaylight.openflowplugin.api.openflow.registry.flow.DeviceFlowRegistry;
25 import org.opendaylight.openflowplugin.impl.registry.flow.FlowDescriptorFactory;
26 import org.opendaylight.openflowplugin.impl.statistics.services.direct.AbstractDirectStatisticsServiceTest;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Counter32;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Counter64;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsInput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsOutput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowId;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowAndStatisticsMap;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapList;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapListBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.multipart.reply.multipart.reply.body.MultipartReplyFlowStatsBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.FlowModFlags;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.multipart.request.multipart.request.body.MultipartRequestFlowStats;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.statistics.types.rev130925.duration.DurationBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartReply;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartReplyBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartRequest;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
45 public class FlowDirectStatisticsServiceTest extends AbstractDirectStatisticsServiceTest {
46 static final Short TABLE_NO = 1;
47 private FlowDirectStatisticsService service;
50 public void setUp() throws Exception {
51 service = new FlowDirectStatisticsService(requestContextStack,
54 multipartWriterProvider);
55 final DeviceFlowRegistry registry = mock(DeviceFlowRegistry.class);
56 when(registry.retrieveDescriptor(any())).thenReturn(FlowDescriptorFactory.create(TABLE_NO, new FlowId("1")));
57 when(deviceContext.getDeviceFlowRegistry()).thenReturn(registry);
61 public void testBuildRequestBody() throws Exception {
62 final GetFlowStatisticsInput input = mock(GetFlowStatisticsInput.class);
64 when(input.getTableId()).thenReturn(TABLE_NO);
66 final MultipartRequestFlowStats body = (MultipartRequestFlowStats) ((MultipartRequest) service
67 .buildRequest(new Xid(42L), input))
68 .getMultipartRequestBody();
70 assertEquals(TABLE_NO, body.getFlowStats().getTableId());
74 public void testBuildReply() throws Exception {
75 final FlowAndStatisticsMapList flowStat = new FlowAndStatisticsMapListBuilder()
76 .setDuration(new DurationBuilder()
77 .setSecond(new Counter32(1L))
78 .setNanosecond(new Counter32(1L))
81 .setByteCount(new Counter64(BigInteger.ONE))
82 .setPacketCount(new Counter64(BigInteger.ONE))
83 .setFlags(new FlowModFlags(true, false, false, false, false))
84 .setMatch(new MatchBuilder().build())
87 final MultipartReply reply = new MultipartReplyBuilder()
88 .setMultipartReplyBody(new MultipartReplyFlowStatsBuilder()
89 .setFlowAndStatisticsMapList(Collections.singletonList(flowStat))
93 final List<MultipartReply> input = Collections.singletonList(reply);
95 final GetFlowStatisticsOutput output = service.buildReply(input, true);
96 assertTrue(output.getFlowAndStatisticsMapList().size() > 0);
98 final FlowAndStatisticsMap stats = output.getFlowAndStatisticsMapList().get(0);
100 assertEquals(stats.getTableId(), TABLE_NO);
104 public void testStoreStatistics() throws Exception {
105 final FlowAndStatisticsMapList stat = mock(FlowAndStatisticsMapList.class);
106 when(stat.getTableId()).thenReturn(TABLE_NO);
107 when(stat.getMatch()).thenReturn(new MatchBuilder().build());
109 final List<FlowAndStatisticsMapList> stats = Collections.singletonList(stat);
110 final GetFlowStatisticsOutput output = mock(GetFlowStatisticsOutput.class);
111 when(output.getFlowAndStatisticsMapList()).thenReturn(stats);
113 multipartWriterProvider.lookup(MultipartType.OFPMPFLOW).get().write(output, true);
114 verify(deviceContext).writeToTransactionWithParentsSlow(eq(LogicalDatastoreType.OPERATIONAL), any(), any());