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
   9 package org.opendaylight.openflowplugin.impl.services.batch;
  10
  11 import com.google.common.collect.Lists;
  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.batch.group.input.update.grouping.OriginalBatchedGroupBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.input.update.grouping.UpdatedBatchedGroupBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.output.list.grouping.BatchFailedGroupsOutput;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groups.service.rev160315.batch.group.output.list.grouping.BatchFailedGroupsOutputBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  41 import org.opendaylight.yangtools.yang.common.RpcError;
  42 import org.opendaylight.yangtools.yang.common.RpcResult;
  43 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  44
  45 /**
  46  * Test for {@link FlatBatchGroupAdapters}.
  47  */
  48 public class FlatBatchGroupAdaptersTest {
  49
  50     private static final NodeId NODE_ID = new NodeId("ut-node-id");
  51     private static final InstanceIdentifier<Node> NODE_II = InstanceIdentifier.create(Nodes.class)
  52             .child(Node.class, new NodeKey(NODE_ID));
  53     private static final NodeRef NODE_REF = new NodeRef(NODE_II);
  54
  55     @Test
  56     public void testAdaptFlatBatchAddGroup() throws Exception {
  57         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_ADD);
  58         planStep.setBarrierAfter(true);
  59         planStep.getTaskBag().addAll(Lists.newArrayList(
  60                 createAddGroupBatch(1L),
  61                 createAddGroupBatch(2L)));
  62
  63         final AddGroupsBatchInput addGroupsBatchInput = FlatBatchGroupAdapters.adaptFlatBatchAddGroup(planStep, NODE_REF);
  64
  65         Assert.assertTrue(addGroupsBatchInput.isBarrierAfter());
  66         Assert.assertEquals(2, addGroupsBatchInput.getBatchAddGroups().size());
  67         Assert.assertEquals(1L, addGroupsBatchInput.getBatchAddGroups().get(0).getGroupId().getValue().longValue());
  68         Assert.assertEquals(2L, addGroupsBatchInput.getBatchAddGroups().get(1).getGroupId().getValue().longValue());
  69     }
  70
  71     private FlatBatchAddGroup createAddGroupBatch(final long groupIdValue) {
  72         return new FlatBatchAddGroupBuilder()
  73                 .setGroupId(new GroupId(groupIdValue))
  74                 .build();
  75     }
  76
  77     private FlatBatchRemoveGroup createRemoveGroupBatch(final long groupIdValue) {
  78         return new FlatBatchRemoveGroupBuilder()
  79                 .setGroupId(new GroupId(groupIdValue))
  80                 .build();
  81     }
  82
  83     private FlatBatchUpdateGroup createUpdateGroupBatch(final long groupIdValue) {
  84         return new FlatBatchUpdateGroupBuilder()
  85                 .setOriginalBatchedGroup(new OriginalBatchedGroupBuilder()
  86                         .setGroupId(new GroupId(groupIdValue))
  87                         .build())
  88                 .setUpdatedBatchedGroup(new UpdatedBatchedGroupBuilder()
  89                         .setGroupId(new GroupId(groupIdValue))
  90                         .build())
  91                 .build();
  92     }
  93
  94     @Test
  95     public void testAdaptFlatBatchRemoveGroup() throws Exception {
  96         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_REMOVE);
  97         planStep.setBarrierAfter(true);
  98         planStep.getTaskBag().addAll(Lists.newArrayList(
  99                 createRemoveGroupBatch(1L),
 100                 createRemoveGroupBatch(2L)));
 101
 102         final RemoveGroupsBatchInput removeGroupsBatchInput = FlatBatchGroupAdapters.adaptFlatBatchRemoveGroup(planStep, NODE_REF);
 103
 104         Assert.assertTrue(removeGroupsBatchInput.isBarrierAfter());
 105         Assert.assertEquals(2, removeGroupsBatchInput.getBatchRemoveGroups().size());
 106         Assert.assertEquals(1L, removeGroupsBatchInput.getBatchRemoveGroups().get(0).getGroupId().getValue().longValue());
 107         Assert.assertEquals(2L, removeGroupsBatchInput.getBatchRemoveGroups().get(1).getGroupId().getValue().longValue());
 108     }
 109
 110     @Test
 111     public void testAdaptFlatBatchUpdateGroup() throws Exception {
 112         final BatchPlanStep planStep = new BatchPlanStep(BatchStepType.FLOW_UPDATE);
 113         planStep.setBarrierAfter(true);
 114         planStep.getTaskBag().addAll(Lists.newArrayList(
 115                 createUpdateGroupBatch(1L),
 116                 createUpdateGroupBatch(2L)));
 117
 118         final UpdateGroupsBatchInput updateGroupsBatchInput = FlatBatchGroupAdapters.adaptFlatBatchUpdateGroup(planStep, NODE_REF);
 119
 120         Assert.assertTrue(updateGroupsBatchInput.isBarrierAfter());
 121         Assert.assertEquals(2, updateGroupsBatchInput.getBatchUpdateGroups().size());
 122         Assert.assertEquals(1L, updateGroupsBatchInput.getBatchUpdateGroups().get(0).getUpdatedBatchedGroup().getGroupId().getValue().longValue());
 123         Assert.assertEquals(2L, updateGroupsBatchInput.getBatchUpdateGroups().get(1).getUpdatedBatchedGroup().getGroupId().getValue().longValue());
 124     }
 125
 126     @Test
 127     public void testCreateBatchGroupChainingFunction_failures() throws Exception {
 128         final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder.<BatchGroupOutputListGrouping>failed()
 129                 .withError(RpcError.ErrorType.APPLICATION, "ut-groupError")
 130                 .withResult(new AddGroupsBatchOutputBuilder()
 131                         .setBatchFailedGroupsOutput(Lists.newArrayList(
 132                                 createBatchFailedGroupsOutput(0, 1L),
 133                                 createBatchFailedGroupsOutput(1, 2L)
 134                         ))
 135                         .build())
 136                 .build();
 137
 138         final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
 139                 .convertBatchGroupResult(3).apply(input);
 140
 141         Assert.assertFalse(rpcResult.isSuccessful());
 142         Assert.assertEquals(1, rpcResult.getErrors().size());
 143         Assert.assertEquals(2, rpcResult.getResult().getBatchFailure().size());
 144         Assert.assertEquals(3, rpcResult.getResult().getBatchFailure().get(0).getBatchOrder().intValue());
 145         Assert.assertEquals(4, rpcResult.getResult().getBatchFailure().get(1).getBatchOrder().intValue());
 146         Assert.assertEquals(2L, ((FlatBatchFailureGroupIdCase) rpcResult.getResult().getBatchFailure().get(1).getBatchItemIdChoice()).getGroupId().getValue().longValue());
 147     }
 148
 149     @Test
 150     public void testCreateBatchGroupChainingFunction_successes() throws Exception {
 151         final RpcResult<BatchGroupOutputListGrouping> input = RpcResultBuilder
 152                 .<BatchGroupOutputListGrouping>success(new AddGroupsBatchOutputBuilder().build())
 153                 .build();
 154
 155         final RpcResult<ProcessFlatBatchOutput> rpcResult = FlatBatchGroupAdapters
 156                 .convertBatchGroupResult(0).apply(input);
 157
 158         Assert.assertTrue(rpcResult.isSuccessful());
 159         Assert.assertEquals(0, rpcResult.getErrors().size());
 160         Assert.assertEquals(0, rpcResult.getResult().getBatchFailure().size());
 161     }
 162
 163     private BatchFailedGroupsOutput createBatchFailedGroupsOutput(final Integer batchOrder, final long groupIdValue) {
 164         return new BatchFailedGroupsOutputBuilder()
 165                 .setGroupId(new GroupId(groupIdValue))
 166                 .setBatchOrder(batchOrder)
 167                 .build();
 168     }
 169
 170     private BatchFailure createChainFailure(final int batchOrder, final long groupIdValue) {
 171         return new BatchFailureBuilder()
 172                 .setBatchOrder(batchOrder)
 173                 .setBatchItemIdChoice(new FlatBatchFailureGroupIdCaseBuilder()
 174                         .setGroupId(new GroupId(groupIdValue))
 175                         .build())
 176                 .build();
 177     }
 178 }