openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/factories/HelloInputMessageFactoryTest.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
  14 import java.util.ArrayList;
  15 import java.util.List;
  16
  17 import org.junit.Assert;
  18 import org.junit.Before;
  19 import org.junit.Test;
  20 import org.opendaylight.openflowjava.protocol.api.extensibility.OFSerializer;
  21 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  22 import org.opendaylight.openflowjava.protocol.api.keys.MessageTypeKey;
  23 import org.opendaylight.openflowjava.protocol.impl.serialization.SerializerRegistryImpl;
  24 import org.opendaylight.openflowjava.protocol.impl.util.BufferHelper;
  25 import org.opendaylight.openflowjava.util.ByteBufUtils;
  26 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.HelloElementType;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.HelloInput;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.HelloInputBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.hello.Elements;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.hello.ElementsBuilder;
  32 import org.slf4j.Logger;
  33 import org.slf4j.LoggerFactory;
  34
  35 /**
  36  * @author michal.polkorab
  37  *
  38  */
  39 public class HelloInputMessageFactoryTest {
  40
  41     private static final Logger LOG = LoggerFactory.getLogger(HelloInputMessageFactoryTest.class);
  42     private SerializerRegistry registry;
  43     private OFSerializer<HelloInput> helloFactory;
  44
  45     /**
  46      * Initializes serializer registry and stores correct factory in field
  47      */
  48     @Before
  49     public void startUp() {
  50         registry = new SerializerRegistryImpl();
  51         registry.init();
  52         helloFactory = registry.getSerializer(
  53                 new MessageTypeKey<>(EncodeConstants.OF13_VERSION_ID, HelloInput.class));
  54     }
  55
  56     /**
  57      * Testing of {@link HelloInputMessageFactory} for correct translation from POJO
  58      * @throws Exception
  59      */
  60     @Test
  61     public void testWithoutElementsSet() throws Exception {
  62         HelloInputBuilder hib = new HelloInputBuilder();
  63         BufferHelper.setupHeader(hib, EncodeConstants.OF13_VERSION_ID);
  64         HelloInput hi = hib.build();
  65
  66         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
  67         helloFactory.serialize(hi, out);
  68
  69         BufferHelper.checkHeaderV13(out,(byte) 0, EncodeConstants.OFHEADER_SIZE);
  70     }
  71
  72     /**
  73      * Testing of {@link HelloInputMessageFactory} for correct translation from POJO
  74      * @throws Exception
  75      */
  76     @Test
  77     public void testWith4BitVersionBitmap() throws Exception {
  78         int lengthOfBitmap = 4;
  79         HelloInputBuilder builder = new HelloInputBuilder();
  80         BufferHelper.setupHeader(builder, EncodeConstants.OF13_VERSION_ID);
  81         List<Elements> expectedElement = createElement(lengthOfBitmap);
  82         builder.setElements(expectedElement);
  83         HelloInput message = builder.build();
  84
  85         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
  86         helloFactory.serialize(message, out);
  87         if (LOG.isDebugEnabled()) {
  88             LOG.debug("bytebuf: {}", ByteBufUtils.byteBufToHexString(out));
  89         }
  90
  91         BufferHelper.checkHeaderV13(out, (byte) 0, 16);
  92         Elements element = readElement(out).get(0);
  93         Assert.assertEquals("Wrong element type", expectedElement.get(0).getType(), element.getType());
  94         Elements comparation = createComparationElement(lengthOfBitmap).get(0);
  95         Assert.assertArrayEquals("Wrong element bitmap", comparation.getVersionBitmap().toArray(), element.getVersionBitmap().toArray());
  96     }
  97
  98     /**
  99      * Testing of {@link HelloInputMessageFactory} for correct translation from POJO
 100      * @throws Exception
 101      */
 102     @Test
 103     public void testWith64BitVersionBitmap() throws Exception {
 104         int lengthOfBitmap = 64;
 105         HelloInputBuilder builder = new HelloInputBuilder();
 106         BufferHelper.setupHeader(builder, EncodeConstants.OF13_VERSION_ID);
 107         List<Elements> expectedElement = createElement(lengthOfBitmap);
 108         builder.setElements(expectedElement);
 109         HelloInput message = builder.build();
 110
 111         ByteBuf out = UnpooledByteBufAllocator.DEFAULT.buffer();
 112         helloFactory.serialize(message, out);
 113         if (LOG.isDebugEnabled()) {
 114             LOG.debug("bytebuf: ", ByteBufUtils.byteBufToHexString(out));
 115         }
 116
 117         BufferHelper.checkHeaderV13(out, (byte) 0, 24);
 118         Elements element = readElement(out).get(0);
 119         Assert.assertEquals("Wrong element type", expectedElement.get(0).getType(), element.getType());
 120         Elements comparation = createComparationElement(lengthOfBitmap).get(0);
 121         Assert.assertArrayEquals("Wrong element bitmap", comparation.getVersionBitmap().toArray(), element.getVersionBitmap().toArray());
 122     }
 123
 124     private static List<Elements> createElement(int lengthOfBitmap) {
 125         ElementsBuilder elementsBuilder = new ElementsBuilder();
 126         List<Elements> elementsList = new ArrayList<>();
 127         List<Boolean> booleanList = new ArrayList<>();
 128         for (int i = 0; i < lengthOfBitmap; i++) {
 129             booleanList.add(true);
 130         }
 131         elementsBuilder.setType(HelloElementType.forValue(1));
 132         elementsBuilder.setVersionBitmap(booleanList);
 133         elementsList.add(elementsBuilder.build());
 134         return elementsList;
 135     }
 136
 137     private static List<Elements> createComparationElement(int lengthOfBitmap) {
 138         ElementsBuilder elementsBuilder = new ElementsBuilder();
 139         List<Elements> elementsList = new ArrayList<>();
 140         List<Boolean> booleanList = new ArrayList<>();
 141         for (int i = 0; i < lengthOfBitmap; i++) {
 142             booleanList.add(true);
 143         }
 144         if ((lengthOfBitmap % Integer.SIZE) != 0) {
 145             for (int i = 0; i < (Integer.SIZE - (lengthOfBitmap % Integer.SIZE)); i++) {
 146                 booleanList.add(false);
 147             }
 148         }
 149         LOG.debug("boolsize {}", booleanList.size());
 150         elementsBuilder.setType(HelloElementType.forValue(1));
 151         elementsBuilder.setVersionBitmap(booleanList);
 152         elementsList.add(elementsBuilder.build());
 153         return elementsList;
 154     }
 155
 156     private static List<Elements> readElement(ByteBuf input) {
 157         List<Elements> elementsList = new ArrayList<>();
 158         while (input.readableBytes() > 0) {
 159             ElementsBuilder elementsBuilder = new ElementsBuilder();
 160             int type = input.readUnsignedShort();
 161             int elementLength = input.readUnsignedShort();
 162             if (type == HelloElementType.VERSIONBITMAP.getIntValue()) {
 163                 elementsBuilder.setType(HelloElementType.forValue(type));
 164                 int[] versionBitmap = new int[(elementLength - 4) / 4];
 165                 for (int i = 0; i < versionBitmap.length; i++) {
 166                     versionBitmap[i] = (int) input.readUnsignedInt();
 167                 }
 168                 elementsBuilder.setVersionBitmap(readVersionBitmap(versionBitmap));
 169                 int paddingRemainder = elementLength % EncodeConstants.PADDING;
 170                 if (paddingRemainder != 0) {
 171                     input.readBytes(EncodeConstants.PADDING - paddingRemainder);
 172                 }
 173             }
 174             elementsList.add(elementsBuilder.build());
 175         }
 176         return elementsList;
 177     }
 178
 179     private static List<Boolean> readVersionBitmap(int[] input){
 180         List<Boolean> versionBitmapList = new ArrayList<>();
 181         for (int i = 0; i < input.length; i++) {
 182             int mask = input[i];
 183             for (int j = 0; j < Integer.SIZE; j++) {
 184                     versionBitmapList.add((mask & (1<<j)) != 0);
 185             }
 186         }
 187         return versionBitmapList;
 188     }
 189
 190 }