opendaylight/md-sal/statistics-manager/src/test/java/test/mock/util/OpendaylightPortStatisticsServiceMock.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.transaction.rev131103.TransactionId;
   9 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  10 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  11 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetAllNodeConnectorsStatisticsInput;
  12 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetAllNodeConnectorsStatisticsOutput;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetAllNodeConnectorsStatisticsOutputBuilder;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetNodeConnectorStatisticsInput;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetNodeConnectorStatisticsOutput;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetNodeConnectorStatisticsOutputBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.NodeConnectorStatisticsUpdateBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.OpendaylightPortStatisticsService;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.node.connector.statistics.and.port.number.map.NodeConnectorStatisticsAndPortNumberMap;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.node.connector.statistics.and.port.number.map.NodeConnectorStatisticsAndPortNumberMapBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.node.connector.statistics.and.port.number.map.NodeConnectorStatisticsAndPortNumberMapKey;
  22 import org.opendaylight.yangtools.yang.common.RpcResult;
  23 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  24
  25 public class OpendaylightPortStatisticsServiceMock implements OpendaylightPortStatisticsService {
  26     NotificationProviderServiceHelper notifService;
  27
  28     public OpendaylightPortStatisticsServiceMock(NotificationProviderServiceHelper notifService) {
  29         this.notifService = notifService;
  30     }
  31
  32     @Override
  33     public Future<RpcResult<GetAllNodeConnectorsStatisticsOutput>> getAllNodeConnectorsStatistics(GetAllNodeConnectorsStatisticsInput input) {
  34         GetAllNodeConnectorsStatisticsOutputBuilder builder = new GetAllNodeConnectorsStatisticsOutputBuilder();
  35         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  36         builder.setTransactionId(transId);
  37         NodeConnectorStatisticsUpdateBuilder ncsuBuilder = new NodeConnectorStatisticsUpdateBuilder();
  38         NodeConnectorStatisticsAndPortNumberMapBuilder ncsapnmBuilder = new NodeConnectorStatisticsAndPortNumberMapBuilder();
  39         List<NodeConnectorStatisticsAndPortNumberMap> nodeConnectorStatisticsAndPortNumberMaps = new ArrayList<>();
  40         ncsapnmBuilder.setKey(new NodeConnectorStatisticsAndPortNumberMapKey(StatisticsManagerTest.getNodeConnectorId()));
  41         ncsapnmBuilder.setReceiveDrops(StatisticsManagerTest.BIG_INTEGER_TEST_VALUE);
  42         nodeConnectorStatisticsAndPortNumberMaps.add(ncsapnmBuilder.build());
  43         ncsuBuilder.setTransactionId(transId);
  44         ncsuBuilder.setId(input.getNode().getValue().firstKeyOf(Node.class, NodeKey.class).getId());
  45         ncsuBuilder.setNodeConnectorStatisticsAndPortNumberMap(nodeConnectorStatisticsAndPortNumberMaps);
  46         ncsuBuilder.setMoreReplies(true);
  47         notifService.pushDelayedNotification(ncsuBuilder.build(), 0); // 1st notification
  48         ncsuBuilder.setMoreReplies(false);
  49         ncsapnmBuilder.setCollisionCount(StatisticsManagerTest.BIG_INTEGER_TEST_VALUE);
  50         nodeConnectorStatisticsAndPortNumberMaps.clear();
  51         nodeConnectorStatisticsAndPortNumberMaps.add(ncsapnmBuilder.build());
  52         ncsuBuilder.setNodeConnectorStatisticsAndPortNumberMap(nodeConnectorStatisticsAndPortNumberMaps);
  53         notifService.pushDelayedNotification(ncsuBuilder.build(), 10); // 2nd notification
  54         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  55     }
  56
  57     @Override
  58     public Future<RpcResult<GetNodeConnectorStatisticsOutput>> getNodeConnectorStatistics(GetNodeConnectorStatisticsInput input) {
  59         GetNodeConnectorStatisticsOutputBuilder builder = new GetNodeConnectorStatisticsOutputBuilder();
  60         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  61         builder.setTransactionId(transId);
  62         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  63     }
  64 }