Update OF header lenght

[openflowjava.git] / openflowjava-util / src / test / java / org / opendaylight / openflowjava / util / ByteBufUtilsTest.java

diff --git a/openflowjava-util/src/test/java/org/opendaylight/openflowjava/util/ByteBufUtilsTest.java b/openflowjava-util/src/test/java/org/opendaylight/openflowjava/util/ByteBufUtilsTest.java
index 8196e69ba095f5f035f1a557e48ef7de0d6dd333..b7f0f593caed035f5481f86576537c4c7979c42a 100644 (file)
--- a/openflowjava-util/src/test/java/org/opendaylight/openflowjava/util/ByteBufUtilsTest.java
+++ b/openflowjava-util/src/test/java/org/opendaylight/openflowjava/util/ByteBufUtilsTest.java
@@ -448,4 +448,22 @@ public class ByteBufUtilsTest {
         Assert.assertTrue(bytes.length == shorts.size()*2);
         Assert.assertArrayEquals(EXPECTEDVALUES1AND255, bytes);
     }
+
+    @Test
+    public void testUpdateHeader() throws IOException {
+        ByteBuf buffer = PooledByteBufAllocator.DEFAULT.buffer();
+        buffer.writeInt(1);
+        int start = buffer.writerIndex();
+        buffer.writeShort(4);
+        buffer.writeShort(EncodeConstants.EMPTY_LENGTH);
+        buffer.writeLong(8);
+        int end = buffer.writerIndex();
+
+        ByteBufUtils.updateOFHeaderLength(buffer, start);
+        Assert.assertEquals(buffer.readInt(), 1);
+        Assert.assertEquals(buffer.readShort(), 4);
+        Assert.assertEquals(buffer.readShort(), 12);
+        Assert.assertEquals(buffer.readLong(), 8l);
+        Assert.assertEquals(buffer.getShort(start + EncodeConstants.OFHEADER_LENGTH_INDEX), end - start);
+    }
 }