bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/AddPathAllPathsTest.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.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertFalse;
  12 import static org.junit.Assert.assertNull;
  13 import static org.junit.Assert.assertTrue;
  14 import static org.opendaylight.protocol.util.CheckUtil.checkEquals;
  15 import static org.opendaylight.protocol.util.CheckUtil.checkReceivedMessages;
  16 import static org.opendaylight.protocol.util.CheckUtil.waitFutureSuccess;
  17
  18 import com.google.common.collect.ImmutableMap;
  19 import io.netty.channel.Channel;
  20 import io.netty.channel.ChannelFuture;
  21 import java.net.InetSocketAddress;
  22 import java.util.Collections;
  23 import java.util.Map;
  24 import java.util.concurrent.ExecutionException;
  25 import org.junit.After;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
  29 import org.opendaylight.protocol.bgp.mode.impl.add.all.paths.AllPathSelection;
  30 import org.opendaylight.protocol.bgp.rib.spi.State;
  31 import org.opendaylight.protocol.bgp.rib.spi.state.BGPAfiSafiState;
  32 import org.opendaylight.protocol.bgp.rib.spi.state.BGPErrorHandlingState;
  33 import org.opendaylight.protocol.bgp.rib.spi.state.BGPGracelfulRestartState;
  34 import org.opendaylight.protocol.bgp.rib.spi.state.BGPPeerMessagesState;
  35 import org.opendaylight.protocol.bgp.rib.spi.state.BGPPeerState;
  36 import org.opendaylight.protocol.bgp.rib.spi.state.BGPRibState;
  37 import org.opendaylight.protocol.bgp.rib.spi.state.BGPSessionState;
  38 import org.opendaylight.protocol.bgp.rib.spi.state.BGPTimersState;
  39 import org.opendaylight.protocol.bgp.rib.spi.state.BGPTransportState;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Update;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerRole;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.RibId;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  47
  48 public class AddPathAllPathsTest extends AbstractAddPathTest {
  49     private RIBImpl ribImpl;
  50     private Channel serverChannel;
  51
  52     @Before
  53     public void setUp() throws Exception {
  54         super.setUp();
  55         final Map<TablesKey, PathSelectionMode> pathTables
  56                 = ImmutableMap.of(TABLES_KEY, new AllPathSelection());
  57
  58         this.ribImpl = new RIBImpl(this.tableRegistry, new RibId("test-rib"), AS_NUMBER, BGP_ID,
  59                 this.ribExtension, this.serverDispatcher, this.codecsRegistry,
  60             getDomBroker(), getDataBroker(), this.policies, TABLES_TYPE, pathTables);
  61
  62         this.ribImpl.instantiateServiceInstance();
  63         this.ribImpl.onGlobalContextUpdated(this.schemaService.getGlobalContext());
  64         final ChannelFuture channelFuture = this.serverDispatcher.createServer(new InetSocketAddress(RIB_ID, PORT));
  65         waitFutureSuccess(channelFuture);
  66         this.serverChannel = channelFuture.channel();
  67     }
  68
  69     @After
  70     public void tearDown() throws ExecutionException, InterruptedException {
  71         waitFutureSuccess(this.serverChannel.close());
  72         super.tearDown();
  73     }
  74     /*
  75      * All-Paths
  76      *                                            ___________________
  77      *                                           | ODL BGP 127.0.0.1 |
  78      * [peer://127.0.0.2; p1, lp100] --(iBGP)--> |                   | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
  79      * [peer://127.0.0.3; p1, lp200] --(iBGP)--> |                   |
  80      * [peer://127.0.0.4; p1, lp50] --(iBGP)-->  |                   | --(RR-client, add-path) --> [Peer://127.0.0.6; (p1, path-id1, lp100), (p1, path-id2, pl50), (p1, path-id3, pl200), (p1, path-id4, pl20)]
  81      * [peer://127.0.0.2; p1, lp20] --(iBGP)-->  |___________________|
  82      * p1 = 1.1.1.1/32
  83      *
  84      *
  85      *
  86      */
  87     @Test
  88     public void testUseCase1() throws Exception {
  89         final BgpParameters nonAddPathParams = createParameter(false);
  90         final BgpParameters addPathParams = createParameter(true);
  91
  92         final BGPPeer peer1 = configurePeer(this.tableRegistry, PEER1, this.ribImpl, nonAddPathParams, PeerRole.Ibgp,
  93                 this.serverRegistry);
  94         final BGPSessionImpl session1 = createPeerSession(PEER1, nonAddPathParams, new SimpleSessionListener());
  95
  96         configurePeer(this.tableRegistry, PEER2, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
  97         final BGPSessionImpl session2 = createPeerSession(PEER2, nonAddPathParams, new SimpleSessionListener());
  98
  99         configurePeer(this.tableRegistry, PEER3, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
 100         final BGPSessionImpl session3 = createPeerSession(PEER3, nonAddPathParams, new SimpleSessionListener());
 101
 102         final SimpleSessionListener listener4 = new SimpleSessionListener();
 103         final BGPPeer peer4 = configurePeer(this.tableRegistry, PEER4, this.ribImpl, nonAddPathParams,
 104                 PeerRole.RrClient, this.serverRegistry);
 105
 106         BGPPeerState peer4State = peer4.getPeerState();
 107         assertNull(peer4State.getGroupId());
 108         assertEquals(new IpAddress(PEER4), peer4State.getNeighborAddress());
 109         assertEquals(0L, peer4State.getTotalPathsCount());
 110         assertEquals(0L, peer4State.getTotalPrefixes());
 111
 112         assertNull(peer4State.getBGPTimersState());
 113         assertNull(peer4State.getBGPTransportState());
 114         assertNull(peer4State.getBGPSessionState());
 115         assertEquals(0L, peer4State.getBGPErrorHandlingState().getErroneousUpdateReceivedCount());
 116
 117         BGPGracelfulRestartState gracefulRestart = peer4State.getBGPGracelfulRestart();
 118         assertFalse(gracefulRestart.isGracefulRestartAdvertized(TABLES_KEY));
 119         assertFalse(gracefulRestart.isGracefulRestartReceived(TABLES_KEY));
 120         assertFalse(gracefulRestart.isLocalRestarting());
 121         assertFalse(gracefulRestart.isPeerRestarting());
 122         assertEquals(0L, gracefulRestart.getPeerRestartTime());
 123
 124         BGPAfiSafiState afiSafiState = peer4State.getBGPAfiSafiState();
 125         assertEquals(AFI_SAFIS_ADVERTIZED, afiSafiState.getAfiSafisAdvertized());
 126         assertEquals(Collections.emptySet(), afiSafiState.getAfiSafisReceived());
 127         assertEquals(0L, afiSafiState.getPrefixesSentCount(TABLES_KEY));
 128         assertEquals(0L, afiSafiState.getPrefixesReceivedCount(TABLES_KEY));
 129         assertEquals(0L, afiSafiState.getPrefixesInstalledCount(TABLES_KEY));
 130
 131         assertFalse(afiSafiState.isGracefulRestartAdvertized(TABLES_KEY));
 132         assertFalse(afiSafiState.isGracefulRestartAdvertized(TABLES_KEY));
 133         assertFalse(afiSafiState.isLocalRestarting());
 134         assertFalse(afiSafiState.isPeerRestarting());
 135         assertFalse(afiSafiState.isAfiSafiSupported(TABLES_KEY));
 136
 137         final BGPSessionImpl session4 = createPeerSession(PEER4, nonAddPathParams, listener4);
 138
 139         final SimpleSessionListener listener5 = new SimpleSessionListener();
 140         configurePeer(this.tableRegistry, PEER5, this.ribImpl, addPathParams, PeerRole.RrClient, this.serverRegistry);
 141         final BGPSessionImpl session5 = createPeerSession(PEER5, addPathParams, listener5);
 142         checkPeersPresentOnDataStore(5);
 143
 144         //the best route
 145         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
 146         checkReceivedMessages(listener4, 1);
 147         checkReceivedMessages(listener5, 1);
 148         assertEquals(UPD_100, listener5.getListMsg().get(0));
 149
 150         final BGPPeerState peer1State = peer1.getPeerState();
 151         assertNull(peer1State.getGroupId());
 152         assertEquals(new IpAddress(PEER1), peer1State.getNeighborAddress());
 153         assertEquals(1L, peer1State.getTotalPathsCount());
 154         assertEquals(1L, peer1State.getTotalPrefixes());
 155
 156         final BGPTimersState timerStatePeer1 = peer1State.getBGPTimersState();
 157         assertEquals(HOLDTIMER, timerStatePeer1.getNegotiatedHoldTime());
 158         assertTrue(timerStatePeer1.getUpTime() > 0L);
 159
 160         final BGPTransportState transportStatePeer1 = peer1State.getBGPTransportState();
 161         assertEquals(new PortNumber(PORT), transportStatePeer1.getLocalPort());
 162         assertEquals(new IpAddress(PEER1), transportStatePeer1.getRemoteAddress());
 163
 164         assertEquals(State.UP, peer1State.getBGPSessionState().getSessionState());
 165         checkEquals(()-> assertEquals(1L,
 166                 peer1State.getBGPPeerMessagesState().getUpdateMessagesReceivedCount()));
 167         checkEquals(()-> assertEquals(0L,
 168                 peer1State.getBGPPeerMessagesState().getUpdateMessagesSentCount()));
 169
 170         final BGPSessionState sessionStatePeer1 = peer1State.getBGPSessionState();
 171         assertFalse(sessionStatePeer1.isAddPathCapabilitySupported());
 172         assertFalse(sessionStatePeer1.isAsn32CapabilitySupported());
 173         assertFalse(sessionStatePeer1.isGracefulRestartCapabilitySupported());
 174         assertTrue(sessionStatePeer1.isMultiProtocolCapabilitySupported());
 175         assertFalse(sessionStatePeer1.isRouterRefreshCapabilitySupported());
 176
 177         final BGPAfiSafiState afiSafiStatePeer1 = peer1State.getBGPAfiSafiState();
 178         assertEquals(AFI_SAFIS_ADVERTIZED, afiSafiStatePeer1.getAfiSafisAdvertized());
 179         assertEquals(AFI_SAFIS_ADVERTIZED, afiSafiStatePeer1.getAfiSafisReceived());
 180         assertEquals(0L, afiSafiStatePeer1.getPrefixesSentCount(TABLES_KEY));
 181         assertEquals(1L, afiSafiStatePeer1.getPrefixesReceivedCount(TABLES_KEY));
 182         assertEquals(1L, afiSafiStatePeer1.getPrefixesInstalledCount(TABLES_KEY));
 183
 184         assertFalse(afiSafiStatePeer1.isGracefulRestartAdvertized(TABLES_KEY));
 185         assertFalse(afiSafiStatePeer1.isGracefulRestartAdvertized(TABLES_KEY));
 186         assertFalse(afiSafiStatePeer1.isLocalRestarting());
 187         assertFalse(afiSafiStatePeer1.isPeerRestarting());
 188         assertTrue(afiSafiStatePeer1.isAfiSafiSupported(TABLES_KEY));
 189
 190         final BGPRibState ribState = this.ribImpl.getRIBState();
 191         assertEquals(1, ribState.getPathsCount().size());
 192         assertEquals(1L,  ribState.getTablesPrefixesCount().size());
 193         assertEquals(BGP_ID, ribState.getRouteId());
 194         assertEquals(AS_NUMBER, ribState.getAs());
 195         assertEquals(1L, ribState.getPathCount(TABLES_KEY));
 196         assertEquals(1L, ribState.getPrefixesCount(TABLES_KEY));
 197         assertEquals(1L, ribState.getTotalPathsCount());
 198         assertEquals(1L, ribState.getTotalPrefixesCount());
 199
 200         final SimpleSessionListener listener6 = new SimpleSessionListener();
 201         final BGPPeer peer6 = configurePeer(this.tableRegistry, PEER6, this.ribImpl, nonAddPathParams,
 202                 PeerRole.RrClient, this.serverRegistry);
 203         final BGPSessionImpl session6 = createPeerSession(PEER6, nonAddPathParams, listener6);
 204         checkPeersPresentOnDataStore(6);
 205         checkReceivedMessages(listener6, 1);
 206         assertEquals(UPD_NA_100, listener6.getListMsg().get(0));
 207         causeBGPError(session6);
 208         checkEquals(()-> assertEquals(1L,
 209                 peer6.getPeerState().getBGPPeerMessagesState().getNotificationMessagesSentCount()));
 210
 211         checkPeersPresentOnDataStore(5);
 212
 213         //the second best route
 214         sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 2);
 215         checkReceivedMessages(listener4, 1);
 216         checkReceivedMessages(listener5, 2);
 217         assertEquals(UPD_50, listener5.getListMsg().get(1));
 218
 219         //new best route
 220         sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 3);
 221         checkReceivedMessages(listener4, 2);
 222         checkReceivedMessages(listener5, 3);
 223         assertEquals(UPD_200, listener5.getListMsg().get(2));
 224
 225         peer4State = peer4.getPeerState();
 226         assertNull(peer4State.getGroupId());
 227         assertEquals(new IpAddress(PEER4), peer4State.getNeighborAddress());
 228         assertEquals(0L, peer4State.getTotalPathsCount());
 229         assertEquals(0L, peer4State.getTotalPrefixes());
 230
 231         final BGPTimersState timerState = peer4State.getBGPTimersState();
 232         assertEquals(HOLDTIMER, timerState.getNegotiatedHoldTime());
 233         assertTrue(timerState.getUpTime() > 0L);
 234
 235         final BGPTransportState transportState = peer4State.getBGPTransportState();
 236         assertEquals(new PortNumber(PORT), transportState.getLocalPort());
 237         assertEquals(new IpAddress(PEER4), transportState.getRemoteAddress());
 238
 239         final BGPPeerMessagesState peerMessagesState = peer4State.getBGPPeerMessagesState();
 240         assertEquals(0L, peerMessagesState.getNotificationMessagesReceivedCount());
 241         assertEquals(0L, peerMessagesState.getNotificationMessagesSentCount());
 242         assertEquals(0L, peerMessagesState.getUpdateMessagesReceivedCount());
 243         assertEquals(2L, peerMessagesState.getUpdateMessagesSentCount());
 244
 245         final BGPSessionState bgpSessionState = peer4State.getBGPSessionState();
 246         assertEquals(State.UP, bgpSessionState.getSessionState());
 247         assertFalse(bgpSessionState.isAddPathCapabilitySupported());
 248         assertFalse(bgpSessionState.isAsn32CapabilitySupported());
 249         assertFalse(bgpSessionState.isGracefulRestartCapabilitySupported());
 250         assertTrue(bgpSessionState.isMultiProtocolCapabilitySupported());
 251         assertFalse(bgpSessionState.isRouterRefreshCapabilitySupported());
 252
 253         final BGPErrorHandlingState errorHandling = peer4State.getBGPErrorHandlingState();
 254         assertEquals(0L, errorHandling.getErroneousUpdateReceivedCount());
 255
 256         gracefulRestart = peer4State.getBGPGracelfulRestart();
 257         assertFalse(gracefulRestart.isGracefulRestartAdvertized(TABLES_KEY));
 258         assertFalse(gracefulRestart.isGracefulRestartReceived(TABLES_KEY));
 259         assertFalse(gracefulRestart.isLocalRestarting());
 260         assertFalse(gracefulRestart.isPeerRestarting());
 261         assertEquals(0L, gracefulRestart.getPeerRestartTime());
 262
 263         afiSafiState = peer4State.getBGPAfiSafiState();
 264         assertEquals(AFI_SAFIS_ADVERTIZED, afiSafiState.getAfiSafisAdvertized());
 265         assertEquals(AFI_SAFIS_ADVERTIZED, afiSafiState.getAfiSafisReceived());
 266         assertEquals(2L, afiSafiState.getPrefixesSentCount(TABLES_KEY));
 267         assertEquals(0L, afiSafiState.getPrefixesReceivedCount(TABLES_KEY));
 268         assertEquals(0L, afiSafiState.getPrefixesInstalledCount(TABLES_KEY));
 269
 270         assertFalse(afiSafiState.isGracefulRestartAdvertized(TABLES_KEY));
 271         assertFalse(afiSafiState.isGracefulRestartAdvertized(TABLES_KEY));
 272         assertFalse(afiSafiState.isLocalRestarting());
 273         assertFalse(afiSafiState.isPeerRestarting());
 274         assertTrue(afiSafiState.isAfiSafiSupported(TABLES_KEY));
 275
 276         //the worst route
 277         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 20, 3);
 278         checkReceivedMessages(listener4, 2);
 279         checkReceivedMessages(listener5, 4);
 280         assertEquals(UPD_200.getAttributes().getLocalPref(),
 281                 ((Update) listener4.getListMsg().get(1)).getAttributes().getLocalPref());
 282         assertEquals(UPD_20, listener5.getListMsg().get(3));
 283
 284         //withdraw second best route, 1 advertisement(1 withdrawal) for add-path supported, none for non add path
 285         sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
 286         checkReceivedMessages(listener4, 2);
 287         checkReceivedMessages(listener5, 5);
 288
 289         //we advertise again to try new test
 290         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 3);
 291         checkReceivedMessages(listener4, 2);
 292         checkReceivedMessages(listener5, 6);
 293         assertEquals(UPD_200, listener5.getListMsg().get(2));
 294
 295         //withdraw second best route, 1 advertisement(1 withdrawal) for add-path supported, 1 for non add path (withdrawal)
 296         sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
 297         checkReceivedMessages(listener4, 3);
 298         checkReceivedMessages(listener5, 7);
 299
 300         sendNotification(session1);
 301         checkEquals(()-> assertEquals(1L,
 302                 peer1.getPeerState().getBGPPeerMessagesState().getNotificationMessagesReceivedCount()));
 303         session1.close();
 304         session2.close();
 305         session3.close();
 306         session4.close();
 307         session5.close();
 308     }
 309 }