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

   1 package test.mock.util;
   2
   3 import java.math.BigInteger;
   4 import java.util.ArrayList;
   5 import java.util.List;
   6 import java.util.concurrent.Future;
   7
   8 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.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 import com.google.common.util.concurrent.Futures;
  26
  27 public class OpendaylightPortStatisticsServiceMock implements OpendaylightPortStatisticsService {
  28     NotificationProviderServiceHelper notifService;
  29
  30     public OpendaylightPortStatisticsServiceMock(NotificationProviderServiceHelper notifService) {
  31         this.notifService = notifService;
  32     }
  33
  34     @Override
  35     public Future<RpcResult<GetAllNodeConnectorsStatisticsOutput>> getAllNodeConnectorsStatistics(GetAllNodeConnectorsStatisticsInput input) {
  36         GetAllNodeConnectorsStatisticsOutputBuilder builder = new GetAllNodeConnectorsStatisticsOutputBuilder();
  37         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  38         builder.setTransactionId(transId);
  39         NodeConnectorStatisticsUpdateBuilder ncsuBuilder = new NodeConnectorStatisticsUpdateBuilder();
  40         NodeConnectorStatisticsAndPortNumberMapBuilder ncsapnmBuilder = new NodeConnectorStatisticsAndPortNumberMapBuilder();
  41         List<NodeConnectorStatisticsAndPortNumberMap> nodeConnectorStatisticsAndPortNumberMaps = new ArrayList<>();
  42         ncsapnmBuilder.setKey(new NodeConnectorStatisticsAndPortNumberMapKey(StatisticsManagerTest.getNodeConnectorId()));
  43         ncsapnmBuilder.setReceiveDrops(StatisticsManagerTest.BIG_INTEGER_TEST_VALUE);
  44         nodeConnectorStatisticsAndPortNumberMaps.add(ncsapnmBuilder.build());
  45         ncsuBuilder.setTransactionId(transId);
  46         ncsuBuilder.setId(input.getNode().getValue().firstKeyOf(Node.class, NodeKey.class).getId());
  47         ncsuBuilder.setNodeConnectorStatisticsAndPortNumberMap(nodeConnectorStatisticsAndPortNumberMaps);
  48         ncsuBuilder.setMoreReplies(true);
  49         notifService.pushDelayedNotification(ncsuBuilder.build(), 0); // 1st notification
  50         ncsuBuilder.setMoreReplies(false);
  51         ncsapnmBuilder.setCollisionCount(StatisticsManagerTest.BIG_INTEGER_TEST_VALUE);
  52         nodeConnectorStatisticsAndPortNumberMaps.clear();
  53         nodeConnectorStatisticsAndPortNumberMaps.add(ncsapnmBuilder.build());
  54         ncsuBuilder.setNodeConnectorStatisticsAndPortNumberMap(nodeConnectorStatisticsAndPortNumberMaps);
  55         notifService.pushDelayedNotification(ncsuBuilder.build(), 10); // 2nd notification
  56         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  57     }
  58
  59     @Override
  60     public Future<RpcResult<GetNodeConnectorStatisticsOutput>> getNodeConnectorStatistics(GetNodeConnectorStatisticsInput input) {
  61         GetNodeConnectorStatisticsOutputBuilder builder = new GetNodeConnectorStatisticsOutputBuilder();
  62         TransactionId transId = new TransactionId(BigInteger.valueOf(TestUtils.getNewTransactionId()));
  63         builder.setTransactionId(transId);
  64         return Futures.immediateFuture(RpcResultBuilder.success(builder.build()).build());
  65     }
  66 }