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