2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.lispflowmapping.implementation.lisp;
10 import static org.junit.Assert.assertEquals;
12 import com.google.common.collect.Lists;
13 import com.google.common.collect.Sets;
14 import java.lang.reflect.Field;
15 import java.util.ArrayList;
16 import java.util.Date;
17 import java.util.HashSet;
18 import java.util.List;
20 import org.junit.Before;
21 import org.junit.Test;
22 import org.junit.runner.RunWith;
23 import org.mockito.ArgumentCaptor;
24 import org.mockito.Mock;
25 import org.mockito.Mockito;
26 import org.mockito.Spy;
27 import org.mockito.runners.MockitoJUnitRunner;
28 import org.opendaylight.controller.md.sal.binding.api.NotificationService;
29 import org.opendaylight.lispflowmapping.config.ConfigIni;
30 import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
31 import org.opendaylight.lispflowmapping.interfaces.dao.Subscriber;
32 import org.opendaylight.lispflowmapping.interfaces.lisp.IMapNotifyHandler;
33 import org.opendaylight.lispflowmapping.interfaces.mappingservice.IMappingService;
34 import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
35 import org.opendaylight.lispflowmapping.lisp.type.MappingData;
36 import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
37 import org.opendaylight.lispflowmapping.lisp.util.SourceDestKeyHelper;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinary;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv4AddressBinary;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapRegister;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.XtrId;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecord;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecordBuilder;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapnotifymessage.MapNotify;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapnotifymessage.MapNotifyBuilder;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkeyBuilder;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemBuilder;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemKey;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapregistermessage.MapRegisterBuilder;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequestnotification.MapRequest;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddress;
60 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddressBuilder;
61 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
63 @RunWith(MockitoJUnitRunner.class)
64 public class MapServerTest {
66 @Mock private static IMappingService mapService;
67 @Mock private static IMapNotifyHandler notifyHandler;
68 @Mock private static NotificationService notificationService;
70 @Spy private static Set<Subscriber> subscriberSetMock_1 = new HashSet<>();
71 @Spy private static Set<Subscriber> subscriberSetMock_2 = new HashSet<>();
72 @Spy private static Set<Subscriber> subscriberSetMock_3 = new HashSet<>();
73 private static MapServer mapServer;
74 private static MapRegister mapRegister;
75 private static MappingData mappingData;
77 private static final String IPV4_STRING_1 = "1.2.3.0";
78 private static final String IPV4_STRING_2 = "1.2.4.0";
79 private static final String IPV4_STRING_3 = "192.168.0.1";
80 private static final String IPV4_STRING_4 = "192.168.0.2";
81 private static final String IPV4_STRING_5 = "192.168.0.3";
82 private static final String IPV4_STRING_6 = "192.168.0.4";
83 private static final String IPV4_SOURCE_STRING_1 = "127.0.0.1";
84 private static final String IPV4_SOURCE_STRING_2 = "127.0.0.2";
85 private static final String IPV4_SOURCE_STRING_3 = "127.0.0.3";
86 private static final String IPV4_SOURCE_STRING_4 = "127.0.0.4";
87 private static final String IPV4_SOURCE_STRING_5 = "127.0.0.5";
88 private static final String IPV4_SOURCE_STRING_6 = "127.0.0.6";
89 private static final String IPV4_PREFIX_1 = "/24";
90 private static final String IPV4_PREFIX_2 = "/16";
91 private static final int MASK = 24;
92 private static final int VNI = 10;
94 private static final Eid IPV4_EID_1 = LispAddressUtil.asIpv4Eid(IPV4_STRING_1);
95 private static final Eid IPV4_EID_2 = LispAddressUtil.asIpv4Eid(IPV4_STRING_2);
96 private static final Eid IPV4_PREFIX_EID_1 = LispAddressUtil.asIpv4PrefixBinaryEid(IPV4_STRING_1 + IPV4_PREFIX_1);
97 private static final Eid IPV4_PREFIX_EID_2 = LispAddressUtil.asIpv4PrefixBinaryEid(IPV4_STRING_1 + IPV4_PREFIX_2);
98 private static final Eid IPV4_SOURCE_EID_1 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_1);
99 private static final Eid IPV4_SOURCE_EID_2 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_2);
100 private static final Eid IPV4_SOURCE_EID_3 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_3);
101 private static final Eid IPV4_SOURCE_EID_4 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_4);
102 private static final Eid IPV4_SOURCE_EID_5 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_5);
103 private static final Eid IPV4_SOURCE_EID_6 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_6);
104 private static final Rloc RLOC_1 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_1);
105 private static final Rloc RLOC_2 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_2);
106 private static final Rloc RLOC_3 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_3);
107 private static final Rloc RLOC_4 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_4);
108 private static final Rloc RLOC_5 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_5);
109 private static final Rloc RLOC_6 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_6);
111 private static final IpAddressBinary IPV4_BINARY_1 =
112 new IpAddressBinary(new Ipv4AddressBinary(new byte[] {1, 2, 3, 0}));
113 private static final IpAddressBinary IPV4_BINARY_2 =
114 new IpAddressBinary(new Ipv4AddressBinary(new byte[] {1, 2, 4, 0}));
116 private static final long TWO_DAYS = 86400000L * 2;
118 private static final Subscriber SUBSCRIBER_RLOC_1 = new Subscriber(RLOC_1, // timedOut() == true
119 IPV4_SOURCE_EID_1, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT,
120 new Date(System.currentTimeMillis() - TWO_DAYS));
121 private static final Subscriber SUBSCRIBER_RLOC_2 = new Subscriber(RLOC_2, // timedOut() == false
122 IPV4_SOURCE_EID_2, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT);
123 private static final Subscriber SUBSCRIBER_RLOC_3 = new Subscriber(RLOC_3, // timedOut() == true
124 IPV4_SOURCE_EID_3, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT,
125 new Date(System.currentTimeMillis() - TWO_DAYS));
126 private static final Subscriber SUBSCRIBER_RLOC_4 = new Subscriber(RLOC_4, // timedOut() == false
127 IPV4_SOURCE_EID_4, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT);
128 private static final Subscriber SUBSCRIBER_RLOC_5 = new Subscriber(RLOC_5, // timedOut() == true
129 IPV4_SOURCE_EID_5, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT,
130 new Date(System.currentTimeMillis() - TWO_DAYS));
131 private static final Subscriber SUBSCRIBER_RLOC_6 = new Subscriber(RLOC_6, // timedOut() == false
132 IPV4_SOURCE_EID_6, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT);
134 private static final Eid SOURCE_DEST_KEY_EID = LispAddressUtil
135 .asSrcDstEid(IPV4_STRING_1, IPV4_STRING_2, MASK, MASK, VNI);
136 private static final MappingAuthkey MAPPING_AUTHKEY = new MappingAuthkeyBuilder().setKeyType(0).build();
137 private static final ConfigIni CONFIG_INI = ConfigIni.getInstance();
139 private static final LocatorRecord LOCATOR_RECORD_1 = new LocatorRecordBuilder().setRloc(RLOC_1).build();
140 private static final LocatorRecord LOCATOR_RECORD_2 = new LocatorRecordBuilder().setRloc(RLOC_2).build();
142 private static final MappingRecord OLD_MAPPING_RECORD_1 = getDefaultMappingRecordBuilder()
143 .setLocatorRecord(Lists.newArrayList(LOCATOR_RECORD_1)).build();
144 private static final MappingRecord OLD_MAPPING_RECORD_2 = getDefaultMappingRecordBuilder()
145 .setLocatorRecord(Lists.newArrayList(LOCATOR_RECORD_2)).build();
146 private static final MappingData OLD_MAPPING_DATA_1 = new MappingData(OLD_MAPPING_RECORD_1);
147 private static final MappingData OLD_MAPPING_DATA_2 = new MappingData(OLD_MAPPING_RECORD_2);
149 private static final Set<IpAddressBinary> DEFAULT_IP_ADDRESS_SET = getDefaultIpAddressSet();
152 public void init() throws NoSuchFieldException, IllegalAccessException {
153 mapServer = new MapServer(mapService, true, notifyHandler, notificationService);
154 subscriberSetMock_1.add(SUBSCRIBER_RLOC_1);
155 subscriberSetMock_1.add(SUBSCRIBER_RLOC_2);
156 subscriberSetMock_2.add(SUBSCRIBER_RLOC_3);
157 subscriberSetMock_2.add(SUBSCRIBER_RLOC_4);
158 subscriberSetMock_3.add(SUBSCRIBER_RLOC_5);
159 subscriberSetMock_3.add(SUBSCRIBER_RLOC_6);
160 mapRegister = getDefaultMapRegisterBuilder().build();
161 mappingData = new MappingData(mapRegister.getMappingRecordItem().iterator().next().getMappingRecord(),
162 System.currentTimeMillis());
163 setConfigIniMappingMergeField(false);
167 public void handleMapRegisterTest_MappingMergeFalse() throws NoSuchFieldException, IllegalAccessException {
168 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1)).thenReturn(OLD_MAPPING_DATA_1);
169 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SUBSCRIBERS))
170 .thenReturn(subscriberSetMock_1);
172 mappingData.setMergeEnabled(false);
173 mapServer.handleMapRegister(mapRegister);
175 final ArgumentCaptor<MappingData> captor = ArgumentCaptor.forClass(MappingData.class);
176 Mockito.verify(mapService).addMapping(Mockito.eq(MappingOrigin.Southbound), Mockito.eq(IPV4_EID_1),
177 Mockito.eq(mapRegister.getSiteId()), captor.capture());
178 assertEquals(captor.getValue().getRecord(), mappingData.getRecord());
179 Mockito.verify(notifyHandler).handleMapNotify(getDefaultMapNotifyBuilder(mapRegister)
180 .setAuthenticationData(null).build(), null);
182 // only 1 subscriber has timed out.
183 assertEquals(2, subscriberSetMock_1.size());
187 public void handleMapRegisterTest_MappingMergeTrue() throws NoSuchFieldException, IllegalAccessException {
188 setConfigIniMappingMergeField(true);
190 final MappingRecordItemBuilder mappingRecordItemBuilder = new MappingRecordItemBuilder()
191 .setMappingRecord(OLD_MAPPING_RECORD_1);
192 final MapNotifyBuilder mapNotifyBuilder = getDefaultMapNotifyBuilder(mapRegister)
193 .setMappingRecordItem(new ArrayList<>());
194 mapNotifyBuilder.getMappingRecordItem().add(mappingRecordItemBuilder.build());
196 // no mapping changes
197 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
198 .thenReturn(OLD_MAPPING_DATA_1);
200 mappingData.setMergeEnabled(true);
201 mapServer.handleMapRegister(mapRegister);
203 final ArgumentCaptor<MappingData> captor = ArgumentCaptor.forClass(MappingData.class);
204 Mockito.verify(mapService).addMapping(Mockito.eq(MappingOrigin.Southbound), Mockito.eq(IPV4_EID_1),
205 Mockito.eq(mapRegister.getSiteId()), captor.capture());
206 assertEquals(captor.getValue().getRecord(), mappingData.getRecord());
207 Mockito.verify(notifyHandler).handleMapNotify(mapNotifyBuilder.setAuthenticationData(null).build(), null);
211 public void handleMapRegisterTest_findNegativeSubscribers() throws NoSuchFieldException, IllegalAccessException {
212 setConfigIniMappingMergeField(true);
214 mapRegister.getMappingRecordItem().clear();
215 mapRegister.getMappingRecordItem().add(getDefaultMappingRecordItemBuilder(IPV4_PREFIX_EID_1).build());
217 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
219 .setEid(IPV4_PREFIX_EID_2);
220 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
221 final Eid maskedEid1 = LispAddressUtil.asIpv4Eid("1.2.0.0");
223 final Subscriber subscriber1 = Mockito.mock(Subscriber.class);
224 Mockito.when(subscriber1.timedOut()).thenReturn(true);
225 Mockito.when(subscriber1.toString()).thenReturn("sub1");
227 final Set<Subscriber> set1 = Sets.newHashSet(subscriber1);
229 Mockito.when(mapService.getAuthenticationKey(IPV4_PREFIX_EID_1)).thenReturn(MAPPING_AUTHKEY);
230 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_PREFIX_EID_1, SubKeys.SRC_RLOCS))
231 .thenReturn(DEFAULT_IP_ADDRESS_SET);
233 Mockito.when(mapService.getParentPrefix(IPV4_PREFIX_EID_1)).thenReturn(maskedEid1);
234 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_PREFIX_EID_1, SubKeys.SUBSCRIBERS))
236 Mockito.when(mapService.getData(MappingOrigin.Southbound, maskedEid1, SubKeys.SUBSCRIBERS)).thenReturn(set1);
238 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_PREFIX_EID_1))
239 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
240 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
243 mapServer.handleMapRegister(mapRegister);
244 Mockito.verify(subscriber1).timedOut();
248 public void handleMapRegisterTest_verifyTransportAddresses() throws NoSuchFieldException, IllegalAccessException {
249 setConfigIniMappingMergeField(true);
252 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
253 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
254 .thenReturn(OLD_MAPPING_DATA_1)
255 .thenReturn(OLD_MAPPING_DATA_2)
256 .thenReturn(getDefaultMappingData(getDefaultMappingRecordBuilder().build()));
257 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SUBSCRIBERS))
258 .thenReturn(subscriberSetMock_1);
259 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
260 .thenReturn(DEFAULT_IP_ADDRESS_SET);
263 final List<TransportAddress> transportAddressList = getTransportAddressList();
264 final MapNotifyBuilder mapNotifyBuilder = getDefaultMapNotifyBuilder(mapRegister);
265 mapNotifyBuilder.setMappingRecordItem(new ArrayList<>());
266 mapNotifyBuilder.getMappingRecordItem().add(new MappingRecordItemBuilder()
267 .setMappingRecord(getDefaultMappingRecordBuilder().build()).build());
269 mapServer.handleMapRegister(mapRegister);
270 Mockito.verify(notifyHandler).handleMapNotify(mapNotifyBuilder.build(), transportAddressList);
274 public void handleMapRegisterTest_withTwoMappingRecords() throws NoSuchFieldException, IllegalAccessException {
275 setConfigIniMappingMergeField(true);
278 // Add a MappingRecord with SrcDestKey Eid Type
279 final MappingRecordItemBuilder mappingRecordItemBuilder = new MappingRecordItemBuilder()
280 .setMappingRecord(getDefaultMappingRecordBuilder().setEid(SOURCE_DEST_KEY_EID).build());
281 final MapRegisterBuilder mapRegisterSrcDstBuilder = getDefaultMapRegisterBuilder();
283 final List<MappingRecordItem> list = mapRegisterSrcDstBuilder.getMappingRecordItem();
284 list.add(mappingRecordItemBuilder.build());
286 // ------------- Stubbing for SourceDestKey type Eid mapping -------------------
288 Mockito.when(mapService.getAuthenticationKey(SOURCE_DEST_KEY_EID)).thenReturn(MAPPING_AUTHKEY);
289 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, SOURCE_DEST_KEY_EID))
290 // ensure mappings are different
291 .thenReturn(OLD_MAPPING_DATA_1)
292 .thenReturn(OLD_MAPPING_DATA_2)
293 .thenReturn(OLD_MAPPING_DATA_2);
294 // return a subscriberSet for SrcDestKeyEid MappingRecord
295 Mockito.when(mapService.getData(MappingOrigin.Southbound, SOURCE_DEST_KEY_EID, SubKeys.SUBSCRIBERS))
296 .thenReturn(subscriberSetMock_1);
298 // return a subscriberSet for SrcDestKeyEid destination MappingRecord
299 Mockito.when(mapService.getData(MappingOrigin.Southbound, SourceDestKeyHelper.getDstBinary(SOURCE_DEST_KEY_EID),
300 SubKeys.SUBSCRIBERS)).thenReturn(subscriberSetMock_2);
302 // ----------------- Stubbing for Ipv4 type Eid mapping ------------------------
304 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
305 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
306 // ensure mappings are different
307 .thenReturn(OLD_MAPPING_DATA_1)
308 .thenReturn(OLD_MAPPING_DATA_2);
309 // return a subscriberSet for Ipv4Eid MappingRecord
310 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1,SubKeys.SUBSCRIBERS))
311 .thenReturn(subscriberSetMock_3);
313 // -----------------------------------------------------------------------------
316 mapServer.handleMapRegister(mapRegisterSrcDstBuilder.build());
318 // for SrcDstKey mapping
319 final ArgumentCaptor<MapRequest> captor_1 = ArgumentCaptor.forClass(MapRequest.class);
320 Mockito.verify(notifyHandler, Mockito.times(1)).handleSMR(captor_1.capture(), Mockito.eq(RLOC_2));
321 final Eid resultEid_1 = captor_1.getValue().getEidItem().iterator().next().getEid();
322 assertEquals(IPV4_SOURCE_EID_2, resultEid_1);
324 // for SrcDst destination mapping
325 final ArgumentCaptor<MapRequest> captor_2 = ArgumentCaptor.forClass(MapRequest.class);
326 Mockito.verify(notifyHandler, Mockito.times(1)).handleSMR(captor_2.capture(), Mockito.eq(RLOC_4));
327 final Eid resultEid_2 = captor_2.getValue().getEidItem().iterator().next().getEid();
328 assertEquals(IPV4_SOURCE_EID_4, resultEid_2);
331 final ArgumentCaptor<MapRequest> captor_3 = ArgumentCaptor.forClass(MapRequest.class);
332 Mockito.verify(notifyHandler, Mockito.times(2)).handleSMR(captor_3.capture(), Mockito.eq(RLOC_6));
333 final Eid resultEid_3 = captor_3.getValue().getEidItem().iterator().next().getEid();
334 assertEquals(IPV4_SOURCE_EID_6, resultEid_3);
338 @SuppressWarnings({ "unchecked", "rawtypes" })
339 public void mappingChangedTest_withDifferentEid() throws NoSuchFieldException, IllegalAccessException {
340 setConfigIniMappingMergeField(true);
342 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
345 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
346 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
348 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
349 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
350 .thenReturn(DEFAULT_IP_ADDRESS_SET);
352 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
353 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
354 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
357 mapServer.handleMapRegister(mapRegister);
358 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
359 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
360 assertEquals(2, captor.getValue().size());
364 @SuppressWarnings({ "unchecked", "rawtypes" })
365 public void mappingChangedTest_withDifferentRLOC() throws NoSuchFieldException, IllegalAccessException {
366 setConfigIniMappingMergeField(true);
368 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder();
370 mappingRecordBuilder_1.getLocatorRecord().add(new LocatorRecordBuilder().build());
371 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
372 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
374 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
375 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
376 .thenReturn(DEFAULT_IP_ADDRESS_SET);
378 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
379 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
380 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
383 mapServer.handleMapRegister(mapRegister);
384 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
385 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
386 assertEquals(2, captor.getValue().size());
390 @SuppressWarnings({ "unchecked", "rawtypes" })
391 public void mappingChangedTest_withDifferentAction() throws NoSuchFieldException, IllegalAccessException {
392 setConfigIniMappingMergeField(true);
394 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
396 .setAction(MappingRecord.Action.NativelyForward);
397 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
398 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
400 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
401 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
402 .thenReturn(DEFAULT_IP_ADDRESS_SET);
404 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
405 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
406 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
409 mapServer.handleMapRegister(mapRegister);
410 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
411 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
412 assertEquals(2, captor.getValue().size());
416 @SuppressWarnings({ "unchecked", "rawtypes" })
417 public void mappingChangedTest_withDifferentTTL() throws NoSuchFieldException, IllegalAccessException {
418 setConfigIniMappingMergeField(true);
420 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
423 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
424 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
426 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
427 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
428 .thenReturn(DEFAULT_IP_ADDRESS_SET);
430 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
431 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
432 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
435 mapServer.handleMapRegister(mapRegister);
436 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
437 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
438 assertEquals(2, captor.getValue().size());
442 @SuppressWarnings({ "unchecked", "rawtypes" })
443 public void mappingChangedTest_withDifferentMapVersion() throws NoSuchFieldException, IllegalAccessException {
444 setConfigIniMappingMergeField(true);
446 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
448 .setMapVersion((short) 10);
449 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
450 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
452 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
453 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
454 .thenReturn(DEFAULT_IP_ADDRESS_SET);
457 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
458 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
459 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
462 mapServer.handleMapRegister(mapRegister);
463 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
464 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
465 assertEquals(2, captor.getValue().size());
469 @SuppressWarnings({ "unchecked", "rawtypes" })
470 public void mappingChangedTest_withNullMap() throws NoSuchFieldException, IllegalAccessException {
471 setConfigIniMappingMergeField(true);
473 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
474 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
476 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
477 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
478 .thenReturn(DEFAULT_IP_ADDRESS_SET);
480 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
482 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
485 mapServer.handleMapRegister(mapRegister);
486 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
487 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
488 assertEquals(2, captor.getValue().size());
491 private static MapRegisterBuilder getDefaultMapRegisterBuilder() {
492 final MapRegisterBuilder mapRegisterBuilder = new MapRegisterBuilder()
493 .setProxyMapReply(true)
494 .setWantMapNotify(true)
496 .setMappingRecordItem(new ArrayList<>())
497 .setMergeEnabled(true)
499 .setSiteId(new SiteId(new byte[]{0, 1, 2, 3, 4, 5, 6, 7}))
500 .setXtrId(new XtrId(new byte[]{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}))
501 .setXtrSiteIdPresent(true);
502 mapRegisterBuilder.getMappingRecordItem().add(getDefaultMappingRecordItemBuilder().build());
504 return mapRegisterBuilder;
507 private static MappingData getDefaultMappingData(MappingRecord mappingRecord) {
508 if (mappingRecord == null) {
509 mappingRecord = getDefaultMappingRecordBuilder().build();
511 return new MappingData(mappingRecord, System.currentTimeMillis());
514 private static MappingRecordItemBuilder getDefaultMappingRecordItemBuilder() {
515 return getDefaultMappingRecordItemBuilder(IPV4_EID_1);
518 private static MappingRecordItemBuilder getDefaultMappingRecordItemBuilder(Eid eid) {
519 return new MappingRecordItemBuilder()
520 .setMappingRecordItemId("mapping-record-item-id")
521 .setKey(new MappingRecordItemKey("mapping-record-item-key"))
522 .setMappingRecord(getDefaultMappingRecordBuilder(eid).build());
525 private static MappingRecordBuilder getDefaultMappingRecordBuilder() {
526 return getDefaultMappingRecordBuilder(IPV4_EID_1);
529 private static MappingRecordBuilder getDefaultMappingRecordBuilder(Eid eid) {
530 return new MappingRecordBuilder()
531 .setAction(MappingRecord.Action.NoAction)
532 .setAuthoritative(false)
533 .setLocatorRecord(new ArrayList<>())
534 .setMapVersion((short) 0)
539 private static MapNotifyBuilder getDefaultMapNotifyBuilder(MapRegister mapRegister) {
540 final MapNotifyBuilder mapNotifyBuilder = new MapNotifyBuilder()
541 .setXtrSiteIdPresent(mapRegister.isXtrSiteIdPresent())
542 .setSiteId(mapRegister.getSiteId())
543 .setXtrId(mapRegister.getXtrId())
544 .setNonce(mapRegister.getNonce())
545 .setKeyId(mapRegister.getKeyId())
546 .setMergeEnabled(mapRegister.isMergeEnabled())
547 .setMappingRecordItem(new ArrayList<>())
548 .setAuthenticationData(new byte[]{});
549 mapNotifyBuilder.getMappingRecordItem().add(getDefaultMappingRecordItemBuilder().build());
551 return mapNotifyBuilder;
554 private static void setConfigIniMappingMergeField(boolean value) throws NoSuchFieldException,
555 IllegalAccessException {
556 final Field mappingMergeField = CONFIG_INI.getClass().getDeclaredField("mappingMerge");
557 mappingMergeField.setAccessible(true);
558 mappingMergeField.setBoolean(CONFIG_INI, value);
561 private static Set<IpAddressBinary> getDefaultIpAddressSet() {
562 final Set<IpAddressBinary> addressSet = Sets.newHashSet(IPV4_BINARY_1, IPV4_BINARY_2);
567 private static List<TransportAddress> getTransportAddressList() {
568 TransportAddressBuilder transportAddressBuilder1 = new TransportAddressBuilder()
569 .setIpAddress(IPV4_BINARY_1)
570 .setPort(new PortNumber(LispMessage.PORT_NUM));
572 TransportAddressBuilder transportAddressBuilder2 = new TransportAddressBuilder()
573 .setIpAddress(IPV4_BINARY_2)
574 .setPort(new PortNumber(LispMessage.PORT_NUM));
576 final List<TransportAddress> transportAddressList = Lists.newArrayList(
577 transportAddressBuilder1.build(),
578 transportAddressBuilder2.build());
580 return transportAddressList;