openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/services/batch/FlatBatchGroupAdaptersTest.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 package org.opendaylight.openflowplugin.impl.services.batch;
   9
  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.binding.util.BindingMap;
  44 import org.opendaylight.yangtools.yang.common.RpcError;
  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;
  49
  50 /**
  51  * Test for {@link FlatBatchGroupAdapters}.
  52  */
  53 public class FlatBatchGroupAdaptersTest {
  54
  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);
  59
  60     @Test
  61     public void testAdaptFlatBatchAddGroup() {
  62         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_ADD);
  63         planStep.setBarrierAfter(true);
  64         planStep.getTaskBag().addAll(Lists.newArrayList(
  65                 createAddGroupBatch(Uint32.ONE),
  66                 createAddGroupBatch(Uint32.TWO)));
  67
  68         final AddGroupsBatchInput addGroupsBatchInput =
  69                 FlatBatchGroupAdapters.adaptFlatBatchAddGroup(planStep, NODE_REF);
  70         Iterator<BatchAddGroups> iterator = addGroupsBatchInput.nonnullBatchAddGroups().values().iterator();
  71
  72         Assert.assertTrue(addGroupsBatchInput.getBarrierAfter());
  73         Assert.assertEquals(2, addGroupsBatchInput.getBatchAddGroups().size());
  74         Assert.assertEquals(1L, iterator.next().getGroupId().getValue().longValue());
  75         Assert.assertEquals(2L, iterator.next().getGroupId().getValue().longValue());
  76     }
  77
  78     private static FlatBatchAddGroup createAddGroupBatch(final Uint32 groupIdValue) {
  79         return new FlatBatchAddGroupBuilder()
  80                 .setGroupId(new GroupId(groupIdValue))
  81                 .setBatchOrder(groupIdValue.toUint16())
  82                 .build();
  83     }
  84
  85     private static FlatBatchRemoveGroup createRemoveGroupBatch(final Uint32 groupIdValue) {
  86         return new FlatBatchRemoveGroupBuilder()
  87                 .setGroupId(new GroupId(groupIdValue))
  88                 .setBatchOrder(groupIdValue.toUint16())
  89                 .build();
  90     }
  91
  92     private static FlatBatchUpdateGroup createUpdateGroupBatch(final Uint32 groupIdValue) {
  93         return new FlatBatchUpdateGroupBuilder()
  94                 .setOriginalBatchedGroup(new OriginalBatchedGroupBuilder()
  95                         .setGroupId(new GroupId(groupIdValue))
  96                         .build())
  97                 .setUpdatedBatchedGroup(new UpdatedBatchedGroupBuilder()
  98                         .setGroupId(new GroupId(groupIdValue))
  99                         .build())
 100                 .setBatchOrder(groupIdValue.toUint16())
 101                 .build();
 102     }
 103
 104     @Test
 105     public void testAdaptFlatBatchRemoveGroup() {
 106         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_REMOVE);
 107         planStep.setBarrierAfter(true);
 108         planStep.getTaskBag().addAll(Lists.newArrayList(
 109                 createRemoveGroupBatch(Uint32.ONE),
 110                 createRemoveGroupBatch(Uint32.TWO)));
 111
 112         final RemoveGroupsBatchInput removeGroupsBatchInput =
 113                 FlatBatchGroupAdapters.adaptFlatBatchRemoveGroup(planStep, NODE_REF);
 114         Iterator<BatchRemoveGroups> iterator = removeGroupsBatchInput.nonnullBatchRemoveGroups().values().iterator();
 115
 116         Assert.assertTrue(removeGroupsBatchInput.getBarrierAfter());
 117         Assert.assertEquals(2, removeGroupsBatchInput.getBatchRemoveGroups().size());
 118         Assert.assertEquals(1L, iterator.next().getGroupId().getValue().longValue());
 119         Assert.assertEquals(2L, iterator.next().getGroupId().getValue().longValue());
 120     }
 121
 122     @Test
 123     public void testAdaptFlatBatchUpdateGroup() {
 124         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_UPDATE);
 125         planStep.setBarrierAfter(true);
 126         planStep.getTaskBag().addAll(Lists.newArrayList(
 127                 createUpdateGroupBatch(Uint32.ONE),
 128                 createUpdateGroupBatch(Uint32.TWO)));
 129
 130         final UpdateGroupsBatchInput updateGroupsBatchInput =
 131                 FlatBatchGroupAdapters.adaptFlatBatchUpdateGroup(planStep, NODE_REF);
 132
 133         Assert.assertTrue(updateGroupsBatchInput.getBarrierAfter());
 134         Assert.assertEquals(2, updateGroupsBatchInput.getBatchUpdateGroups().size());
 135         Assert.assertEquals(1L, updateGroupsBatchInput.getBatchUpdateGroups().get(0)
 136                 .getUpdatedBatchedGroup().getGroupId().getValue().longValue());
 137         Assert.assertEquals(2L, updateGroupsBatchInput.getBatchUpdateGroups().get(1)
 138                 .getUpdatedBatchedGroup().getGroupId().getValue().longValue());
 139     }
 140
 141     @Test
 142     public void testCreateBatchGroupChainingFunction_failures() {
 143         final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder.<BatchGroupOutputListGrouping>failed()
 144                 .withError(RpcError.ErrorType.APPLICATION, "ut-groupError")
 145                 .withResult(new AddGroupsBatchOutputBuilder()
 146                         .setBatchFailedGroupsOutput(BindingMap.ordered(
 147                                 createBatchFailedGroupsOutput(Uint16.ZERO, Uint32.ONE),
 148                                 createBatchFailedGroupsOutput(Uint16.ONE, Uint32.TWO)
 149                         ))
 150                         .build())
 151                 .build();
 152
 153         final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
 154                 .convertBatchGroupResult(3).apply(input);
 155         Iterator<BatchFailure> iterator = rpcResult.getResult().nonnullBatchFailure().values().iterator();
 156
 157         Assert.assertFalse(rpcResult.isSuccessful());
 158         Assert.assertEquals(1, rpcResult.getErrors().size());
 159         Assert.assertEquals(2, rpcResult.getResult().nonnullBatchFailure().size());
 160         Assert.assertEquals(3, iterator.next().getBatchOrder().intValue());
 161         BatchFailure secondBatchFailure = iterator.next();
 162         Assert.assertEquals(4, secondBatchFailure.getBatchOrder().intValue());
 163         Assert.assertEquals(2L, ((FlatBatchFailureGroupIdCase) secondBatchFailure
 164                 .getBatchItemIdChoice()).getGroupId().getValue().longValue());
 165     }
 166
 167     @Test
 168     public void testCreateBatchGroupChainingFunction_successes() {
 169         final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder
 170                 .<BatchGroupOutputListGrouping>success(new AddGroupsBatchOutputBuilder().build())
 171                 .build();
 172
 173         final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
 174                 .convertBatchGroupResult(0).apply(input);
 175
 176         Assert.assertTrue(rpcResult.isSuccessful());
 177         Assert.assertEquals(0, rpcResult.getErrors().size());
 178         Assert.assertEquals(0, rpcResult.getResult().nonnullBatchFailure().size());
 179     }
 180
 181     private static BatchFailedGroupsOutput createBatchFailedGroupsOutput(final Uint16 batchOrder,
 182             final Uint32 groupIdValue) {
 183         return new BatchFailedGroupsOutputBuilder()
 184                 .setGroupId(new GroupId(groupIdValue))
 185                 .setBatchOrder(batchOrder)
 186                 .build();
 187     }
 188
 189     private static BatchFailure createChainFailure(final Uint16 batchOrder, final Uint32 groupIdValue) {
 190         return new BatchFailureBuilder()
 191                 .setBatchOrder(batchOrder)
 192                 .setBatchItemIdChoice(new FlatBatchFailureGroupIdCaseBuilder()
 193                         .setGroupId(new GroupId(groupIdValue))
 194                         .build())
 195                 .build();
 196     }
 197 }