1 package test.mock.util;
3 import java.math.BigInteger;
4 import java.util.ArrayList;
6 import java.util.concurrent.Future;
8 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsUpdateBuilder;
9 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.GetFlowTablesStatisticsInput;
10 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.GetFlowTablesStatisticsOutput;
11 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.GetFlowTablesStatisticsOutputBuilder;
12 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.OpendaylightFlowTableStatisticsService;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.and.statistics.map.FlowTableAndStatisticsMap;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.and.statistics.map.FlowTableAndStatisticsMapBuilder;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.and.statistics.map.FlowTableAndStatisticsMapKey;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.TransactionId;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
19 import org.opendaylight.yangtools.yang.common.RpcResult;
20 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
22 import com.google.common.util.concurrent.Futures;
24 public class OpendaylightFlowTableStatisticsServiceMock implements OpendaylightFlowTableStatisticsService {
25 NotificationProviderServiceHelper notifService;
27 public OpendaylightFlowTableStatisticsServiceMock(NotificationProviderServiceHelper notifService) {
28 this.notifService = notifService;
32 public Future<RpcResult<GetFlowTablesStatisticsOutput>> getFlowTablesStatistics(GetFlowTablesStatisticsInput input) {
33 GetFlowTablesStatisticsOutputBuilder builder = new GetFlowTablesStatisticsOutputBuilder();
34 TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
35 builder.setTransactionId(transId);
36 FlowTableStatisticsUpdateBuilder ftsBuilder = new FlowTableStatisticsUpdateBuilder();
37 FlowTableAndStatisticsMapBuilder ftasmBuilder = new FlowTableAndStatisticsMapBuilder();
38 List<FlowTableAndStatisticsMap> tableAndStatisticsMaps = new ArrayList<>();
39 ftasmBuilder.setKey(new FlowTableAndStatisticsMapKey(StatisticsManagerTest.getTableId()));
40 ftasmBuilder.setActiveFlows(StatisticsManagerTest.COUNTER_32_TEST_VALUE);
41 tableAndStatisticsMaps.add(ftasmBuilder.build());
42 ftsBuilder.setTransactionId(transId);
43 ftsBuilder.setId(input.getNode().getValue().firstKeyOf(Node.class, NodeKey.class).getId());
44 ftsBuilder.setFlowTableAndStatisticsMap(tableAndStatisticsMaps);
45 ftsBuilder.setMoreReplies(true);
46 notifService.pushDelayedNotification(ftsBuilder.build(), 0); // 1st notification
47 ftsBuilder.setMoreReplies(false);
48 ftasmBuilder.setPacketsLookedUp(StatisticsManagerTest.COUNTER_64_TEST_VALUE);
49 tableAndStatisticsMaps.clear();
50 tableAndStatisticsMaps.add(ftasmBuilder.build());
51 ftsBuilder.setFlowTableAndStatisticsMap(tableAndStatisticsMaps);
52 notifService.pushDelayedNotification(ftsBuilder.build(), 0); // 2nd notification
53 return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());