bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/AbstractAddPathTest.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.protocol.bgp.rib.impl;
   9
  10 import static org.mockito.Matchers.any;
  11 import static org.mockito.Mockito.doReturn;
  12 import static org.opendaylight.protocol.util.CheckUtil.readDataOperational;
  13 import static org.opendaylight.protocol.util.CheckUtil.waitFutureSuccess;
  14
  15 import com.google.common.base.Optional;
  16 import com.google.common.collect.ImmutableList;
  17 import com.google.common.collect.Lists;
  18 import io.netty.channel.epoll.Epoll;
  19 import io.netty.channel.nio.NioEventLoopGroup;
  20 import io.netty.util.concurrent.Future;
  21 import java.net.InetSocketAddress;
  22 import java.util.Collections;
  23 import java.util.List;
  24 import java.util.Set;
  25 import java.util.concurrent.ExecutionException;
  26 import java.util.concurrent.TimeUnit;
  27 import javassist.ClassPool;
  28 import org.junit.After;
  29 import org.junit.Assert;
  30 import org.junit.Before;
  31 import org.mockito.Mock;
  32 import org.mockito.Mockito;
  33 import org.mockito.MockitoAnnotations;
  34 import org.opendaylight.controller.md.sal.binding.test.AbstractConcurrentDataBrokerTest;
  35 import org.opendaylight.mdsal.binding.dom.adapter.BindingToNormalizedNodeCodec;
  36 import org.opendaylight.mdsal.binding.dom.codec.gen.impl.StreamWriterGenerator;
  37 import org.opendaylight.mdsal.binding.dom.codec.impl.BindingNormalizedNodeCodecRegistry;
  38 import org.opendaylight.mdsal.binding.generator.impl.GeneratedClassLoadingStrategy;
  39 import org.opendaylight.mdsal.binding.generator.impl.ModuleInfoBackedContext;
  40 import org.opendaylight.mdsal.binding.generator.util.JavassistUtils;
  41 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonService;
  42 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
  43 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration;
  44 import org.opendaylight.protocol.bgp.inet.RIBActivator;
  45 import org.opendaylight.protocol.bgp.parser.BGPError;
  46 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  47 import org.opendaylight.protocol.bgp.parser.impl.BGPActivator;
  48 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
  49 import org.opendaylight.protocol.bgp.parser.spi.pojo.SimpleBGPExtensionProviderContext;
  50 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
  51 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
  52 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext;
  53 import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext;
  54 import org.opendaylight.protocol.util.InetSocketAddressUtil;
  55 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  56 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  57 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  58 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv4.prefixes.DestinationIpv4Builder;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv4.prefixes.destination.ipv4.Ipv4PrefixesBuilder;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv4.routes.ipv4.routes.Ipv4Route;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4Case;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4CaseBuilder;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.NotifyBuilder;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.Open;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.OpenBuilder;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.PathId;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.Update;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.UpdateBuilder;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.BgpParameters;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.BgpParametersBuilder;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.OptionalCapabilities;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.AttributesBuilder;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.AsPathBuilder;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.ClusterIdBuilder;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.LocalPrefBuilder;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.OriginBuilder;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.OriginatorIdBuilder;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.as.path.SegmentsBuilder;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.update.message.WithdrawnRoutesBuilder;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.Attributes1;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.Attributes1Builder;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.BgpTableType;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.CParameters1;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.CParameters1Builder;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.SendReceive;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.AddPathCapabilityBuilder;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.MultiprotocolCapability;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.MultiprotocolCapabilityBuilder;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  94 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.update.attributes.MpReachNlri;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.update.attributes.MpReachNlriBuilder;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.update.attributes.mp.reach.nlri.AdvertizedRoutes;
  97 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.update.attributes.mp.reach.nlri.AdvertizedRoutesBuilder;
  98 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.BgpRib;
  99 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.PeerRole;
 100 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey;
 101 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
 102 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpOrigin;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
 104 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
 105 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
 106 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
 107 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
 108 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 109 import org.opendaylight.yangtools.yang.binding.Notification;
 110 import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
 111 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 112
 113 public abstract class AbstractAddPathTest extends AbstractConcurrentDataBrokerTest {
 114     private static final int RETRY_TIMER = 10;
 115     static final String RIB_ID = "127.0.0.1";
 116     static final BgpId BGP_ID = new BgpId(RIB_ID);
 117     static final Ipv4Address PEER1 = new Ipv4Address("127.0.0.2");
 118     static final Ipv4Address PEER2 = new Ipv4Address("127.0.0.3");
 119     static final Ipv4Address PEER3 = new Ipv4Address("127.0.0.4");
 120     static final Ipv4Address PEER4 = new Ipv4Address("127.0.0.5");
 121     static final Ipv4Address PEER5 = new Ipv4Address("127.0.0.6");
 122     static final Ipv4Address PEER6 = new Ipv4Address("127.0.0.7");
 123     static final AsNumber AS_NUMBER = new AsNumber(72L);
 124     static final int PORT = InetSocketAddressUtil.getRandomPort();
 125     static final Ipv4Prefix PREFIX1 = new Ipv4Prefix("1.1.1.1/32");
 126     private static final ClusterIdentifier CLUSTER_ID = new ClusterIdentifier(RIB_ID);
 127     static final int HOLDTIMER = 2180;
 128     private static final Ipv4Address NH1 = new Ipv4Address("2.2.2.2");
 129     static final Update UPD_100 = createSimpleUpdate(PREFIX1, new PathId(1L), CLUSTER_ID, 100);
 130     static final Update UPD_50 = createSimpleUpdate(PREFIX1, new PathId(2L), CLUSTER_ID, 50);
 131     static final Update UPD_200 = createSimpleUpdate(PREFIX1, new PathId(3L), CLUSTER_ID, 200);
 132     static final Update UPD_20 = createSimpleUpdate(PREFIX1, new PathId(1L), CLUSTER_ID, 20);
 133     static final Update UPD_NA_100 = createSimpleUpdate(PREFIX1, null, CLUSTER_ID, 100);
 134     static final Update UPD_NA_100_EBGP = createSimpleUpdateEbgp(PREFIX1);
 135     static final Update UPD_NA_200 = createSimpleUpdate(PREFIX1, null, CLUSTER_ID, 200);
 136     static final Update UPD_NA_200_EBGP = createSimpleUpdateEbgp(PREFIX1);
 137     static final TablesKey TABLES_KEY = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
 138     static final List<BgpTableType> TABLES_TYPE = ImmutableList.of(new BgpTableTypeImpl(TABLES_KEY.getAfi(),
 139         TABLES_KEY.getSafi()));
 140     static final Set<TablesKey> AFI_SAFIS_ADVERTIZED = Collections.singleton(TABLES_KEY);
 141     private BGPExtensionProviderContext context;
 142     private static final InstanceIdentifier<BgpRib> BGP_IID = InstanceIdentifier.create(BgpRib.class);
 143     protected SchemaContext schemaContext;
 144     @Mock
 145     protected ClusterSingletonServiceProvider clusterSingletonServiceProvider;
 146     BindingToNormalizedNodeCodec mappingService;
 147     BGPDispatcherImpl serverDispatcher;
 148     RIBExtensionProviderContext ribExtension;
 149     private RIBActivator ribActivator;
 150     private BGPActivator bgpActivator;
 151     private NioEventLoopGroup worker;
 152     private NioEventLoopGroup boss;
 153     private org.opendaylight.protocol.bgp.inet.BGPActivator inetActivator;
 154     protected StrictBGPPeerRegistry serverRegistry;
 155     protected BGPPeerTrackerImpl peerTracker = new BGPPeerTrackerImpl();
 156     protected CodecsRegistryImpl codecsRegistry;
 157
 158     @Before
 159     public void setUp() throws Exception {
 160         MockitoAnnotations.initMocks(this);
 161         this.ribActivator = new RIBActivator();
 162         this.ribExtension = new SimpleRIBExtensionProviderContext();
 163
 164         this.ribActivator.startRIBExtensionProvider(this.ribExtension);
 165
 166         this.bgpActivator = new BGPActivator();
 167         this.inetActivator = new org.opendaylight.protocol.bgp.inet.BGPActivator();
 168         this.context = new SimpleBGPExtensionProviderContext();
 169         this.bgpActivator.start(this.context);
 170         this.inetActivator.start(this.context);
 171
 172         this.mappingService = new BindingToNormalizedNodeCodec(
 173             GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy(),
 174             new BindingNormalizedNodeCodecRegistry(StreamWriterGenerator.create(
 175                 JavassistUtils.forClassPool(ClassPool.getDefault()))));
 176         final ModuleInfoBackedContext moduleInfoBackedContext = ModuleInfoBackedContext.create();
 177         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(BgpParameters.class));
 178         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(MultiprotocolCapability.class));
 179         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(DestinationIpv4Case.class));
 180         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(AdvertizedRoutes.class));
 181         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(BgpRib.class));
 182         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(Attributes1.class));
 183         moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(MpReachNlri.class));
 184         this.mappingService.onGlobalContextUpdated(moduleInfoBackedContext.tryToCreateSchemaContext().get());
 185         this.schemaContext = moduleInfoBackedContext.getSchemaContext();
 186         if (!Epoll.isAvailable()) {
 187             this.worker = new NioEventLoopGroup();
 188             this.boss = new NioEventLoopGroup();
 189         }
 190         this.serverRegistry = new StrictBGPPeerRegistry();
 191         this.serverDispatcher = new BGPDispatcherImpl(this.context.getMessageRegistry(), this.boss, this.worker,
 192             this.serverRegistry);
 193         doReturn(Mockito.mock(ClusterSingletonServiceRegistration.class)).when(this.clusterSingletonServiceProvider)
 194             .registerClusterSingletonService(any(ClusterSingletonService.class));
 195
 196         this.codecsRegistry = CodecsRegistryImpl.create(this.mappingService.getCodecFactory(),
 197                 this.ribExtension.getClassLoadingStrategy());
 198     }
 199
 200     @After
 201     public void tearDown() throws ExecutionException, InterruptedException {
 202         this.serverDispatcher.close();
 203         if (!Epoll.isAvailable()) {
 204             this.worker.shutdownGracefully(0, 0, TimeUnit.SECONDS);
 205             this.boss.shutdownGracefully(0, 0, TimeUnit.SECONDS);
 206         }
 207         this.mappingService.close();
 208         this.ribActivator.close();
 209         this.inetActivator.close();
 210         this.bgpActivator.close();
 211     }
 212
 213     void sendRouteAndCheckIsOnLocRib(final BGPSessionImpl session, final Ipv4Prefix prefix, final long localPreference,
 214         final int expectedRoutesOnDS) throws Exception {
 215         waitFutureSuccess(session.writeAndFlush(createSimpleUpdate(prefix, null, null, localPreference)));
 216         checkLocRib(expectedRoutesOnDS);
 217     }
 218
 219     void sendWithdrawalRouteAndCheckIsOnLocRib(final BGPSessionImpl session, final Ipv4Prefix prefix,
 220         final long localPreference, final int expectedRoutesOnDS) throws Exception {
 221         waitFutureSuccess(session.writeAndFlush(createSimpleWithdrawalUpdate(prefix, localPreference)));
 222         checkLocRib(expectedRoutesOnDS);
 223     }
 224
 225     void sendNotification(final BGPSessionImpl session) {
 226         final Notification notMsg = new NotifyBuilder().setErrorCode(BGPError.OPT_PARAM_NOT_SUPPORTED.getCode())
 227             .setErrorSubcode(BGPError.OPT_PARAM_NOT_SUPPORTED.getSubcode()).setData(new byte[] { 4, 9 }).build();
 228         waitFutureSuccess(session.writeAndFlush(notMsg));
 229     }
 230
 231     void causeBGPError(final BGPSessionImpl session) {
 232         final Open openObj = new OpenBuilder().setBgpIdentifier(new Ipv4Address("1.1.1.1"))
 233             .setHoldTimer(50).setMyAsNumber(72).build();
 234         waitFutureSuccess(session.writeAndFlush(openObj));
 235     }
 236
 237     private void checkLocRib(final int expectedRoutesOnDS) throws Exception {
 238         Thread.sleep(100);
 239         readDataOperational(getDataBroker(), BGP_IID, bgpRib -> {
 240             final Ipv4RoutesCase routes = (Ipv4RoutesCase) bgpRib.getRib().get(0).getLocRib().getTables().get(0)
 241                 .getRoutes();
 242             final List<Ipv4Route> routeList = routes.getIpv4Routes().getIpv4Route();
 243             Assert.assertEquals(expectedRoutesOnDS, routeList.size());
 244             return bgpRib;
 245         });
 246     }
 247
 248     void checkPeersPresentOnDataStore(final int numberOfPeers) throws Exception {
 249         readDataOperational(getDataBroker(), BGP_IID, bgpRib -> {
 250             Assert.assertEquals(numberOfPeers, bgpRib.getRib().get(0).getPeer().size());
 251             return bgpRib;
 252         });
 253     }
 254
 255     BGPSessionImpl createPeerSession(final Ipv4Address peer, final BgpParameters bgpParameters,
 256         final SimpleSessionListener sessionListener) throws InterruptedException {
 257         final StrictBGPPeerRegistry clientRegistry = new StrictBGPPeerRegistry();
 258         final BGPDispatcherImpl clientDispatcher = new BGPDispatcherImpl(this.context.getMessageRegistry(), this.boss,
 259             this.worker, clientRegistry);
 260         clientRegistry.addPeer(new IpAddress(new Ipv4Address(RIB_ID)), sessionListener,
 261             new BGPSessionPreferences(AS_NUMBER, HOLDTIMER, new BgpId(peer),
 262                 AS_NUMBER, Lists.newArrayList(bgpParameters), Optional.absent()));
 263
 264         return connectPeer(peer, clientDispatcher);
 265     }
 266
 267     protected static BGPPeer configurePeer(final Ipv4Address peerAddress, final RIBImpl ribImpl,
 268         final BgpParameters bgpParameters, final PeerRole peerRole, final BGPPeerRegistry bgpPeerRegistry) {
 269         final IpAddress ipAddress = new IpAddress(peerAddress);
 270
 271         final BGPPeer bgpPeer = new BGPPeer(new IpAddress(peerAddress), ribImpl, peerRole, null,
 272             AFI_SAFIS_ADVERTIZED, Collections.emptySet());
 273         final List<BgpParameters> tlvs = Lists.newArrayList(bgpParameters);
 274         bgpPeerRegistry.addPeer(ipAddress, bgpPeer,
 275             new BGPSessionPreferences(AS_NUMBER, HOLDTIMER, new BgpId(RIB_ID), AS_NUMBER, tlvs, Optional.absent()));
 276         bgpPeer.instantiateServiceInstance();
 277         return bgpPeer;
 278     }
 279
 280     private static BGPSessionImpl connectPeer(final Ipv4Address localAddress, final BGPDispatcherImpl dispatcherImpl)
 281         throws InterruptedException {
 282         final Future<BGPSessionImpl> future = dispatcherImpl.createClient(new InetSocketAddress(localAddress.getValue(), PORT),
 283             new InetSocketAddress(RIB_ID, PORT), RETRY_TIMER, true);
 284         Thread.sleep(200);
 285         waitFutureSuccess(future);
 286         Thread.sleep(100);
 287         return future.getNow();
 288     }
 289
 290     protected static BgpParameters createParameter(final boolean addPath) {
 291         final OptionalCapabilities mp = new OptionalCapabilitiesBuilder().setCParameters(
 292             new CParametersBuilder().addAugmentation(CParameters1.class,
 293                 new CParameters1Builder().setMultiprotocolCapability(
 294                     new MultiprotocolCapabilityBuilder().setAfi(Ipv4AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class)
 295                         .build()).build()).build()).build();
 296         final List<OptionalCapabilities> capabilities = Lists.newArrayList(mp);
 297         if (addPath) {
 298             final OptionalCapabilities addPathCapa = new OptionalCapabilitiesBuilder().setCParameters(
 299                 new CParametersBuilder().addAugmentation(CParameters1.class,
 300                     new CParameters1Builder().setAddPathCapability(
 301                         new AddPathCapabilityBuilder().setAddressFamilies(Lists.newArrayList(
 302                             new AddressFamiliesBuilder()
 303                                 .setAfi(Ipv4AddressFamily.class)
 304                                 .setSafi(UnicastSubsequentAddressFamily.class)
 305                                 .setSendReceive(SendReceive.Both)
 306                                 .build()))
 307                             .build()).build()).build()).build();
 308             capabilities.add(addPathCapa);
 309         }
 310         return new BgpParametersBuilder().setOptionalCapabilities(capabilities).build();
 311     }
 312
 313     private static Update createSimpleUpdate(final Ipv4Prefix prefix, final PathId pathId, final ClusterIdentifier clusterId,
 314         final long localPreference) {
 315         final AttributesBuilder attBuilder = new AttributesBuilder();
 316         attBuilder.setLocalPref(new LocalPrefBuilder().setPref(localPreference).build());
 317         attBuilder.setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build());
 318         attBuilder.setAsPath(new AsPathBuilder().setSegments(Collections.emptyList()).build());
 319         if (clusterId != null) {
 320             attBuilder.setClusterId(new ClusterIdBuilder().setCluster(Collections.singletonList(clusterId)).build());
 321             attBuilder.setOriginatorId(new OriginatorIdBuilder().setOriginator(new Ipv4Address(clusterId)).build());
 322         }
 323         addAttributeAugmentation(attBuilder, prefix, pathId);
 324         return new UpdateBuilder().setAttributes(attBuilder.build()).build();
 325     }
 326
 327     private static Update createSimpleUpdateEbgp(final Ipv4Prefix prefix) {
 328         final AttributesBuilder attBuilder = new AttributesBuilder();
 329         attBuilder.setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build());
 330         attBuilder.setAsPath(new AsPathBuilder().setSegments(Collections.singletonList(
 331             new SegmentsBuilder().setAsSequence(Collections.singletonList(AS_NUMBER)).build())).build());
 332         addAttributeAugmentation(attBuilder, prefix, null);
 333
 334         return new UpdateBuilder().setAttributes(attBuilder.build()).build();
 335     }
 336
 337     private static void addAttributeAugmentation(final AttributesBuilder attBuilder, final Ipv4Prefix prefix,
 338         final PathId pathId) {
 339         attBuilder.setUnrecognizedAttributes(Collections.emptyList());
 340         attBuilder.addAugmentation(Attributes1.class,
 341             new Attributes1Builder().setMpReachNlri(
 342                 new MpReachNlriBuilder()
 343                     .setCNextHop(new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(NH1)
 344                         .build()).build())
 345                     .setAfi(Ipv4AddressFamily.class)
 346                     .setSafi(UnicastSubsequentAddressFamily.class)
 347                     .setAdvertizedRoutes(new AdvertizedRoutesBuilder().setDestinationType(
 348                         new DestinationIpv4CaseBuilder().setDestinationIpv4(
 349                             new DestinationIpv4Builder().setIpv4Prefixes(Collections.singletonList(
 350                                 new Ipv4PrefixesBuilder().setPathId(pathId).setPrefix(new Ipv4Prefix(prefix)).build()))
 351                                 .build())
 352                             .build()).build())
 353                     .build()).build());
 354     }
 355
 356     private static Update createSimpleWithdrawalUpdate(final Ipv4Prefix prefix, final long localPreference) {
 357         final AttributesBuilder attBuilder = new AttributesBuilder();
 358         attBuilder.setLocalPref(new LocalPrefBuilder().setPref(localPreference).build());
 359         attBuilder.setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build());
 360         attBuilder.setAsPath(new AsPathBuilder().setSegments(Collections.emptyList()).build());
 361         attBuilder.setUnrecognizedAttributes(Collections.emptyList());
 362         return new UpdateBuilder()
 363                 .setWithdrawnRoutes(Collections.singletonList(new WithdrawnRoutesBuilder().setPrefix(prefix).build()))
 364                 .build();
 365     }
 366 }