1356cc82ad431110b4d132fa1d3de95793afca28
[bgpcep.git] / bgp / rib-impl / src / test / java / org / opendaylight / protocol / bgp / rib / impl / AddPathNPathsTest.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 import static org.opendaylight.protocol.util.CheckUtil.checkReceivedMessages;
13 import static org.opendaylight.protocol.util.CheckUtil.waitFutureSuccess;
14
15 import com.google.common.collect.ImmutableMap;
16 import io.netty.channel.Channel;
17 import io.netty.channel.ChannelFuture;
18 import java.net.InetSocketAddress;
19 import java.util.Map;
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;
33
34 public class AddPathNPathsTest extends AbstractAddPathTest {
35     private RIBImpl ribImpl;
36     private Channel serverChannel;
37
38     @Before
39     public void setUp() throws Exception {
40         super.setUp();
41         final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
42         final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk, new AddPathBestNPathSelection(2L));
43
44         this.ribImpl = new RIBImpl(this.tableRegistry, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID),
45                 this.ribExtension,
46                 this.serverDispatcher, this.codecsRegistry, getDomBroker(), getDataBroker(), this.policies,
47                 TABLES_TYPE,  pathTables);
48
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();
54     }
55
56     @After
57     public void tearDown() throws ExecutionException, InterruptedException {
58         waitFutureSuccess(this.serverChannel.close());
59         super.tearDown();
60     }
61     /*
62      * N-Paths
63      *                                            ___________________
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)-->  |___________________|
69      * p1 = 1.1.1.1/32
70      */
71     @Test
72     public void testUseCase1() throws Exception {
73         final BgpParameters nonAddPathParams = createParameter(false);
74         final BgpParameters addPathParams = createParameter(true);
75
76         configurePeer(this.tableRegistry, PEER1, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
77         final BGPSessionImpl session1 = createPeerSession(PEER1, nonAddPathParams, new SimpleSessionListener());
78
79         configurePeer(this.tableRegistry, PEER2, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
80        final BGPSessionImpl session2 = createPeerSession(PEER2, nonAddPathParams, new SimpleSessionListener());
81
82         configurePeer(this.tableRegistry, PEER3, this.ribImpl, nonAddPathParams, PeerRole.Ibgp, this.serverRegistry);
83         final BGPSessionImpl session3 = createPeerSession(PEER3, nonAddPathParams, new SimpleSessionListener());
84
85         final SimpleSessionListener listener4 = new SimpleSessionListener();
86         configurePeer(this.tableRegistry, PEER4, this.ribImpl, nonAddPathParams, PeerRole.RrClient,
87                 this.serverRegistry);
88         final BGPSessionImpl session4 = createPeerSession(PEER4, nonAddPathParams, listener4);
89
90         final SimpleSessionListener listener5 = new SimpleSessionListener();
91         configurePeer(this.tableRegistry, PEER5, this.ribImpl, addPathParams, PeerRole.RrClient, this.serverRegistry);
92         final BGPSessionImpl session5 = createPeerSession(PEER5, addPathParams, listener5);
93         checkPeersPresentOnDataStore(5);
94
95         //new best route so far
96         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 1);
97         checkReceivedMessages(listener4, 1);
98         checkReceivedMessages(listener5, 1);
99         assertEquals(UPD_100, listener5.getListMsg().get(0));
100
101         final SimpleSessionListener listener6 = new SimpleSessionListener();
102         configurePeer(this.tableRegistry, PEER6, this.ribImpl, nonAddPathParams, PeerRole.RrClient,
103                 this.serverRegistry);
104         final BGPSessionImpl session6 = createPeerSession(PEER6, nonAddPathParams, listener6);
105         checkPeersPresentOnDataStore(6);
106         checkReceivedMessages(listener6, 1);
107         assertEquals(UPD_NA_100, listener6.getListMsg().get(0));
108         session6.close();
109         checkPeersPresentOnDataStore(5);
110
111         //the second best route
112         sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 50, 2);
113         checkReceivedMessages(listener4, 1);
114         checkReceivedMessages(listener5, 2);
115         assertEquals(UPD_50, listener5.getListMsg().get(1));
116
117         //new best route
118         sendRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
119         checkReceivedMessages(listener4, 2);
120         checkReceivedMessages(listener5, 3);
121         assertEquals(UPD_200, listener5.getListMsg().get(2));
122
123         //the worst prefix, no changes
124         sendRouteAndCheckIsOnLocRib(session2, PREFIX1, 20, 2);
125         checkReceivedMessages(listener4, 2);
126         checkReceivedMessages(listener5, 3);
127
128         //withdraw second best route, 2 advertisement (1 withdrawal) for add-path supported, none for non add path
129         sendWithdrawalRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
130         checkReceivedMessages(listener4, 2);
131         checkReceivedMessages(listener5, 5);
132
133         //we advertise again to try new test
134         sendRouteAndCheckIsOnLocRib(session1, PREFIX1, 100, 2);
135         checkReceivedMessages(listener4, 2);
136         checkReceivedMessages(listener5, 6);
137
138         //withdraw second best route, 2 advertisement (1 withdrawal) for add-path supported,
139         // 1 withdrawal for non add path
140         sendWithdrawalRouteAndCheckIsOnLocRib(session3, PREFIX1, 200, 2);
141         checkReceivedMessages(listener4, 3);
142         checkReceivedMessages(listener5, 8);
143
144         session1.close();
145         session2.close();
146         session3.close();
147         session4.close();
148         session5.close();
149     }
150 }