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
8 package org.opendaylight.protocol.bgp.rib.impl;
10 import static org.junit.Assert.assertEquals;
11 import static org.opendaylight.protocol.bgp.rib.impl.CheckUtil.checkReceivedMessages;
12 import static org.opendaylight.protocol.bgp.rib.impl.CheckUtil.waitFutureSuccess;
14 import com.google.common.collect.ImmutableList;
15 import com.google.common.collect.ImmutableMap;
16 import java.net.InetSocketAddress;
17 import java.util.List;
19 import org.junit.Test;
20 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
21 import org.opendaylight.protocol.bgp.mode.impl.add.all.paths.AllPathSelection;
22 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Update;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
33 public class AddPathAllPathsTest extends AbstractAddPathTest {
37 * | ODL BGP 127.0.0.1 |
38 * [peer://127.0.0.2; p1, lp100] --(iBGP)--> | | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
39 * [peer://127.0.0.3; p1, lp200] --(iBGP)--> | |
40 * [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), (p1, path-id4, pl20)]
41 * [peer://127.0.0.2; p1, lp20] --(iBGP)--> |___________________|
45 public void testUseCase1() throws Exception {
47 final List<BgpTableType> tables = ImmutableList.of(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
48 final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(TABLES_KEY, new AllPathSelection());
51 final RIBImpl ribImpl = new RIBImpl(this.clusterSingletonServiceProvider, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID), null, this.ribExtension,
52 this.dispatcher, this.mappingService.getCodecFactory(), getDomBroker(), tables, pathTables, this.ribExtension.getClassLoadingStrategy(), null);
54 ribImpl.instantiateServiceInstance();
55 ribImpl.onGlobalContextUpdated(this.schemaContext);
57 waitFutureSuccess(this.dispatcher.createServer(StrictBGPPeerRegistry.GLOBAL, new InetSocketAddress(RIB_ID, PORT)));
58 final BgpParameters nonAddPathParams = createParameter(false);
59 final BgpParameters addPathParams = createParameter(true);
61 final BGPSessionImpl session1 = createPeerSession(PEER1, PeerRole.Ibgp, nonAddPathParams, ribImpl, new SimpleSessionListener());
62 final BGPSessionImpl session2 = createPeerSession(PEER2, PeerRole.Ibgp, nonAddPathParams, ribImpl, new SimpleSessionListener());
63 final BGPSessionImpl session3 = createPeerSession(PEER3, PeerRole.Ibgp, nonAddPathParams, ribImpl, new SimpleSessionListener());
64 final SimpleSessionListener listener4 = new SimpleSessionListener();
65 final BGPSessionImpl session4 = createPeerSession(PEER4, PeerRole.RrClient, nonAddPathParams, ribImpl, listener4);
66 final SimpleSessionListener listener5 = new SimpleSessionListener();
67 final BGPSessionImpl session5 = createPeerSession(PEER5, PeerRole.RrClient, addPathParams, ribImpl, listener5);
68 checkPeersPresentOnDataStore(5);
71 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
72 checkReceivedMessages(listener4, 1);
73 checkReceivedMessages(listener5, 1);
74 assertEquals(UPD_100, listener5.getListMsg().get(0));
76 final SimpleSessionListener listener6 = new SimpleSessionListener();
77 final BGPSessionImpl session6 = createPeerSession(PEER6, PeerRole.RrClient, nonAddPathParams, ribImpl, listener6);
78 checkPeersPresentOnDataStore(6);
79 checkReceivedMessages(listener6, 1);
80 assertEquals(UPD_NA_100, listener6.getListMsg().get(0));
82 checkPeersPresentOnDataStore(5);
84 //the second best route
85 sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 2);
86 checkReceivedMessages(listener4, 1);
87 checkReceivedMessages(listener5, 2);
88 assertEquals(UPD_50, listener5.getListMsg().get(1));
91 sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 3);
92 checkReceivedMessages(listener4, 2);
93 checkReceivedMessages(listener5, 3);
94 assertEquals(UPD_200, listener5.getListMsg().get(2));
97 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 20, 3);
98 checkReceivedMessages(listener4, 2);
99 checkReceivedMessages(listener5, 4);
100 assertEquals(UPD_200.getAttributes().getLocalPref(), ((Update) listener4.getListMsg().get(1)).getAttributes().getLocalPref());
101 assertEquals(UPD_20, listener5.getListMsg().get(3));
103 //withdraw second best route, 1 advertisement(1 withdrawal) for add-path supported, none for non add path
104 sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
105 checkReceivedMessages(listener4, 2);
106 checkReceivedMessages(listener5, 5);
108 //we advertise again to try new test
109 sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 3);
110 checkReceivedMessages(listener4, 2);
111 checkReceivedMessages(listener5, 6);
112 assertEquals(UPD_200, listener5.getListMsg().get(2));
114 //withdraw second best route, 1 advertisement(1 withdrawal) for add-path supported, 1 for non add path (withdrawal)
115 sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
116 checkReceivedMessages(listener4, 3);
117 checkReceivedMessages(listener5, 7);