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.state;
11 import static org.junit.Assert.assertEquals;
12 import static org.mockito.Matchers.any;
13 import static org.mockito.Matchers.eq;
14 import static org.mockito.Mockito.doAnswer;
15 import static org.mockito.Mockito.doReturn;
16 import static org.opendaylight.protocol.util.CheckUtil.checkNotPresentOperational;
17 import static org.opendaylight.protocol.util.CheckUtil.readDataOperational;
19 import java.math.BigDecimal;
20 import java.math.BigInteger;
21 import java.util.ArrayList;
22 import java.util.Arrays;
23 import java.util.Collections;
24 import java.util.List;
25 import java.util.Optional;
26 import java.util.concurrent.atomic.LongAdder;
27 import org.junit.Assert;
28 import org.junit.Before;
29 import org.junit.Test;
30 import org.mockito.Mock;
31 import org.mockito.MockitoAnnotations;
32 import org.opendaylight.controller.md.sal.binding.test.AbstractConcurrentDataBrokerTest;
33 import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
34 import org.opendaylight.protocol.bgp.rib.spi.State;
35 import org.opendaylight.protocol.bgp.rib.spi.state.BGPAfiSafiState;
36 import org.opendaylight.protocol.bgp.rib.spi.state.BGPErrorHandlingState;
37 import org.opendaylight.protocol.bgp.rib.spi.state.BGPGracelfulRestartState;
38 import org.opendaylight.protocol.bgp.rib.spi.state.BGPPeerMessagesState;
39 import org.opendaylight.protocol.bgp.rib.spi.state.BGPPeerState;
40 import org.opendaylight.protocol.bgp.rib.spi.state.BGPRibState;
41 import org.opendaylight.protocol.bgp.rib.spi.state.BGPSessionState;
42 import org.opendaylight.protocol.bgp.rib.spi.state.BGPStateConsumer;
43 import org.opendaylight.protocol.bgp.rib.spi.state.BGPTimersState;
44 import org.opendaylight.protocol.bgp.rib.spi.state.BGPTransportState;
45 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
46 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
47 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.GracefulRestartBuilder;
48 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.graceful.restart.StateBuilder;
49 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.operational.rev151009.BgpNeighborState;
50 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.operational.rev151009.bgp.neighbor.prefix.counters_state.PrefixesBuilder;
51 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.graceful.restart.GracefulRestart;
52 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafis;
53 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder;
54 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ErrorHandling;
55 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ErrorHandlingBuilder;
56 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Timers;
57 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TimersBuilder;
58 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Transport;
59 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TransportBuilder;
60 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
61 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroup;
62 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroupBuilder;
63 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
64 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
65 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
66 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
67 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.ADDPATHS;
68 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.ASN32;
69 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.BgpCapability;
70 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.CommunityType;
71 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.GRACEFULRESTART;
72 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV4UNICAST;
73 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.MPBGP;
74 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.ROUTEREFRESH;
75 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
76 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
77 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
78 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
79 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
80 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
81 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
82 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
83 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
84 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
85 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
86 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Timeticks;
87 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.BgpNeighborStateAugmentation;
88 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.BgpNeighborStateAugmentationBuilder;
89 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.GlobalAfiSafiStateAugmentation;
90 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.GlobalAfiSafiStateAugmentationBuilder;
91 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborAfiSafiGracefulRestartStateAugmentation;
92 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborAfiSafiGracefulRestartStateAugmentationBuilder;
93 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborAfiSafiStateAugmentation;
94 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborAfiSafiStateAugmentationBuilder;
95 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborErrorHandlingStateAugmentation;
96 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborErrorHandlingStateAugmentationBuilder;
97 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborGracefulRestartStateAugmentation;
98 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborGracefulRestartStateAugmentationBuilder;
99 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborStateAugmentation;
100 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborTimersStateAugmentation;
101 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborTimersStateAugmentationBuilder;
102 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborTransportStateAugmentation;
103 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.NeighborTransportStateAugmentationBuilder;
104 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.PeerGroupStateAugmentation;
105 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.PeerGroupStateAugmentationBuilder;
106 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.Protocol1;
107 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.MessagesBuilder;
108 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.messages.ReceivedBuilder;
109 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.network.instances.network.instance.protocols.protocol.bgp.neighbors.neighbor.state.messages.SentBuilder;
110 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.BgpRib;
111 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.RibId;
112 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.Rib;
113 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.RibKey;
114 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey;
115 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
116 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
117 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
118 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
119 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
121 public class StateProviderImplTest extends AbstractConcurrentDataBrokerTest {
122 private final LongAdder totalPathsCounter = new LongAdder();
123 private final LongAdder totalPrefixesCounter = new LongAdder();
124 private final PortNumber localPort = new PortNumber(1790);
125 private final PortNumber remotePort = new PortNumber(179);
126 private final int restartTime = 15;
127 private final String ribId = "identifier-test";
128 private final InstanceIdentifier<Bgp> bgpInstanceIdentifier = InstanceIdentifier.create(NetworkInstances.class)
129 .child(NetworkInstance.class, new NetworkInstanceKey("global-bgp")).child(Protocols.class)
130 .child(Protocol.class, new ProtocolKey(BGP.class, this.ribId)).augmentation(Protocol1.class).child(Bgp.class);
131 static final TablesKey TABLES_KEY = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
132 private final AsNumber as = new AsNumber(72L);
133 private final BgpId bgpId = new BgpId("127.0.0.1");
134 private final IpAddress neighborAddress = new IpAddress(new Ipv4Address("127.0.0.2"));
135 private final List<Class<? extends BgpCapability>> supportedCap = Arrays.asList(ASN32.class, ROUTEREFRESH.class,
136 MPBGP.class, ADDPATHS.class, GRACEFULRESTART.class);
138 private BGPStateConsumer stateCollector;
140 private BGPTableTypeRegistryConsumer tableTypeRegistry;
142 private BGPRibState bgpRibState;
144 private BGPPeerState bgpPeerState;
146 private BGPSessionState bgpSessionState;
148 private BGPPeerMessagesState bgpPeerMessagesState;
150 private BGPTimersState timersState;
152 private BGPTransportState bgpTransportState;
154 private BGPErrorHandlingState bgpErrorHandlingState;
156 private BGPGracelfulRestartState bgpGracelfulRestartState;
158 private BGPAfiSafiState bgpAfiSafiState;
160 private final List<BGPPeerState> bgpPeerStates = new ArrayList<>();
161 private final List<BGPRibState> bgpRibStates = new ArrayList<>();
164 public void setUp() throws Exception {
165 MockitoAnnotations.initMocks(this);
167 doReturn(Optional.of(IPV4UNICAST.class))
168 .when(this.tableTypeRegistry).getAfiSafiType(eq(TABLES_KEY));
170 doReturn(this.bgpRibStates).when(this.stateCollector).getRibStats();
171 doReturn(this.bgpPeerStates).when(this.stateCollector).getPeerStats();
173 final KeyedInstanceIdentifier<Rib, RibKey> iid = InstanceIdentifier.create(BgpRib.class)
174 .child(Rib.class, new RibKey(new RibId(this.ribId)));
175 doReturn(iid).when(this.bgpRibState).getInstanceIdentifier();
176 doReturn(this.as).when(this.bgpRibState).getAs();
177 doReturn(this.bgpId).when(this.bgpRibState).getRouteId();
179 doAnswer(invocation -> this.totalPathsCounter.longValue())
180 .when(this.bgpRibState).getTotalPathsCount();
181 doAnswer(invocation -> this.totalPrefixesCounter.longValue())
182 .when(this.bgpRibState).getTotalPrefixesCount();
183 doAnswer(invocation -> this.totalPathsCounter.longValue())
184 .when(this.bgpRibState).getPathCount(eq(TABLES_KEY));
185 doAnswer(invocation -> this.totalPrefixesCounter.longValue())
186 .when(this.bgpRibState).getPrefixesCount(eq(TABLES_KEY));
187 doAnswer(invocation -> Collections.singletonMap(TABLES_KEY,
188 this.totalPrefixesCounter.longValue())).when(this.bgpRibState).getTablesPrefixesCount();
189 doAnswer(invocation -> Collections.singletonMap(TABLES_KEY,
190 this.totalPathsCounter.longValue())).when(this.bgpRibState).getPathsCount();
193 doReturn("test-group").when(this.bgpPeerState).getGroupId();
194 doReturn(iid).when(this.bgpPeerState).getInstanceIdentifier();
195 doAnswer(invocation -> this.totalPrefixesCounter.longValue()).when(this.bgpPeerState).getTotalPrefixes();
196 doAnswer(invocation -> this.totalPathsCounter.longValue()).when(this.bgpPeerState).getTotalPathsCount();
197 doReturn(this.neighborAddress).when(this.bgpPeerState).getNeighborAddress();
198 doReturn(this.bgpSessionState).when(this.bgpPeerState).getBGPSessionState();
199 doReturn(this.bgpPeerMessagesState).when(this.bgpPeerState).getBGPPeerMessagesState();
201 doReturn(1L).when(this.bgpPeerMessagesState).getNotificationMessagesReceivedCount();
202 doReturn(1L).when(this.bgpPeerMessagesState).getNotificationMessagesSentCount();
203 doReturn(1L).when(this.bgpPeerMessagesState).getUpdateMessagesReceivedCount();
204 doReturn(1L).when(this.bgpPeerMessagesState).getUpdateMessagesSentCount();
205 doReturn(State.UP).when(this.bgpSessionState).getSessionState();
206 doReturn(true).when(this.bgpSessionState).isAddPathCapabilitySupported();
207 doReturn(true).when(this.bgpSessionState).isAsn32CapabilitySupported();
208 doReturn(true).when(this.bgpSessionState).isGracefulRestartCapabilitySupported();
209 doReturn(true).when(this.bgpSessionState).isMultiProtocolCapabilitySupported();
210 doReturn(true).when(this.bgpSessionState).isRouterRefreshCapabilitySupported();
212 doReturn(this.timersState).when(this.bgpPeerState).getBGPTimersState();
213 doReturn(10L).when(this.timersState).getNegotiatedHoldTime();
214 doReturn(1L).when(this.timersState).getUpTime();
216 doReturn(this.bgpTransportState).when(this.bgpPeerState).getBGPTransportState();
217 doReturn(this.localPort).when(this.bgpTransportState).getLocalPort();
218 doReturn(this.neighborAddress).when(this.bgpTransportState).getRemoteAddress();
219 doReturn(this.remotePort).when(this.bgpTransportState).getRemotePort();
221 doReturn(this.bgpErrorHandlingState).when(this.bgpPeerState).getBGPErrorHandlingState();
222 doReturn(1L).when(this.bgpErrorHandlingState).getErroneousUpdateReceivedCount();
224 doReturn(this.bgpGracelfulRestartState).when(this.bgpPeerState).getBGPGracelfulRestart();
225 doReturn(true).when(this.bgpGracelfulRestartState).isGracefulRestartAdvertized(any());
226 doReturn(true).when(this.bgpGracelfulRestartState).isGracefulRestartReceived(any());
227 doReturn(true).when(this.bgpGracelfulRestartState).isLocalRestarting();
228 doReturn(true).when(this.bgpGracelfulRestartState).isPeerRestarting();
229 doReturn(this.restartTime).when(this.bgpGracelfulRestartState).getPeerRestartTime();
231 doReturn(this.bgpAfiSafiState).when(this.bgpPeerState).getBGPAfiSafiState();
232 doReturn(Collections.singleton(TABLES_KEY)).when(this.bgpAfiSafiState).getAfiSafisAdvertized();
233 doReturn(Collections.singleton(TABLES_KEY)).when(this.bgpAfiSafiState).getAfiSafisReceived();
234 doReturn(1L).when(this.bgpAfiSafiState).getPrefixesInstalledCount(any());
235 doReturn(2L).when(this.bgpAfiSafiState).getPrefixesReceivedCount(any());
236 doReturn(1L).when(this.bgpAfiSafiState).getPrefixesSentCount(any());
237 doReturn(true).when(this.bgpAfiSafiState).isAfiSafiSupported(any());
238 doReturn(true).when(this.bgpAfiSafiState).isGracefulRestartAdvertized(any());
239 doReturn(true).when(this.bgpAfiSafiState).isGracefulRestartReceived(any());
243 public void testActiveStateProvider() throws Exception {
244 doReturn(true).when(this.bgpRibState).isActive();
245 doReturn(true).when(this.bgpPeerState).isActive();
247 final StateProviderImpl stateProvider = new StateProviderImpl(getDataBroker(), 1, this.tableTypeRegistry,
248 this.stateCollector, "global-bgp");
249 stateProvider.init();
251 final Global globalExpected = buildGlobalExpected(0);
252 this.bgpRibStates.add(this.bgpRibState);
253 readDataOperational(getDataBroker(), this.bgpInstanceIdentifier, bgpRib -> {
254 final Global global = bgpRib.getGlobal();
255 assertEquals(globalExpected, global);
259 this.totalPathsCounter.increment();
260 this.totalPrefixesCounter.increment();
262 final Global globalExpected2 = buildGlobalExpected(1);
263 readDataOperational(getDataBroker(), this.bgpInstanceIdentifier, bgpRib -> {
264 final Global global = bgpRib.getGlobal();
265 assertEquals(globalExpected2, global);
269 this.totalPathsCounter.decrement();
270 this.totalPrefixesCounter.decrement();
272 final Global globalExpected3 = buildGlobalExpected(0);
273 readDataOperational(getDataBroker(), this.bgpInstanceIdentifier, bgpRib -> {
274 final Global global = bgpRib.getGlobal();
275 assertEquals(globalExpected3, global);
276 Assert.assertNull(bgpRib.getNeighbors());
277 Assert.assertNull(bgpRib.getPeerGroups());
281 this.bgpPeerStates.add(this.bgpPeerState);
282 final PeerGroup peerGroupExpected = buildGroupExpected();
284 this.totalPathsCounter.increment();
285 this.totalPrefixesCounter.increment();
287 final AfiSafis expectedAfiSafis = buildAfiSafis();
288 final ErrorHandling expectedErrorHandling = buildErrorHandling();
289 final GracefulRestart expectedGracefulRestart = buildGracefulRestart();
290 final Transport expectedTransport = buildTransport();
291 final Timers expectedTimers = buildTimers();
292 final BgpNeighborStateAugmentation expectedBgpNeighborState = buildBgpNeighborStateAugmentation();
294 readDataOperational(getDataBroker(), this.bgpInstanceIdentifier, bgpRib -> {
295 final Neighbors neighbors = bgpRib.getNeighbors();
296 Assert.assertNotNull(neighbors);
297 assertEquals(peerGroupExpected, bgpRib.getPeerGroups().getPeerGroup().get(0));
298 final Neighbor neighborResult = neighbors.getNeighbor().get(0);
299 assertEquals(this.neighborAddress, neighborResult.getNeighborAddress());
300 assertEquals(expectedAfiSafis, neighborResult.getAfiSafis());
301 assertEquals(expectedErrorHandling, neighborResult.getErrorHandling());
302 assertEquals(expectedGracefulRestart, neighborResult.getGracefulRestart());
303 assertEquals(expectedTransport, neighborResult.getTransport());
304 assertEquals(expectedTimers, neighborResult.getTimers());
305 final org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group
306 .State stateResult = neighborResult.getState();
307 assertEquals(expectedBgpNeighborState, stateResult.getAugmentation(BgpNeighborStateAugmentation.class));
308 assertEquals(BgpNeighborState.SessionState.ESTABLISHED, stateResult
309 .getAugmentation(NeighborStateAugmentation.class).getSessionState());
310 final List<Class<? extends BgpCapability>> supportedCapabilitiesResult = stateResult
311 .getAugmentation(NeighborStateAugmentation.class).getSupportedCapabilities();
312 Assert.assertTrue(supportedCapabilitiesResult.containsAll(this.supportedCap));
316 this.bgpRibStates.clear();
317 checkNotPresentOperational(getDataBroker(), this.bgpInstanceIdentifier);
319 stateProvider.close();
323 public void testInactiveStateProvider() throws Exception {
324 doReturn(false).when(this.bgpRibState).isActive();
325 doReturn(false).when(this.bgpPeerState).isActive();
327 final StateProviderImpl stateProvider = new StateProviderImpl(getDataBroker(), 1, this.tableTypeRegistry,
328 this.stateCollector, "global-bgp");
329 stateProvider.init();
331 this.bgpRibStates.add(this.bgpRibState);
332 checkNotPresentOperational(getDataBroker(), this.bgpInstanceIdentifier);
334 this.bgpPeerStates.add(this.bgpPeerState);
335 checkNotPresentOperational(getDataBroker(), this.bgpInstanceIdentifier);
337 this.bgpRibStates.clear();
338 checkNotPresentOperational(getDataBroker(), this.bgpInstanceIdentifier);
340 stateProvider.close();
343 private static BgpNeighborStateAugmentation buildBgpNeighborStateAugmentation() {
344 final BgpNeighborStateAugmentation augmentation = new BgpNeighborStateAugmentationBuilder()
345 .setMessages(new MessagesBuilder().setReceived(new ReceivedBuilder()
346 .setNOTIFICATION(BigInteger.ONE).setUPDATE(BigInteger.ONE).build())
347 .setSent(new SentBuilder().setNOTIFICATION(BigInteger.ONE).setUPDATE(BigInteger.ONE).build())
352 private static AfiSafis buildAfiSafis() {
353 final NeighborAfiSafiStateAugmentationBuilder neighborAfiSafiStateAugmentation =
354 new NeighborAfiSafiStateAugmentationBuilder().setActive(true).setPrefixes(
355 new PrefixesBuilder().setSent(1L).setReceived(2L).setInstalled(1L).build());
356 final AfiSafi afiSafi = new AfiSafiBuilder()
357 .setAfiSafiName(IPV4UNICAST.class)
358 .setGracefulRestart(new GracefulRestartBuilder().setState(new StateBuilder().setEnabled(false)
359 .addAugmentation(NeighborAfiSafiGracefulRestartStateAugmentation.class,
360 new NeighborAfiSafiGracefulRestartStateAugmentationBuilder()
361 .setAdvertised(true).setReceived(true).build())
363 .setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp
364 .common.afi.safi.list.afi.safi.StateBuilder().setEnabled(false).addAugmentation(
365 NeighborAfiSafiStateAugmentation.class, neighborAfiSafiStateAugmentation.build())
369 return new AfiSafisBuilder().setAfiSafi(Collections.singletonList(afiSafi)).build();
372 private static ErrorHandling buildErrorHandling() {
373 final ErrorHandling errorHandling = new ErrorHandlingBuilder().setState(
374 new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.error
375 .handling.StateBuilder().setTreatAsWithdraw(false)
376 .addAugmentation(NeighborErrorHandlingStateAugmentation.class,
377 new NeighborErrorHandlingStateAugmentationBuilder()
378 .setErroneousUpdateMessages(1L).build()).build()).build();
379 return errorHandling;
382 private static Timers buildTimers() {
383 final Timers timers = new TimersBuilder().setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang
384 .bgp.rev151009.bgp.neighbor.group.timers.StateBuilder()
385 .setConnectRetry(BigDecimal.valueOf(30))
386 .setHoldTime(BigDecimal.valueOf(90))
387 .setKeepaliveInterval(BigDecimal.valueOf(30))
388 .setMinimumAdvertisementInterval(BigDecimal.valueOf(30))
389 .addAugmentation(NeighborTimersStateAugmentation.class, new NeighborTimersStateAugmentationBuilder()
390 .setNegotiatedHoldTime(BigDecimal.TEN).setUptime(new Timeticks(1L)).build())
395 private Transport buildTransport() {
396 final Transport transport = new TransportBuilder().setState(new org.opendaylight.yang.gen.v1.http.openconfig
397 .net.yang.bgp.rev151009.bgp.neighbor.group.transport.StateBuilder()
398 .setMtuDiscovery(false)
399 .setPassiveMode(false)
400 .addAugmentation(NeighborTransportStateAugmentation.class,
401 new NeighborTransportStateAugmentationBuilder().setLocalPort(this.localPort)
402 .setRemotePort(this.remotePort)
403 .setRemoteAddress(this.neighborAddress).build())
408 private GracefulRestart buildGracefulRestart() {
409 final NeighborGracefulRestartStateAugmentationBuilder gracefulAugmentation
410 = new NeighborGracefulRestartStateAugmentationBuilder();
411 gracefulAugmentation.setPeerRestarting(false);
412 gracefulAugmentation.setLocalRestarting(false);
413 gracefulAugmentation.setPeerRestartTime(0);
414 gracefulAugmentation.setLocalRestarting(true)
415 .setPeerRestarting(true).setPeerRestartTime(this.restartTime);
416 final GracefulRestart gracefulRestart = new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp
417 .rev151009.bgp.graceful.restart.GracefulRestartBuilder().setState(new org.opendaylight.yang.gen.v1.http
418 .openconfig.net.yang.bgp.rev151009.bgp.graceful.restart.graceful.restart.StateBuilder()
419 .addAugmentation(NeighborGracefulRestartStateAugmentation.class,
420 gracefulAugmentation.build()).build()).build();
421 return gracefulRestart;
424 private Global buildGlobalExpected(final long prefixesAndPaths) {
425 return new GlobalBuilder()
426 .setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base
427 .StateBuilder().setRouterId(new Ipv4Address(this.bgpId.getValue()))
428 .setTotalPrefixes(prefixesAndPaths).setTotalPaths(prefixesAndPaths).setAs(this.as).build())
429 .setAfiSafis(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base
430 .AfiSafisBuilder().setAfiSafi(Collections.singletonList(new AfiSafiBuilder()
431 .setAfiSafiName(IPV4UNICAST.class).setState(new org.opendaylight.yang.gen.v1.http.openconfig
432 .net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.afi.safi.StateBuilder()
434 .addAugmentation(GlobalAfiSafiStateAugmentation.class,
435 new GlobalAfiSafiStateAugmentationBuilder()
436 .setTotalPaths(prefixesAndPaths).setTotalPrefixes(prefixesAndPaths)
437 .build()).build()).build()))
441 private static PeerGroup buildGroupExpected() {
442 return new PeerGroupBuilder().setPeerGroupName("test-group").setState(new org.opendaylight.yang.gen.v1.http
443 .openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.StateBuilder()
444 .setSendCommunity(CommunityType.NONE)
445 .setRouteFlapDamping(false)
446 .addAugmentation(PeerGroupStateAugmentation.class,
447 new PeerGroupStateAugmentationBuilder().setTotalPaths(1L).setTotalPrefixes(1L)