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.config;
11 import static org.junit.Assert.assertEquals;
12 import static org.mockito.Matchers.any;
13 import static org.mockito.Matchers.anyString;
14 import static org.mockito.Matchers.eq;
15 import static org.mockito.Mockito.mock;
16 import static org.mockito.Mockito.times;
17 import static org.mockito.Mockito.verify;
18 import static org.opendaylight.controller.md.sal.binding.api.DataObjectModification.ModificationType.DELETE;
19 import static org.opendaylight.controller.md.sal.binding.api.DataObjectModification.ModificationType.WRITE;
20 import static org.opendaylight.protocol.bgp.rib.impl.config.AbstractConfig.TABLES_KEY;
21 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createAddPath;
22 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createConfig;
23 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createRR;
24 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createTimers;
25 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createTransport;
26 import static org.powermock.api.mockito.PowerMockito.spy;
27 import static org.powermock.api.mockito.PowerMockito.verifyPrivate;
29 import com.google.common.primitives.Shorts;
30 import com.google.common.util.concurrent.CheckedFuture;
31 import java.util.ArrayList;
32 import java.util.Collection;
33 import java.util.Collections;
34 import java.util.Dictionary;
35 import java.util.Optional;
36 import org.junit.Before;
37 import org.junit.Test;
38 import org.junit.runner.RunWith;
39 import org.mockito.Mock;
40 import org.mockito.Mockito;
41 import org.mockito.MockitoAnnotations;
42 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
43 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
44 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
45 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
46 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
47 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
48 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
49 import org.opendaylight.controller.md.sal.dom.api.DOMDataBrokerExtension;
50 import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService;
51 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
52 import org.opendaylight.controller.sal.core.api.model.SchemaService;
53 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTreeFactory;
54 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
55 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration;
56 import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
57 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
58 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
59 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
60 import org.opendaylight.protocol.bgp.rib.impl.spi.BgpDeployer;
61 import org.opendaylight.protocol.bgp.rib.impl.spi.InstanceType;
62 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionConsumerContext;
63 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
64 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
65 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafis;
66 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
67 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.Config;
68 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.ConfigBuilder;
69 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
70 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborBuilder;
71 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
72 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
73 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
74 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
75 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
76 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV4UNICAST;
77 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV6UNICAST;
78 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
79 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
80 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceBuilder;
81 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
82 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
83 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.ProtocolsBuilder;
84 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
85 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
86 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
87 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
88 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
89 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
90 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
91 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi1;
92 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi1Builder;
93 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi2;
94 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi2Builder;
95 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Protocol1;
96 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
97 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
98 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
99 import org.opendaylight.yangtools.concepts.ListenerRegistration;
100 import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
101 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
102 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
103 import org.osgi.framework.BundleContext;
104 import org.osgi.framework.ServiceRegistration;
105 import org.osgi.service.blueprint.container.BlueprintContainer;
106 import org.powermock.core.classloader.annotations.PrepareForTest;
107 import org.powermock.modules.junit4.PowerMockRunner;
109 @RunWith(PowerMockRunner.class)
110 @PrepareForTest({BgpDeployerImpl.class})
111 public class BgpDeployerImplTest {
112 private static final String NETWORK_INSTANCE_NAME = "network-test";
113 private static final AsNumber AS = new AsNumber(72L);
114 private static final IpAddress IPADDRESS = new IpAddress(new Ipv4Address("127.0.0.1"));
115 private static final BgpId BGP_ID = new BgpId(IPADDRESS.getIpv4Address());
116 private static final BgpTableType TABLE_TYPE = new BgpTableTypeImpl(Ipv4AddressFamily.class,
117 UnicastSubsequentAddressFamily.class);
119 private static final short SHORT = 0;
122 DataObjectModification dObject;
124 private BlueprintContainer blueprintContainer;
126 private BundleContext bundleContext;
128 private DataBroker dataBroker;
130 private BGPTableTypeRegistryConsumer tableTypeRegistry;
132 private WriteTransaction wTx;
134 private DataTreeModification<Bgp> modification;
136 private ListenerRegistration dataTreeRegistration;
138 private ServiceRegistration registration;
140 private Collection<DataTreeModification<Bgp>> collection = Collections.singleton(this.modification);
143 public void setUp() throws Exception {
144 MockitoAnnotations.initMocks(this);
145 final DOMDataBroker domDataBroker = mock(DOMDataBroker.class);
147 Mockito.doReturn(this.wTx).when(this.dataBroker).newWriteOnlyTransaction();
148 Mockito.doReturn("mapping").when(this.tableTypeRegistry).toString();
150 Mockito.doReturn(null).when(domDataBroker).createTransactionChain(any());
151 Mockito.doReturn(Collections.singletonMap(DOMDataTreeChangeService.class, mock(DOMDataBrokerExtension.class)))
152 .when(domDataBroker).getSupportedExtensions();
154 Mockito.doReturn(Optional.of(TABLE_TYPE)).when(this.tableTypeRegistry).getTableType(any());
155 Mockito.doReturn(Optional.of(TABLES_KEY)).when(this.tableTypeRegistry).getTableKey(any());
156 Mockito.doNothing().when(this.registration).unregister();
157 Mockito.doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any()
158 , any(Dictionary.class));
159 Mockito.doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any()
160 , any(Dictionary.class));
163 Mockito.doNothing().when(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(NetworkInstance.class));
164 final CheckedFuture future = mock(CheckedFuture.class);
165 Mockito.doReturn(future).when(this.wTx).submit();
166 Mockito.doNothing().when(future).addListener(any(), any());
167 Mockito.doReturn(this.dataTreeRegistration).when(this.dataBroker).registerDataTreeChangeListener(any(), any());
168 Mockito.doNothing().when(this.dataTreeRegistration).close();
170 final InstanceIdentifier<Bgp> bgpIID = InstanceIdentifier.create(NetworkInstances.class)
171 .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME)).child(Protocols.class)
172 .child(Protocol.class, new ProtocolKey(BGP.class, "bgp")).augmentation(Protocol1.class).child(Bgp.class);
174 Mockito.doReturn(new DataTreeIdentifier<>(LogicalDatastoreType.OPERATIONAL, bgpIID)).when(this.modification).getRootPath();
175 Mockito.doReturn(this.dObject).when(this.modification).getRootNode();
176 Mockito.doReturn("bgpPeer").when(this.modification).toString();
178 Mockito.doReturn(Collections.singleton(this.dObject)).when(this.dObject).getModifiedChildren();
179 Mockito.doReturn("dObject").when(this.dObject).toString();
181 final RIBExtensionConsumerContext extension = mock(RIBExtensionConsumerContext.class);
182 Mockito.doReturn(mock(GeneratedClassLoadingStrategy.class)).when(extension).getClassLoadingStrategy();
184 final ClusterSingletonServiceProvider singletonServiceProvider = mock(ClusterSingletonServiceProvider.class);
185 final ClusterSingletonServiceRegistration serviceRegistration = mock(ClusterSingletonServiceRegistration.class);
186 Mockito.doReturn(serviceRegistration).when(singletonServiceProvider).registerClusterSingletonService(any());
187 Mockito.doNothing().when(serviceRegistration).close();
189 final SchemaService schemaService = mock(SchemaService.class);
190 Mockito.doNothing().when(this.dataTreeRegistration).close();
192 Mockito.doReturn(this.dataTreeRegistration).when(schemaService).registerSchemaContextListener(any());
194 final RibImpl ribImpl = new RibImpl(singletonServiceProvider, extension,
195 mock(BGPDispatcher.class), mock(BindingCodecTreeFactory.class), domDataBroker, schemaService);
196 Mockito.doReturn(ribImpl).when(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
198 final BgpPeer bgpPeer = new BgpPeer(mock(RpcProviderRegistry.class), mock(BGPPeerRegistry.class));
199 Mockito.doReturn(bgpPeer).when(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
200 this.collection = Collections.singleton(this.modification);
204 public void testDeployerRib() throws Exception {
205 Mockito.doReturn(Global.class).when(this.dObject).getDataType();
206 final BgpDeployerImpl deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.blueprintContainer, this.bundleContext, this.dataBroker, this.tableTypeRegistry);
207 final BgpDeployerImpl spyDeployer = spy(deployer);
208 configureGlobal(IPV4UNICAST.class);
209 Mockito.doReturn(WRITE).when(this.dObject).getModificationType();
211 final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> networkInstanceIId = InstanceIdentifier.create(NetworkInstances.class)
212 .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
213 final NetworkInstance netII = new NetworkInstanceBuilder().setName(networkInstanceIId.firstKeyOf(NetworkInstance.class).getName())
214 .setProtocols(new ProtocolsBuilder().build()).build();
215 verify(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), Mockito.eq(netII));
216 verify(this.dataBroker).registerDataTreeChangeListener(Mockito.any(DataTreeIdentifier.class), Mockito.any(BgpDeployerImpl.class));
218 assertEquals(networkInstanceIId, spyDeployer.getInstanceIdentifier());
219 assertEquals(this.tableTypeRegistry, spyDeployer.getTableTypeRegistry());
221 spyDeployer.onDataTreeChanged(this.collection);
222 verifyPrivate(spyDeployer).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
223 verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
224 verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
225 verify(spyDeployer).onDataTreeChanged(any(Collection.class));
226 verify(spyDeployer).onGlobalModified(any(InstanceIdentifier.class), any(), any());
227 verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class), any(BgpDeployer.WriteConfiguration.class));
228 verifyPrivate(spyDeployer).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer
229 .WriteConfiguration.class));
230 verifyPrivate(spyDeployer).invoke("registerRibInstance", any(RibImpl.class), anyString());
231 verify(spyDeployer).onDataTreeChanged(any(Collection.class));
233 //change with same rib already existing
234 spyDeployer.onDataTreeChanged(this.collection);
235 verifyPrivate(spyDeployer, times(2)).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
236 verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
237 verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
238 verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
239 verify(spyDeployer, times(2)).onGlobalModified(any(InstanceIdentifier.class), any(), any());
240 verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class), any(BgpDeployer
241 .WriteConfiguration.class));
242 verifyPrivate(spyDeployer).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer
243 .WriteConfiguration.class));
244 verifyPrivate(spyDeployer).invoke("registerRibInstance", any(RibImpl.class), anyString());
245 verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
247 //Update for existing rib
248 configureGlobal(IPV6UNICAST.class);
249 spyDeployer.onDataTreeChanged(this.collection);
250 verifyPrivate(spyDeployer, times(3)).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
251 verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
252 verify(this.bundleContext, times(2)).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
253 verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
254 verify(spyDeployer, times(3)).onGlobalModified(any(InstanceIdentifier.class), any(), any());
255 verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class),
256 any(BgpDeployer.WriteConfiguration.class));
257 verifyPrivate(spyDeployer).invoke("onGlobalUpdated", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer.WriteConfiguration.class));
258 verify(this.dataTreeRegistration).close();
259 verify(this.registration).unregister();
260 verifyPrivate(spyDeployer).invoke("closeAllBindedPeers", any(InstanceIdentifier.class));
261 verifyPrivate(spyDeployer, times(2)).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any
262 (BgpDeployer.WriteConfiguration.class));
263 verifyPrivate(spyDeployer, times(2)).invoke("registerRibInstance", any(RibImpl.class), anyString());
264 verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
266 //Delete for existing rib
267 Mockito.doReturn(DELETE).when(this.dObject).getModificationType();
269 spyDeployer.onDataTreeChanged(this.collection);
270 verifyPrivate(spyDeployer, times(4)).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
271 verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
272 verify(this.bundleContext, times(2)).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
273 verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
274 verify(spyDeployer, times(3)).onGlobalModified(any(InstanceIdentifier.class), any(), any());
275 verify(spyDeployer).onGlobalRemoved(any(InstanceIdentifier.class));
276 verify(this.dataTreeRegistration, times(2)).close();
277 verify(this.registration, times(2)).unregister();
278 verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class),
279 any(BgpDeployer.WriteConfiguration.class));
280 verifyPrivate(spyDeployer).invoke("onGlobalUpdated", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer.WriteConfiguration.class));
281 verifyPrivate(spyDeployer).invoke("closeAllBindedPeers", any(InstanceIdentifier.class));
282 verifyPrivate(spyDeployer, times(2)).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any
283 (BgpDeployer.WriteConfiguration.class));
284 verifyPrivate(spyDeployer, times(2)).invoke("registerRibInstance", any(RibImpl.class), anyString());
285 verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
290 private void configureGlobal(final Class<? extends AfiSafiType> afi) {
291 final Config config = new ConfigBuilder().setAs(AS).setRouterId(BGP_ID).build();
292 final ArrayList<AfiSafi> afiSafiList = new ArrayList<>();
293 afiSafiList.add(new AfiSafiBuilder().setAfiSafiName(afi).addAugmentation(AfiSafi2.class, new AfiSafi2Builder().setReceive(true)
294 .setSendMax(Shorts.checkedCast(0L)).build()).build());
295 final AfiSafis afiSafi = new AfiSafisBuilder().setAfiSafi(afiSafiList).build();
296 Mockito.doReturn(new GlobalBuilder().setConfig(config).setAfiSafis(afiSafi).build()).when(this.dObject).getDataAfter();
303 public void testDeployerCreateNeighbor() throws Exception {
305 final BgpDeployerImpl deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.blueprintContainer, this.bundleContext, this.dataBroker, this.tableTypeRegistry);
306 final BgpDeployerImpl spyDeployer = spy(deployer);
309 Mockito.doReturn(Global.class).when(this.dObject).getDataType();
310 Mockito.doReturn(WRITE).when(this.dObject).getModificationType();
311 configureGlobal(IPV4UNICAST.class);
313 spyDeployer.onDataTreeChanged(this.collection);
314 verifyPrivate(spyDeployer).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
315 verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
316 verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
317 verify(spyDeployer).onDataTreeChanged(any(Collection.class));
318 verify(spyDeployer).onGlobalModified(any(InstanceIdentifier.class), any(), any());
319 verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class), any(BgpDeployer.WriteConfiguration.class));
320 verifyPrivate(spyDeployer).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer
321 .WriteConfiguration.class));
322 verifyPrivate(spyDeployer).invoke("registerRibInstance", any(RibImpl.class), anyString());
323 verify(spyDeployer).onDataTreeChanged(any(Collection.class));
326 Mockito.doReturn(Neighbors.class).when(this.dObject).getDataType();
327 Mockito.doReturn(WRITE).when(this.dObject).getModificationType();
328 configureNeighbor(IPV4UNICAST.class);
331 final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> networkInstanceIId = InstanceIdentifier.create(NetworkInstances.class)
332 .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
333 final NetworkInstance netII = new NetworkInstanceBuilder().setName(networkInstanceIId.firstKeyOf(NetworkInstance.class).getName())
334 .setProtocols(new ProtocolsBuilder().build()).build();
335 verify(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), Mockito.eq(netII));
336 verify(this.dataBroker).registerDataTreeChangeListener(Mockito.any(DataTreeIdentifier.class), Mockito.any(BgpDeployerImpl.class));
338 assertEquals(networkInstanceIId, spyDeployer.getInstanceIdentifier());
339 assertEquals(this.tableTypeRegistry, spyDeployer.getTableTypeRegistry());
341 spyDeployer.onDataTreeChanged(this.collection);
342 verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
343 verifyPrivate(spyDeployer).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
344 verify(spyDeployer).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
345 verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
346 verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
347 verifyPrivate(spyDeployer).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
348 any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
349 verifyPrivate(spyDeployer).invoke("registerPeerInstance", any(BgpPeer.class), anyString());
350 verify(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
352 //change with same peer already existing
353 spyDeployer.onDataTreeChanged(this.collection);
354 verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
355 verifyPrivate(spyDeployer, times(2)).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
356 verify(spyDeployer, times(2)).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
357 verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
358 verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
359 verifyPrivate(spyDeployer).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
360 any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
361 verifyPrivate(spyDeployer).invoke("registerPeerInstance", any(Neighbor.class), anyString());
362 verify(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
364 //Update for existing rib
365 configureNeighbor(IPV6UNICAST.class);
366 spyDeployer.onDataTreeChanged(this.collection);
367 verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
368 verifyPrivate(spyDeployer, times(3)).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
369 verify(spyDeployer, times(3)).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
370 verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
371 verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
372 verifyPrivate(spyDeployer, times(2)).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
373 any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
374 verifyPrivate(spyDeployer, times(2)).invoke("registerPeerInstance", any(Neighbor.class), anyString());
375 verify(this.bundleContext, times(2)).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
377 //Delete existing Peer
378 Mockito.doReturn(DELETE).when(this.dObject).getModificationType();
380 spyDeployer.onDataTreeChanged(this.collection);
381 verify(spyDeployer, times(5)).onDataTreeChanged(any(Collection.class));
382 verify( this.dObject).getDataBefore();
383 verifyPrivate(spyDeployer, times(4)).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
384 verify(spyDeployer, times(3)).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
385 verify(spyDeployer, times(1)).onNeighborRemoved(any(InstanceIdentifier.class), any(Neighbor.class));
386 verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
387 verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
388 verifyPrivate(spyDeployer, times(2)).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
389 any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
390 verifyPrivate(spyDeployer, times(2)).invoke("registerPeerInstance", any(Neighbor.class), anyString());
391 verify(this.bundleContext, times(2)).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
396 private void configureNeighbor(final Class<? extends AfiSafiType> afi) {
397 Mockito.doReturn(createNeighborExpected(afi)).when(this.dObject).getDataAfter();
398 Mockito.doReturn(createNeighborExpected(afi)).when(this.dObject).getDataBefore();
402 private Neighbor createNeighborExpected(final Class<? extends AfiSafiType> afi) {
403 return new NeighborBuilder()
404 .setAfiSafis(createAfiSafi(afi))
405 .setConfig(createConfig())
406 .setNeighborAddress(IPADDRESS)
407 .setRouteReflector(createRR())
408 .setTimers(createTimers())
409 .setTransport(createTransport())
410 .setAddPaths(createAddPath())
414 private static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafis createAfiSafi(final Class<? extends AfiSafiType> afi) {
415 return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder()
416 .setAfiSafi(Collections.singletonList(new AfiSafiBuilder().setAfiSafiName(afi)
417 .addAugmentation(AfiSafi1.class, new AfiSafi1Builder().setReceive(true).setSendMax(SHORT).build()).build())).build();