integrationtest/src/test/java/org/opendaylight/lispflowmapping/integrationtest/MultiSiteScenario.java

   1 /**
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.lispflowmapping.integrationtest;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.junit.Assert.fail;
  14 import static org.opendaylight.lispflowmapping.integrationtest.MappingServiceIntegrationTest.ourAddress;
  15 import static org.opendaylight.lispflowmapping.integrationtest.MultiSiteScenarioUtil.SITE_A;
  16 import static org.opendaylight.lispflowmapping.integrationtest.MultiSiteScenarioUtil.SITE_D5;
  17
  18 import java.nio.ByteBuffer;
  19 import java.util.ArrayList;
  20 import java.util.Arrays;
  21 import java.util.Collections;
  22 import java.util.HashSet;
  23 import java.util.List;
  24 import java.util.Set;
  25 import org.apache.commons.lang3.ArrayUtils;
  26 import org.opendaylight.lispflowmapping.integrationtest.MultiSiteScenarioUtil.Site;
  27 import org.opendaylight.lispflowmapping.interfaces.lisp.IFlowMapping;
  28 import org.opendaylight.lispflowmapping.interfaces.mappingservice.IMappingService;
  29 import org.opendaylight.lispflowmapping.lisp.serializer.MapRequestSerializer;
  30 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
  31 import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.InstanceIdType;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.Address;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv4AddressBinary;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv4Binary;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapReply;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapRequest;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.XtrId;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.list.EidItem;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.list.EidItemBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecord;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecordBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkeyBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapregisternotification.MapRegisterBuilder;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequest.ItrRloc;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequest.ItrRlocBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequest.SourceEid;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequest.SourceEidBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequestnotification.MapRequestBuilder;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
  59 import org.slf4j.Logger;
  60 import org.slf4j.LoggerFactory;
  61 import com.google.common.net.InetAddresses;
  62
  63 /**
  64  * Contains methods for:
  65  * - checking sites visibility (availability) - simulation of ping command
  66  * - for preparing and processing data structures necessary for creating map request, map register LISP messages.
  67  *
  68  */
  69 class MultiSiteScenario {
  70
  71     private final int DEFAULT_NETWORK_MASK = 24;
  72     private final int IP_MASK = 32;
  73
  74     private final Integer TTL = 1440;
  75     private final String MAP_RECORD_A = "MAP_RECORD_1";
  76
  77     private final Short DEFAULT_MULTICAST_PRIORITY = 255;
  78     private final Short DEFAULT_MULTICAST_WEIGHT = 0;
  79     private final Boolean DEFAULT_LOCAL_LOCATOR = true;
  80     private final Boolean DEFAULT_RLOC_PROBED = false;
  81     private final Boolean DEFAULT_ROUTED = true;
  82     private final byte[] DEFAULT_XTR_ID = new byte[]{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
  83     private final byte[] DEFAULT_SITE_ID = new byte[]{0, 1, 2, 3, 4, 5, 6, 7};
  84
  85     private final MappingAuthkey NULL_AUTH_KEY = new MappingAuthkeyBuilder().setKeyType(0).build();
  86     private final IMappingService mapService;
  87     private final IFlowMapping lms;
  88
  89     private static final Logger LOG = LoggerFactory.getLogger(MultiSiteScenario.class);
  90
  91     enum EidType {
  92         EID_SRC_DST, EID_WITH_PREFIX
  93     }
  94
  95     MultiSiteScenario(final IMappingService mapService, final IFlowMapping lms) {
  96         this.mapService = mapService;
  97         this.lms = lms;
  98     }
  99
 100     private Ipv4AddressBinary verifyIpv4Address(final LocatorRecord locatorRecord) {
 101         assertNotNull(locatorRecord);
 102         final Rloc rloc = locatorRecord.getRloc();
 103         assertNotNull(rloc);
 104         final Address address = rloc.getAddress();
 105         assertTrue(address instanceof Ipv4Binary);
 106         return ((Ipv4Binary) address).getIpv4Binary();
 107     }
 108
 109     private List<LocatorRecord> verifyLocatorRecordExists(final MappingRecord mappingRecord) {
 110         final List<LocatorRecord> locatorRecords = mappingRecord.getLocatorRecord();
 111         assertNotNull(locatorRecords);
 112         return locatorRecords;
 113     }
 114
 115     private MappingRecord verifyMappingRecord(MapReply mapReply) {
 116         assertNotNull(mapReply);
 117         final List<MappingRecordItem> mappingRecordItems = mapReply.getMappingRecordItem();
 118         assertNotNull(mappingRecordItems);
 119         assertEquals(1, mappingRecordItems.size());
 120         final MappingRecordItem mappingRecordItem = mappingRecordItems.get(0);
 121         assertNotNull(mappingRecordItem);
 122         final MappingRecord mappingRecord = mappingRecordItem.getMappingRecord();
 123         assertNotNull(mappingRecord);
 124         return mappingRecord;
 125     }
 126
 127     private void emitMapRegisterMessage(final Site dstSite, final boolean merge) {
 128         final MapRegisterBuilder mapRegisterBuilder = new MapRegisterBuilder();
 129         mapRegisterBuilder.setXtrSiteIdPresent(true);
 130         mapRegisterBuilder.setXtrId(new XtrId(DEFAULT_XTR_ID));
 131         mapRegisterBuilder.setSiteId(new SiteId(DEFAULT_SITE_ID));
 132         mapRegisterBuilder.setMergeEnabled(merge);
 133         final MappingRecordItemBuilder mappingRecordItemBuilder = new MappingRecordItemBuilder();
 134         mappingRecordItemBuilder.setMappingRecordItemId(MAP_RECORD_A);
 135
 136         final MappingRecordBuilder mrb = prepareMappingRecord(EidType.EID_WITH_PREFIX, null, dstSite);
 137         mappingRecordItemBuilder.setMappingRecord(mrb.build());
 138         mapRegisterBuilder.setMappingRecordItem(Collections.singletonList(mappingRecordItemBuilder.build()));
 139
 140         lms.handleMapRegister(mapRegisterBuilder.build());
 141     }
 142
 143     void setCommonAuthentication() {
 144         Eid eid = LispAddressUtil.asIpv4PrefixBinaryEid("0.0.0.0/0", SITE_A.getVNI());
 145         mapService.addAuthenticationKey(eid, NULL_AUTH_KEY);
 146
 147         eid = LispAddressUtil.asIpv4PrefixBinaryEid("0.0.0.0/0", SITE_D5.getVNI());
 148         mapService.addAuthenticationKey(eid, NULL_AUTH_KEY);
 149         sleepForSeconds(1);
 150     }
 151
 152     private Eid toEid(final String destSiteEidPrefix, final InstanceIdType vniValue, final int mask) {
 153         return LispAddressUtil.asIpv4PrefixBinaryEid(destSiteEidPrefix + "/" + mask, vniValue);
 154
 155     }
 156
 157     private MapReply emitMapRequestMessage(final String siteFromEidPrefix, final String siteToEidPrefix, final
 158     InstanceIdType vniValue) {
 159         final MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
 160         final EidItemBuilder eidItemBuilder = new EidItemBuilder();
 161         Eid dstEid = toEid(siteToEidPrefix, vniValue, IP_MASK);
 162
 163         eidItemBuilder.setEid(dstEid);
 164         eidItemBuilder.setEidItemId(siteFromEidPrefix + siteToEidPrefix);
 165         final List<EidItem> eidItem = Collections.singletonList(eidItemBuilder.build());
 166         final Eid srcEid = toEid(siteFromEidPrefix, vniValue, IP_MASK);
 167         mapRequestBuilder.setSourceEid(new SourceEidBuilder().setEid(srcEid).build());
 168         mapRequestBuilder.setEidItem(eidItem);
 169         mapRequestBuilder.setItrRloc(prepareDummyItrRloc());
 170         mapRequestBuilder.setSmrInvoked(false);
 171         return lms.handleMapRequest(mapRequestBuilder.build());
 172     }
 173
 174     private List<ItrRloc> prepareDummyItrRloc() {
 175         List<ItrRloc> itrRlocs = new ArrayList<>();
 176         final ItrRlocBuilder itrRlocBuilder = new ItrRlocBuilder();
 177         itrRlocBuilder.setItrRlocId(ourAddress);
 178         itrRlocBuilder.setRloc(LispAddressUtil.asIpv4Rloc(ourAddress));
 179         itrRlocs.add(itrRlocBuilder.build());
 180         return itrRlocs;
 181     }
 182
 183     void storeNorthMappingNegative(final Site dstSite, final MappingRecord.Action action) {
 184         final Eid eidAsIpv4Prefix = LispAddressUtil.asIpv4PrefixBinaryEid(
 185                 dstSite.getEidPrefix() + "/" + DEFAULT_NETWORK_MASK, dstSite.getVNI());
 186
 187         final MappingRecordBuilder mrbNegative = prepareMappingRecord(EidType.EID_WITH_PREFIX, null, dstSite);
 188         mrbNegative.setEid(eidAsIpv4Prefix);
 189         mrbNegative.setAction(action);
 190
 191         mapService.addMapping(MappingOrigin.Northbound, eidAsIpv4Prefix, dstSite.getSiteId(), mrbNegative.build(),
 192                 false);
 193     }
 194
 195     void deleteNorthMappingNegative(final Site dstSite) {
 196         final Eid eidAsIpv4Prefix = LispAddressUtil.asIpv4PrefixBinaryEid(
 197                 dstSite.getEidPrefix() + "/" + DEFAULT_NETWORK_MASK, dstSite.getVNI());
 198
 199         mapService.removeMapping(MappingOrigin.Northbound, eidAsIpv4Prefix);
 200     }
 201
 202     void storeNorthMappingSrcDst(final Site srcSite, final Site ... dstSite) {
 203         final MappingRecordBuilder mrb = prepareMappingRecord(EidType.EID_SRC_DST, srcSite,
 204                 dstSite);
 205         mapService.addMapping(MappingOrigin.Northbound, mrb.getEid(), dstSite[0].getSiteId(), mrb.build(), false);
 206     }
 207
 208     void storeNorthMappingIpPrefix(final Site... dstSite) {
 209         final MappingRecordBuilder mrb = prepareMappingRecord(EidType.EID_WITH_PREFIX, null, dstSite);
 210         mapService.addMapping(MappingOrigin.Northbound, mrb.getEid(), dstSite[0].getSiteId(), mrb.build(), false);
 211     }
 212
 213     private void storeDestinationSiteMappingViaSouthbound(final Site dstSite, final boolean merge) {
 214         emitMapRegisterMessage(dstSite, merge);
 215     }
 216
 217     private MappingRecordBuilder prepareMappingRecordGeneral(final EidType eidType,
 218                                                              final Site srcSite, final Site dstSite) {
 219         final MappingRecordBuilder mrb = provideCommonMapRecordBuilder();
 220         mrb.setXtrId(new XtrId(ArrayUtils.addAll(dstSite.getSiteId().getValue(), dstSite.getSiteId().getValue())));
 221
 222         Eid eid = null;
 223         if (EidType.EID_SRC_DST.equals(eidType)) {
 224             if (srcSite != null && dstSite != null && srcSite.getEidPrefix() != null && dstSite.getEidPrefix() !=
 225                     null) {
 226                 eid = LispAddressUtil.asSrcDstEid(srcSite.getEidPrefix(), dstSite.getEidPrefix(), DEFAULT_NETWORK_MASK,
 227                         DEFAULT_NETWORK_MASK, dstSite.getVNI().getValue().intValue());
 228             }
 229         }
 230
 231         mrb.setEid(eid == null ? toEid(dstSite.getEidPrefix(), dstSite.getVNI(), DEFAULT_NETWORK_MASK) : eid);
 232         return mrb;
 233     }
 234
 235     private MappingRecordBuilder prepareMappingRecord(final EidType eidType, final Site srcSite, final Site...
 236             dstSites) {
 237         assertTrue(isEidEquals(dstSites));
 238         final MappingRecordBuilder mrb = prepareMappingRecordGeneral(eidType, srcSite, dstSites[0]);
 239         final List<LocatorRecord> locatorRecords = new ArrayList<>();
 240         if (!dstSites[0].isForDeletion) {
 241             for (Site dstSite : dstSites) {
 242                 if (dstSite.getRloc() != null) {
 243                     locatorRecords.add(provideLocatorRecord(LispAddressUtil.asIpv4Rloc(dstSite.getRloc()),
 244                             dstSite.getRloc(), dstSite.getWeight(), dstSite.getPriority()));
 245                 }
 246             }
 247         }
 248         mrb.setLocatorRecord(locatorRecords);
 249
 250         return mrb;
 251     }
 252
 253     private boolean isEidEquals(final Site... sites) {
 254         String commonEid = null;
 255         for (Site site : sites) {
 256             final String concreteEidPrefix = site.getEidPrefix();
 257             if (!concreteEidPrefix.equals(commonEid)) {
 258                 if (commonEid == null) {
 259                     commonEid = concreteEidPrefix;
 260                 } else {
 261                     return false;
 262                 }
 263             }
 264         }
 265         return true;
 266     }
 267
 268     private LocatorRecord provideLocatorRecord(final Rloc rloc, final String rlocStr, final short weight, final short
 269             priority) {
 270         final LocatorRecordBuilder locatorRecordBuilder = new LocatorRecordBuilder();
 271         locatorRecordBuilder.setRloc(rloc);
 272         locatorRecordBuilder.setLocatorId(rlocStr);
 273         locatorRecordBuilder.setPriority(priority);
 274         locatorRecordBuilder.setWeight(weight);
 275         locatorRecordBuilder.setMulticastPriority(DEFAULT_MULTICAST_PRIORITY);
 276         locatorRecordBuilder.setMulticastWeight(DEFAULT_MULTICAST_WEIGHT);
 277         locatorRecordBuilder.setLocalLocator(DEFAULT_LOCAL_LOCATOR);
 278         locatorRecordBuilder.setRlocProbed(DEFAULT_RLOC_PROBED);
 279         locatorRecordBuilder.setRouted(DEFAULT_ROUTED);
 280         return locatorRecordBuilder.build();
 281     }
 282
 283     private MappingRecordBuilder provideCommonMapRecordBuilder() {
 284         final MappingRecordBuilder mappingRecordBuilder = new MappingRecordBuilder();
 285         mappingRecordBuilder.setRecordTtl(TTL);
 286         mappingRecordBuilder.setAction(MappingRecord.Action.NoAction);
 287         mappingRecordBuilder.setAuthoritative(true);
 288         mappingRecordBuilder.setTimestamp(System.currentTimeMillis());
 289         return mappingRecordBuilder;
 290     }
 291
 292     void deleteNorthMapingSrcDst(final Site srcSite, final Site dstSite) {
 293         final Eid eid = LispAddressUtil.asSrcDstEid(srcSite.getEidPrefix(), dstSite.getEidPrefix(),
 294                 DEFAULT_NETWORK_MASK, DEFAULT_NETWORK_MASK, dstSite.getVNI().getValue().intValue());
 295         mapService.removeMapping(MappingOrigin.Northbound, eid);
 296     }
 297
 298     void deleteSouthboundMappings(final Site dstSite) {
 299         emitMapRegisterMessage(dstSite, false);
 300
 301     }
 302
 303     void storeSouthboundMappings(final boolean merge, final Site ... sites) {
 304         for (Site site : sites) {
 305             storeDestinationSiteMappingViaSouthbound(site, merge);
 306         }
 307     }
 308
 309     boolean isPossibleAssertPingResultImmediately(final boolean expectedPingWorks, final boolean isPartialyWorking,
 310                                                   final String  msg) {
 311         //ping fail is unwanted. ping definitely failed
 312         if (expectedPingWorks && !isPartialyWorking) {
 313             fail(msg);
 314         }
 315
 316         //ping fail is wanted. still can fail later
 317         if (!expectedPingWorks && isPartialyWorking) {
 318             return false;
 319         }
 320
 321         //ping fail is unwanted. still can fail later
 322         if (expectedPingWorks && isPartialyWorking) {
 323             return false;
 324         }
 325
 326         //ping fail is wanted. ping definitely failed.
 327         if (!expectedPingWorks && !isPartialyWorking) {
 328             return true;
 329         }
 330         return false;
 331     }
 332
 333     boolean checkActionAndRloc(final Site dstSite, boolean expectedPingWorks, MapReply mapReplyFromSrcToDst, final
 334                                 Site  ... additionalSitesFromMapping) {
 335         final MappingRecord mappingRecord = verifyMappingRecord(mapReplyFromSrcToDst);
 336         final boolean isNotDroppendSrcDst = !MappingRecord.Action.Drop.equals(mappingRecord.getAction());
 337
 338         if (isPossibleAssertPingResultImmediately(expectedPingWorks, isNotDroppendSrcDst, "Drop action has appeared " +
 339                 "during ping")) {
 340             return true;
 341         }
 342
 343         final List<LocatorRecord> locatorRecords = verifyLocatorRecordExists(mappingRecord);
 344         for (Site expectedTargetSite : concatenateSites(dstSite, additionalSitesFromMapping)) {
 345             boolean expectedTargetFound = false;
 346             for (LocatorRecord locatorRecord : locatorRecords) {
 347                 if (expectedTargetSite.getRloc().equals(rlocToString(locatorRecord))) {
 348                     final byte[] ipv4AddressSrcDst = verifyIpv4Address(locatorRecord).getValue();
 349                     final byte[] rloc = InetAddresses.forString((expectedTargetSite.getRloc())).getAddress();
 350                     final boolean isRlocSrcDstEqual = Arrays.equals(ipv4AddressSrcDst, rloc);
 351                     if (isPossibleAssertPingResultImmediately(expectedPingWorks, isRlocSrcDstEqual, "Unexpected RLOC." +
 352                             "Expected value " + rloc + ". Real value " + ipv4AddressSrcDst +
 353                             ".")) {
 354                         return true;
 355                     }
 356
 357                     final boolean isWeightEquals = expectedTargetSite.getWeight() == locatorRecord.getWeight();
 358                     if (isPossibleAssertPingResultImmediately(expectedPingWorks, isWeightEquals, "Weight isn't equal." +
 359                             "Expected value " + expectedTargetSite.getWeight() + ". Value from mapping" +
 360                             locatorRecord.getWeight() + ".")) {
 361                         return true;
 362                     }
 363
 364                     final boolean isPriorityEquals = expectedTargetSite.getPriority() == locatorRecord.getPriority();
 365                     if (isPossibleAssertPingResultImmediately(expectedPingWorks, isPriorityEquals, "Priority isn't " +
 366                             "equal. Expected value " + expectedTargetSite.getPriority() + ". Value from mapping" +
 367                             locatorRecord.getPriority() + ".")) {
 368                         return true;
 369                     }
 370
 371                     expectedTargetFound = true;
 372                     break;
 373                 }
 374             }
 375             if (isPossibleAssertPingResultImmediately(expectedPingWorks, expectedTargetFound, "Mapping for " +
 376                 expectedTargetSite.getRloc() + " was expected but wasn't returned from mapping service." +
 377                 expectedTargetFound)) {
 378                 return true;
 379             }
 380
 381         }
 382
 383         return false;
 384     }
 385
 386     private String rlocToString(final LocatorRecord locatorRecord) {
 387         return LispAddressStringifier.getString(locatorRecord.getRloc());
 388     }
 389
 390     private Iterable<Site> concatenateSites(final Site dstSite, final Site... additionalSitesFromMapping) {
 391         final List<Site> sites = new ArrayList<>();
 392         sites.add(dstSite);
 393         for (Site additionalSite : additionalSitesFromMapping) {
 394             sites.add(additionalSite);
 395         }
 396         return sites;
 397     }
 398
 399     private void assertPing(final Site srcSite, final int srcHostIndex, final Site dstSite, final int dstHostIndex,
 400                          boolean expectedPingWorks, final Site ... additionalSitesFromMapping) {
 401         final MapReply mapReplyFromSrcToDst = emitMapRequestMessage(srcSite.getHost(srcHostIndex), dstSite.getHost
 402                 (dstHostIndex), dstSite.getVNI());
 403         if (checkActionAndRloc(dstSite, expectedPingWorks, mapReplyFromSrcToDst, additionalSitesFromMapping)) {
 404             return;
 405         }
 406
 407         final MapReply mapReplyFromDstToSrc = emitMapRequestMessage(dstSite.getHost(dstHostIndex), srcSite.getHost
 408                 (srcHostIndex), srcSite.getVNI());
 409         if (checkActionAndRloc(srcSite, expectedPingWorks, mapReplyFromDstToSrc)) {
 410             return;
 411         }
 412
 413         final InstanceIdType iidDst = mapReplyFromSrcToDst.getMappingRecordItem().get(0).getMappingRecord().getEid().
 414                 getVirtualNetworkId();
 415         final InstanceIdType iidSrc = mapReplyFromDstToSrc.getMappingRecordItem().get(0).getMappingRecord().getEid().
 416                 getVirtualNetworkId();
 417
 418         final boolean isIIDEqual = iidDst.equals(iidSrc);
 419
 420         if (expectedPingWorks != isIIDEqual) {
 421             fail("IID problem. Dst value " + iidDst.getValue() + ". Src value " + iidSrc.getValue() + ".");
 422         }
 423     }
 424
 425     void assertPingWorks(final Site srcSite, final int srcHostIndex, final Site dstSite, final int dstHostIndex,
 426                          final Site ... additionalSitesFromMapping) {
 427         assertPing(srcSite, srcHostIndex, dstSite, dstHostIndex, true, additionalSitesFromMapping);
 428     }
 429
 430     void assertPingFails(final Site srcSite, final int srcHostIndex, final Site dstSite, final int dstHostIndex) {
 431         assertPing(srcSite, srcHostIndex, dstSite, dstHostIndex, false);
 432     }
 433
 434     private void sleepForSeconds(int seconds) {
 435         try {
 436             Thread.sleep(seconds * 1000);
 437         } catch (InterruptedException e) {
 438             LOG.trace("Interrupted while sleeping", e);
 439         }
 440     }
 441
 442
 443     private List<MapRequest> translateBuffersToMapRequest(byte[][] buffers) {
 444         final List<MapRequest> mapRequests = new ArrayList<>();
 445         for (byte[] buffer : buffers) {
 446             final MapRequest mapRequest = MapRequestSerializer.getInstance().deserialize(ByteBuffer.wrap(buffer), null);
 447             assertNotNull(mapRequest);
 448             mapRequests.add(mapRequest);
 449         }
 450         return mapRequests;
 451     }
 452
 453     private Set<Eid> prepareExpectedEid(final String ... hosts) {
 454         final Set<Eid> eids = new HashSet<>();
 455         for (String host : hosts) {
 456             eids.add(LispAddressUtil.asIpv4PrefixBinaryEid(host + "/" + IP_MASK,
 457                     new InstanceIdType(MultiSiteScenarioUtil.VNI2)));
 458         }
 459         return eids;
 460     }
 461
 462     private SourceEid prepareSourceEid(final String eidPrefix) {
 463         final SourceEidBuilder sourceEidBuilder = new SourceEidBuilder();
 464         final Eid eid = LispAddressUtil.asIpv4Eid(eidPrefix, MultiSiteScenarioUtil.VNI2);
 465         return sourceEidBuilder.setEid(eid).build();
 466     }
 467
 468     void checkSMR(final SocketReader socketReader, final String site, final String ... hosts) {
 469         List<MapRequest> mapRequests = translateBuffersToMapRequest(socketReader.getBuffers(hosts.length));
 470         final Set<Eid> eids = prepareExpectedEid(hosts);
 471         final SourceEid expectedSourceEid = prepareSourceEid(site);
 472         for(MapRequest mapRequest : mapRequests) {
 473             assertTrue(mapRequest.isSmr());
 474             final SourceEid receivedSourceEid = mapRequest.getSourceEid();
 475             assertEquals(expectedSourceEid, receivedSourceEid);
 476             final List<EidItem> currentEidItems = mapRequest.getEidItem();
 477             assertNotNull(currentEidItems);
 478             assertTrue(SMRContainsExpectedEid(eids, currentEidItems));
 479         }
 480         //all expected eids should be after looping via mapRequests matched.
 481         assertTrue("Expected eids wasn't/weren't found " + eids, eids.isEmpty());
 482     }
 483
 484     private boolean SMRContainsExpectedEid(Set<Eid> eids, List<EidItem> currentEidItems) {
 485         for (EidItem eidItem : currentEidItems) {
 486             //if eid from map request is matched then it is removed from set of expected eids
 487             if (!eids.remove(eidItem.getEid())) {
 488                  fail("SMR contained " + eidItem.getEid() + " which wasn't expected.");
 489             }
 490         }
 491         return true;
 492     }
 493
 494 }