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.Ignore;
22 import org.junit.Test;
23 import org.junit.runner.RunWith;
24 import org.mockito.ArgumentCaptor;
25 import org.mockito.Mock;
26 import org.mockito.Mockito;
27 import org.mockito.Spy;
28 import org.mockito.runners.MockitoJUnitRunner;
29 import org.opendaylight.controller.md.sal.binding.api.NotificationService;
30 import org.opendaylight.lispflowmapping.config.ConfigIni;
31 import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
32 import org.opendaylight.lispflowmapping.interfaces.dao.Subscriber;
33 import org.opendaylight.lispflowmapping.interfaces.lisp.IMapNotifyHandler;
34 import org.opendaylight.lispflowmapping.interfaces.mappingservice.IMappingService;
35 import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
36 import org.opendaylight.lispflowmapping.lisp.type.MappingData;
37 import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
38 import org.opendaylight.lispflowmapping.lisp.util.SourceDestKeyHelper;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinary;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv4AddressBinary;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapRegister;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.XtrId;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecord;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecordBuilder;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapnotifymessage.MapNotify;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapnotifymessage.MapNotifyBuilder;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkeyBuilder;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemBuilder;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemKey;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapregistermessage.MapRegisterBuilder;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequestnotification.MapRequest;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
60 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddress;
61 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddressBuilder;
62 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
64 @RunWith(MockitoJUnitRunner.class)
65 public class MapServerTest {
67 @Mock private static IMappingService mapService;
68 @Mock private static IMapNotifyHandler notifyHandler;
69 @Mock private static NotificationService notificationService;
71 @Spy private static Set<Subscriber> subscriberSetMock_1 = new HashSet<>();
72 @Spy private static Set<Subscriber> subscriberSetMock_2 = new HashSet<>();
73 @Spy private static Set<Subscriber> subscriberSetMock_3 = new HashSet<>();
74 private static MapServer mapServer;
75 private static MapRegister mapRegister;
76 private static MappingData mappingData;
78 private static final String IPV4_STRING_1 = "1.2.3.0";
79 private static final String IPV4_STRING_2 = "1.2.4.0";
80 private static final String IPV4_STRING_3 = "192.168.0.1";
81 private static final String IPV4_STRING_4 = "192.168.0.2";
82 private static final String IPV4_STRING_5 = "192.168.0.3";
83 private static final String IPV4_STRING_6 = "192.168.0.4";
84 private static final String IPV4_SOURCE_STRING_1 = "127.0.0.1";
85 private static final String IPV4_SOURCE_STRING_2 = "127.0.0.2";
86 private static final String IPV4_SOURCE_STRING_3 = "127.0.0.3";
87 private static final String IPV4_SOURCE_STRING_4 = "127.0.0.4";
88 private static final String IPV4_SOURCE_STRING_5 = "127.0.0.5";
89 private static final String IPV4_SOURCE_STRING_6 = "127.0.0.6";
90 private static final String IPV4_PREFIX_1 = "/24";
91 private static final String IPV4_PREFIX_2 = "/16";
92 private static final int MASK = 24;
93 private static final int VNI = 10;
95 private static final Eid IPV4_EID_1 = LispAddressUtil.asIpv4Eid(IPV4_STRING_1);
96 private static final Eid IPV4_EID_2 = LispAddressUtil.asIpv4Eid(IPV4_STRING_2);
97 private static final Eid IPV4_PREFIX_EID_1 = LispAddressUtil.asIpv4PrefixBinaryEid(IPV4_STRING_1 + IPV4_PREFIX_1);
98 private static final Eid IPV4_PREFIX_EID_2 = LispAddressUtil.asIpv4PrefixBinaryEid(IPV4_STRING_1 + IPV4_PREFIX_2);
99 private static final Eid IPV4_SOURCE_EID_1 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_1);
100 private static final Eid IPV4_SOURCE_EID_2 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_2);
101 private static final Eid IPV4_SOURCE_EID_3 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_3);
102 private static final Eid IPV4_SOURCE_EID_4 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_4);
103 private static final Eid IPV4_SOURCE_EID_5 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_5);
104 private static final Eid IPV4_SOURCE_EID_6 = LispAddressUtil.asIpv4Eid(IPV4_SOURCE_STRING_6);
105 private static final Rloc RLOC_1 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_1);
106 private static final Rloc RLOC_2 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_2);
107 private static final Rloc RLOC_3 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_3);
108 private static final Rloc RLOC_4 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_4);
109 private static final Rloc RLOC_5 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_5);
110 private static final Rloc RLOC_6 = LispAddressUtil.asIpv4Rloc(IPV4_STRING_6);
112 private static final IpAddressBinary IPV4_BINARY_1 =
113 new IpAddressBinary(new Ipv4AddressBinary(new byte[] {1, 2, 3, 0}));
114 private static final IpAddressBinary IPV4_BINARY_2 =
115 new IpAddressBinary(new Ipv4AddressBinary(new byte[] {1, 2, 4, 0}));
117 private static final long TWO_DAYS = 86400000L * 2;
119 // timedOut() == true
120 private static final Subscriber SUBSCRIBER_RLOC_1 = new Subscriber(RLOC_1,
121 IPV4_SOURCE_EID_1, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT,
122 new Date(System.currentTimeMillis() - TWO_DAYS));
123 // timedOut() == false
124 private static final Subscriber SUBSCRIBER_RLOC_2 = new Subscriber(RLOC_2,
125 IPV4_SOURCE_EID_2, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT);
126 // timedOut() == true
127 private static final Subscriber SUBSCRIBER_RLOC_3 = new Subscriber(RLOC_3,
128 IPV4_SOURCE_EID_3, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT,
129 new Date(System.currentTimeMillis() - TWO_DAYS));
130 // timedOut() == false
131 private static final Subscriber SUBSCRIBER_RLOC_4 = new Subscriber(RLOC_4,
132 IPV4_SOURCE_EID_4, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT);
133 // timedOut() == true
134 private static final Subscriber SUBSCRIBER_RLOC_5 = new Subscriber(RLOC_5,
135 IPV4_SOURCE_EID_5, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT,
136 new Date(System.currentTimeMillis() - TWO_DAYS));
137 // timedOut() == false
138 private static final Subscriber SUBSCRIBER_RLOC_6 = new Subscriber(RLOC_6,
139 IPV4_SOURCE_EID_6, Subscriber.DEFAULT_SUBSCRIBER_TIMEOUT);
141 private static final Eid SOURCE_DEST_KEY_EID = LispAddressUtil
142 .asSrcDstEid(IPV4_STRING_1, IPV4_STRING_2, MASK, MASK, VNI);
143 private static final MappingAuthkey MAPPING_AUTHKEY = new MappingAuthkeyBuilder().setKeyType(0).build();
144 private static final ConfigIni CONFIG_INI = ConfigIni.getInstance();
146 private static final LocatorRecord LOCATOR_RECORD_1 = new LocatorRecordBuilder().setRloc(RLOC_1).build();
147 private static final LocatorRecord LOCATOR_RECORD_2 = new LocatorRecordBuilder().setRloc(RLOC_2).build();
149 private static final MappingRecord OLD_MAPPING_RECORD_1 = getDefaultMappingRecordBuilder()
150 .setLocatorRecord(Lists.newArrayList(LOCATOR_RECORD_1)).build();
151 private static final MappingRecord OLD_MAPPING_RECORD_2 = getDefaultMappingRecordBuilder()
152 .setLocatorRecord(Lists.newArrayList(LOCATOR_RECORD_2)).build();
153 private static final MappingData OLD_MAPPING_DATA_1 = new MappingData(OLD_MAPPING_RECORD_1);
154 private static final MappingData OLD_MAPPING_DATA_2 = new MappingData(OLD_MAPPING_RECORD_2);
156 private static final Set<IpAddressBinary> DEFAULT_IP_ADDRESS_SET = getDefaultIpAddressSet();
159 public void init() throws NoSuchFieldException, IllegalAccessException {
160 mapServer = new MapServer(mapService, true, notifyHandler, notificationService);
161 subscriberSetMock_1.add(SUBSCRIBER_RLOC_1);
162 subscriberSetMock_1.add(SUBSCRIBER_RLOC_2);
163 subscriberSetMock_2.add(SUBSCRIBER_RLOC_3);
164 subscriberSetMock_2.add(SUBSCRIBER_RLOC_4);
165 subscriberSetMock_3.add(SUBSCRIBER_RLOC_5);
166 subscriberSetMock_3.add(SUBSCRIBER_RLOC_6);
167 mapRegister = getDefaultMapRegisterBuilder().build();
168 mappingData = new MappingData(mapRegister.getMappingRecordItem().iterator().next().getMappingRecord(),
169 System.currentTimeMillis());
170 setConfigIniMappingMergeField(false);
174 public void handleMapRegisterTest_MappingMergeFalse() throws NoSuchFieldException, IllegalAccessException {
175 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1)).thenReturn(OLD_MAPPING_DATA_1);
176 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SUBSCRIBERS))
177 .thenReturn(subscriberSetMock_1);
179 mappingData.setMergeEnabled(false);
180 mapServer.handleMapRegister(mapRegister);
182 final ArgumentCaptor<MappingData> captor = ArgumentCaptor.forClass(MappingData.class);
183 Mockito.verify(mapService).addMapping(Mockito.eq(MappingOrigin.Southbound), Mockito.eq(IPV4_EID_1),
184 Mockito.eq(mapRegister.getSiteId()), captor.capture());
185 assertEquals(captor.getValue().getRecord(), mappingData.getRecord());
186 Mockito.verify(notifyHandler).handleMapNotify(getDefaultMapNotifyBuilder(mapRegister)
187 .setAuthenticationData(null).build(), null);
189 // only 1 subscriber has timed out.
190 assertEquals(2, subscriberSetMock_1.size());
194 public void handleMapRegisterTest_MappingMergeTrue() throws NoSuchFieldException, IllegalAccessException {
195 setConfigIniMappingMergeField(true);
197 final MappingRecordItemBuilder mappingRecordItemBuilder = new MappingRecordItemBuilder()
198 .setMappingRecord(OLD_MAPPING_RECORD_1);
199 final MapNotifyBuilder mapNotifyBuilder = getDefaultMapNotifyBuilder(mapRegister)
200 .setMappingRecordItem(new ArrayList<>());
201 mapNotifyBuilder.getMappingRecordItem().add(mappingRecordItemBuilder.build());
203 // no mapping changes
204 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
205 .thenReturn(OLD_MAPPING_DATA_1);
207 mappingData.setMergeEnabled(true);
208 mapServer.handleMapRegister(mapRegister);
210 final ArgumentCaptor<MappingData> captor = ArgumentCaptor.forClass(MappingData.class);
211 Mockito.verify(mapService).addMapping(Mockito.eq(MappingOrigin.Southbound), Mockito.eq(IPV4_EID_1),
212 Mockito.eq(mapRegister.getSiteId()), captor.capture());
213 assertEquals(captor.getValue().getRecord(), mappingData.getRecord());
214 Mockito.verify(notifyHandler).handleMapNotify(mapNotifyBuilder.setAuthenticationData(null).build(), null);
219 public void handleMapRegisterTest_findNegativeSubscribers() throws NoSuchFieldException, IllegalAccessException {
220 setConfigIniMappingMergeField(true);
222 mapRegister.getMappingRecordItem().clear();
223 mapRegister.getMappingRecordItem().add(getDefaultMappingRecordItemBuilder(IPV4_PREFIX_EID_1).build());
225 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
227 .setEid(IPV4_PREFIX_EID_2);
228 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
229 final Eid maskedEid1 = LispAddressUtil.asIpv4Eid("1.2.0.0");
231 final Subscriber subscriber1 = Mockito.mock(Subscriber.class);
232 Mockito.when(subscriber1.timedOut()).thenReturn(true);
233 Mockito.when(subscriber1.toString()).thenReturn("sub1");
235 final Set<Subscriber> set1 = Sets.newHashSet(subscriber1);
237 Mockito.when(mapService.getAuthenticationKey(IPV4_PREFIX_EID_1)).thenReturn(MAPPING_AUTHKEY);
238 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_PREFIX_EID_1, SubKeys.SRC_RLOCS))
239 .thenReturn(DEFAULT_IP_ADDRESS_SET);
241 Mockito.when(mapService.getParentPrefix(IPV4_PREFIX_EID_1)).thenReturn(maskedEid1);
242 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_PREFIX_EID_1, SubKeys.SUBSCRIBERS))
244 Mockito.when(mapService.getData(MappingOrigin.Southbound, maskedEid1, SubKeys.SUBSCRIBERS)).thenReturn(set1);
246 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_PREFIX_EID_1))
247 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
248 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
251 mapServer.handleMapRegister(mapRegister);
252 Mockito.verify(subscriber1).timedOut();
256 public void handleMapRegisterTest_verifyTransportAddresses() throws NoSuchFieldException, IllegalAccessException {
257 setConfigIniMappingMergeField(true);
260 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
261 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
262 .thenReturn(OLD_MAPPING_DATA_1)
263 .thenReturn(OLD_MAPPING_DATA_2)
264 .thenReturn(getDefaultMappingData(getDefaultMappingRecordBuilder().build()));
265 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SUBSCRIBERS))
266 .thenReturn(subscriberSetMock_1);
267 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
268 .thenReturn(DEFAULT_IP_ADDRESS_SET);
271 final List<TransportAddress> transportAddressList = getTransportAddressList();
272 final MapNotifyBuilder mapNotifyBuilder = getDefaultMapNotifyBuilder(mapRegister);
273 mapNotifyBuilder.setMappingRecordItem(new ArrayList<>());
274 mapNotifyBuilder.getMappingRecordItem().add(new MappingRecordItemBuilder()
275 .setMappingRecord(getDefaultMappingRecordBuilder().build()).build());
277 mapServer.handleMapRegister(mapRegister);
278 Mockito.verify(notifyHandler).handleMapNotify(mapNotifyBuilder.build(), transportAddressList);
283 public void handleMapRegisterTest_withTwoMappingRecords() throws NoSuchFieldException, IllegalAccessException {
284 setConfigIniMappingMergeField(true);
287 // Add a MappingRecord with SrcDestKey Eid Type
288 final MappingRecordItemBuilder mappingRecordItemBuilder = new MappingRecordItemBuilder()
289 .setMappingRecord(getDefaultMappingRecordBuilder().setEid(SOURCE_DEST_KEY_EID).build());
290 final MapRegisterBuilder mapRegisterSrcDstBuilder = getDefaultMapRegisterBuilder();
292 final List<MappingRecordItem> list = mapRegisterSrcDstBuilder.getMappingRecordItem();
293 list.add(mappingRecordItemBuilder.build());
295 // ------------- Stubbing for SourceDestKey type Eid mapping -------------------
297 Mockito.when(mapService.getAuthenticationKey(SOURCE_DEST_KEY_EID)).thenReturn(MAPPING_AUTHKEY);
298 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, SOURCE_DEST_KEY_EID))
299 // ensure mappings are different
300 .thenReturn(OLD_MAPPING_DATA_1)
301 .thenReturn(OLD_MAPPING_DATA_2)
302 .thenReturn(OLD_MAPPING_DATA_2);
303 // return a subscriberSet for SrcDestKeyEid MappingRecord
304 Mockito.when(mapService.getData(MappingOrigin.Southbound, SOURCE_DEST_KEY_EID, SubKeys.SUBSCRIBERS))
305 .thenReturn(subscriberSetMock_1);
307 // return a subscriberSet for SrcDestKeyEid destination MappingRecord
308 Mockito.when(mapService.getData(MappingOrigin.Southbound, SourceDestKeyHelper.getDstBinary(SOURCE_DEST_KEY_EID),
309 SubKeys.SUBSCRIBERS)).thenReturn(subscriberSetMock_2);
311 // ----------------- Stubbing for Ipv4 type Eid mapping ------------------------
313 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
314 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
315 // ensure mappings are different
316 .thenReturn(OLD_MAPPING_DATA_1)
317 .thenReturn(OLD_MAPPING_DATA_2);
318 // return a subscriberSet for Ipv4Eid MappingRecord
319 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1,SubKeys.SUBSCRIBERS))
320 .thenReturn(subscriberSetMock_3);
322 // -----------------------------------------------------------------------------
325 mapServer.handleMapRegister(mapRegisterSrcDstBuilder.build());
327 // for SrcDstKey mapping
328 final ArgumentCaptor<MapRequest> captor_1 = ArgumentCaptor.forClass(MapRequest.class);
329 Mockito.verify(notifyHandler, Mockito.times(1)).handleSMR(captor_1.capture(), Mockito.eq(RLOC_2));
330 final Eid resultEid_1 = captor_1.getValue().getEidItem().iterator().next().getEid();
331 assertEquals(IPV4_SOURCE_EID_2, resultEid_1);
333 // for SrcDst destination mapping
334 final ArgumentCaptor<MapRequest> captor_2 = ArgumentCaptor.forClass(MapRequest.class);
335 Mockito.verify(notifyHandler, Mockito.times(1)).handleSMR(captor_2.capture(), Mockito.eq(RLOC_4));
336 final Eid resultEid_2 = captor_2.getValue().getEidItem().iterator().next().getEid();
337 assertEquals(IPV4_SOURCE_EID_4, resultEid_2);
340 final ArgumentCaptor<MapRequest> captor_3 = ArgumentCaptor.forClass(MapRequest.class);
341 Mockito.verify(notifyHandler, Mockito.times(2)).handleSMR(captor_3.capture(), Mockito.eq(RLOC_6));
342 final Eid resultEid_3 = captor_3.getValue().getEidItem().iterator().next().getEid();
343 assertEquals(IPV4_SOURCE_EID_6, resultEid_3);
347 @SuppressWarnings({ "unchecked", "rawtypes" })
348 public void mappingChangedTest_withDifferentEid() throws NoSuchFieldException, IllegalAccessException {
349 setConfigIniMappingMergeField(true);
351 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
354 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
355 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
357 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
358 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
359 .thenReturn(DEFAULT_IP_ADDRESS_SET);
361 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
362 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
363 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
366 mapServer.handleMapRegister(mapRegister);
367 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
368 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
369 assertEquals(2, captor.getValue().size());
373 @SuppressWarnings({ "unchecked", "rawtypes" })
374 public void mappingChangedTest_withDifferentRLOC() throws NoSuchFieldException, IllegalAccessException {
375 setConfigIniMappingMergeField(true);
377 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder();
379 mappingRecordBuilder_1.getLocatorRecord().add(new LocatorRecordBuilder().build());
380 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
381 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
383 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
384 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
385 .thenReturn(DEFAULT_IP_ADDRESS_SET);
387 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
388 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
389 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
392 mapServer.handleMapRegister(mapRegister);
393 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
394 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
395 assertEquals(2, captor.getValue().size());
399 @SuppressWarnings({ "unchecked", "rawtypes" })
400 public void mappingChangedTest_withDifferentAction() throws NoSuchFieldException, IllegalAccessException {
401 setConfigIniMappingMergeField(true);
403 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
405 .setAction(MappingRecord.Action.NativelyForward);
406 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
407 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
409 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
410 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
411 .thenReturn(DEFAULT_IP_ADDRESS_SET);
413 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
414 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
415 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
418 mapServer.handleMapRegister(mapRegister);
419 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
420 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
421 assertEquals(2, captor.getValue().size());
425 @SuppressWarnings({ "unchecked", "rawtypes" })
426 public void mappingChangedTest_withDifferentTTL() throws NoSuchFieldException, IllegalAccessException {
427 setConfigIniMappingMergeField(true);
429 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
432 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
433 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
435 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
436 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
437 .thenReturn(DEFAULT_IP_ADDRESS_SET);
439 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
440 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
441 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
444 mapServer.handleMapRegister(mapRegister);
445 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
446 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
447 assertEquals(2, captor.getValue().size());
451 @SuppressWarnings({ "unchecked", "rawtypes" })
452 public void mappingChangedTest_withDifferentMapVersion() throws NoSuchFieldException, IllegalAccessException {
453 setConfigIniMappingMergeField(true);
455 final MappingRecordBuilder mappingRecordBuilder_1 = getDefaultMappingRecordBuilder()
457 .setMapVersion((short) 10);
458 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
459 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
461 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
462 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
463 .thenReturn(DEFAULT_IP_ADDRESS_SET);
466 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
467 .thenReturn(getDefaultMappingData(mappingRecordBuilder_1.build()))
468 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
471 mapServer.handleMapRegister(mapRegister);
472 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
473 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
474 assertEquals(2, captor.getValue().size());
478 @SuppressWarnings({ "unchecked", "rawtypes" })
479 public void mappingChangedTest_withNullMap() throws NoSuchFieldException, IllegalAccessException {
480 setConfigIniMappingMergeField(true);
482 final MappingRecordBuilder mappingRecordBuilder_2 = getDefaultMappingRecordBuilder();
483 final ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
485 Mockito.when(mapService.getAuthenticationKey(IPV4_EID_1)).thenReturn(MAPPING_AUTHKEY);
486 Mockito.when(mapService.getData(MappingOrigin.Southbound, IPV4_EID_1, SubKeys.SRC_RLOCS))
487 .thenReturn(DEFAULT_IP_ADDRESS_SET);
489 Mockito.when(mapService.getMapping(MappingOrigin.Southbound, IPV4_EID_1))
491 .thenReturn(getDefaultMappingData(mappingRecordBuilder_2.build()))
494 mapServer.handleMapRegister(mapRegister);
495 Mockito.verify(notifyHandler).handleMapNotify(Mockito.any(MapNotify.class), captor.capture());
496 // verify that a list of transport addresses has 2 values - happens only if mappingUpdated == true
497 assertEquals(2, captor.getValue().size());
500 private static MapRegisterBuilder getDefaultMapRegisterBuilder() {
501 final MapRegisterBuilder mapRegisterBuilder = new MapRegisterBuilder()
502 .setProxyMapReply(true)
503 .setWantMapNotify(true)
505 .setMappingRecordItem(new ArrayList<>())
506 .setMergeEnabled(true)
508 .setSiteId(new SiteId(new byte[]{0, 1, 2, 3, 4, 5, 6, 7}))
509 .setXtrId(new XtrId(new byte[]{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}))
510 .setXtrSiteIdPresent(true);
511 mapRegisterBuilder.getMappingRecordItem().add(getDefaultMappingRecordItemBuilder().build());
513 return mapRegisterBuilder;
516 private static MappingData getDefaultMappingData(MappingRecord mappingRecord) {
517 if (mappingRecord == null) {
518 mappingRecord = getDefaultMappingRecordBuilder().build();
520 return new MappingData(mappingRecord, System.currentTimeMillis());
523 private static MappingRecordItemBuilder getDefaultMappingRecordItemBuilder() {
524 return getDefaultMappingRecordItemBuilder(IPV4_EID_1);
527 private static MappingRecordItemBuilder getDefaultMappingRecordItemBuilder(Eid eid) {
528 return new MappingRecordItemBuilder()
529 .setMappingRecordItemId("mapping-record-item-id")
530 .setKey(new MappingRecordItemKey("mapping-record-item-key"))
531 .setMappingRecord(getDefaultMappingRecordBuilder(eid).build());
534 private static MappingRecordBuilder getDefaultMappingRecordBuilder() {
535 return getDefaultMappingRecordBuilder(IPV4_EID_1);
538 private static MappingRecordBuilder getDefaultMappingRecordBuilder(Eid eid) {
539 return new MappingRecordBuilder()
540 .setAction(MappingRecord.Action.NoAction)
541 .setAuthoritative(false)
542 .setLocatorRecord(new ArrayList<>())
543 .setMapVersion((short) 0)
548 private static MapNotifyBuilder getDefaultMapNotifyBuilder(MapRegister mr) {
549 final MapNotifyBuilder mapNotifyBuilder = new MapNotifyBuilder()
550 .setXtrSiteIdPresent(mr.isXtrSiteIdPresent())
551 .setSiteId(mr.getSiteId())
552 .setXtrId(mr.getXtrId())
553 .setNonce(mr.getNonce())
554 .setKeyId(mr.getKeyId())
555 .setMergeEnabled(mr.isMergeEnabled())
556 .setMappingRecordItem(new ArrayList<>())
557 .setAuthenticationData(new byte[]{});
558 mapNotifyBuilder.getMappingRecordItem().add(getDefaultMappingRecordItemBuilder().build());
560 return mapNotifyBuilder;
563 private static void setConfigIniMappingMergeField(boolean value) throws NoSuchFieldException,
564 IllegalAccessException {
565 final Field mappingMergeField = CONFIG_INI.getClass().getDeclaredField("mappingMerge");
566 mappingMergeField.setAccessible(true);
567 mappingMergeField.setBoolean(CONFIG_INI, value);
570 private static Set<IpAddressBinary> getDefaultIpAddressSet() {
571 final Set<IpAddressBinary> addressSet = Sets.newHashSet(IPV4_BINARY_1, IPV4_BINARY_2);
576 private static List<TransportAddress> getTransportAddressList() {
577 TransportAddressBuilder transportAddressBuilder1 = new TransportAddressBuilder()
578 .setIpAddress(IPV4_BINARY_1)
579 .setPort(new PortNumber(LispMessage.PORT_NUM));
581 TransportAddressBuilder transportAddressBuilder2 = new TransportAddressBuilder()
582 .setIpAddress(IPV4_BINARY_2)
583 .setPort(new PortNumber(LispMessage.PORT_NUM));
585 final List<TransportAddress> transportAddressList = Lists.newArrayList(
586 transportAddressBuilder1.build(),
587 transportAddressBuilder2.build());
589 return transportAddressList;