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.process.flat.batch.input.batch.batch.choice.flat.batch.add.group._case.FlatBatchAddGroup;
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.FlatBatchAddGroupBuilder;
17 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;
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.FlatBatchRemoveGroupBuilder;
19 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;
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.FlatBatchUpdateGroupBuilder;
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.FlatBatchFailureGroupIdCase;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flat.batch.service.rev160321.process.flat.batch.output.batch.failure.batch.item.id.choice.FlatBatchFailureGroupIdCaseBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.AddGroupsBatchInput;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.AddGroupsBatchOutputBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.BatchGroupOutputListGrouping;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.RemoveGroupsBatchInput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.UpdateGroupsBatchInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.add.groups.batch.input.BatchAddGroups;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.input.update.grouping.OriginalBatchedGroupBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.input.update.grouping.UpdatedBatchedGroupBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.output.list.grouping.BatchFailedGroupsOutput;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.output.list.grouping.BatchFailedGroupsOutputBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.remove.groups.batch.input.BatchRemoveGroups;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
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 import org.opendaylight.yangtools.yang.common.Uint16;
47 import org.opendaylight.yangtools.yang.common.Uint32;
50 * Test for {@link FlatBatchGroupAdapters}.
52 public class FlatBatchGroupAdaptersTest {
54 private static final NodeId NODE_ID = new NodeId("ut-node-id");
55 private static final InstanceIdentifier<Node> NODE_II = InstanceIdentifier.create(Nodes.class)
56 .child(Node.class, new NodeKey(NODE_ID));
57 private static final NodeRef NODE_REF = new NodeRef(NODE_II);
60 public void testAdaptFlatBatchAddGroup() {
61 final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_ADD);
62 planStep.setBarrierAfter(true);
63 planStep.getTaskBag().addAll(Lists.newArrayList(
64 createAddGroupBatch(Uint32.ONE),
65 createAddGroupBatch(Uint32.TWO)));
67 final AddGroupsBatchInput addGroupsBatchInput =
68 FlatBatchGroupAdapters.adaptFlatBatchAddGroup(planStep, NODE_REF);
69 Iterator<BatchAddGroups> iterator = addGroupsBatchInput.nonnullBatchAddGroups().values().iterator();
71 Assert.assertTrue(addGroupsBatchInput.isBarrierAfter());
72 Assert.assertEquals(2, addGroupsBatchInput.getBatchAddGroups().size());
73 Assert.assertEquals(1L, iterator.next().getGroupId().getValue().longValue());
74 Assert.assertEquals(2L, iterator.next().getGroupId().getValue().longValue());
77 private static FlatBatchAddGroup createAddGroupBatch(final Uint32 groupIdValue) {
78 return new FlatBatchAddGroupBuilder()
79 .setGroupId(new GroupId(groupIdValue))
83 private static FlatBatchRemoveGroup createRemoveGroupBatch(final Uint32 groupIdValue) {
84 return new FlatBatchRemoveGroupBuilder()
85 .setGroupId(new GroupId(groupIdValue))
89 private static FlatBatchUpdateGroup createUpdateGroupBatch(final Uint32 groupIdValue) {
90 return new FlatBatchUpdateGroupBuilder()
91 .setOriginalBatchedGroup(new OriginalBatchedGroupBuilder()
92 .setGroupId(new GroupId(groupIdValue))
94 .setUpdatedBatchedGroup(new UpdatedBatchedGroupBuilder()
95 .setGroupId(new GroupId(groupIdValue))
101 public void testAdaptFlatBatchRemoveGroup() {
102 final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_REMOVE);
103 planStep.setBarrierAfter(true);
104 planStep.getTaskBag().addAll(Lists.newArrayList(
105 createRemoveGroupBatch(Uint32.ONE),
106 createRemoveGroupBatch(Uint32.TWO)));
108 final RemoveGroupsBatchInput removeGroupsBatchInput =
109 FlatBatchGroupAdapters.adaptFlatBatchRemoveGroup(planStep, NODE_REF);
110 Iterator<BatchRemoveGroups> iterator = removeGroupsBatchInput.nonnullBatchRemoveGroups().values().iterator();
112 Assert.assertTrue(removeGroupsBatchInput.isBarrierAfter());
113 Assert.assertEquals(2, removeGroupsBatchInput.getBatchRemoveGroups().size());
114 Assert.assertEquals(1L, iterator.next().getGroupId().getValue().longValue());
115 Assert.assertEquals(2L, iterator.next().getGroupId().getValue().longValue());
119 public void testAdaptFlatBatchUpdateGroup() {
120 final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_UPDATE);
121 planStep.setBarrierAfter(true);
122 planStep.getTaskBag().addAll(Lists.newArrayList(
123 createUpdateGroupBatch(Uint32.ONE),
124 createUpdateGroupBatch(Uint32.TWO)));
126 final UpdateGroupsBatchInput updateGroupsBatchInput =
127 FlatBatchGroupAdapters.adaptFlatBatchUpdateGroup(planStep, NODE_REF);
129 Assert.assertTrue(updateGroupsBatchInput.isBarrierAfter());
130 Assert.assertEquals(2, updateGroupsBatchInput.getBatchUpdateGroups().size());
131 Assert.assertEquals(1L, updateGroupsBatchInput.getBatchUpdateGroups().get(0)
132 .getUpdatedBatchedGroup().getGroupId().getValue().longValue());
133 Assert.assertEquals(2L, updateGroupsBatchInput.getBatchUpdateGroups().get(1)
134 .getUpdatedBatchedGroup().getGroupId().getValue().longValue());
138 public void testCreateBatchGroupChainingFunction_failures() {
139 final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder.<BatchGroupOutputListGrouping>failed()
140 .withError(RpcError.ErrorType.APPLICATION, "ut-groupError")
141 .withResult(new AddGroupsBatchOutputBuilder()
142 .setBatchFailedGroupsOutput(Lists.newArrayList(
143 createBatchFailedGroupsOutput(Uint16.ZERO, Uint32.ONE),
144 createBatchFailedGroupsOutput(Uint16.ONE, Uint32.TWO)
149 final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
150 .convertBatchGroupResult(3).apply(input);
151 Iterator<BatchFailure> iterator = rpcResult.getResult().nonnullBatchFailure().values().iterator();
153 Assert.assertFalse(rpcResult.isSuccessful());
154 Assert.assertEquals(1, rpcResult.getErrors().size());
155 Assert.assertEquals(2, rpcResult.getResult().nonnullBatchFailure().size());
156 Assert.assertEquals(3, iterator.next().getBatchOrder().intValue());
157 BatchFailure secondBatchFailure = iterator.next();
158 Assert.assertEquals(4, secondBatchFailure.getBatchOrder().intValue());
159 Assert.assertEquals(2L, ((FlatBatchFailureGroupIdCase) secondBatchFailure
160 .getBatchItemIdChoice()).getGroupId().getValue().longValue());
164 public void testCreateBatchGroupChainingFunction_successes() {
165 final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder
166 .<BatchGroupOutputListGrouping>success(new AddGroupsBatchOutputBuilder().build())
169 final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
170 .convertBatchGroupResult(0).apply(input);
172 Assert.assertTrue(rpcResult.isSuccessful());
173 Assert.assertEquals(0, rpcResult.getErrors().size());
174 Assert.assertEquals(0, rpcResult.getResult().nonnullBatchFailure().size());
177 private static BatchFailedGroupsOutput createBatchFailedGroupsOutput(final Uint16 batchOrder,
178 final Uint32 groupIdValue) {
179 return new BatchFailedGroupsOutputBuilder()
180 .setGroupId(new GroupId(groupIdValue))
181 .setBatchOrder(batchOrder)
185 private static BatchFailure createChainFailure(final Uint16 batchOrder, final Uint32 groupIdValue) {
186 return new BatchFailureBuilder()
187 .setBatchOrder(batchOrder)
188 .setBatchItemIdChoice(new FlatBatchFailureGroupIdCaseBuilder()
189 .setGroupId(new GroupId(groupIdValue))