applications/statistics-manager/src/test/java/test/mock/util/OpendaylightFlowStatisticsServiceMock.java

   1 /*
   2  * Copyright (c) 2014, 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package test.mock.util;
   9
  10 import java.math.BigInteger;
  11 import java.util.ArrayList;
  12 import java.util.List;
  13 import java.util.concurrent.Future;
  14
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.AggregateFlowStatisticsUpdateBuilder;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowsStatisticsUpdateBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForAllFlowsInput;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutput;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutputBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForGivenMatchInput;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutput;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutputBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowStatisticsFromFlowTableInput;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowStatisticsFromFlowTableOutput;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowStatisticsFromFlowTableOutputBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowsStatisticsFromAllFlowTablesInput;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowsStatisticsFromAllFlowTablesOutput;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowsStatisticsFromAllFlowTablesOutputBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetFlowStatisticsFromFlowTableInput;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetFlowStatisticsFromFlowTableOutput;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetFlowStatisticsFromFlowTableOutputBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.OpendaylightFlowStatisticsService;
  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.transaction.rev150304.TransactionId;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  38 import org.opendaylight.yangtools.yang.common.RpcResult;
  39 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  40
  41 import com.google.common.util.concurrent.Futures;
  42
  43 public class OpendaylightFlowStatisticsServiceMock implements OpendaylightFlowStatisticsService {
  44     NotificationProviderServiceHelper notifService;
  45
  46     public OpendaylightFlowStatisticsServiceMock(NotificationProviderServiceHelper notifService) {
  47         this.notifService = notifService;
  48     }
  49
  50     @Override
  51     public Future<RpcResult<GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutput>> getAggregateFlowStatisticsFromFlowTableForAllFlows(GetAggregateFlowStatisticsFromFlowTableForAllFlowsInput input) {
  52         GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutputBuilder builder = new GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutputBuilder();
  53         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  54         builder.setTransactionId(transId);
  55         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  56     }
  57
  58     @Override
  59     public Future<RpcResult<GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutput>> getAggregateFlowStatisticsFromFlowTableForGivenMatch(GetAggregateFlowStatisticsFromFlowTableForGivenMatchInput input) {
  60         GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutputBuilder builder = new GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutputBuilder();
  61         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  62         builder.setTransactionId(transId);
  63         AggregateFlowStatisticsUpdateBuilder afsuBuilder = new AggregateFlowStatisticsUpdateBuilder();
  64         afsuBuilder.setMoreReplies(false);
  65         afsuBuilder.setTransactionId(transId);
  66         afsuBuilder.setByteCount(StatisticsManagerTest.COUNTER_64_TEST_VALUE);
  67         afsuBuilder.setId(input.getNode().getValue().firstKeyOf(Node.class, NodeKey.class).getId());
  68         notifService.pushDelayedNotification(afsuBuilder.build(), 100);
  69         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  70     }
  71
  72     @Override
  73     public Future<RpcResult<GetAllFlowStatisticsFromFlowTableOutput>> getAllFlowStatisticsFromFlowTable(GetAllFlowStatisticsFromFlowTableInput input) {
  74         GetAllFlowStatisticsFromFlowTableOutputBuilder builder = new GetAllFlowStatisticsFromFlowTableOutputBuilder();
  75         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  76         builder.setTransactionId(transId);
  77         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  78     }
  79
  80     @Override
  81     public Future<RpcResult<GetAllFlowsStatisticsFromAllFlowTablesOutput>> getAllFlowsStatisticsFromAllFlowTables(GetAllFlowsStatisticsFromAllFlowTablesInput input) {
  82         GetAllFlowsStatisticsFromAllFlowTablesOutputBuilder builder = new GetAllFlowsStatisticsFromAllFlowTablesOutputBuilder();
  83         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  84         builder.setTransactionId(transId);
  85         List<FlowAndStatisticsMapList> flowAndStatisticsMapLists = new ArrayList<>();
  86         FlowsStatisticsUpdateBuilder flowsStatisticsUpdateBuilder = new FlowsStatisticsUpdateBuilder();
  87         flowsStatisticsUpdateBuilder.setTransactionId(transId);
  88         flowsStatisticsUpdateBuilder.setMoreReplies(false);
  89         flowsStatisticsUpdateBuilder.setId(input.getNode().getValue().firstKeyOf(Node.class, NodeKey.class).getId());
  90         FlowAndStatisticsMapListBuilder flowAndStatisticsMapListBuilder = new FlowAndStatisticsMapListBuilder(StatisticsManagerTest.getFlow());
  91         flowAndStatisticsMapListBuilder.setTableId(StatisticsManagerTest.getFlow().getTableId());
  92         flowAndStatisticsMapListBuilder.setContainerName(StatisticsManagerTest.getFlow().getContainerName());
  93         flowAndStatisticsMapListBuilder.setBarrier(StatisticsManagerTest.getFlow().isBarrier());
  94         flowAndStatisticsMapListBuilder.setByteCount(StatisticsManagerTest.COUNTER_64_TEST_VALUE);
  95         flowAndStatisticsMapLists.add(flowAndStatisticsMapListBuilder.build());
  96         flowsStatisticsUpdateBuilder.setFlowAndStatisticsMapList(flowAndStatisticsMapLists);
  97         notifService.pushDelayedNotification(flowsStatisticsUpdateBuilder.build(), 100);
  98         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  99     }
 100
 101     @Override
 102     public Future<RpcResult<GetFlowStatisticsFromFlowTableOutput>> getFlowStatisticsFromFlowTable(GetFlowStatisticsFromFlowTableInput input) {
 103         GetFlowStatisticsFromFlowTableOutputBuilder builder = new GetFlowStatisticsFromFlowTableOutputBuilder();
 104         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
 105         builder.setTransactionId(transId);
 106         List<FlowAndStatisticsMapList> flowAndStatisticsMapLists = new ArrayList<>();
 107         FlowsStatisticsUpdateBuilder flowsStatisticsUpdateBuilder = new FlowsStatisticsUpdateBuilder();
 108         flowsStatisticsUpdateBuilder.setTransactionId(transId);
 109         flowsStatisticsUpdateBuilder.setMoreReplies(false);
 110         flowsStatisticsUpdateBuilder.setId(input.getNode().getValue().firstKeyOf(Node.class, NodeKey.class).getId());
 111         FlowAndStatisticsMapListBuilder flowAndStatisticsMapListBuilder = new FlowAndStatisticsMapListBuilder(input);
 112         flowAndStatisticsMapListBuilder.setTableId(input.getTableId());
 113         flowAndStatisticsMapListBuilder.setContainerName(input.getContainerName());
 114         flowAndStatisticsMapListBuilder.setBarrier(input.isBarrier());
 115         flowAndStatisticsMapListBuilder.setByteCount(StatisticsManagerTest.COUNTER_64_TEST_VALUE);
 116         flowAndStatisticsMapLists.add(flowAndStatisticsMapListBuilder.build());
 117         flowsStatisticsUpdateBuilder.setFlowAndStatisticsMapList(flowAndStatisticsMapLists);
 118         notifService.pushDelayedNotification(flowsStatisticsUpdateBuilder.build(), 100);
 119         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
 120     }
 121
 122 }