Migrate implementation/neutron/southbound to IETF YANG model
[lispflowmapping.git] / mappingservice / implementation / src / test / java / org / opendaylight / lispflowmapping / implementation / lisp / MapResolverTest.java
index f6fd64771e598428604d72ead162d5d206d9d0f8..5f74e2569e4a1136a152a23b62f632e192bde7e4 100644 (file)
@@ -22,21 +22,22 @@ import org.opendaylight.lispflowmapping.implementation.MappingService;
 import org.opendaylight.lispflowmapping.implementation.MappingSystem;
 import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
 import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
-import org.opendaylight.lispflowmapping.lisp.util.LispAFIConvertor;
+import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
 import org.opendaylight.lispflowmapping.tools.junit.BaseTestCase;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.EidToLocatorRecord.Action;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapReply;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapRequest;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eidrecords.EidRecord;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eidrecords.EidRecordBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eidtolocatorrecords.EidToLocatorRecord;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eidtolocatorrecords.EidToLocatorRecordBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.lispaddress.LispAddressContainer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.list.EidItem;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.list.EidItemBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecord;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord.Action;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequest.ItrRloc;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequest.SourceEidBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequestnotification.MapRequestBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
 
 public class MapResolverTest extends BaseTestCase {
 
@@ -45,10 +46,10 @@ public class MapResolverTest extends BaseTestCase {
     private ILispDAO dao;
     private MappingService mapService;
     private MapRequestBuilder mapRequest;
-    private LispAddressContainer v4Address;
-    private LispAddressContainer v6Address;
+    private Eid v4Address;
+    private Eid v6Address;
 
-    private HashMap<LispAddressContainer, Map<String, EidToLocatorRecord>> daoResults;
+    private HashMap<Eid, Map<String, MappingRecord>> daoResults;
 
     @Override
     @Before
@@ -70,24 +71,22 @@ public class MapResolverTest extends BaseTestCase {
         testedMapResolver.basicInit();
 
         mapRequest = new MapRequestBuilder();
-        v4Address = LispAFIConvertor.asIPv4Address("1.2.3.4");
-        v6Address = LispAFIConvertor.asIPv6Address("0:0:0:0:0:0:0:1");
-        daoResults = new HashMap<LispAddressContainer, Map<String, EidToLocatorRecord>>();
+        v4Address = LispAddressUtil.asIpv4PrefixEid("1.2.3.4/32");
+        v6Address = LispAddressUtil.asIpv6PrefixEid("0:0:0:0:0:0:0:1/128");
+        daoResults = new HashMap<Eid, Map<String, MappingRecord>>();
     }
 
     @Test
     public void handleMapRequest__ReplyWithSingleLocator() throws Exception {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32)
-                        .setLispAddressContainer(LispAFIConvertor.asIPv4Address("1.2.3.4")).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(v4Address);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(v4Address);
 
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
         locator.setLocalLocator(false);
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         locator.setRouted(true);
         locator.setMulticastPriority((short) 5);
         locator.setWeight((short) 17);
@@ -97,9 +96,9 @@ public class MapResolverTest extends BaseTestCase {
 
         MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
 
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
         assertEquals(1, eidToLocators.getLocatorRecord().size());
-        LocatorRecord resultLocator = mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0);
+        LocatorRecord resultLocator = mapReply.getMappingRecordItem().get(0).getMappingRecord().getLocatorRecord().get(0);
         assertEquals(locator.isLocalLocator(), resultLocator.isLocalLocator());
         assertEquals(locator.isRouted(), resultLocator.isRouted());
         assertEquals(locator.getMulticastPriority(), resultLocator.getMulticastPriority());
@@ -107,31 +106,29 @@ public class MapResolverTest extends BaseTestCase {
         assertEquals(locator.getPriority(), resultLocator.getPriority());
         assertEquals(locator.getWeight(), resultLocator.getWeight());
 
-        assertLocator(LispAFIConvertor.asIPv4Address("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
+        assertLocator(LispAddressUtil.asIpv4Rloc("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
     }
 
     @Test
     public void handleMapRequest__VerifyBasicFields() throws Exception {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(v4Address);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(v4Address);
         record.setRecordTtl(100);
 
         record.setAuthoritative(true);
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record.getLocatorRecord().add(locator.build());
         prepareMapping(record.build());
 
         MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
 
         assertEquals(mapRequest.getNonce(), mapReply.getNonce());
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
-        assertEquals((byte) 32, eidToLocators.getMaskLength().byteValue());
-        assertEquals(v4Address, eidToLocators.getLispAddressContainer());
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
+        assertEquals(v4Address, eidToLocators.getEid());
         assertEquals(record.isAuthoritative(), eidToLocators.isAuthoritative());
         assertEquals(record.getAction(), eidToLocators.getAction());
         assertEquals(record.getRecordTtl(), eidToLocators.getRecordTtl());
@@ -140,15 +137,13 @@ public class MapResolverTest extends BaseTestCase {
     @Test
     public void handleMapRequest__VerifyMask() throws Exception {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(LispAFIConvertor.asIPv4Prefix("1.2.3.0", 24));
-        record.setMaskLength((short) 24);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(LispAddressUtil.asIpv4PrefixEid("1.2.3.0/24"));
 
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record.getLocatorRecord().add(locator.build());
         prepareMapping(record.build());
 
@@ -156,23 +151,20 @@ public class MapResolverTest extends BaseTestCase {
         MapReply mapReply = testedMapResolver.handleMapRequest(mr);
 
         assertEquals(mr.getNonce(), mapReply.getNonce());
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
-        assertEquals((byte) 24, eidToLocators.getMaskLength().byteValue());
-        assertEquals(LispAFIConvertor.asIPv4Prefix("1.2.3.0", 24), eidToLocators.getLispAddressContainer());
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
+        assertEquals(LispAddressUtil.asIpv4PrefixEid("1.2.3.0/24"), eidToLocators.getEid());
     }
 
     @Test
     public void handleMapRequest__VerifyMaskIPv6() throws Exception {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(v6Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v6Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:0")));
-        record.setMaskLength((short) 128);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(LispAddressUtil.asIpv6PrefixEid("0:0:0:0:0:0:0:0/128"));
 
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record.getLocatorRecord().add(locator.build());
         prepareMapping(record.build());
 
@@ -181,51 +173,48 @@ public class MapResolverTest extends BaseTestCase {
         MapReply mapReply = testedMapResolver.handleMapRequest(mr);
 
         assertEquals(mr.getNonce(), mapReply.getNonce());
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
-        assertEquals((byte) 128, eidToLocators.getMaskLength().byteValue());
-        assertEquals(v6Address, eidToLocators.getLispAddressContainer());
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
+        assertEquals(v6Address, eidToLocators.getEid());
     }
 
     @Test
     public void handleMapRequest__VerifyMaskIPv6NoMatch() throws Exception {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(v6Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v6Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(LispAFIConvertor.asIPv6Prefix("0:1:0:0:0:0:0:1", 112));
-        record.setMaskLength((short) 112);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(LispAddressUtil.asIpv6PrefixEid("0:1:0:0:0:0:0:1/112"));
 
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record.getLocatorRecord().add(locator.build());
         prepareMapping(record.build());
 
         MapRequest mr = mapRequest.build();
 
         MapReply mapReply = testedMapResolver.handleMapRequest(mr);
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
         assertEquals(null, eidToLocators.getLocatorRecord());
     }
 
     @Test
-    public void handleMapRequest_VerifyNativelyForwardAutherized() {
+    public void handleMapRequest_VerifyNativelyForwardAuthorized() {
         MapRequest mr = getDefaultMapRequest();
 
-        Map<String, EidToLocatorRecord> result = new HashMap<String, EidToLocatorRecord>();
+        Map<String, MappingRecord> result = new HashMap<String, MappingRecord>();
         result.put(SubKeys.RECORD, null);
 
         MapReply mapReply = getNativelyForwardMapReply(mr, result);
 
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
         assertEquals(1, eidToLocators.getRecordTtl().intValue());
         assertEquals(Action.NativelyForward, eidToLocators.getAction());
     }
 
-    private MapReply getNativelyForwardMapReply(MapRequest mr, Map<String, EidToLocatorRecord> result) {
-        allowing(dao).get(wany(LispAddressContainer.class));
+    private MapReply getNativelyForwardMapReply(MapRequest mr, Map<String, MappingRecord> result) {
+        allowing(dao).get(wany(Eid.class));
         ret(result);
-        allowing(dao).getSpecific(wany(LispAddressContainer.class), with(SubKeys.AUTH_KEY));
+        allowing(dao).getSpecific(wany(Eid.class), with(SubKeys.AUTH_KEY));
         ret("pass");
         MapReply mapReply = testedMapResolver.handleMapRequest(mr);
         return mapReply;
@@ -233,8 +222,7 @@ public class MapResolverTest extends BaseTestCase {
 
     private MapRequest getDefaultMapRequest() {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
         MapRequest mr = mapRequest.build();
         return mr;
     }
@@ -242,15 +230,13 @@ public class MapResolverTest extends BaseTestCase {
     @Test
     public void handleMapRequest__VerifyMaskNoMatch() throws Exception {
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(LispAFIConvertor.asIPv4Prefix("1.2.4.0", 24));
-        record.setMaskLength((short) 24);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(LispAddressUtil.asIpv4PrefixEid("1.2.4.0/24"));
 
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record.getLocatorRecord().add(locator.build());
         prepareMapping(record.build());
 
@@ -258,7 +244,7 @@ public class MapResolverTest extends BaseTestCase {
 
         MapReply mapReply = testedMapResolver.handleMapRequest(mr);
 
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
         assertEquals(null, eidToLocators.getLocatorRecord());
     }
 
@@ -266,93 +252,87 @@ public class MapResolverTest extends BaseTestCase {
     public void handleMapRequest__ReplyWithMultipleLocators() throws Exception {
 
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
 
-        EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
-        record.setLispAddressContainer(v4Address);
+        MappingRecordBuilder record = getDefaultMappingRecordBuilder();
+        record.setEid(v4Address);
 
         LocatorRecordBuilder locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record.getLocatorRecord().add(locator.build());
         locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv6Address("0:0:0:0:0:0:0:1"));
+        locator.setRloc(LispAddressUtil.asIpv6Rloc("0:0:0:0:0:0:0:1"));
         record.getLocatorRecord().add(locator.build());
         locator = getDefaultLocatorBuilder();
-        locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("1.8.2.7"));
+        locator.setRloc(LispAddressUtil.asIpv4Rloc("1.8.2.7"));
         record.getLocatorRecord().add(locator.build());
 
         prepareMapping(record.build());
 
         MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
 
-        EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+        MappingRecord eidToLocators = mapReply.getMappingRecordItem().get(0).getMappingRecord();
         assertEquals(3, eidToLocators.getLocatorRecord().size());
 
-        assertLocator(LispAFIConvertor.asIPv4Address("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
-        assertLocator(LispAFIConvertor.asIPv6Address("0:0:0:0:0:0:0:1"), eidToLocators.getLocatorRecord().get(1));
-        assertLocator(LispAFIConvertor.asIPv4Address("1.8.2.7"), eidToLocators.getLocatorRecord().get(2));
+        assertLocator(LispAddressUtil.asIpv4Rloc("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
+        assertLocator(LispAddressUtil.asIpv6Rloc("0:0:0:0:0:0:0:1"), eidToLocators.getLocatorRecord().get(1));
+        assertLocator(LispAddressUtil.asIpv4Rloc("1.8.2.7"), eidToLocators.getLocatorRecord().get(2));
     }
 
     @Test
     public void handleMapRequest__MultipleEIDs() throws Exception {
 
         mapRequest = getDefaultMapRequestBuilder();
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
-        mapRequest.getEidRecord().add(
-                new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(v6Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v4Address).build());
+        mapRequest.getEidItem().add(new EidItemBuilder().setEid(v6Address).build());
 
-        EidToLocatorRecordBuilder record1 = getDefaultEidToLocatorBuilder();
-        record1.setLispAddressContainer(v4Address);
+        MappingRecordBuilder record1 = getDefaultMappingRecordBuilder();
+        record1.setEid(v4Address);
         record1.setRecordTtl(100);
 
         LocatorRecordBuilder locator1 = getDefaultLocatorBuilder();
-        locator1.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+        locator1.setRloc(LispAddressUtil.asIpv4Rloc("4.3.2.1"));
         record1.getLocatorRecord().add(locator1.build());
 
-        EidToLocatorRecordBuilder record2 = getDefaultEidToLocatorBuilder();
-        record2.setLispAddressContainer(v6Address);
-        record2.setMaskLength((short) 128);
+        MappingRecordBuilder record2 = getDefaultMappingRecordBuilder();
+        record2.setEid(v6Address);
         record2.setRecordTtl(100);
 
         LocatorRecordBuilder locator2 = getDefaultLocatorBuilder();
-        locator2.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1")));
+        locator2.setRloc(LispAddressUtil.asIpv6Rloc("0:0:0:0:0:0:0:1"));
         record2.getLocatorRecord().add(locator2.build());
 
         prepareMapping(record1.build(), record2.build());
 
         MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
 
-        EidToLocatorRecord eidToLocators1 = mapReply.getEidToLocatorRecord().get(0);
+        MappingRecord eidToLocators1 = mapReply.getMappingRecordItem().get(0).getMappingRecord();
         assertEquals(1, eidToLocators1.getLocatorRecord().size());
-        assertLocator(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")), eidToLocators1
-                .getLocatorRecord().get(0));
+        assertLocator(LispAddressUtil.asIpv4Rloc("4.3.2.1"), eidToLocators1.getLocatorRecord().get(0));
 
-        EidToLocatorRecord eidToLocators2 = mapReply.getEidToLocatorRecord().get(1);
+        MappingRecord eidToLocators2 = mapReply.getMappingRecordItem().get(1).getMappingRecord();
         assertEquals(1, eidToLocators2.getLocatorRecord().size());
-        assertLocator(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1")), eidToLocators2
-                .getLocatorRecord().get(0));
+        assertLocator(LispAddressUtil.asIpv6Rloc("0:0:0:0:0:0:0:1"), eidToLocators2.getLocatorRecord().get(0));
     }
 
-    private void assertLocator(LispAddressContainer expectedAddress, LocatorRecord locatorRecord) {
-        assertEquals(expectedAddress, locatorRecord.getLispAddressContainer());
+    private void assertLocator(Rloc expectedAddress, LocatorRecord locatorRecord) {
+        assertEquals(expectedAddress, locatorRecord.getRloc());
         Assert.assertTrue(locatorRecord.isRouted());
     }
 
-    private Map<String, EidToLocatorRecord> prepareMapping(EidToLocatorRecord... records) {
+    private Map<String, MappingRecord> prepareMapping(MappingRecord... records) {
         if (records.length > 0) {
-            for (EidToLocatorRecord eidToLocatorRecord : records) {
-                Map<String, EidToLocatorRecord> result = new HashMap<String, EidToLocatorRecord>();
+            for (MappingRecord eidToLocatorRecord : records) {
+                Map<String, MappingRecord> result = new HashMap<String, MappingRecord>();
                 result.put(SubKeys.RECORD, eidToLocatorRecord);
 
-                daoResults.put(eidToLocatorRecord.getLispAddressContainer(), result);
+                daoResults.put(eidToLocatorRecord.getEid(), result);
             }
         }
 
-        ValueSaverAction<LispAddressContainer> daoGetSaverAction = new ValueSaverAction<LispAddressContainer>() {
+        ValueSaverAction<Eid> daoGetSaverAction = new ValueSaverAction<Eid>() {
             @Override
-            protected boolean validate(LispAddressContainer value) {
+            protected boolean validate(Eid value) {
                 return true;
             }
 
@@ -364,7 +344,7 @@ public class MapResolverTest extends BaseTestCase {
 
         allowing(dao).get(with(daoGetSaverAction));
         will(daoGetSaverAction);
-        allowing(dao).getSpecific(wany(LispAddressContainer.class), with(SubKeys.AUTH_KEY));
+        allowing(dao).getSpecific(wany(Eid.class), with(SubKeys.AUTH_KEY));
 
         return daoResults.get(v4Address);
     }
@@ -372,7 +352,7 @@ public class MapResolverTest extends BaseTestCase {
     private MapRequestBuilder getDefaultMapRequestBuilder() {
         MapRequestBuilder mrBuilder = new MapRequestBuilder();
         mrBuilder.setAuthoritative(false);
-        mrBuilder.setEidRecord(new ArrayList<EidRecord>());
+        mrBuilder.setEidItem(new ArrayList<EidItem>());
         mrBuilder.setItrRloc(new ArrayList<ItrRloc>());
         mrBuilder.setMapDataPresent(true);
         mrBuilder.setNonce((long) 0);
@@ -380,18 +360,16 @@ public class MapResolverTest extends BaseTestCase {
         mrBuilder.setProbe(false);
         mrBuilder.setSmr(false);
         mrBuilder.setSmrInvoked(true);
-        mrBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(
-                LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("127.0.0.1"))).build());
+        mrBuilder.setSourceEid(new SourceEidBuilder().setEid(LispAddressUtil.asIpv4Eid("127.0.0.1")).build());
         return mrBuilder;
     }
 
-    private EidToLocatorRecordBuilder getDefaultEidToLocatorBuilder() {
-        EidToLocatorRecordBuilder builder = new EidToLocatorRecordBuilder();
+    private MappingRecordBuilder getDefaultMappingRecordBuilder() {
+        MappingRecordBuilder builder = new MappingRecordBuilder();
         builder.setAction(Action.NoAction);
         builder.setAuthoritative(false);
         builder.setLocatorRecord(new ArrayList<LocatorRecord>());
         builder.setMapVersion((short) 0);
-        builder.setMaskLength((short) 32);
         builder.setRecordTtl(60);
         return builder;
     }