openflowjava/openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/GroupModInputMessageFactoryTest.java

   1 /*
   2  * Copyright (c) 2013 Pantheon Technologies s.r.o. 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.openflowjava.protocol.impl.serialization.factories;
  10
  11 import io.netty.buffer.ByteBuf;
  12 import io.netty.buffer.UnpooledByteBufAllocator;
  13 import java.util.ArrayList;
  14 import java.util.List;
  15 import org.junit.Assert;
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  19 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  20 import org.opendaylight.openflowjava.protocol.impl.serialization.SerializerRegistryImpl;
  21 import org.opendaylight.openflowjava.protocol.impl.util.BufferHelper;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupId;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupModCommand;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupType;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.PortNumber;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GroupModInput;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GroupModInputBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.buckets.grouping.BucketsList;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.buckets.grouping.BucketsListBuilder;
  30 import org.opendaylight.yangtools.yang.common.Uint16;
  31 import org.opendaylight.yangtools.yang.common.Uint32;
  32
  33 /**
  34  * Unit tests for GroupModInputMessageFactory.
  35  *
  36  * @author timotej.kubas
  37  */
  38 public class GroupModInputMessageFactoryTest {
  39     private static final byte MESSAGE_TYPE = 15;
  40     private static final byte PADDING_IN_GROUP_MOD_MESSAGE = 1;
  41     private SerializerRegistry registry;
  42     private GroupModInputMessageFactory groupModFactory;
  43
  44     /**
  45      * Initializes serializer registry and stores correct factory in field.
  46      */
  47     @Before
  48     public void startUp() {
  49         registry = new SerializerRegistryImpl();
  50         registry.init();
  51         groupModFactory = new GroupModInputMessageFactory(false);
  52     }
  53
  54     /**
  55      * Testing of {@link GroupModInputMessageFactory} for correct translation from POJO.
  56      */
  57     @Test
  58     public void testGroupModInputMessage() throws Exception {
  59         GroupModInputBuilder builder = new GroupModInputBuilder();
  60         BufferHelper.setupHeader(builder, EncodeConstants.OF13_VERSION_ID);
  61         builder.setCommand(GroupModCommand.forValue(2));
  62         builder.setType(GroupType.forValue(3));
  63         builder.setGroupId(new GroupId(Uint32.valueOf(256)));
  64         List<BucketsList> exp = createBucketsList();
  65         builder.setBucketsList(exp);
  66         final GroupModInput message = builder.build();
  67
  68         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
  69
  70         // simulate parent message
  71         out.writeInt(1);
  72         out.writeZero(2);
  73         out.writeShort(3);
  74
  75         groupModFactory.serialize(message, out);
  76         // read parent message
  77         out.readInt();
  78         out.skipBytes(2);
  79         out.readShort();
  80
  81         BufferHelper.checkHeaderV13(out, MESSAGE_TYPE, 32);
  82         Assert.assertEquals("Wrong command", message.getCommand().getIntValue(), out.readUnsignedShort());
  83         Assert.assertEquals("Wrong type", message.getType().getIntValue(), out.readUnsignedByte());
  84         out.skipBytes(PADDING_IN_GROUP_MOD_MESSAGE);
  85         Assert.assertEquals("Wrong groupId", message.getGroupId().getValue().intValue(), out.readUnsignedInt());
  86         List<BucketsList> rec = createBucketsListFromBufer(out);
  87         Assert.assertArrayEquals("Wrong bucketList", exp.toArray(), rec.toArray());
  88     }
  89
  90     private static List<BucketsList> createBucketsList() {
  91         final List<BucketsList> bucketsList = new ArrayList<>();
  92         BucketsListBuilder bucketsBuilder = new BucketsListBuilder();
  93         bucketsBuilder.setWeight(Uint16.TEN);
  94         bucketsBuilder.setWatchPort(new PortNumber(Uint32.valueOf(65)));
  95         bucketsBuilder.setWatchGroup(Uint32.valueOf(22));
  96         BucketsList bucket = bucketsBuilder.build();
  97         bucketsList.add(bucket);
  98         return bucketsList;
  99     }
 100
 101     private static List<BucketsList> createBucketsListFromBufer(ByteBuf out) {
 102         final List<BucketsList> bucketsList = new ArrayList<>();
 103         BucketsListBuilder bucketsBuilder = new BucketsListBuilder();
 104         out.skipBytes(Short.BYTES);
 105         bucketsBuilder.setWeight(Uint16.fromShortBits(out.readShort()));
 106         bucketsBuilder.setWatchPort(new PortNumber(Uint32.fromIntBits(out.readInt())));
 107         bucketsBuilder.setWatchGroup(Uint32.fromIntBits(out.readInt()));
 108         out.skipBytes(4);
 109         BucketsList bucket = bucketsBuilder.build();
 110         bucketsList.add(bucket);
 111         return bucketsList;
 112     }
 113
 114     /**
 115      * Testing of {@link GroupModInputMessageFactory} for correct translation from POJO.
 116      */
 117     @Test
 118     public void testGroupModInputWithNoBuckets() throws Exception {
 119         GroupModInputBuilder builder = new GroupModInputBuilder();
 120         BufferHelper.setupHeader(builder, EncodeConstants.OF13_VERSION_ID);
 121         builder.setCommand(GroupModCommand.forValue(2));
 122         builder.setType(GroupType.forValue(3));
 123         builder.setGroupId(new GroupId(Uint32.valueOf(256)));
 124         GroupModInput message = builder.build();
 125
 126         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
 127         groupModFactory.serialize(message, out);
 128
 129         BufferHelper.checkHeaderV13(out, MESSAGE_TYPE, 16);
 130         Assert.assertEquals("Wrong command", message.getCommand().getIntValue(), out.readUnsignedShort());
 131         Assert.assertEquals("Wrong type", message.getType().getIntValue(), out.readUnsignedByte());
 132         out.skipBytes(PADDING_IN_GROUP_MOD_MESSAGE);
 133         Assert.assertEquals("Wrong groupId", message.getGroupId().getValue().intValue(), out.readUnsignedInt());
 134         Assert.assertTrue("Unexpected data", out.readableBytes() == 0);
 135     }
 136 }