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
9 package org.opendaylight.openflowplugin.impl.services.batch;
11 import com.google.common.collect.Lists;
12 import java.util.Iterator;
13 import org.junit.Assert;
14 import org.junit.Test;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.ProcessFlatBatchOutput;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.input.batch.batch.choice.flat.batch.add.group._case.FlatBatchAddGroup;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.input.batch.batch.choice.flat.batch.add.group._case.FlatBatchAddGroupBuilder;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.input.batch.batch.choice.flat.batch.remove.group._case.FlatBatchRemoveGroup;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.input.batch.batch.choice.flat.batch.remove.group._case.FlatBatchRemoveGroupBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.input.batch.batch.choice.flat.batch.update.group._case.FlatBatchUpdateGroup;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.input.batch.batch.choice.flat.batch.update.group._case.FlatBatchUpdateGroupBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.BatchFailure;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.BatchFailureBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.batch.failure.batch.item.id.choice.FlatBatchFailureGroupIdCase;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.batch.failure.batch.item.id.choice.FlatBatchFailureGroupIdCaseBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.AddGroupsBatchInput;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.AddGroupsBatchOutputBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.BatchGroupOutputListGrouping;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.RemoveGroupsBatchInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.UpdateGroupsBatchInput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.add.groups.batch.input.BatchAddGroups;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.input.update.grouping.OriginalBatchedGroupBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.input.update.grouping.UpdatedBatchedGroupBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.output.list.grouping.BatchFailedGroupsOutput;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.output.list.grouping.BatchFailedGroupsOutputBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.remove.groups.batch.input.BatchRemoveGroups;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
44 import org.opendaylight.yangtools.yang.common.RpcError;
45 import org.opendaylight.yangtools.yang.common.RpcResult;
46 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
49 * Test for {@link FlatBatchGroupAdapters}.
51 public class FlatBatchGroupAdaptersTest {
53 private static final NodeId NODE_ID = new NodeId("ut-node-id");
54 private static final InstanceIdentifier<Node> NODE_II = InstanceIdentifier.create(Nodes.class)
55 .child(Node.class, new NodeKey(NODE_ID));
56 private static final NodeRef NODE_REF = new NodeRef(NODE_II);
59 public void testAdaptFlatBatchAddGroup() {
60 final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_ADD);
61 planStep.setBarrierAfter(true);
62 planStep.getTaskBag().addAll(Lists.newArrayList(
63 createAddGroupBatch(1L),
64 createAddGroupBatch(2L)));
66 final AddGroupsBatchInput addGroupsBatchInput =
67 FlatBatchGroupAdapters.adaptFlatBatchAddGroup(planStep, NODE_REF);
68 Iterator<BatchAddGroups> iterator = addGroupsBatchInput.nonnullBatchAddGroups().values().iterator();
70 Assert.assertTrue(addGroupsBatchInput.isBarrierAfter());
71 Assert.assertEquals(2, addGroupsBatchInput.getBatchAddGroups().size());
72 Assert.assertEquals(1L, iterator.next().getGroupId().getValue().longValue());
73 Assert.assertEquals(2L, iterator.next().getGroupId().getValue().longValue());
76 private FlatBatchAddGroup createAddGroupBatch(final long groupIdValue) {
77 return new FlatBatchAddGroupBuilder()
78 .setGroupId(new GroupId(groupIdValue))
82 private FlatBatchRemoveGroup createRemoveGroupBatch(final long groupIdValue) {
83 return new FlatBatchRemoveGroupBuilder()
84 .setGroupId(new GroupId(groupIdValue))
88 private FlatBatchUpdateGroup createUpdateGroupBatch(final long groupIdValue) {
89 return new FlatBatchUpdateGroupBuilder()
90 .setOriginalBatchedGroup(new OriginalBatchedGroupBuilder()
91 .setGroupId(new GroupId(groupIdValue))
93 .setUpdatedBatchedGroup(new UpdatedBatchedGroupBuilder()
94 .setGroupId(new GroupId(groupIdValue))
100 public void testAdaptFlatBatchRemoveGroup() {
101 final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_REMOVE);
102 planStep.setBarrierAfter(true);
103 planStep.getTaskBag().addAll(Lists.newArrayList(
104 createRemoveGroupBatch(1L),
105 createRemoveGroupBatch(2L)));
107 final RemoveGroupsBatchInput removeGroupsBatchInput =
108 FlatBatchGroupAdapters.adaptFlatBatchRemoveGroup(planStep, NODE_REF);
109 Iterator<BatchRemoveGroups> iterator = removeGroupsBatchInput.nonnullBatchRemoveGroups().values().iterator();
111 Assert.assertTrue(removeGroupsBatchInput.isBarrierAfter());
112 Assert.assertEquals(2, removeGroupsBatchInput.getBatchRemoveGroups().size());
113 Assert.assertEquals(1L, iterator.next().getGroupId().getValue().longValue());
114 Assert.assertEquals(2L, iterator.next().getGroupId().getValue().longValue());
118 public void testAdaptFlatBatchUpdateGroup() {
119 final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_UPDATE);
120 planStep.setBarrierAfter(true);
121 planStep.getTaskBag().addAll(Lists.newArrayList(
122 createUpdateGroupBatch(1L),
123 createUpdateGroupBatch(2L)));
125 final UpdateGroupsBatchInput updateGroupsBatchInput =
126 FlatBatchGroupAdapters.adaptFlatBatchUpdateGroup(planStep, NODE_REF);
128 Assert.assertTrue(updateGroupsBatchInput.isBarrierAfter());
129 Assert.assertEquals(2, updateGroupsBatchInput.getBatchUpdateGroups().size());
130 Assert.assertEquals(1L, updateGroupsBatchInput.getBatchUpdateGroups().get(0)
131 .getUpdatedBatchedGroup().getGroupId().getValue().longValue());
132 Assert.assertEquals(2L, updateGroupsBatchInput.getBatchUpdateGroups().get(1)
133 .getUpdatedBatchedGroup().getGroupId().getValue().longValue());
137 public void testCreateBatchGroupChainingFunction_failures() {
138 final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder.<BatchGroupOutputListGrouping>failed()
139 .withError(RpcError.ErrorType.APPLICATION, "ut-groupError")
140 .withResult(new AddGroupsBatchOutputBuilder()
141 .setBatchFailedGroupsOutput(Lists.newArrayList(
142 createBatchFailedGroupsOutput(0, 1L),
143 createBatchFailedGroupsOutput(1, 2L)
148 final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
149 .convertBatchGroupResult(3).apply(input);
150 Iterator<BatchFailure> iterator = rpcResult.getResult().nonnullBatchFailure().values().iterator();
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, ((FlatBatchFailureGroupIdCase) secondBatchFailure
159 .getBatchItemIdChoice()).getGroupId().getValue().longValue());
163 public void testCreateBatchGroupChainingFunction_successes() {
164 final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder
165 .<BatchGroupOutputListGrouping>success(new AddGroupsBatchOutputBuilder().build())
168 final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
169 .convertBatchGroupResult(0).apply(input);
171 Assert.assertTrue(rpcResult.isSuccessful());
172 Assert.assertEquals(0, rpcResult.getErrors().size());
173 Assert.assertEquals(0, rpcResult.getResult().nonnullBatchFailure().size());
176 private BatchFailedGroupsOutput createBatchFailedGroupsOutput(final Integer batchOrder, final long groupIdValue) {
177 return new BatchFailedGroupsOutputBuilder()
178 .setGroupId(new GroupId(groupIdValue))
179 .setBatchOrder(batchOrder)
183 private BatchFailure createChainFailure(final int batchOrder, final long groupIdValue) {
184 return new BatchFailureBuilder()
185 .setBatchOrder(batchOrder)
186 .setBatchItemIdChoice(new FlatBatchFailureGroupIdCaseBuilder()
187 .setGroupId(new GroupId(groupIdValue))