+/*
+ * Copyright (c) 2014 Contextream, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
package org.opendaylight.lispflowmapping.implementation.lisp;
import static org.junit.Assert.assertEquals;
+import java.util.ArrayList;
import java.util.HashMap;
+import java.util.List;
import java.util.Map;
import junit.framework.Assert;
+import org.jmock.api.Invocation;
import org.junit.Before;
import org.junit.Test;
+import org.opendaylight.lispflowmapping.implementation.LispMappingService;
+import org.opendaylight.lispflowmapping.implementation.dao.MappingServiceKeyUtil;
+import org.opendaylight.lispflowmapping.implementation.util.LispAFIConvertor;
import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
-import org.opendaylight.lispflowmapping.type.lisp.EidRecord;
-import org.opendaylight.lispflowmapping.type.lisp.EidToLocatorRecord;
-import org.opendaylight.lispflowmapping.type.lisp.LocatorRecord;
-import org.opendaylight.lispflowmapping.type.lisp.MapReply;
-import org.opendaylight.lispflowmapping.type.lisp.MapRequest;
-import org.opendaylight.lispflowmapping.type.lisp.address.LispAddress;
-import org.opendaylight.lispflowmapping.type.lisp.address.LispIpv4Address;
-import org.opendaylight.lispflowmapping.type.lisp.address.LispIpv6Address;
+import org.opendaylight.lispflowmapping.interfaces.dao.IMappingServiceKey;
+import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceRLOCGroup;
import org.opendaylight.lispflowmapping.tools.junit.BaseTestCase;
-
-/*
- * Copyright (c) 2013 Contextream, Inc. and others. All rights reserved.
- *
- * This program and the accompanying materials are made available under the
- * terms of the Eclipse Public License v1.0 which accompanies this distribution,
- * and is available at http://www.eclipse.org/legal/epl-v10.html
- */
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.EidToLocatorRecord.Action;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.LispAFIAddress;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.MapReply;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.MapRequest;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidrecords.EidRecord;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidrecords.EidRecordBuilder;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecord;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.locatorrecords.LocatorRecord;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.locatorrecords.LocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.maprequest.ItrRloc;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.maprequest.SourceEidBuilder;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.maprequestnotification.MapRequestBuilder;
public class MapResolverTest extends BaseTestCase {
- private MapResolver testedMapResolver;
+ // private MapResolver testedMapResolver;
+ private LispMappingService testedMapResolver;
private ILispDAO lispDAO;
- private MapRequest mapRequest;
- private LispIpv4Address v4Address;
+ private MapRequestBuilder mapRequest;
+ private LispAFIAddress v4Address;
+ private LispAFIAddress v6Address;
+
+ private HashMap<IMappingServiceKey, Map<String, MappingServiceRLOCGroup>> daoResults;
@Override
@Before
public void before() throws Exception {
super.before();
lispDAO = context.mock(ILispDAO.class);
- testedMapResolver = new MapResolver(lispDAO);
+ testedMapResolver = new LispMappingService();
+ testedMapResolver.basicInit(lispDAO);
- mapRequest = new MapRequest();
- v4Address = new LispIpv4Address(0x01020304);
+ mapRequest = new MapRequestBuilder();
+ v4Address = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
+ v6Address = LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1");
+ daoResults = new HashMap<IMappingServiceKey, Map<String, MappingServiceRLOCGroup>>();
}
@Test
- public void handleMapRequest__NumRLOCsMismatch() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- Map<String, Object> rlocs = prepareMapping(new LispIpv4Address(0x44332211));
- rlocs.put("NumRLOCs", 5);
-
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
-
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(1, eidToLocators.getLocators().size());
- assertLocator(new LispIpv4Address(0x44332211), eidToLocators.getLocators().get(0));
+ public void handleMapRequest__ReplyWithSingleLocator() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32)
+ .setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("1.2.3.4"))).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(v4Address));
+
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLocalLocator(false);
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ locator.setRouted(true);
+ locator.setMulticastPriority((short) 5);
+ locator.setWeight((short) 17);
+ locator.setPriority((short) 16);
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
+
+ MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
+
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(1, eidToLocators.getLocatorRecord().size());
+ LocatorRecord resultLocator = mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0);
+ assertEquals(locator.isLocalLocator(), resultLocator.isLocalLocator());
+ assertEquals(locator.isRouted(), resultLocator.isRouted());
+ assertEquals(locator.getMulticastPriority(), resultLocator.getMulticastPriority());
+ assertEquals(locator.getMulticastWeight(), resultLocator.getMulticastWeight());
+ assertEquals(locator.getPriority(), resultLocator.getPriority());
+ assertEquals(locator.getWeight(), resultLocator.getWeight());
+
+ assertLocator(LispAFIConvertor.asIPAfiAddress("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
}
@Test
- public void handleMapRequest__BadRLOCType() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- Map<String, Object> rlocs = prepareMapping(new LispIpv4Address(0x71717171), new LispIpv4Address(0x44332211));
- rlocs.put("RLOC0", "Ooga booga");
+ public void handleMapRequest__VerifyBasicFields() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(v4Address)).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(v4Address));
+ record.setRecordTtl(100);
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ record.setAuthoritative(true);
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(1, eidToLocators.getLocators().size());
- assertLocator(new LispIpv4Address(0x44332211), eidToLocators.getLocators().get(0));
+ 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, LispAFIConvertor.toAFI(eidToLocators.getLispAddressContainer()));
+ assertEquals(record.isAuthoritative(), eidToLocators.isAuthoritative());
+ assertEquals(record.getAction(), eidToLocators.getAction());
+ assertEquals(record.getRecordTtl(), eidToLocators.getRecordTtl());
}
@Test
- public void handleMapRequest__NoNumRLOCs() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- allowing(lispDAO).get(wany(LispAddress.class));
- ret(new HashMap<String, Object>());
+ public void handleMapRequest__VerifyMask() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(v4Address)).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("1.2.3.0")));
+ record.setMaskLength((short) 24);
+
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
+
+ MapRequest mr = mapRequest.build();
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(0, eidToLocators.getLocators().size());
+ assertEquals(mr.getNonce(), mapReply.getNonce());
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals((byte) 32, eidToLocators.getMaskLength().byteValue());
+ assertEquals(v4Address, LispAFIConvertor.toAFI(eidToLocators.getLispAddressContainer()));
}
@Test
- public void handleMapRequest__IllegalNumRLOCs() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- Map<String, Object> rlocs = prepareMapping(new LispIpv4Address(0x44332211));
- rlocs.put("NumRLOCs", "Bla");
+ public void handleMapRequest__VerifyMaskIPv6() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(LispAFIConvertor.toContainer(v6Address)).build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:0")));
+ record.setMaskLength((short) 128);
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(0, eidToLocators.getLocators().size());
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
+
+ MapRequest mr = mapRequest.build();
+
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
+
+ assertEquals(mr.getNonce(), mapReply.getNonce());
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals((byte) 128, eidToLocators.getMaskLength().byteValue());
+ assertEquals(v6Address, LispAFIConvertor.toAFI(eidToLocators.getLispAddressContainer()));
}
@Test
- public void handleMapRequest__ReplyWithSingleLocator() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
+ public void handleMapRequest__VerifyMaskIPv6NoMatch() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(LispAFIConvertor.toContainer(v6Address)).build());
- prepareMapping(new LispIpv4Address(0x04030201));
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:1:0:0:0:0:0:1")));
+ record.setMaskLength((short) 112);
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(1, eidToLocators.getLocators().size());
+ MapRequest mr = mapRequest.build();
- assertLocator(new LispIpv4Address(0x04030201), eidToLocators.getLocators().get(0));
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(0, eidToLocators.getLocatorRecord().size());
}
- private Map<String, Object> prepareMapping(LispAddress... address) {
- Map<String, Object> result = null;
- if (address.length > 0) {
- result = new HashMap<String, Object>();
- result.put("NumRLOCs", address.length);
- for (int i = 0; i < address.length; i++) {
- result.put("RLOC" + i, address[i]);
- }
- }
+ @Test
+ public void handleMapRequest_VerifyNativelyForwardAutherized() {
+ MapRequest mr = getDefaultMapRequest();
+
+ Map<String, List<MappingServiceRLOCGroup>> result = new HashMap<String, List<MappingServiceRLOCGroup>>();
+ result.put(AbstractLispComponent.ADDRESS_SUBKEY, new ArrayList<MappingServiceRLOCGroup>());
- oneOf(lispDAO).get(weq(v4Address));
+ MapReply mapReply = getNativelyForwardMapReply(mr, result);
+
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(1, eidToLocators.getRecordTtl().intValue());
+ assertEquals(Action.NativelyForward, eidToLocators.getAction());
+ }
+
+ private MapReply getNativelyForwardMapReply(MapRequest mr, Map<String, List<MappingServiceRLOCGroup>> result) {
+ allowing(lispDAO).get(wany(IMappingServiceKey.class));
ret(result);
+ allowing(lispDAO).getSpecific(wany(IMappingServiceKey.class), with(AbstractLispComponent.PASSWORD_SUBKEY));
+ ret("pass");
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
+ return mapReply;
+ }
- return result;
+ private MapRequest getDefaultMapRequest() {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(v4Address)).build());
+ MapRequest mr = mapRequest.build();
+ return mr;
}
@Test
- public void handleMapRequest__VerifyBasicFields() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0x20, v4Address));
- prepareMapping();
+ public void handleMapRequest__VerifyMaskNoMatch() throws Exception {
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(v4Address)).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("1.2.4.0")));
+ record.setMaskLength((short) 24);
+
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("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);
+ assertEquals(0, eidToLocators.getLocatorRecord().size());
- assertEquals(mapRequest.getNonce(), mapReply.getNonce());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals((byte) 0x20, eidToLocators.getMaskLength());
- assertEquals(v4Address, eidToLocators.getPrefix());
}
@Test
public void handleMapRequest__ReplyWithMultipleLocators() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- prepareMapping(new LispIpv4Address(0x04030201), new LispIpv6Address("::1"), new LispIpv4Address("1.8.2.7"));
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(v4Address)).build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(v4Address));
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(3, eidToLocators.getLocators().size());
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record.getLocatorRecord().add(locator.build());
+ locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1")));
+ record.getLocatorRecord().add(locator.build());
+ locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("1.8.2.7")));
+ record.getLocatorRecord().add(locator.build());
- assertLocator(new LispIpv4Address(0x04030201), eidToLocators.getLocators().get(0));
- assertLocator(new LispIpv6Address("::1"), eidToLocators.getLocators().get(1));
- assertLocator(new LispIpv4Address("1.8.2.7"), eidToLocators.getLocators().get(2));
+ prepareMapping(record.build());
+
+ MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
+
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(3, eidToLocators.getLocatorRecord().size());
+
+ assertLocator(LispAFIConvertor.asIPAfiAddress("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
+ assertLocator(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1"), eidToLocators.getLocatorRecord().get(1));
+ assertLocator(LispAFIConvertor.asIPAfiAddress("1.8.2.7"), eidToLocators.getLocatorRecord().get(2));
}
- private void assertLocator(LispAddress expectedAddress, LocatorRecord locatorRecord) {
- assertEquals(expectedAddress, locatorRecord.getLocator());
+ @Test
+ public void handleMapRequest__MultipleEIDs() throws Exception {
+
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(v4Address)).build());
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(LispAFIConvertor.toContainer(v6Address)).build());
+
+ EidToLocatorRecordBuilder record1 = getDefaultEidToLocatorBuilder();
+ record1.setLispAddressContainer(LispAFIConvertor.toContainer(v4Address));
+ record1.setRecordTtl(100);
+
+ LocatorRecordBuilder locator1 = getDefaultLocatorBuilder();
+ locator1.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record1.getLocatorRecord().add(locator1.build());
+
+ EidToLocatorRecordBuilder record2 = getDefaultEidToLocatorBuilder();
+ record2.setLispAddressContainer(LispAFIConvertor.toContainer(v6Address));
+ record2.setMaskLength((short) 128);
+ record2.setRecordTtl(100);
+
+ LocatorRecordBuilder locator2 = getDefaultLocatorBuilder();
+ locator2.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("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);
+ assertEquals(1, eidToLocators1.getLocatorRecord().size());
+ assertLocator(LispAFIConvertor.asIPAfiAddress("4.3.2.1"), eidToLocators1.getLocatorRecord().get(0));
+
+ EidToLocatorRecord eidToLocators2 = mapReply.getEidToLocatorRecord().get(1);
+ assertEquals(1, eidToLocators2.getLocatorRecord().size());
+ assertLocator(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1"), eidToLocators2.getLocatorRecord().get(0));
+ }
+
+ private void assertLocator(LispAFIAddress expectedAddress, LocatorRecord locatorRecord) {
+ assertEquals(LispAFIConvertor.toContainer(expectedAddress), locatorRecord.getLispAddressContainer());
Assert.assertTrue(locatorRecord.isRouted());
}
+
+ private Map<String, MappingServiceRLOCGroup> prepareMapping(EidToLocatorRecord... records) {
+ if (records.length > 0) {
+ for (EidToLocatorRecord eidToLocatorRecord : records) {
+ Map<String, MappingServiceRLOCGroup> result = new HashMap<String, MappingServiceRLOCGroup>();
+ MappingServiceRLOCGroup rlocs = new MappingServiceRLOCGroup(eidToLocatorRecord.getRecordTtl(), eidToLocatorRecord.getAction(),
+ eidToLocatorRecord.isAuthoritative());
+ for (LocatorRecord locator : eidToLocatorRecord.getLocatorRecord()) {
+ rlocs.addRecord(locator);
+ }
+ result.put(AbstractLispComponent.ADDRESS_SUBKEY, rlocs);
+
+ daoResults.put(
+ MappingServiceKeyUtil.generateMappingServiceKey(eidToLocatorRecord.getLispAddressContainer(),
+ (short) eidToLocatorRecord.getMaskLength()), result);
+ }
+ }
+
+ ValueSaverAction<IMappingServiceKey> daoGetSaverAction = new ValueSaverAction<IMappingServiceKey>() {
+ @Override
+ protected boolean validate(IMappingServiceKey value) {
+ return true;
+ }
+
+ @Override
+ public Object invoke(Invocation invocation) throws Throwable {
+ return daoResults.get(lastValue);
+ }
+ };
+
+ allowing(lispDAO).get(with(daoGetSaverAction));
+ will(daoGetSaverAction);
+ allowing(lispDAO).getSpecific(wany(IMappingServiceKey.class), with(AbstractLispComponent.PASSWORD_SUBKEY));
+
+ return daoResults.get(MappingServiceKeyUtil.generateMappingServiceKey(LispAFIConvertor.toContainer(v4Address)));
+ }
+
+ private MapRequestBuilder getDefaultMapRequestBuilder() {
+ MapRequestBuilder mrBuilder = new MapRequestBuilder();
+ mrBuilder.setAuthoritative(false);
+ mrBuilder.setEidRecord(new ArrayList<EidRecord>());
+ mrBuilder.setItrRloc(new ArrayList<ItrRloc>());
+ mrBuilder.setMapDataPresent(true);
+ mrBuilder.setNonce((long) 0);
+ mrBuilder.setPitr(false);
+ mrBuilder.setProbe(false);
+ mrBuilder.setSmr(false);
+ mrBuilder.setSmrInvoked(true);
+ mrBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(
+ LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("127.0.0.1"))).build());
+ return mrBuilder;
+ }
+
+ private EidToLocatorRecordBuilder getDefaultEidToLocatorBuilder() {
+ EidToLocatorRecordBuilder builder = new EidToLocatorRecordBuilder();
+ 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;
+ }
+
+ private LocatorRecordBuilder getDefaultLocatorBuilder() {
+ LocatorRecordBuilder builder = new LocatorRecordBuilder();
+ builder.setLocalLocator(false);
+ builder.setMulticastPriority((short) 0);
+ builder.setMulticastWeight((short) 0);
+ builder.setPriority((short) 0);
+ builder.setRlocProbed(false);
+ builder.setRouted(true);
+ builder.setWeight((short) 0);
+ return builder;
+ }
+
}
Code Review / lispflowmapping.git / blobdiff