opendaylight/md-sal/statistics-manager/src/test/java/test/mock/util/OpendaylightFlowStatisticsServiceMock.java

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