+/*
+ * 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.Assert;
import org.junit.Before;
-import org.junit.Ignore;
import org.junit.Test;
+import org.opendaylight.lispflowmapping.implementation.LispMappingService;
import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
-import org.opendaylight.lispflowmapping.interfaces.dao.IMappingServiceKey;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceKeyUtil;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceRLOC;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceValue;
+import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
+import org.opendaylight.lispflowmapping.lisp.util.LispAFIConvertor;
import org.opendaylight.lispflowmapping.tools.junit.BaseTestCase;
-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;
-
-/*
- * 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.urn.opendaylight.lfm.lisp.proto.rev150820.EidToLocatorRecord.Action;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.MapReply;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.MapRequest;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidrecords.EidRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidrecords.EidRecordBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lispaddress.LispAddressContainer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.locatorrecords.LocatorRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.locatorrecords.LocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.maprequest.ItrRloc;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.maprequest.SourceEidBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.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 LispIpv6Address v6Address;
+ private MapRequestBuilder mapRequest;
+ private LispAddressContainer v4Address;
+ private LispAddressContainer v6Address;
- private HashMap<IMappingServiceKey, Map<String, MappingServiceValue>> daoResults;
+ private HashMap<LispAddressContainer, Map<String, EidToLocatorRecord>> 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("1.2.3.4");
- v6Address = new LispIpv6Address("0:0:0:0:0:0:0:1");
- daoResults = new HashMap<IMappingServiceKey, Map<String, MappingServiceValue>>();
+ 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>>();
}
@Test
- @Ignore
- public void handleMapRequest__NumRLOCsMismatch() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(v4Address);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x44332211)));
- // Map<String, Object> rlocs = prepareMapping(record);
- // 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.asIPv4Address("1.2.3.4")).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(v4Address);
+
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLocalLocator(false);
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("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.asIPv4Address("4.3.2.1"), eidToLocators.getLocatorRecord().get(0));
}
@Test
- @Ignore
- public void handleMapRequest__BadRLOCType() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(v4Address);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x71717171)));
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x44332211)));
- // Map<String, Object> rlocs = prepareMapping(record);
- // rlocs.put("RLOC0", "Ooga booga");
-
- 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__VerifyBasicFields() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
- @Test
- @Ignore
- public void handleMapRequest__NoNumRLOCs() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- allowing(lispDAO).get(wany(IMappingServiceKey.class));
- ret(new HashMap<String, Object>());
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(v4Address);
+ record.setRecordTtl(100);
+
+ record.setAuthoritative(true);
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(0, eidToLocators.getLocators().size());
+ assertEquals(mapRequest.getNonce(), mapReply.getNonce());
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals((byte) 32, eidToLocators.getMaskLength().byteValue());
+ assertEquals(v4Address, eidToLocators.getLispAddressContainer());
+ assertEquals(record.isAuthoritative(), eidToLocators.isAuthoritative());
+ assertEquals(record.getAction(), eidToLocators.getAction());
+ assertEquals(record.getRecordTtl(), eidToLocators.getRecordTtl());
}
@Test
- @Ignore
- public void handleMapRequest__IllegalNumRLOCs() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(v4Address);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x44332211)));
- // Map<String, Object> rlocs = prepareMapping(record);
- // rlocs.put("NumRLOCs", "Bla");
-
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
-
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(0, eidToLocators.getLocators().size());
+ public void handleMapRequest__VerifyMask() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.asIPv4Prefix("1.2.3.0", 24));
+ 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);
+
+ 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());
}
@Test
- public void handleMapRequest__ReplyWithSingleLocator() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, new LispIpv4Address("1.2.3.4")));
+ public void handleMapRequest__VerifyMaskIPv6() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(v6Address).build());
+
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:0")));
+ record.setMaskLength((short) 128);
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(v4Address);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- prepareMapping(record);
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ MapRequest mr = mapRequest.build();
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(1, eidToLocators.getLocators().size());
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
- assertLocator(new LispIpv4Address(0x04030201), eidToLocators.getLocators().get(0));
+ assertEquals(mr.getNonce(), mapReply.getNonce());
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals((byte) 128, eidToLocators.getMaskLength().byteValue());
+ assertEquals(v6Address, eidToLocators.getLispAddressContainer());
}
- private Map<String, MappingServiceValue> prepareMapping(EidToLocatorRecord... records) {
- if (records.length > 0) {
- for (EidToLocatorRecord eidToLocatorRecord : records) {
- MappingServiceValue value = new MappingServiceValue();
- Map<String, MappingServiceValue> result = new HashMap<String, MappingServiceValue>();
- result.put("value", value);
- List<MappingServiceRLOC> rlocs = new ArrayList<MappingServiceRLOC>();
- for (LocatorRecord locator : eidToLocatorRecord.getLocators()) {
- rlocs.add(new MappingServiceRLOC(locator, eidToLocatorRecord.getRecordTtl()));
- }
- value.setRlocs(rlocs);
-
- daoResults.put(
- MappingServiceKeyUtil.generateMappingServiceKey(eidToLocatorRecord.getPrefix(), (byte) eidToLocatorRecord.getMaskLength()),
- result);
- }
- }
+ @Test
+ public void handleMapRequest__VerifyMaskIPv6NoMatch() throws Exception {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(v6Address).build());
- ValueSaverAction<IMappingServiceKey> daoGetSaverAction = new ValueSaverAction<IMappingServiceKey>() {
- @Override
- protected boolean validate(IMappingServiceKey value) {
- return true;
- }
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.asIPv6Prefix("0:1:0:0:0:0:0:1", 112));
+ record.setMaskLength((short) 112);
- @Override
- public Object invoke(Invocation invocation) throws Throwable {
- return daoResults.get(lastValue);
- }
- };
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+ record.getLocatorRecord().add(locator.build());
+ prepareMapping(record.build());
- allowing(lispDAO).get(with(daoGetSaverAction));
- will(daoGetSaverAction);
+ MapRequest mr = mapRequest.build();
- return daoResults.get(MappingServiceKeyUtil.generateMappingServiceKey(v4Address));
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(null, eidToLocators.getLocatorRecord());
}
@Test
- public void handleMapRequest__VerifyBasicFields() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
+ public void handleMapRequest_VerifyNativelyForwardAutherized() {
+ MapRequest mr = getDefaultMapRequest();
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(v4Address);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- prepareMapping(record);
+ Map<String, EidToLocatorRecord> result = new HashMap<String, EidToLocatorRecord>();
+ result.put(SubKeys.RECORD, null);
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ MapReply mapReply = getNativelyForwardMapReply(mr, result);
- assertEquals(mapRequest.getNonce(), mapReply.getNonce());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals((byte) 0, eidToLocators.getMaskLength());
- assertEquals(v4Address, eidToLocators.getPrefix());
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(1, eidToLocators.getRecordTtl().intValue());
+ assertEquals(Action.NativelyForward, eidToLocators.getAction());
+ }
+
+ private MapReply getNativelyForwardMapReply(MapRequest mr, Map<String, EidToLocatorRecord> result) {
+ allowing(lispDAO).get(wany(LispAddressContainer.class));
+ ret(result);
+ allowing(lispDAO).getSpecific(wany(LispAddressContainer.class), with(SubKeys.PASSWORD));
+ ret("pass");
+ MapReply mapReply = testedMapResolver.handleMapRequest(mr);
+ return mapReply;
+ }
+
+ private MapRequest getDefaultMapRequest() {
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+ MapRequest mr = mapRequest.build();
+ return mr;
}
@Test
- public void handleMapRequest__VerifyMask() throws Exception {
- EidRecord mapRequestRecord = new EidRecord((byte) 0, v4Address);
- mapRequestRecord.setMaskLength(32);
- mapRequest.addEidRecord(mapRequestRecord);
+ public void handleMapRequest__VerifyMaskNoMatch() throws Exception {
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(new LispIpv4Address("1.2.3.0")).setMaskLength(24);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- prepareMapping(record);
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(LispAFIConvertor.asIPv4Prefix("1.2.4.0", 24));
+ record.setMaskLength((short) 24);
- assertEquals(mapRequest.getNonce(), mapReply.getNonce());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals((byte) 32, eidToLocators.getMaskLength());
- assertEquals(v4Address, eidToLocators.getPrefix());
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("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(null, eidToLocators.getLocatorRecord());
}
@Test
- public void handleMapRequest__VerifyMaskIPv6() throws Exception {
- EidRecord mapRequestRecord = new EidRecord((byte) 0, v6Address);
- mapRequestRecord.setMaskLength(128);
- mapRequest.addEidRecord(mapRequestRecord);
+ public void handleMapRequest__ReplyWithMultipleLocators() throws Exception {
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(new LispIpv6Address("0:0:0:0:0:0:0:0")).setMaskLength(128);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- prepareMapping(record);
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ EidToLocatorRecordBuilder record = getDefaultEidToLocatorBuilder();
+ record.setLispAddressContainer(v4Address);
- assertEquals(mapRequest.getNonce(), mapReply.getNonce());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals((byte) 128, eidToLocators.getMaskLength());
- assertEquals(v6Address, eidToLocators.getPrefix());
- }
+ LocatorRecordBuilder locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("4.3.2.1"));
+ record.getLocatorRecord().add(locator.build());
+ locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv6Address("0:0:0:0:0:0:0:1"));
+ record.getLocatorRecord().add(locator.build());
+ locator = getDefaultLocatorBuilder();
+ locator.setLispAddressContainer(LispAFIConvertor.asIPv4Address("1.8.2.7"));
+ record.getLocatorRecord().add(locator.build());
- @Test
- public void handleMapRequest__VerifyMaskIPv6NoMatch() throws Exception {
- EidRecord mapRequestRecord = new EidRecord((byte) 128, v6Address);
- mapRequest.addEidRecord(mapRequestRecord);
+ prepareMapping(record.build());
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(new LispIpv6Address("0:0:0:0:0:0:0:1")).setMaskLength(112);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- prepareMapping(record);
+ MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(3, eidToLocators.getLocatorRecord().size());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(0, eidToLocators.getLocators().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));
}
@Test
- public void handleMapRequest__VerifyMaskNoMatch() throws Exception {
- EidRecord mapRequestRecord = new EidRecord((byte) 0, v4Address);
- mapRequestRecord.setMaskLength(32);
- mapRequest.addEidRecord(mapRequestRecord);
+ public void handleMapRequest__MultipleEIDs() throws Exception {
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(new LispIpv4Address("1.2.4.0")).setMaskLength(24);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- prepareMapping(record);
+ mapRequest = getDefaultMapRequestBuilder();
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(v4Address).build());
+ mapRequest.getEidRecord().add(
+ new EidRecordBuilder().setMask((short) 128).setLispAddressContainer(v6Address).build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ EidToLocatorRecordBuilder record1 = getDefaultEidToLocatorBuilder();
+ record1.setLispAddressContainer(v4Address);
+ record1.setRecordTtl(100);
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(0, eidToLocators.getLocators().size());
+ LocatorRecordBuilder locator1 = getDefaultLocatorBuilder();
+ locator1.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
+ record1.getLocatorRecord().add(locator1.build());
- }
+ EidToLocatorRecordBuilder record2 = getDefaultEidToLocatorBuilder();
+ record2.setLispAddressContainer(v6Address);
+ record2.setMaskLength((short) 128);
+ record2.setRecordTtl(100);
- @Test
- public void handleMapRequest__ReplyWithMultipleLocators() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
+ LocatorRecordBuilder locator2 = getDefaultLocatorBuilder();
+ locator2.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1")));
+ record2.getLocatorRecord().add(locator2.build());
- EidToLocatorRecord record = new EidToLocatorRecord().setPrefix(v4Address);
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
- record.addLocator(new LocatorRecord().setLocator(new LispIpv6Address("::1")));
- record.addLocator(new LocatorRecord().setLocator(new LispIpv4Address("1.8.2.7")));
- prepareMapping(record);
+ prepareMapping(record1.build(), record2.build());
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest.build());
- EidToLocatorRecord eidToLocators = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(3, eidToLocators.getLocators().size());
+ EidToLocatorRecord eidToLocators1 = mapReply.getEidToLocatorRecord().get(0);
+ assertEquals(1, eidToLocators1.getLocatorRecord().size());
+ assertLocator(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")), eidToLocators1
+ .getLocatorRecord().get(0));
- 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));
+ EidToLocatorRecord eidToLocators2 = mapReply.getEidToLocatorRecord().get(1);
+ assertEquals(1, eidToLocators2.getLocatorRecord().size());
+ assertLocator(LispAFIConvertor.toContainer(LispAFIConvertor.asIPv6AfiAddress("0:0:0:0:0:0:0:1")), eidToLocators2
+ .getLocatorRecord().get(0));
}
- private void assertLocator(LispAddress expectedAddress, LocatorRecord locatorRecord) {
- assertEquals(expectedAddress, locatorRecord.getLocator());
+ private void assertLocator(LispAddressContainer expectedAddress, LocatorRecord locatorRecord) {
+ assertEquals(expectedAddress, locatorRecord.getLispAddressContainer());
Assert.assertTrue(locatorRecord.isRouted());
}
- @Test
- public void handleMapRequest__MultipleEIDs() throws Exception {
- mapRequest.addEidRecord(new EidRecord((byte) 0, v4Address));
- mapRequest.addEidRecord(new EidRecord((byte) 0, v6Address));
+ private Map<String, EidToLocatorRecord> prepareMapping(EidToLocatorRecord... records) {
+ if (records.length > 0) {
+ for (EidToLocatorRecord eidToLocatorRecord : records) {
+ Map<String, EidToLocatorRecord> result = new HashMap<String, EidToLocatorRecord>();
+ result.put(SubKeys.RECORD, eidToLocatorRecord);
- EidToLocatorRecord record1 = new EidToLocatorRecord().setPrefix(v4Address);
- record1.addLocator(new LocatorRecord().setLocator(new LispIpv4Address(0x04030201)));
+ daoResults.put(eidToLocatorRecord.getLispAddressContainer(), result);
+ }
+ }
- EidToLocatorRecord record2 = new EidToLocatorRecord().setPrefix(v6Address);
- record2.addLocator(new LocatorRecord().setLocator(new LispIpv6Address("0:0:0:0:0:0:0:1")));
- // prepareMapping(record2);
- prepareMapping(record1, record2);
+ ValueSaverAction<LispAddressContainer> daoGetSaverAction = new ValueSaverAction<LispAddressContainer>() {
+ @Override
+ protected boolean validate(LispAddressContainer value) {
+ return true;
+ }
- MapReply mapReply = testedMapResolver.handleMapRequest(mapRequest);
+ @Override
+ public Object invoke(Invocation invocation) throws Throwable {
+ return daoResults.get(lastValue);
+ }
+ };
+
+ allowing(lispDAO).get(with(daoGetSaverAction));
+ will(daoGetSaverAction);
+ allowing(lispDAO).getSpecific(wany(LispAddressContainer.class), with(SubKeys.PASSWORD));
+
+ return daoResults.get(v4Address);
+ }
- EidToLocatorRecord eidToLocators1 = mapReply.getEidToLocatorRecords().get(0);
- assertEquals(1, eidToLocators1.getLocators().size());
- assertLocator(new LispIpv4Address(0x04030201), eidToLocators1.getLocators().get(0));
+ 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;
+ }
- EidToLocatorRecord eidToLocators2 = mapReply.getEidToLocatorRecords().get(1);
- assertEquals(1, eidToLocators2.getLocators().size());
- assertLocator(new LispIpv6Address("0:0:0:0:0:0:0:1"), eidToLocators2.getLocators().get(0));
+ 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