2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.protocol.bgp.rib.impl;
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;
15 import com.google.common.collect.ImmutableMap;
16 import io.netty.channel.Channel;
17 import io.netty.channel.ChannelFuture;
18 import java.net.InetSocketAddress;
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.add.n.paths.AddPathBestNPathSelection;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerRole;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.RibId;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.BgpId;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.UnicastSubsequentAddressFamily;
34 public class AddPathNPathsTest extends AbstractAddPathTest {
35 private RIBImpl ribImpl;
36 private Channel serverChannel;
39 public void setUp() throws Exception {
41 final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
42 final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk,
43 new AddPathBestNPathSelection(2L, this.peerTracker));
45 this.ribImpl = new RIBImpl(new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID), this.ribExtension,
46 this.serverDispatcher, this.codecsRegistry, getDomBroker(), getDataBroker(), this.policies,
47 this.peerTracker, TABLES_TYPE, pathTables);
49 this.ribImpl.instantiateServiceInstance();
50 this.ribImpl.onGlobalContextUpdated(this.schemaService.getGlobalContext());
51 final ChannelFuture channelFuture = this.serverDispatcher.createServer(new InetSocketAddress(RIB_ID, PORT));
52 waitFutureSuccess(channelFuture);
53 this.serverChannel = channelFuture.channel();
57 public void tearDown() throws ExecutionException, InterruptedException {
58 waitFutureSuccess(this.serverChannel.close());
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)--> | | --(RR-client, 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)--> |___________________|
72 public void testUseCase1() throws Exception {
73 final BgpParameters nonAddPathParams = createParameter(false);
74 final BgpParameters addPathParams = createParameter(true);
76 configurePeer(PEER1, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
77 final BGPSessionImpl session1 = createPeerSession(PEER1, nonAddPathParams, new SimpleSessionListener());
79 configurePeer(PEER2, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
80 final BGPSessionImpl session2 = createPeerSession(PEER2, nonAddPathParams, new SimpleSessionListener());
82 configurePeer(PEER3, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
83 final BGPSessionImpl session3 = createPeerSession(PEER3, nonAddPathParams, new SimpleSessionListener());
85 final SimpleSessionListener listener4 = new SimpleSessionListener();
86 configurePeer(PEER4, this.ribImpl, nonAddPathParams, PeerRole.RrClient, this.serverRegistry);
87 final BGPSessionImpl session4 = createPeerSession(PEER4, nonAddPathParams, listener4);
89 final SimpleSessionListener listener5 = new SimpleSessionListener();
90 configurePeer(PEER5, this.ribImpl, addPathParams, PeerRole.RrClient, this.serverRegistry);
91 final BGPSessionImpl session5 = createPeerSession(PEER5, addPathParams, listener5);
92 checkPeersPresentOnDataStore(5);
94 //new best route so far
95 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
96 checkReceivedMessages(listener4, 1);
97 checkReceivedMessages(listener5, 1);
98 assertEquals(UPD_100, listener5.getListMsg().get(0));
100 final SimpleSessionListener listener6 = new SimpleSessionListener();
101 configurePeer(PEER6, this.ribImpl, nonAddPathParams, PeerRole.RrClient, this.serverRegistry);
102 final BGPSessionImpl session6 = createPeerSession(PEER6, nonAddPathParams, listener6);
103 checkPeersPresentOnDataStore(6);
104 checkReceivedMessages(listener6, 1);
105 assertEquals(UPD_NA_100, listener6.getListMsg().get(0));
107 checkPeersPresentOnDataStore(5);
109 //the second best route
110 sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 2);
111 checkReceivedMessages(listener4, 1);
112 checkReceivedMessages(listener5, 2);
113 assertEquals(UPD_50, listener5.getListMsg().get(1));
116 sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
117 checkReceivedMessages(listener4, 2);
118 checkReceivedMessages(listener5, 3);
119 assertEquals(UPD_200, listener5.getListMsg().get(2));
121 //the worst prefix, no changes
122 sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 20, 2);
123 checkReceivedMessages(listener4, 2);
124 checkReceivedMessages(listener5, 3);
126 //withdraw second best route, 2 advertisement (1 withdrawal) for add-path supported, none for non add path
127 sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
128 checkReceivedMessages(listener4, 2);
129 checkReceivedMessages(listener5, 5);
131 //we advertise again to try new test
132 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
133 checkReceivedMessages(listener4, 2);
134 checkReceivedMessages(listener5, 6);
136 //withdraw second best route, 2 advertisement (1 withdrawal) for add-path supported, 1 withdrawal for non add path
137 sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
138 checkReceivedMessages(listener4, 3);
139 checkReceivedMessages(listener5, 8);