JUnit Test - LispAddressStringifierTest update

Change-Id: I7f5e3de8de21f14334ec004d1e38634315c084a6
Signed-off-by: Miroslav Toth <mirtoth@cisco.com>

diff --git a/mappingservice/lisp-proto/src/test/java/org/opendaylight/lispflowmapping/lisp/util/LispAddressStringifierTest.java b/mappingservice/lisp-proto/src/test/java/org/opendaylight/lispflowmapping/lisp/util/LispAddressStringifierTest.java
index 24eac9414d91cfc843b01c867a41103bee9f13c1..7b2be320ee09ff4d536afe1ca5935bae2ca671c6 100644 (file)
--- a/mappingservice/lisp-proto/src/test/java/org/opendaylight/lispflowmapping/lisp/util/LispAddressStringifierTest.java
+++ b/mappingservice/lisp-proto/src/test/java/org/opendaylight/lispflowmapping/lisp/util/LispAddressStringifierTest.java
@@ -94,8 +94,18 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev16
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv6AddressBinary;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv4Binary;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv4BinaryBuilder;
+
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address
+        .address.Ipv4PrefixBinary;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address
+        .address.Ipv4PrefixBinaryBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv6Binary;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv6BinaryBuilder;
+
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address
+        .address.Ipv6PrefixBinary;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address
+        .address.Ipv6PrefixBinaryBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.XtrId;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.EidBuilder;
 
@@ -147,6 +157,18 @@ public class LispAddressStringifierTest {
             .setIpv6Prefix(new Ipv6Prefix(IPV6_STRING + PREFIX)).build();
     private static final LispAddress LISP_IPV6_PREFIX = new EidBuilder().setAddress(IPV6_PREFIX).build();
 
+    // Ipv4PrefixBinary
+    private static final Ipv4PrefixBinary IPV4_PREFIX_BINARY = new Ipv4PrefixBinaryBuilder()
+            .setIpv4AddressBinary(IPV4_ADDRESS_BINARY)
+            .setIpv4MaskLength((short) 24).build();
+    private static final LispAddress LISP_IPV4_PREFIX_BINARY = new EidBuilder().setAddress(IPV4_PREFIX_BINARY).build();
+
+    // Ipv6PrefixBinary
+    private static final Ipv6PrefixBinary IPV6_PREFIX_BINARY = new Ipv6PrefixBinaryBuilder()
+            .setIpv6AddressBinary(IPV6_ADDRESS_BINARY)
+            .setIpv6MaskLength((short) 24).build();
+    private static final LispAddress LISP_IPV6_PREFIX_BINARY = new EidBuilder().setAddress(IPV6_PREFIX_BINARY).build();
+
     // Mac
     private static final MacAddress MAC_ADDRESS = new MacAddress("01:23:45:67:89:ab");
     private static final Mac MAC = new MacBuilder().setMac(MAC_ADDRESS).build();
@@ -311,6 +333,14 @@ public class LispAddressStringifierTest {
         assertEquals(IPV4_STRING + PREFIX, LispAddressStringifier.getString(LISP_IPV4_PREFIX));
     }
 
+    /**
+     * Tests {@link LispAddressStringifier#getString} with Ipv4PrefixBinary address type.
+     */
+    @Test
+    public void getStringTest_withIpv4PrefixBinary() {
+        assertEquals(IPV4_STRING + PREFIX, LispAddressStringifier.getString(LISP_IPV4_PREFIX_BINARY));
+    }
+
     /**
      * Tests {@link LispAddressStringifier#getString} with Ipv6 address type.
      */
@@ -335,6 +365,14 @@ public class LispAddressStringifierTest {
         assertEquals(IPV6_STRING + PREFIX, LispAddressStringifier.getString(LISP_IPV6_PREFIX));
     }
 
+    /**
+     * Tests {@link LispAddressStringifier#getString} with Ipv6PrefixBinary address type.
+     */
+    @Test
+    public void getStringTest_withIpv6PrefixBinary() {
+        assertEquals(IPV6_STRING + PREFIX, LispAddressStringifier.getString(LISP_IPV6_PREFIX_BINARY));
+    }
+
     /**
      * Tests {@link LispAddressStringifier#getString} with Mac address type.
      */