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