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.util;
10 import com.google.common.base.Function;
11 import com.google.common.collect.Lists;
12 import java.util.Collections;
13 import java.util.List;
15 import org.apache.commons.lang3.tuple.Pair;
16 import org.junit.Assert;
17 import org.junit.Test;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.SendBarrierOutput;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterRef;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchOutput;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchOutputBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.BatchMeterOutputListGrouping;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutputBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutputKey;
33 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
34 import org.opendaylight.yangtools.yang.common.RpcError;
35 import org.opendaylight.yangtools.yang.common.RpcResult;
36 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
37 import org.opendaylight.yangtools.yang.common.Uint32;
40 * Test for {@link MeterUtil}.
42 public class MeterUtilTest {
44 public static final NodeId DUMMY_NODE_ID = new NodeId("dummyNodeId");
45 private static final MeterId DUMMY_METER_ID = new MeterId(Uint32.valueOf(42));
46 private static final MeterId DUMMY_METER_ID_2 = new MeterId(Uint32.valueOf(43));
49 public void testBuildGroupPath() {
50 final InstanceIdentifier<Node> nodePath = InstanceIdentifier
52 .child(Node.class, new NodeKey(DUMMY_NODE_ID));
54 final MeterRef meterRef = MeterUtil.buildMeterPath(nodePath, DUMMY_METER_ID);
55 final InstanceIdentifier<?> meterRefValue = meterRef.getValue();
56 Assert.assertEquals(DUMMY_NODE_ID, meterRefValue.firstKeyOf(Node.class).getId());
57 Assert.assertEquals(DUMMY_METER_ID, meterRefValue.firstKeyOf(Meter.class).getMeterId());
61 public void testCreateCumulatingFunction() {
62 final Function<List<RpcResult<String>>, RpcResult<List<BatchFailedMetersOutput>>> function =
63 MeterUtil.createCumulativeFunction(Lists.newArrayList(
64 createBatchMeter(DUMMY_METER_ID),
65 createBatchMeter(DUMMY_METER_ID_2)));
67 final RpcResult<List<BatchFailedMetersOutput>> output = function.apply(Lists.newArrayList(
68 RpcResultBuilder.success("a").build(),
69 RpcResultBuilder.<String>failed()
70 .withError(RpcError.ErrorType.APPLICATION, "ut-meter-error")
73 Assert.assertFalse(output.isSuccessful());
74 Assert.assertEquals(1, output.getResult().size());
75 Assert.assertEquals(DUMMY_METER_ID_2, output.getResult().get(0).getMeterId());
76 Assert.assertEquals(1, output.getResult().get(0).getBatchOrder().intValue());
79 private static org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.Meter createBatchMeter(
80 final MeterId meterId) {
81 return new MeterBuilder()
87 public void testMeterAddTransformFailure() {
88 final RpcResult<List<BatchFailedMetersOutput>> input = createBatchOutcomeWithError();
89 checkBatchErrorOutcomeTransformation(MeterUtil.METER_ADD_TRANSFORM.apply(input));
93 public void testMeterAddTransformSuccess() {
94 final RpcResult<List<BatchFailedMetersOutput>> input = createEmptyBatchOutcome();
95 checkBatchSuccessOutcomeTransformation(MeterUtil.METER_ADD_TRANSFORM.apply(input));
99 public void testMeterRemoveTransformFailure() {
100 final RpcResult<List<BatchFailedMetersOutput>> input = createBatchOutcomeWithError();
101 checkBatchErrorOutcomeTransformation(MeterUtil.METER_REMOVE_TRANSFORM.apply(input));
105 public void testFlowRemoveTransformSuccess() {
106 final RpcResult<List<BatchFailedMetersOutput>> input = createEmptyBatchOutcome();
107 checkBatchSuccessOutcomeTransformation(MeterUtil.METER_REMOVE_TRANSFORM.apply(input));
111 public void testFlowUpdateTransformFailure() {
112 final RpcResult<List<BatchFailedMetersOutput>> input = createBatchOutcomeWithError();
113 checkBatchErrorOutcomeTransformation(MeterUtil.METER_UPDATE_TRANSFORM.apply(input));
117 public void testFlowUpdateTransformSuccess() {
118 final RpcResult<List<BatchFailedMetersOutput>> input = createEmptyBatchOutcome();
119 checkBatchSuccessOutcomeTransformation(MeterUtil.METER_UPDATE_TRANSFORM.apply(input));
122 private static <T extends BatchMeterOutputListGrouping> void checkBatchSuccessOutcomeTransformation(
123 final RpcResult<T> output) {
124 Assert.assertTrue(output.isSuccessful());
125 Assert.assertEquals(0, output.getResult().nonnullBatchFailedMetersOutput().size());
126 Assert.assertEquals(0, output.getErrors().size());
129 private static RpcResult<List<BatchFailedMetersOutput>> createEmptyBatchOutcome() {
130 return RpcResultBuilder
131 .success(Collections.<BatchFailedMetersOutput>emptyList())
135 private static RpcResult<List<BatchFailedMetersOutput>> createBatchOutcomeWithError() {
136 return RpcResultBuilder.<List<BatchFailedMetersOutput>>failed()
137 .withError(RpcError.ErrorType.APPLICATION, "ut-flowAddFail")
138 .withResult(Collections.singletonList(new BatchFailedMetersOutputBuilder()
139 .setMeterId(DUMMY_METER_ID)
144 private static <T extends BatchMeterOutputListGrouping> void checkBatchErrorOutcomeTransformation(
145 final RpcResult<T> output) {
146 Assert.assertFalse(output.isSuccessful());
147 Assert.assertEquals(1, output.getResult().nonnullBatchFailedMetersOutput().size());
148 Assert.assertEquals(DUMMY_METER_ID,
149 output.getResult().nonnullBatchFailedMetersOutput().values().iterator().next().getMeterId());
151 Assert.assertEquals(1, output.getErrors().size());
155 public void testCreateComposingFunction_success_success() {
156 final Function<Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>>,
157 RpcResult<AddMetersBatchOutput>> compositeFunction = MeterUtil.createComposingFunction();
159 final RpcResult<AddMetersBatchOutput> addGroupBatchOutput = createAddMetersBatchSuccessOutput();
160 final RpcResult<SendBarrierOutput> barrierOutput = RpcResultBuilder.<SendBarrierOutput>success().build();
161 final Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>> input =
162 Pair.of(addGroupBatchOutput, barrierOutput);
163 final RpcResult<AddMetersBatchOutput> composite = compositeFunction.apply(input);
165 Assert.assertTrue(composite.isSuccessful());
166 Assert.assertEquals(0, composite.getErrors().size());
167 Map<BatchFailedMetersOutputKey, BatchFailedMetersOutput> failedMeters
168 = composite.getResult().nonnullBatchFailedMetersOutput();
169 Assert.assertEquals(0, failedMeters.size());
173 public void testCreateComposingFunction_failure_success() {
174 final Function<Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>>,
175 RpcResult<AddMetersBatchOutput>> compositeFunction = MeterUtil.createComposingFunction();
177 final RpcResult<AddMetersBatchOutput> addGroupBatchOutput = createAddMetersBatchFailureOutcome();
178 final RpcResult<SendBarrierOutput> barrierOutput = RpcResultBuilder.<SendBarrierOutput>success().build();
179 final Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>> input =
180 Pair.of(addGroupBatchOutput, barrierOutput);
181 final RpcResult<AddMetersBatchOutput> composite = compositeFunction.apply(input);
183 Assert.assertFalse(composite.isSuccessful());
184 Assert.assertEquals(1, composite.getErrors().size());
185 Assert.assertEquals(1, composite.getResult().getBatchFailedMetersOutput().size());
189 public void testCreateComposingFunction_success_failure() {
190 final Function<Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>>,
191 RpcResult<AddMetersBatchOutput>> compositeFunction = MeterUtil.createComposingFunction();
193 final RpcResult<AddMetersBatchOutput> addGroupBatchOutput = createAddMetersBatchSuccessOutput();
194 final RpcResult<SendBarrierOutput> barrierOutput = createBarrierFailureOutcome();
195 final Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>> input =
196 Pair.of(addGroupBatchOutput, barrierOutput);
197 final RpcResult<AddMetersBatchOutput> composite = compositeFunction.apply(input);
199 Assert.assertFalse(composite.isSuccessful());
200 Assert.assertEquals(1, composite.getErrors().size());
201 Map<BatchFailedMetersOutputKey, BatchFailedMetersOutput> failedMeters
202 = composite.getResult().nonnullBatchFailedMetersOutput();
203 Assert.assertEquals(0, failedMeters.size());
207 public void testCreateComposingFunction_failure_failure() {
208 final Function<Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>>,
209 RpcResult<AddMetersBatchOutput>> compositeFunction = MeterUtil.createComposingFunction();
211 final RpcResult<AddMetersBatchOutput> addGroupBatchOutput = createAddMetersBatchFailureOutcome();
212 final RpcResult<SendBarrierOutput> barrierOutput = createBarrierFailureOutcome();
213 final Pair<RpcResult<AddMetersBatchOutput>, RpcResult<SendBarrierOutput>> input =
214 Pair.of(addGroupBatchOutput, barrierOutput);
215 final RpcResult<AddMetersBatchOutput> composite = compositeFunction.apply(input);
217 Assert.assertFalse(composite.isSuccessful());
218 Assert.assertEquals(2, composite.getErrors().size());
219 Assert.assertEquals(1, composite.getResult().getBatchFailedMetersOutput().size());
222 private static RpcResult<SendBarrierOutput> createBarrierFailureOutcome() {
223 return RpcResultBuilder.<SendBarrierOutput>failed()
224 .withError(RpcError.ErrorType.APPLICATION, "ut-barrier-error")
228 private static RpcResult<AddMetersBatchOutput> createAddMetersBatchSuccessOutput() {
229 return RpcResultBuilder
230 .success(new AddMetersBatchOutputBuilder()
231 .setBatchFailedMetersOutput(Collections.emptyMap())
236 private static RpcResult<AddMetersBatchOutput> createAddMetersBatchFailureOutcome() {
237 final RpcResult<List<BatchFailedMetersOutput>> batchOutcomeWithError = createBatchOutcomeWithError();
238 return RpcResultBuilder.<AddMetersBatchOutput>failed()
239 .withResult(new AddMetersBatchOutputBuilder()
240 .setBatchFailedMetersOutput(batchOutcomeWithError.getResult())
242 .withRpcErrors(batchOutcomeWithError.getErrors())