* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.protocol.bgp.rib.impl;
import static org.junit.Assert.assertEquals;
+import static org.opendaylight.protocol.util.CheckUtil.checkReceivedMessages;
+import static org.opendaylight.protocol.util.CheckUtil.waitFutureSuccess;
-import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import io.netty.channel.Channel;
+import io.netty.channel.ChannelFuture;
import java.net.InetSocketAddress;
-import java.util.List;
import java.util.Map;
+import org.junit.After;
+import org.junit.Before;
import org.junit.Test;
import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
import org.opendaylight.protocol.bgp.mode.impl.base.BasePathSelectionModeFactory;
-import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerRole;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.RibId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.BgpId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.UnicastSubsequentAddressFamily;
public class AddPathBasePathsTest extends AbstractAddPathTest {
- /*
- * Base-Paths
- * ___________________
- * | ODL BGP 127.0.0.1 |
- * [peer://127.0.0.2; p1, lp100] --(iBGP)--> | | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
- * [peer://127.0.0.3; p1, lp200] --(iBGP)--> | |
- * [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)]
- * [peer://127.0.0.2; p1, lp20] --(iBGP)--> |___________________|
- * p1 = 1.1.1.1/32
- */
- @Test
- public void testUseCase1() throws Exception {
+ private RIBImpl ribImpl;
+ private Channel serverChannel;
- final List<BgpTableType> tables = ImmutableList.of(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
+ @Override
+ @Before
+ public void setUp() throws Exception {
+ super.setUp();
final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
- final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk, BasePathSelectionModeFactory.createBestPathSelectionStrategy());
+ final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk,
+ BasePathSelectionModeFactory.createBestPathSelectionStrategy());
+
+ this.ribImpl = new RIBImpl(this.tableRegistry, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID),
+ this.ribExtension,
+ this.serverDispatcher, this.codecsRegistry, getDomBroker(), getDataBroker(), this.policies,
+ TABLES_TYPE, pathTables);
+ this.ribImpl.instantiateServiceInstance();
+ this.ribImpl.onGlobalContextUpdated(this.schemaService.getGlobalContext());
+ final ChannelFuture channelFuture = this.serverDispatcher.createServer(new InetSocketAddress(RIB_ID, PORT));
+ waitFutureSuccess(channelFuture);
+ this.serverChannel = channelFuture.channel();
+ }
- final RIBImpl ribImpl = new RIBImpl(this.clusterSingletonServiceProvider, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID), null, this.ribExtension,
- this.dispatcher, this.mappingService.getCodecFactory(), getDomBroker(), tables, pathTables, this.ribExtension.getClassLoadingStrategy(), null);
+ @Override
+ @After
+ public void tearDown() throws Exception {
+ waitFutureSuccess(this.serverChannel.close());
+ super.tearDown();
+ }
- ribImpl.instantiateServiceInstance();
- ribImpl.onGlobalContextUpdated(this.schemaContext);
+ /*
+ * Base-Paths
+ * ___________________
+ * | ODL BGP 127.0.0.1 |
+ * [peer://127.0.0.2; p1, lp100] --(iBGP)--> | | --(RR-client, non add-path) -->
+ * [peer://127.0.0.3; p1, lp200] --(iBGP)--> | | [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
+ * [peer://127.0.0.4; p1, lp50] --(iBGP)--> | | --(eBgp, non add-path) -->
+ * [peer://127.0.0.2; p1, lp20] --(iBGP)--> |___________________| [Peer://127.0.0.6; (p1, path-id1, lp100),
+ * p1 = 1.1.1.1/32 (p1, path-id2, pl50), (p1, path-id3, pl200)]
+ */
+ @Test
+ public void testUseCase1() throws Exception {
+ final BgpParameters nonAddPathParams = createParameter(false);
- this.dispatcher.createServer(StrictBGPPeerRegistry.GLOBAL, new InetSocketAddress(RIB_ID, PORT)).sync();
- Thread.sleep(1000);
+ configurePeer(this.tableRegistry, PEER1, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
+ final BGPSessionImpl session1 = createPeerSession(PEER1, nonAddPathParams, new SimpleSessionListener());
- final BGPHandlerFactory hf = new BGPHandlerFactory(this.context.getMessageRegistry());
- final BgpParameters nonAddPathParams = createParameter(false);
+ configurePeer(this.tableRegistry, PEER2, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
+ final BGPSessionImpl session2 = createPeerSession(PEER2, nonAddPathParams, new SimpleSessionListener());
+
+ configurePeer(this.tableRegistry, PEER3, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
+ final BGPSessionImpl session3 = createPeerSession(PEER3,nonAddPathParams, new SimpleSessionListener());
- final Channel session1 = createPeerSession(PEER1, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
- final Channel session2 = createPeerSession(PEER2, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
- final Channel session3 = createPeerSession(PEER3, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
final SimpleSessionListener listener4 = new SimpleSessionListener();
- final Channel session4 = createPeerSession(PEER4, PeerRole.RrClient, nonAddPathParams, ribImpl, hf, listener4);
+ configurePeer(this.tableRegistry, PEER4, this.ribImpl, nonAddPathParams, PeerRole.RrClient,
+ this.serverRegistry);
+ final BGPSessionImpl session4 = createPeerSession(PEER4, nonAddPathParams, listener4);
+
final SimpleSessionListener listener5 = new SimpleSessionListener();
- final Channel session5 = createPeerSession(PEER5, PeerRole.Ebgp, nonAddPathParams, ribImpl, hf, listener5);
- Thread.sleep(1000);
+ configurePeer(this.tableRegistry, PEER5, this.ribImpl, nonAddPathParams, PeerRole.Ebgp, this.serverRegistry);
+ final BGPSessionImpl session5 = createPeerSession(PEER5, nonAddPathParams, listener5);
checkPeersPresentOnDataStore(5);
//new best route so far
sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
- assertEquals(1, listener4.getListMsg().size());
- assertEquals(1, listener5.getListMsg().size());
- assertEquals(UPD_NA_100, listener4.getListMsg().get(0));
- assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(0));
+ checkReceivedMessages(listener4, 2);
+ checkReceivedMessages(listener5, 2);
+ assertEquals(UPD_NA_100, listener4.getListMsg().get(1));
+ assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(1));
//the second best route
- sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 1);
- assertEquals(1, listener4.getListMsg().size());
- assertEquals(1, listener5.getListMsg().size());
+ sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 100, 1);
+ checkReceivedMessages(listener4, 2);
+ checkReceivedMessages(listener5, 2);
//new best route
sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 1);
- assertEquals(2, listener4.getListMsg().size());
- assertEquals(2, listener5.getListMsg().size());
- assertEquals(UPD_NA_200, listener4.getListMsg().get(1));
- assertEquals(UPD_NA_200_EBGP, listener5.getListMsg().get(1));
+ checkReceivedMessages(listener4, 3);
+ checkReceivedMessages(listener5, 3);
+ assertEquals(UPD_NA_200, listener4.getListMsg().get(2));
+ assertEquals(UPD_NA_200_EBGP, listener5.getListMsg().get(2));
final SimpleSessionListener listener6 = new SimpleSessionListener();
- final Channel session6 = createPeerSession(PEER6, PeerRole.RrClient, nonAddPathParams, ribImpl, hf, listener6);
- Thread.sleep(1000);
+ configurePeer(this.tableRegistry, PEER6, this.ribImpl, nonAddPathParams, PeerRole.RrClient,
+ this.serverRegistry);
+ final BGPSessionImpl session6 = createPeerSession(PEER6, nonAddPathParams, listener6);
+
checkPeersPresentOnDataStore(6);
- assertEquals(1, listener6.getListMsg().size());
- assertEquals(UPD_NA_200, listener6.getListMsg().get(0));
- Thread.sleep(1000);
+ checkReceivedMessages(listener6, 2);
+ assertEquals(UPD_NA_200, listener6.getListMsg().get(1));
session6.close();
- Thread.sleep(1000);
+ checkPeersPresentOnDataStore(5);
//best route updated to be the worse one
sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 20, 1);
- assertEquals(3, listener4.getListMsg().size());
- assertEquals(3, listener5.getListMsg().size());
- assertEquals(UPD_NA_100, listener4.getListMsg().get(2));
- assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(2));
+ checkReceivedMessages(listener4, 4);
+ checkReceivedMessages(listener5, 4);
+ assertEquals(UPD_NA_100, listener4.getListMsg().get(3));
+ assertEquals(UPD_NA_100_EBGP, listener5.getListMsg().get(3));
//Remove second best, no advertisement should be done
sendWithdrawalRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 1);
- assertEquals(3, listener4.getListMsg().size());
- assertEquals(3, listener5.getListMsg().size());
+ checkReceivedMessages(listener4, 4);
+ checkReceivedMessages(listener5, 4);
//Remove best, 1 advertisement
sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
- assertEquals(4, listener4.getListMsg().size());
- assertEquals(4, listener5.getListMsg().size());
+ checkReceivedMessages(listener4, 5);
+ checkReceivedMessages(listener5, 5);
//Remove best, 1 withdrawal
sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 20, 0);
- assertEquals(5, listener4.getListMsg().size());
- assertEquals(5, listener5.getListMsg().size());
+ checkReceivedMessages(listener4, 6);
+ checkReceivedMessages(listener5, 6);
session1.close();
session2.close();
session3.close();
session4.close();
session5.close();
- Thread.sleep(1000);
}
}
Code Review / bgpcep.git / blobdiff