bgp/extensions/mvpn/src/test/java/org/opendaylight/protocol/bgp/mvpn/impl/attributes/tunnel/identifier/PAddressPMulticastGroupUtilTest.java

   1 /*
   2  * Copyright (c) 2016 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.bgp.mvpn.impl.attributes.tunnel.identifier;
   9
  10 import static org.junit.Assert.assertArrayEquals;
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNull;
  13
  14 import io.netty.buffer.ByteBuf;
  15 import io.netty.buffer.Unpooled;
  16 import org.junit.Test;
  17 import org.opendaylight.protocol.util.ByteArray;
  18 import org.opendaylight.protocol.util.Ipv4Util;
  19 import org.opendaylight.protocol.util.Ipv6Util;
  20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6AddressNoZone;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pmsi.tunnel.rev180329.PAddressPMulticastGroup;
  24
  25 public final class PAddressPMulticastGroupUtilTest {
  26     private static final String IPV6_MODEL = "2001::1";
  27     private static final IpAddressNoZone IPV6 = new IpAddressNoZone(new Ipv6AddressNoZone(IPV6_MODEL));
  28     private static final byte[] IPV4_ADDRESS_EXPECTED = {(byte) 0x01, (byte) 0x01, (byte) 0x01, (byte) 0x01};
  29     private static final byte[] IPV6_ADDRESS_EXPECTED = {
  30         0x20, (byte) 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01
  31     };
  32     private static final byte[] SENDER_P_MULTICAST_GROUP_EXPECTED = {
  33         (byte) 0x01, (byte) 0x01, (byte) 0x01, (byte) 0x01, (byte) 0x17, (byte) 0x01, (byte) 0x01, (byte) 0x01
  34     };
  35     private static final IpAddressNoZone IP_ADDRESS = new IpAddressNoZone(new Ipv4AddressNoZone("1.1.1.1"));
  36     private static final IpAddressNoZone P_MULTICAST = new IpAddressNoZone(new Ipv4AddressNoZone("23.1.1.1"));
  37
  38     @Test
  39     public void parseIpAddress() {
  40         final ByteBuf ipv4Actual = Unpooled.buffer();
  41         PAddressPMulticastGroupUtil.serializeIpAddress(IP_ADDRESS, ipv4Actual);
  42         assertArrayEquals(IPV4_ADDRESS_EXPECTED, ByteArray.readAllBytes(ipv4Actual));
  43         final ByteBuf ipv6Actual = Unpooled.buffer();
  44         PAddressPMulticastGroupUtil.serializeIpAddress(IPV6, ipv6Actual);
  45         assertArrayEquals(IPV6_ADDRESS_EXPECTED, ByteArray.readAllBytes(ipv6Actual));
  46         assertEquals(IP_ADDRESS, PAddressPMulticastGroupUtil.parseIpAddress(Ipv4Util.IP4_LENGTH,
  47                 Unpooled.wrappedBuffer(IPV4_ADDRESS_EXPECTED)));
  48         assertEquals(IPV6, PAddressPMulticastGroupUtil.parseIpAddress(Ipv6Util.IPV6_LENGTH,
  49                 Unpooled.wrappedBuffer(IPV6_ADDRESS_EXPECTED)));
  50         assertNull(PAddressPMulticastGroupUtil.parseIpAddress(6,
  51                 Unpooled.wrappedBuffer(IPV4_ADDRESS_EXPECTED)));
  52     }
  53
  54     @Test
  55     public void parseIpAddressPMulticastGroup() {
  56         final PAddressPMulticastGroup pAddressPMulticastGroup = new MyPAddressPMulticastGroup();
  57         final ByteBuf pAddressPMulticastGroupActual = Unpooled.buffer();
  58         PAddressPMulticastGroupUtil.serializeSenderPMulticastGroup(pAddressPMulticastGroup,
  59                 pAddressPMulticastGroupActual);
  60         assertArrayEquals(SENDER_P_MULTICAST_GROUP_EXPECTED, ByteArray.readAllBytes(pAddressPMulticastGroupActual));
  61
  62         final PAddressPMulticastGroup actual = PAddressPMulticastGroupUtil
  63                 .parseSenderPMulticastGroup(Unpooled.wrappedBuffer(SENDER_P_MULTICAST_GROUP_EXPECTED));
  64         assertEquals(IP_ADDRESS, actual.getPAddress());
  65         assertEquals(P_MULTICAST, actual.getPMulticastGroup());
  66     }
  67
  68     private static class MyPAddressPMulticastGroup implements PAddressPMulticastGroup {
  69         @Override
  70         public IpAddressNoZone getPAddress() {
  71             return IP_ADDRESS;
  72         }
  73
  74         @Override
  75         public IpAddressNoZone getPMulticastGroup() {
  76             return P_MULTICAST;
  77         }
  78
  79         @Override
  80         public Class<MyPAddressPMulticastGroup> implementedInterface() {
  81             return MyPAddressPMulticastGroup.class;
  82         }
  83     }
  84 }