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 org.junit.After;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
24 import org.opendaylight.protocol.bgp.mode.impl.add.n.paths.AddPathBestNPathSelection;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerRole;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.RibId;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.BgpId;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.UnicastSubsequentAddressFamily;
33 public class AddPathNPathsTest extends AbstractAddPathTest {
34 private RIBImpl ribImpl;
35 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, new AddPathBestNPathSelection(2));
44 this.ribImpl = new RIBImpl(this.tableRegistry, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID),
46 this.serverDispatcher, this.codecsRegistry, getDomBroker(), getDataBroker(), this.policies,
47 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();
58 public void tearDown() throws Exception {
59 waitFutureSuccess(this.serverChannel.close());
66 * | ODL BGP 127.0.0.1 |
67 * [peer://127.0.0.2; p1, lp100] --(iBGP)--> | | --(RR-client, non add-path) -->
68 * [peer://127.0.0.3; p1, lp200] --(iBGP)--> | | [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
69 * [peer://127.0.0.4; p1, lp50] --(iBGP)--> | | --(RR-client, add-path) -->
70 * [peer://127.0.0.2; p1, lp20] --(iBGP)--> |___________________| [Peer://127.0.0.6; (p1, path-id1, lp100),
71 * p1 = 1.1.1.1/32 (p1, path-id2, pl50), (p1, path-id3, pl200)]
74 public void testUseCase1() throws Exception {
75 final BgpParameters nonAddPathParams = createParameter(false);
76 final BgpParameters addPathParams = createParameter(true);
78 configurePeer(this.tableRegistry, PEER1, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
79 final BGPSessionImpl session1 = createPeerSession(PEER1, nonAddPathParams, new SimpleSessionListener());
81 configurePeer(this.tableRegistry, PEER2, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
82 final BGPSessionImpl session2 = createPeerSession(PEER2, nonAddPathParams, new SimpleSessionListener());
84 configurePeer(this.tableRegistry, PEER3, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
85 final BGPSessionImpl session3 = createPeerSession(PEER3, nonAddPathParams, new SimpleSessionListener());
87 final SimpleSessionListener listener4 = new SimpleSessionListener();
88 configurePeer(this.tableRegistry, PEER4, this.ribImpl, nonAddPathParams, PeerRole.RrClient,
90 final BGPSessionImpl session4 = createPeerSession(PEER4, nonAddPathParams, listener4);
92 final SimpleSessionListener listener5 = new SimpleSessionListener();
93 configurePeer(this.tableRegistry, PEER5, this.ribImpl, addPathParams, PeerRole.RrClient, this.serverRegistry);
94 final BGPSessionImpl session5 = createPeerSession(PEER5, addPathParams, listener5);
95 checkPeersPresentOnDataStore(5);
97 //new best route so far
98 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
99 checkReceivedMessages(listener4, 2);
100 checkReceivedMessages(listener5, 2);
101 assertEquals(UPD_100, listener5.getListMsg().get(1));
103 final SimpleSessionListener listener6 = new SimpleSessionListener();
104 configurePeer(this.tableRegistry, PEER6, this.ribImpl, nonAddPathParams, PeerRole.RrClient,
105 this.serverRegistry);
106 final BGPSessionImpl session6 = createPeerSession(PEER6, nonAddPathParams, listener6);
107 checkPeersPresentOnDataStore(6);
108 checkReceivedMessages(listener6, 2);
109 assertEquals(UPD_NA_100, listener6.getListMsg().get(1));
111 checkPeersPresentOnDataStore(5);
113 //the second best route
114 sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 2);
115 checkReceivedMessages(listener4, 2);
116 checkReceivedMessages(listener5, 3);
117 assertEquals(UPD_50, listener5.getListMsg().get(2));
120 sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
121 checkReceivedMessages(listener4, 3);
122 checkReceivedMessages(listener5, 4);
123 assertEquals(UPD_200, listener5.getListMsg().get(3));
125 //the worst prefix, no changes
126 sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 20, 2);
127 checkReceivedMessages(listener4, 3);
128 checkReceivedMessages(listener5, 4);
130 //withdraw second best route, 2 advertisement (1 withdrawal) for add-path supported, none for non add path
131 sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
132 checkReceivedMessages(listener4, 3);
133 checkReceivedMessages(listener5, 6);
135 //we advertise again to try new test
136 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
137 checkReceivedMessages(listener4, 3);
138 checkReceivedMessages(listener5, 7);
140 //withdraw second best route, 2 advertisement (1 withdrawal) for add-path supported,
141 // 1 withdrawal for non add path
142 sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
143 checkReceivedMessages(listener4, 4);
144 checkReceivedMessages(listener5, 9);