openflow-protocol-impl/src/test/java/org/opendaylight/openflowjava/protocol/impl/serialization/match/OxmTcpSrcSerializerTest.java

   1 /*\r
   2  * Copyright (c) 2014 Pantheon Technologies s.r.o. and others. All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 \r
   9 package org.opendaylight.openflowjava.protocol.impl.serialization.match;\r
  10 \r
  11 import static org.junit.Assert.assertEquals;\r
  12 import static org.junit.Assert.assertTrue;\r
  13 import io.netty.buffer.ByteBuf;\r
  14 import io.netty.buffer.PooledByteBufAllocator;\r
  15 \r
  16 import org.junit.Test;\r
  17 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;\r
  18 import org.opendaylight.openflowjava.protocol.api.util.OxmMatchConstants;\r
  19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;\r
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.augments.rev131002.PortMatchEntry;\r
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.augments.rev131002.PortMatchEntryBuilder;\r
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.OpenflowBasicClass;\r
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.TcpSrc;\r
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.oxm.fields.grouping.MatchEntriesBuilder;\r
  25 \r
  26 /**\r
  27  * @author michal.polkorab\r
  28  *\r
  29  */\r
  30 public class OxmTcpSrcSerializerTest {\r
  31 \r
  32     OxmTcpSrcSerializer serializer = new OxmTcpSrcSerializer();\r
  33 \r
  34     /**\r
  35      * Test correct serialization\r
  36      */\r
  37     @Test\r
  38     public void testSerialize() {\r
  39         MatchEntriesBuilder builder = prepareMatchEntry(512);\r
  40         \r
  41         ByteBuf buffer = PooledByteBufAllocator.DEFAULT.buffer();\r
  42         serializer.serialize(builder.build(), buffer);\r
  43 \r
  44         checkHeader(buffer, false);\r
  45         assertEquals("Wrong value", 512, buffer.readUnsignedShort());\r
  46         assertTrue("Unexpected data", buffer.readableBytes() == 0);\r
  47     }\r
  48 \r
  49     /**\r
  50      * Test correct header serialization\r
  51      */\r
  52     @Test\r
  53     public void testSerializeHeader() {\r
  54         MatchEntriesBuilder builder = prepareHeader(false);\r
  55         \r
  56         ByteBuf buffer = PooledByteBufAllocator.DEFAULT.buffer();\r
  57         serializer.serializeHeader(builder.build(), buffer);\r
  58 \r
  59         checkHeader(buffer, false);\r
  60         assertTrue("Unexpected data", buffer.readableBytes() == 0);\r
  61     }\r
  62 \r
  63     /**\r
  64      * Test correct oxm-class return value\r
  65      */\r
  66     @Test\r
  67     public void testGetOxmClassCode() {\r
  68         assertEquals("Wrong oxm-class", OxmMatchConstants.OPENFLOW_BASIC_CLASS, serializer.getOxmClassCode());\r
  69     }\r
  70 \r
  71     /**\r
  72      * Test correct oxm-field return value\r
  73      */\r
  74     @Test\r
  75     public void getOxmFieldCode() {\r
  76         assertEquals("Wrong oxm-class", OxmMatchConstants.TCP_SRC, serializer.getOxmFieldCode());\r
  77     }\r
  78 \r
  79     /**\r
  80      * Test correct value length return value\r
  81      */\r
  82     @Test\r
  83     public void testGetValueLength() {\r
  84         assertEquals("Wrong value length", EncodeConstants.SIZE_OF_SHORT_IN_BYTES, serializer.getValueLength());\r
  85     }\r
  86 \r
  87     private static MatchEntriesBuilder prepareMatchEntry(int value) {\r
  88         MatchEntriesBuilder builder = prepareHeader(false);\r
  89         PortMatchEntryBuilder portBuilder = new PortMatchEntryBuilder();\r
  90         portBuilder.setPort(new PortNumber(value));\r
  91         builder.addAugmentation(PortMatchEntry.class, portBuilder.build());\r
  92         return builder;\r
  93     }\r
  94 \r
  95     private static MatchEntriesBuilder prepareHeader(boolean hasMask) {\r
  96         MatchEntriesBuilder builder = new MatchEntriesBuilder();\r
  97         builder.setOxmClass(OpenflowBasicClass.class);\r
  98         builder.setOxmMatchField(TcpSrc.class);\r
  99         builder.setHasMask(hasMask);\r
 100         return builder;\r
 101     }\r
 102 \r
 103     private static void checkHeader(ByteBuf buffer, boolean hasMask) {\r
 104         assertEquals("Wrong oxm-class", OxmMatchConstants.OPENFLOW_BASIC_CLASS, buffer.readUnsignedShort());\r
 105         short fieldAndMask = buffer.readUnsignedByte();\r
 106         assertEquals("Wrong oxm-field", OxmMatchConstants.TCP_SRC, fieldAndMask >>> 1);\r
 107         assertEquals("Wrong hasMask", hasMask, (fieldAndMask & 1) != 0);\r
 108         assertEquals("Wrong length", EncodeConstants.SIZE_OF_SHORT_IN_BYTES, buffer.readUnsignedByte());\r
 109     }\r
 110 }