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
  13 import com.google.common.collect.ImmutableList;
  14 import com.google.common.collect.ImmutableMap;
  15 import io.netty.channel.Channel;
  16 import java.net.InetSocketAddress;
  17 import java.util.List;
  18 import java.util.Map;
  19 import org.junit.Test;
  20 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
  21 import org.opendaylight.protocol.bgp.mode.impl.base.BasePathSelectionModeFactory;
  22 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  31
  32 public class AddPathBasePathsTest extends AbstractAddPathTest {
  33     /*
  34     * Base-Paths
  35     *                                            ___________________
  36     *                                           | ODL BGP 127.0.0.1 |
  37     * [peer://127.0.0.2; p1, lp100] --(iBGP)--> |                   | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
  38     * [peer://127.0.0.3; p1, lp200] --(iBGP)--> |                   |
  39     * [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)]
  40     * [peer://127.0.0.2; p1, lp20] --(iBGP)-->  |___________________|
  41     * p1 = 1.1.1.1/32
  42     */
  43     @Test
  44     public void testUseCase1() throws Exception {
  45
  46         final List<BgpTableType> tables = ImmutableList.of(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
  47         final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
  48         final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk, BasePathSelectionModeFactory.createBestPathSelectionStrategy());
  49
  50         final RIBImpl ribImpl = new RIBImpl(this.clusterSingletonServiceProvider, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID), null, this.ribExtension,
  51             this.dispatcher, this.mappingService.getCodecFactory(), getDomBroker(), tables, pathTables, this.ribExtension.getClassLoadingStrategy(), null);
  52
  53         ribImpl.instantiateServiceInstance();
  54         ribImpl.onGlobalContextUpdated(this.schemaContext);
  55
  56         this.dispatcher.createServer(StrictBGPPeerRegistry.GLOBAL, new InetSocketAddress(RIB_ID, PORT)).sync();
  57         Thread.sleep(1000);
  58
  59         final BGPHandlerFactory hf = new BGPHandlerFactory(this.context.getMessageRegistry());
  60         final BgpParameters nonAddPathParams = createParameter(false);
  61
  62         final Channel session1 = createPeerSession(PEER1, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
  63         final Channel session2 = createPeerSession(PEER2, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
  64         final Channel session3 = createPeerSession(PEER3, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
  65         final SimpleSessionListener listener4 = new SimpleSessionListener();
  66         final Channel session4 = createPeerSession(PEER4, PeerRole.RrClient, nonAddPathParams, ribImpl, hf, listener4);
  67         final SimpleSessionListener listener5 = new SimpleSessionListener();
  68         final Channel session5 = createPeerSession(PEER5, PeerRole.Ebgp, nonAddPathParams, ribImpl, hf, listener5);
  69         Thread.sleep(1000);
  70         checkPeersPresentOnDataStore(5);
  71
  72         //new best route so far
  73         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
  74         assertEquals(1, listener4.getListMsg().size());
  75         assertEquals(1, listener5.getListMsg().size());
  76         assertEquals(UPD_NA_100, listener4.getListMsg().get(0));
  77         assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(0));
  78
  79         //the second best route
  80         sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 1);
  81         assertEquals(1, listener4.getListMsg().size());
  82         assertEquals(1, listener5.getListMsg().size());
  83
  84         //new best route
  85         sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 1);
  86         assertEquals(2, listener4.getListMsg().size());
  87         assertEquals(2, listener5.getListMsg().size());
  88         assertEquals(UPD_NA_200, listener4.getListMsg().get(1));
  89         assertEquals(UPD_NA_200_EBGP, listener5.getListMsg().get(1));
  90
  91         final SimpleSessionListener listener6 = new SimpleSessionListener();
  92         final Channel session6 = createPeerSession(PEER6, PeerRole.RrClient, nonAddPathParams, ribImpl, hf, listener6);
  93         Thread.sleep(1000);
  94         checkPeersPresentOnDataStore(6);
  95         assertEquals(1, listener6.getListMsg().size());
  96         assertEquals(UPD_NA_200, listener6.getListMsg().get(0));
  97         Thread.sleep(1000);
  98         session6.close();
  99         Thread.sleep(1000);
 100
 101         //best route updated to be the worse one
 102         sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 20, 1);
 103         assertEquals(3, listener4.getListMsg().size());
 104         assertEquals(3, listener5.getListMsg().size());
 105         assertEquals(UPD_NA_100, listener4.getListMsg().get(2));
 106         assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(2));
 107
 108         //Remove second best, no advertisement should be done
 109         sendWithdrawalRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 1);
 110         assertEquals(3, listener4.getListMsg().size());
 111         assertEquals(3, listener5.getListMsg().size());
 112
 113         //Remove best, 1 advertisement
 114         sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
 115         assertEquals(4, listener4.getListMsg().size());
 116         assertEquals(4, listener5.getListMsg().size());
 117
 118         //Remove best, 1 withdrawal
 119         sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 20, 0);
 120         assertEquals(5, listener4.getListMsg().size());
 121         assertEquals(5, listener5.getListMsg().size());
 122
 123         session1.close();
 124         session2.close();
 125         session3.close();
 126         session4.close();
 127         session5.close();
 128         Thread.sleep(1000);
 129     }
 130 }