openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/services/batch/FlatBatchMeterAdaptersTest.java

   1 /*
   2  * Copyright (c) 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 org.opendaylight.openflowplugin.impl.services.batch;
   9
  10 import com.google.common.collect.Lists;
  11 import java.util.Iterator;
  12 import org.junit.Assert;
  13 import org.junit.Test;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.ProcessFlatBatchOutput;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.failure.ids.aug.FlatBatchFailureMeterIdCase;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.failure.ids.aug.FlatBatchFailureMeterIdCaseBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.meter.crud._case.aug.flat.batch.add.meter._case.FlatBatchAddMeter;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.meter.crud._case.aug.flat.batch.add.meter._case.FlatBatchAddMeterBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.meter.crud._case.aug.flat.batch.remove.meter._case.FlatBatchRemoveMeter;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.meter.crud._case.aug.flat.batch.remove.meter._case.FlatBatchRemoveMeterBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.meter.crud._case.aug.flat.batch.update.meter._case.FlatBatchUpdateMeter;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.flat.batch.meter.crud._case.aug.flat.batch.update.meter._case.FlatBatchUpdateMeterBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.BatchFailure;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.BatchFailureBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchInput;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchOutputBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.BatchMeterOutputListGrouping;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchInput;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchInput;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.add.meters.batch.input.BatchAddMeters;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.input.update.grouping.OriginalBatchedMeterBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.input.update.grouping.UpdatedBatchedMeterBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutput;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutputBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.remove.meters.batch.input.BatchRemoveMeters;
  42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  43 import org.opendaylight.yangtools.yang.binding.util.BindingMap;
  44 import org.opendaylight.yangtools.yang.common.ErrorType;
  45 import org.opendaylight.yangtools.yang.common.RpcResult;
  46 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  47 import org.opendaylight.yangtools.yang.common.Uint16;
  48 import org.opendaylight.yangtools.yang.common.Uint32;
  49
  50 /**
  51  * Test for {@link FlatBatchMeterAdapters}.
  52  */
  53 public class FlatBatchMeterAdaptersTest {
  54
  55     private static final NodeId NODE_ID = new NodeId("ut-node-id");
  56     private static final InstanceIdentifier<Node> NODE_II = InstanceIdentifier.create(Nodes.class)
  57             .child(Node.class, new NodeKey(NODE_ID));
  58     private static final NodeRef NODE_REF = new NodeRef(NODE_II);
  59
  60     @Test
  61     public void testAdaptFlatBatchAddMeter() {
  62         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_ADD);
  63         planStep.setBarrierAfter(true);
  64         planStep.getTaskBag().addAll(Lists.newArrayList(
  65                 createAddMeterBatch(Uint32.ONE),
  66                 createAddMeterBatch(Uint32.TWO)));
  67
  68         final AddMetersBatchInput addMetersBatchInput =
  69                 FlatBatchMeterAdapters.adaptFlatBatchAddMeter(planStep, NODE_REF);
  70
  71         Assert.assertTrue(addMetersBatchInput.getBarrierAfter());
  72         Assert.assertEquals(2, addMetersBatchInput.nonnullBatchAddMeters().size());
  73         final Iterator<BatchAddMeters> it = addMetersBatchInput.nonnullBatchAddMeters().values().iterator();
  74
  75         Assert.assertEquals(1L, it.next().getMeterId().getValue().longValue());
  76         Assert.assertEquals(2L, it.next().getMeterId().getValue().longValue());
  77     }
  78
  79     private static FlatBatchAddMeter createAddMeterBatch(final Uint32 groupIdValue) {
  80         return new FlatBatchAddMeterBuilder()
  81                 .setMeterId(new MeterId(groupIdValue))
  82                 .setBatchOrder(groupIdValue.toUint16())
  83                 .build();
  84     }
  85
  86     private static FlatBatchRemoveMeter createRemoveMeterBatch(final Uint32 groupIdValue) {
  87         return new FlatBatchRemoveMeterBuilder()
  88                 .setMeterId(new MeterId(groupIdValue))
  89                 .setBatchOrder(groupIdValue.toUint16())
  90                 .build();
  91     }
  92
  93     private static FlatBatchUpdateMeter createUpdateMeterBatch(final Uint32 groupIdValue) {
  94         return new FlatBatchUpdateMeterBuilder()
  95                 .setOriginalBatchedMeter(new OriginalBatchedMeterBuilder()
  96                         .setMeterId(new MeterId(groupIdValue))
  97                         .build())
  98                 .setUpdatedBatchedMeter(new UpdatedBatchedMeterBuilder()
  99                         .setMeterId(new MeterId(groupIdValue))
 100                         .build())
 101                 .setBatchOrder(groupIdValue.toUint16())
 102                 .build();
 103     }
 104
 105     @Test
 106     public void testAdaptFlatBatchRemoveMeter() {
 107         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_REMOVE);
 108         planStep.setBarrierAfter(true);
 109         planStep.getTaskBag().addAll(Lists.newArrayList(
 110                 createRemoveMeterBatch(Uint32.ONE),
 111                 createRemoveMeterBatch(Uint32.TWO)));
 112
 113         final RemoveMetersBatchInput removeMetersBatchInput =
 114                 FlatBatchMeterAdapters.adaptFlatBatchRemoveMeter(planStep, NODE_REF);
 115         Iterator<BatchRemoveMeters> iterator = removeMetersBatchInput.nonnullBatchRemoveMeters().values().iterator();
 116
 117         Assert.assertTrue(removeMetersBatchInput.getBarrierAfter());
 118         Assert.assertEquals(2, removeMetersBatchInput.nonnullBatchRemoveMeters().size());
 119         Assert.assertEquals(1L, iterator.next().getMeterId().getValue().longValue());
 120         Assert.assertEquals(2L, iterator.next().getMeterId().getValue().longValue());
 121     }
 122
 123     @Test
 124     public void testAdaptFlatBatchUpdateMeter() {
 125         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_UPDATE);
 126         planStep.setBarrierAfter(true);
 127         planStep.getTaskBag().addAll(Lists.newArrayList(
 128                 createUpdateMeterBatch(Uint32.ONE),
 129                 createUpdateMeterBatch(Uint32.TWO)));
 130
 131         final UpdateMetersBatchInput updateMetersBatchInput =
 132                 FlatBatchMeterAdapters.adaptFlatBatchUpdateMeter(planStep, NODE_REF);
 133
 134         Assert.assertTrue(updateMetersBatchInput.getBarrierAfter());
 135         Assert.assertEquals(2, updateMetersBatchInput.getBatchUpdateMeters().size());
 136         Assert.assertEquals(1L, updateMetersBatchInput.getBatchUpdateMeters().get(0)
 137                 .getUpdatedBatchedMeter().getMeterId().getValue().longValue());
 138         Assert.assertEquals(2L, updateMetersBatchInput.getBatchUpdateMeters().get(1)
 139                 .getUpdatedBatchedMeter().getMeterId().getValue().longValue());
 140     }
 141
 142     @Test
 143     public void testCreateBatchMeterChainingFunction_failures() {
 144         final RpcResult<BatchMeterOutputListGrouping> input = RpcResultBuilder.<BatchMeterOutputListGrouping>failed()
 145                 .withError(ErrorType.APPLICATION, "ut-meterError")
 146                 .withResult(new AddMetersBatchOutputBuilder()
 147                         .setBatchFailedMetersOutput(BindingMap.ordered(
 148                                 createBatchFailedMetersOutput(Uint16.ZERO, Uint32.ONE),
 149                                 createBatchFailedMetersOutput(Uint16.ONE, Uint32.TWO)
 150                         ))
 151                         .build())
 152                 .build();
 153
 154         final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchMeterAdapters
 155                 .convertBatchMeterResult(3).apply(input);
 156         Iterator<BatchFailure> iterator = rpcResult.getResult().nonnullBatchFailure().values().iterator();
 157
 158         Assert.assertFalse(rpcResult.isSuccessful());
 159         Assert.assertEquals(1, rpcResult.getErrors().size());
 160         Assert.assertEquals(2, rpcResult.getResult().nonnullBatchFailure().size());
 161         Assert.assertEquals(3, iterator.next().getBatchOrder().intValue());
 162         BatchFailure secondBatchFailure = iterator.next();
 163         Assert.assertEquals(4, secondBatchFailure.getBatchOrder().intValue());
 164         Assert.assertEquals(2L, ((FlatBatchFailureMeterIdCase) secondBatchFailure
 165                 .getBatchItemIdChoice()).getMeterId().getValue().longValue());
 166     }
 167
 168     @Test
 169     public void testCreateBatchMeterChainingFunction_successes() {
 170         final RpcResult<BatchMeterOutputListGrouping> input = RpcResultBuilder
 171                 .<BatchMeterOutputListGrouping>success(new AddMetersBatchOutputBuilder().build())
 172                 .build();
 173
 174         final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchMeterAdapters
 175                 .convertBatchMeterResult(0).apply(input);
 176
 177         Assert.assertTrue(rpcResult.isSuccessful());
 178         Assert.assertEquals(0, rpcResult.getErrors().size());
 179         Assert.assertEquals(0, rpcResult.getResult().nonnullBatchFailure().size());
 180     }
 181
 182     private static BatchFailedMetersOutput createBatchFailedMetersOutput(final Uint16 batchOrder,
 183             final Uint32 groupIdValue) {
 184         return new BatchFailedMetersOutputBuilder()
 185                 .setMeterId(new MeterId(groupIdValue))
 186                 .setBatchOrder(batchOrder)
 187                 .build();
 188     }
 189
 190     private static BatchFailure createChainFailure(final Uint16 batchOrder, final Uint32 groupIdValue) {
 191         return new BatchFailureBuilder()
 192                 .setBatchOrder(batchOrder)
 193                 .setBatchItemIdChoice(new FlatBatchFailureMeterIdCaseBuilder()
 194                         .setMeterId(new MeterId(groupIdValue))
 195                         .build())
 196                 .build();
 197     }
 198 }