mappingservice/yangmodel/src/test/java/org/opendaylight/lispflowmapping/serializer/address/LispTrafficEngineeringLCAFAddressTest.java

   1 /*
   2  * Copyright (c) 2014 Contextream, 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.lispflowmapping.serializer.address;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.nio.ByteBuffer;
  13 import java.util.ArrayList;
  14 import java.util.List;
  15
  16 import junitx.framework.ArrayAssert;
  17
  18 import org.junit.Test;
  19 import org.opendaylight.lispflowmapping.lisp.serializer.address.LispAddressSerializer;
  20 import org.opendaylight.lispflowmapping.lisp.serializer.exception.LispSerializationException;
  21 import org.opendaylight.lispflowmapping.lisp.type.AddressFamilyNumberEnum;
  22 import org.opendaylight.lispflowmapping.lisp.type.LispCanonicalAddressFormatEnum;
  23 import org.opendaylight.lispflowmapping.lisp.util.LispAFIConvertor;
  24 import org.opendaylight.lispflowmapping.tools.junit.BaseTestCase;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafTrafficEngineeringAddress;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispAFIAddress;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcaftrafficengineeringaddress.Hops;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcaftrafficengineeringaddress.HopsBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcaftrafficengineering.LcafTrafficEngineeringAddrBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.reencaphop.HopBuilder;
  31
  32 public class LispTrafficEngineeringLCAFAddressTest extends BaseTestCase {
  33
  34     @Test
  35     public void deserialize__Simple() throws Exception {
  36         LispAFIAddress address = LispAddressSerializer.getInstance().deserialize(hexToByteBuffer("40 03 00 00 " + //
  37                 "0A 00 00 10 " + //
  38                 "00 00 00 01 AA BB CC DD " + // IPv4
  39                 "00 00 00 01 11 22 33 44")); // IPv4
  40
  41         assertEquals(AddressFamilyNumberEnum.LCAF.getIanaCode(), address.getAfi().shortValue());
  42         LcafTrafficEngineeringAddress trafficEngineering = (LcafTrafficEngineeringAddress) address;
  43
  44         assertEquals(LispCanonicalAddressFormatEnum.TRAFFIC_ENGINEERING.getLispCode(), trafficEngineering.getLcafType().byteValue());
  45
  46         List<Hops> hops = trafficEngineering.getHops();
  47         assertEquals(2, hops.size());
  48
  49         assertEquals(LispAFIConvertor.asPrimitiveIPAfiAddress("170.187.204.221"), hops.get(0).getHop().getPrimitiveAddress());
  50         assertEquals(LispAFIConvertor.asPrimitiveIPAfiAddress("17.34.51.68"), hops.get(1).getHop().getPrimitiveAddress());
  51     }
  52
  53     @Test
  54     public void deserialize__Bits() throws Exception {
  55         LispAFIAddress address = LispAddressSerializer.getInstance().deserialize(hexToByteBuffer("40 03 00 00 " + //
  56                 "0A 00 00 10 " + //
  57                 "00 05 00 01 AA BB CC DD " + // IPv4
  58                 "00 02 00 01 11 22 33 44")); // IPv4
  59
  60         assertEquals(AddressFamilyNumberEnum.LCAF.getIanaCode(), address.getAfi().shortValue());
  61         LcafTrafficEngineeringAddress trafficEngineering = (LcafTrafficEngineeringAddress) address;
  62
  63         assertEquals(LispCanonicalAddressFormatEnum.TRAFFIC_ENGINEERING.getLispCode(), trafficEngineering.getLcafType().byteValue());
  64
  65         List<Hops> hops = trafficEngineering.getHops();
  66         assertEquals(2, hops.size());
  67
  68         assertEquals(LispAFIConvertor.asPrimitiveIPAfiAddress("170.187.204.221"), hops.get(0).getHop().getPrimitiveAddress());
  69         assertEquals(true, hops.get(0).isLookup().booleanValue());
  70         assertEquals(false, hops.get(0).isRLOCProbe().booleanValue());
  71         assertEquals(true, hops.get(0).isStrict().booleanValue());
  72         assertEquals(LispAFIConvertor.asPrimitiveIPAfiAddress("17.34.51.68"), hops.get(1).getHop().getPrimitiveAddress());
  73         assertEquals(false, hops.get(1).isLookup().booleanValue());
  74         assertEquals(true, hops.get(1).isRLOCProbe().booleanValue());
  75         assertEquals(false, hops.get(1).isStrict().booleanValue());
  76     }
  77
  78     @Test
  79     public void deserialize__NoAddresses() throws Exception {
  80         LispAFIAddress address = LispAddressSerializer.getInstance().deserialize(hexToByteBuffer("40 03 00 00 " + //
  81                 "0A 00 00 00 "));
  82
  83         assertEquals(AddressFamilyNumberEnum.LCAF.getIanaCode(), address.getAfi().shortValue());
  84         LcafTrafficEngineeringAddress trafficEngineering = (LcafTrafficEngineeringAddress) address;
  85
  86         assertEquals(LispCanonicalAddressFormatEnum.TRAFFIC_ENGINEERING.getLispCode(), trafficEngineering.getLcafType().byteValue());
  87
  88         List<Hops> hops = trafficEngineering.getHops();
  89         assertEquals(0, hops.size());
  90     }
  91
  92     @Test(expected = LispSerializationException.class)
  93     public void deserialize__ShorterBuffer() throws Exception {
  94         LispAddressSerializer.getInstance().deserialize(hexToByteBuffer("40 03 00 00 " + //
  95                 "0A 00 00 18 " + //
  96                 "00 01 AA BB CC DD " + // IPv4
  97                 "00 02 11 22 33 44 11 22 33 44 11 22 33 44"));
  98     }
  99
 100     @Test(expected = LispSerializationException.class)
 101     public void deserialize__ShorterBuffer2() throws Exception {
 102         LispAddressSerializer.getInstance().deserialize(hexToByteBuffer("40 03 00 00 " + //
 103                 "0A 00 00 18 "));
 104     }
 105
 106     @Test
 107     public void serialize__Simple() throws Exception {
 108         List<Hops> hops = new ArrayList<Hops>();
 109         hops.add(new HopsBuilder().setHop(new HopBuilder().setPrimitiveAddress(LispAFIConvertor.asPrimitiveIPAfiAddress("170.187.204.221")).build())
 110                 .build());
 111         hops.add(new HopsBuilder().setHop(new HopBuilder().setPrimitiveAddress(LispAFIConvertor.asPrimitiveIPAfiAddress("17.34.51.68")).build())
 112                 .build());
 113         LcafTrafficEngineeringAddrBuilder trafficBuilder = new LcafTrafficEngineeringAddrBuilder();
 114         trafficBuilder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode()).setLcafType(
 115                 (short) LispCanonicalAddressFormatEnum.TRAFFIC_ENGINEERING.getLispCode());
 116         trafficBuilder.setHops(hops);
 117
 118         ByteBuffer buf = ByteBuffer.allocate(LispAddressSerializer.getInstance().getAddressSize(trafficBuilder.build()));
 119         LispAddressSerializer.getInstance().serialize(buf, trafficBuilder.build());
 120         ByteBuffer expectedBuf = hexToByteBuffer("40 03 00 00 " + //
 121                 "0A 00 00 10 " + //
 122                 "00 00 00 01 AA BB CC DD " + // IPv4
 123                 "00 00 00 01 11 22 33 44"); // IPv4
 124         ArrayAssert.assertEquals(expectedBuf.array(), buf.array());
 125     }
 126
 127     @Test
 128     public void serialize__NoAddresses() throws Exception {
 129         LcafTrafficEngineeringAddrBuilder addressBuilder = new LcafTrafficEngineeringAddrBuilder();
 130         addressBuilder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode()).setLcafType(
 131                 (short) LispCanonicalAddressFormatEnum.TRAFFIC_ENGINEERING.getLispCode());
 132
 133         ByteBuffer buf = ByteBuffer.allocate(LispAddressSerializer.getInstance().getAddressSize(addressBuilder.build()));
 134         LispAddressSerializer.getInstance().serialize(buf, addressBuilder.build());
 135         ByteBuffer expectedBuf = hexToByteBuffer("40 03 00 00 " + //
 136                 "0A 00 00 00");
 137         ArrayAssert.assertEquals(expectedBuf.array(), buf.array());
 138     }
 139 }