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