bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/AddPathBasePathsTest.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
   9 package org.opendaylight.protocol.bgp.rib.impl;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.opendaylight.protocol.util.CheckUtil.checkReceivedMessages;
  13 import static org.opendaylight.protocol.util.CheckUtil.waitFutureSuccess;
  14
  15 import com.google.common.collect.ImmutableMap;
  16 import io.netty.channel.Channel;
  17 import io.netty.channel.ChannelFuture;
  18 import java.net.InetSocketAddress;
  19 import java.util.Map;
  20 import java.util.concurrent.ExecutionException;
  21 import org.junit.After;
  22 import org.junit.Before;
  23 import org.junit.Test;
  24 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
  25 import org.opendaylight.protocol.bgp.mode.impl.base.BasePathSelectionModeFactory;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.BgpParameters;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.PeerRole;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.RibId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  33
  34 public class AddPathBasePathsTest extends AbstractAddPathTest {
  35     private RIBImpl ribImpl;
  36     private Channel serverChannel;
  37
  38     @Before
  39     public void setUp() throws Exception {
  40         super.setUp();
  41         final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
  42         final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk,
  43             BasePathSelectionModeFactory.createBestPathSelectionStrategy(this.peerTracker));
  44
  45         this.ribImpl = new RIBImpl(new RibId("test-rib"),
  46             AS_NUMBER, new BgpId(RIB_ID), null, this.ribExtension, this.serverDispatcher,
  47             this.codecsRegistry, getDomBroker(), this.policies, this.peerTracker, TABLES_TYPE, pathTables,
  48                 this.mappingService);
  49         this.ribImpl.instantiateServiceInstance();
  50         this.ribImpl.onGlobalContextUpdated(this.schemaContext);
  51         final ChannelFuture channelFuture = this.serverDispatcher.createServer(new InetSocketAddress(RIB_ID, PORT));
  52         waitFutureSuccess(channelFuture);
  53         this.serverChannel = channelFuture.channel();
  54     }
  55
  56     @After
  57     public void tearDown() throws ExecutionException, InterruptedException {
  58         waitFutureSuccess(this.serverChannel.close());
  59         super.tearDown();
  60     }
  61     /*
  62     * Base-Paths
  63     *                                            ___________________
  64     *                                           | ODL BGP 127.0.0.1 |
  65     * [peer://127.0.0.2; p1, lp100] --(iBGP)--> |                   | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
  66     * [peer://127.0.0.3; p1, lp200] --(iBGP)--> |                   |
  67     * [peer://127.0.0.4; p1, lp50] --(iBGP)-->  |                   | --(eBgp, non add-path) --> [Peer://127.0.0.6; (p1, path-id1, lp100), (p1, path-id2, pl50), (p1, path-id3, pl200)]
  68     * [peer://127.0.0.2; p1, lp20] --(iBGP)-->  |___________________|
  69     * p1 = 1.1.1.1/32
  70     */
  71     @Test
  72     public void testUseCase1() throws Exception {
  73         final BgpParameters nonAddPathParams = createParameter(false);
  74
  75         configurePeer(PEER1, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
  76         final BGPSessionImpl session1 = createPeerSession(PEER1, nonAddPathParams, new SimpleSessionListener());
  77
  78         configurePeer(PEER2, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
  79         final BGPSessionImpl session2 = createPeerSession(PEER2, nonAddPathParams, new SimpleSessionListener());
  80
  81         configurePeer(PEER3, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
  82         final BGPSessionImpl session3 = createPeerSession(PEER3,nonAddPathParams, new SimpleSessionListener());
  83
  84         final SimpleSessionListener listener4 = new SimpleSessionListener();
  85         configurePeer(PEER4, this.ribImpl, nonAddPathParams, PeerRole.RrClient, this.serverRegistry);
  86         final BGPSessionImpl session4 = createPeerSession(PEER4, nonAddPathParams, listener4);
  87
  88         final SimpleSessionListener listener5 = new SimpleSessionListener();
  89         configurePeer(PEER5, this.ribImpl, nonAddPathParams, PeerRole.Ebgp, this.serverRegistry);
  90         final BGPSessionImpl session5 = createPeerSession(PEER5, nonAddPathParams, listener5);
  91         checkPeersPresentOnDataStore(5);
  92
  93         //new best route so far
  94         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
  95         checkReceivedMessages(listener4, 1);
  96         checkReceivedMessages(listener5, 1);
  97         assertEquals(UPD_NA_100, listener4.getListMsg().get(0));
  98         assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(0));
  99
 100         //the second best route
 101         sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 1);
 102         checkReceivedMessages(listener4, 1);
 103         checkReceivedMessages(listener5, 1);
 104
 105         //new best route
 106         sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 1);
 107         checkReceivedMessages(listener4, 2);
 108         checkReceivedMessages(listener5, 2);
 109         assertEquals(UPD_NA_200, listener4.getListMsg().get(1));
 110         assertEquals(UPD_NA_200_EBGP, listener5.getListMsg().get(1));
 111
 112         final SimpleSessionListener listener6 = new SimpleSessionListener();
 113         configurePeer(PEER6, this.ribImpl, nonAddPathParams, PeerRole.RrClient, this.serverRegistry);
 114         final BGPSessionImpl session6 = createPeerSession(PEER6, nonAddPathParams, listener6);
 115
 116         checkPeersPresentOnDataStore(6);
 117         checkReceivedMessages(listener6, 1);
 118         assertEquals(UPD_NA_200, listener6.getListMsg().get(0));
 119         session6.close();
 120         checkPeersPresentOnDataStore(5);
 121
 122         //best route updated to be the worse one
 123         sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 20, 1);
 124         checkReceivedMessages(listener4, 3);
 125         checkReceivedMessages(listener5, 3);
 126         assertEquals(UPD_NA_100, listener4.getListMsg().get(2));
 127         assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(2));
 128
 129         //Remove second best, no advertisement should be done
 130         sendWithdrawalRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 1);
 131         checkReceivedMessages(listener4, 3);
 132         checkReceivedMessages(listener5, 3);
 133
 134         //Remove best, 1 advertisement
 135         sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
 136         checkReceivedMessages(listener4, 4);
 137         checkReceivedMessages(listener5, 4);
 138
 139         //Remove best, 1 withdrawal
 140         sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 20, 0);
 141         checkReceivedMessages(listener4, 5);
 142         checkReceivedMessages(listener5, 5);
 143
 144         session1.close();
 145         session2.close();
 146         session3.close();
 147         session4.close();
 148         session5.close();
 149     }
 150 }