2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.openflowplugin.impl.services.batch;
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.binding.util.BindingMap;
43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
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;
51 * Test for {@link FlatBatchMeterAdapters}.
53 public class FlatBatchMeterAdaptersTest {
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.toIdentifier());
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)));
68 final AddMetersBatchInput addMetersBatchInput =
69 FlatBatchMeterAdapters.adaptFlatBatchAddMeter(planStep, NODE_REF);
71 Assert.assertTrue(addMetersBatchInput.getBarrierAfter());
72 Assert.assertEquals(2, addMetersBatchInput.nonnullBatchAddMeters().size());
73 final Iterator<BatchAddMeters> it = addMetersBatchInput.nonnullBatchAddMeters().values().iterator();
75 Assert.assertEquals(1L, it.next().getMeterId().getValue().longValue());
76 Assert.assertEquals(2L, it.next().getMeterId().getValue().longValue());
79 private static FlatBatchAddMeter createAddMeterBatch(final Uint32 groupIdValue) {
80 return new FlatBatchAddMeterBuilder()
81 .setMeterId(new MeterId(groupIdValue))
82 .setBatchOrder(groupIdValue.toUint16())
86 private static FlatBatchRemoveMeter createRemoveMeterBatch(final Uint32 groupIdValue) {
87 return new FlatBatchRemoveMeterBuilder()
88 .setMeterId(new MeterId(groupIdValue))
89 .setBatchOrder(groupIdValue.toUint16())
93 private static FlatBatchUpdateMeter createUpdateMeterBatch(final Uint32 groupIdValue) {
94 return new FlatBatchUpdateMeterBuilder()
95 .setOriginalBatchedMeter(new OriginalBatchedMeterBuilder()
96 .setMeterId(new MeterId(groupIdValue))
98 .setUpdatedBatchedMeter(new UpdatedBatchedMeterBuilder()
99 .setMeterId(new MeterId(groupIdValue))
101 .setBatchOrder(groupIdValue.toUint16())
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)));
113 final RemoveMetersBatchInput removeMetersBatchInput =
114 FlatBatchMeterAdapters.adaptFlatBatchRemoveMeter(planStep, NODE_REF);
115 Iterator<BatchRemoveMeters> iterator = removeMetersBatchInput.nonnullBatchRemoveMeters().values().iterator();
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());
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)));
131 final UpdateMetersBatchInput updateMetersBatchInput =
132 FlatBatchMeterAdapters.adaptFlatBatchUpdateMeter(planStep, NODE_REF);
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());
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)
154 final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchMeterAdapters
155 .convertBatchMeterResult(3).apply(input);
156 Iterator<BatchFailure> iterator = rpcResult.getResult().nonnullBatchFailure().values().iterator();
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());
169 public void testCreateBatchMeterChainingFunction_successes() {
170 final RpcResult<BatchMeterOutputListGrouping> input = RpcResultBuilder
171 .<BatchMeterOutputListGrouping>success(new AddMetersBatchOutputBuilder().build())
174 final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchMeterAdapters
175 .convertBatchMeterResult(0).apply(input);
177 Assert.assertTrue(rpcResult.isSuccessful());
178 Assert.assertEquals(0, rpcResult.getErrors().size());
179 Assert.assertEquals(0, rpcResult.getResult().nonnullBatchFailure().size());
182 private static BatchFailedMetersOutput createBatchFailedMetersOutput(final Uint16 batchOrder,
183 final Uint32 groupIdValue) {
184 return new BatchFailedMetersOutputBuilder()
185 .setMeterId(new MeterId(groupIdValue))
186 .setBatchOrder(batchOrder)
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))