util/src/test/java/org/opendaylight/protocol/util/ByteBufWriteUtilTest.java

   1 /*
   2  * Copyright (c) 2014 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 package org.opendaylight.protocol.util;
   9
  10 import static org.junit.Assert.assertArrayEquals;
  11 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeFloat32;
  12 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeIpv4Address;
  13 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeIpv4Prefix;
  14 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeIpv6Address;
  15 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeIpv6Prefix;
  16 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeMedium;
  17
  18 import io.netty.buffer.ByteBuf;
  19 import io.netty.buffer.Unpooled;
  20 import org.junit.Test;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Address;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Prefix;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ieee754.rev130819.Float32;
  26
  27 public class ByteBufWriteUtilTest {
  28     private static final byte[] FOUR_BYTE_ZEROS = { 0, 0, 0, 0 };
  29
  30     @Test
  31     public void testWriteMediumValue() {
  32         final byte[] result = { 0, 0, 5 };
  33         final ByteBuf output = Unpooled.buffer(3);
  34         writeMedium(5, output);
  35         assertArrayEquals(result, output.array());
  36
  37         output.clear();
  38         final byte[] resultZero = { 0, 0, 0 };
  39         writeMedium(null, output);
  40         assertArrayEquals(resultZero, output.array());
  41     }
  42
  43     @Test
  44     public void testWriteIpv4Address() {
  45         final byte[] result = { 127, 0, 0, 1 };
  46         final ByteBuf output = Unpooled.buffer(Ipv4Util.IP4_LENGTH);
  47         writeIpv4Address(new Ipv4Address("127.0.0.1"), output);
  48         assertArrayEquals(result, output.array());
  49
  50         output.clear();
  51         writeIpv4Address(null, output);
  52         assertArrayEquals(FOUR_BYTE_ZEROS, output.array());
  53     }
  54
  55     @Test
  56     public void testWriteIpv4Prefix() {
  57         final byte[] result = { 123, 122, 4, 5, 8 };
  58         final ByteBuf output = Unpooled.buffer(ByteBufWriteUtil.IPV4_PREFIX_BYTE_LENGTH);
  59         writeIpv4Prefix(new Ipv4Prefix("123.122.4.5/8"), output);
  60         assertArrayEquals(result, output.array());
  61
  62         output.clear();
  63         final byte[] zeroResult = { 0, 0, 0, 0, 0 };
  64         writeIpv4Prefix(null, output);
  65         assertArrayEquals(zeroResult, output.array());
  66     }
  67
  68     @Test
  69     public void testWriteIpv6Address() {
  70         final byte[] result = { 0x20, (byte) 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  71             0x00, 0x00, 0x01 };
  72         final ByteBuf output = Unpooled.buffer(Ipv6Util.IPV6_LENGTH);
  73         writeIpv6Address(new Ipv6Address("2001::1"), output);
  74         assertArrayEquals(result, output.array());
  75
  76         output.clear();
  77         final byte[] zeroResult = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,};
  78         writeIpv6Address(null, output);
  79         assertArrayEquals(zeroResult, output.array());
  80     }
  81
  82     @Test
  83     public void testWriteIpv6Prefix() {
  84         final byte[] result = { 0x20, (byte) 0x01, 0x0d, (byte) 0xb8, 0x00, 0x01, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00,
  85             0x00, 0x00, 0x00, 0x00, 0x40 };
  86         final ByteBuf output = Unpooled.buffer(ByteBufWriteUtil.IPV6_PREFIX_BYTE_LENGTH);
  87         writeIpv6Prefix(new Ipv6Prefix("2001:db8:1:2::/64"), output);
  88         assertArrayEquals(result, output.array());
  89
  90         output.clear();
  91         final byte[] zeroResult = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
  92         writeIpv6Prefix(null, output);
  93         assertArrayEquals(zeroResult, output.array());
  94     }
  95
  96     @Test
  97     public void testWriteFloat32() {
  98         final byte[] result = { 0, 0, 0, 5 };
  99         final ByteBuf output = Unpooled.buffer(Float.BYTES);
 100         writeFloat32(new Float32(result), output);
 101         assertArrayEquals(result, output.array());
 102
 103         output.clear();
 104         writeFloat32(null, output);
 105         assertArrayEquals(FOUR_BYTE_ZEROS, output.array());
 106     }
 107 }