bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/config/BgpDeployerImplTest.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.config;
  10
  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.BgpPeerTest.createAddPath;
  21 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createConfig;
  22 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createRR;
  23 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createTimers;
  24 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createTransport;
  25 import static org.powermock.api.mockito.PowerMockito.spy;
  26 import static org.powermock.api.mockito.PowerMockito.verifyPrivate;
  27
  28 import com.google.common.primitives.Shorts;
  29 import com.google.common.util.concurrent.CheckedFuture;
  30 import java.math.BigDecimal;
  31 import java.util.ArrayList;
  32 import java.util.Collection;
  33 import java.util.Collections;
  34 import java.util.Dictionary;
  35 import java.util.List;
  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.mode.impl.add.all.paths.AllPathSelection;
  57 import org.opendaylight.protocol.bgp.openconfig.spi.BGPOpenConfigMappingService;
  58 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  59 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
  60 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
  61 import org.opendaylight.protocol.bgp.rib.impl.spi.BgpDeployer;
  62 import org.opendaylight.protocol.bgp.rib.impl.spi.InstanceType;
  63 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionConsumerContext;
  64 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
  65 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
  66 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafis;
  67 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
  68 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.Config;
  69 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.ConfigBuilder;
  70 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AddPaths;
  71 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AddPathsBuilder;
  72 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.RouteReflector;
  73 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.RouteReflectorBuilder;
  74 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Timers;
  75 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TimersBuilder;
  76 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Transport;
  77 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TransportBuilder;
  78 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
  79 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborBuilder;
  80 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
  81 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
  82 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
  83 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
  84 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
  85 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.CommunityType;
  86 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV4UNICAST;
  87 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV6UNICAST;
  88 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.PeerType;
  89 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.RrClusterIdTypeBuilder;
  90 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
  91 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
  92 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceBuilder;
  93 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
  94 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
  95 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.ProtocolsBuilder;
  96 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
  97 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
  98 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
  99 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
 100 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
 101 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
 102 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
 104 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.SendReceive;
 105 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
 106 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi1;
 107 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi1Builder;
 108 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi2;
 109 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi2Builder;
 110 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config1;
 111 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config1Builder;
 112 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Protocol1;
 113 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
 114 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
 115 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
 116 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
 117 import org.opendaylight.yangtools.concepts.ListenerRegistration;
 118 import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
 119 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 120 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
 121 import org.osgi.framework.BundleContext;
 122 import org.osgi.framework.ServiceRegistration;
 123 import org.osgi.service.blueprint.container.BlueprintContainer;
 124 import org.powermock.core.classloader.annotations.PrepareForTest;
 125 import org.powermock.modules.junit4.PowerMockRunner;
 126
 127 @RunWith(PowerMockRunner.class)
 128 @PrepareForTest({BgpDeployerImpl.class})
 129 public class BgpDeployerImplTest {
 130     private static final String NETWORK_INSTANCE_NAME = "network-test";
 131     private static final AsNumber AS = new AsNumber(72L);
 132     private static final IpAddress IPADDRESS = new IpAddress(new Ipv4Address("127.0.0.1"));
 133     private static final BgpId BGP_ID = new BgpId(IPADDRESS.getIpv4Address());
 134     private static final BgpTableType TABLE_TYPE = new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
 135     private static final short SHORT = 0;
 136
 137     @Mock
 138     DataObjectModification dObject;
 139     @Mock
 140     private BlueprintContainer blueprintContainer;
 141     @Mock
 142     private BundleContext bundleContext;
 143     @Mock
 144     private DataBroker dataBroker;
 145     @Mock
 146     private BGPOpenConfigMappingService mappingService;
 147     @Mock
 148     private WriteTransaction wTx;
 149     @Mock
 150     private DataTreeModification<Bgp> modification;
 151     @Mock
 152     private ListenerRegistration dataTreeRegistration;
 153     @Mock
 154     private ServiceRegistration registration;
 155
 156     private Collection<DataTreeModification<Bgp>> collection = Collections.singleton(modification);
 157
 158     @Before
 159     public void setUp() throws Exception {
 160         MockitoAnnotations.initMocks(this);
 161         final DOMDataBroker domDataBroker = mock(DOMDataBroker.class);
 162
 163         Mockito.doReturn(this.wTx).when(this.dataBroker).newWriteOnlyTransaction();
 164         Mockito.doReturn("mapping").when(this.mappingService).toString();
 165         Mockito.doReturn(false).when(this.mappingService).isApplicationPeer(any(Neighbor.class));
 166         Mockito.doReturn(PeerRole.RrClient).when(this.mappingService).toPeerRole(any(Neighbor.class));
 167         Mockito.doReturn(Collections.singletonList(new AddressFamiliesBuilder().setAfi(Ipv4AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class)
 168             .setSendReceive(SendReceive.Both).build())).when(this.mappingService).toAddPathCapability(any(List.class));
 169
 170         Mockito.doReturn(null).when(domDataBroker).createTransactionChain(any());
 171         Mockito.doReturn(Collections.singletonMap(DOMDataTreeChangeService.class, mock(DOMDataBrokerExtension.class)))
 172             .when(domDataBroker).getSupportedExtensions();
 173
 174         Mockito.doReturn(Collections.singletonList(TABLE_TYPE)).when(this.mappingService).toTableTypes(any());
 175         Mockito.doReturn(Collections.singletonMap(TABLE_TYPE, new AllPathSelection()))
 176             .when(this.mappingService).toPathSelectionMode(any());
 177         Mockito.doNothing().when(this.registration).unregister();
 178         Mockito.doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any()
 179             , any(Dictionary.class));
 180         Mockito.doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any()
 181             , any(Dictionary.class));
 182
 183
 184         Mockito.doNothing().when(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(NetworkInstance.class));
 185         final CheckedFuture future = mock(CheckedFuture.class);
 186         Mockito.doReturn(future).when(this.wTx).submit();
 187         Mockito.doNothing().when(future).addListener(any(), any());
 188         Mockito.doReturn(this.dataTreeRegistration).when(this.dataBroker).registerDataTreeChangeListener(any(), any());
 189         Mockito.doNothing().when(this.dataTreeRegistration).close();
 190
 191         final InstanceIdentifier<Bgp> bgpIID = InstanceIdentifier.create(NetworkInstances.class)
 192             .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME)).child(Protocols.class)
 193             .child(Protocol.class, new ProtocolKey(BGP.class, "bgp")).augmentation(Protocol1.class).child(Bgp.class);
 194
 195         Mockito.doReturn(new DataTreeIdentifier<>(LogicalDatastoreType.OPERATIONAL, bgpIID)).when(this.modification).getRootPath();
 196         Mockito.doReturn(this.dObject).when(this.modification).getRootNode();
 197         Mockito.doReturn("bgpPeer").when(this.modification).toString();
 198
 199         Mockito.doReturn(Collections.singleton(this.dObject)).when(this.dObject).getModifiedChildren();
 200         Mockito.doReturn("dObject").when(this.dObject).toString();
 201
 202         final RIBExtensionConsumerContext extension = mock(RIBExtensionConsumerContext.class);
 203         Mockito.doReturn(mock(GeneratedClassLoadingStrategy.class)).when(extension).getClassLoadingStrategy();
 204
 205         final ClusterSingletonServiceProvider singletonServiceProvider = mock(ClusterSingletonServiceProvider.class);
 206         final ClusterSingletonServiceRegistration serviceRegistration = mock(ClusterSingletonServiceRegistration.class);
 207         Mockito.doReturn(serviceRegistration).when(singletonServiceProvider).registerClusterSingletonService(any());
 208         Mockito.doNothing().when(serviceRegistration).close();
 209
 210         final SchemaService schemaService = mock(SchemaService.class);
 211         Mockito.doNothing().when(this.dataTreeRegistration).close();
 212
 213         Mockito.doReturn(this.dataTreeRegistration).when(schemaService).registerSchemaContextListener(any());
 214
 215         final RibImpl ribImpl = new RibImpl(singletonServiceProvider, extension,
 216             mock(BGPDispatcher.class), mock(BindingCodecTreeFactory.class), domDataBroker, schemaService);
 217         Mockito.doReturn(ribImpl).when(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
 218
 219         final BgpPeer bgpPeer = new BgpPeer(mock(RpcProviderRegistry.class), mock(BGPPeerRegistry.class));
 220         Mockito.doReturn(bgpPeer).when(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
 221         this.collection = Collections.singleton(modification);
 222     }
 223
 224     @Test
 225     public void testDeployerRib() throws Exception {
 226         Mockito.doReturn(Global.class).when(this.dObject).getDataType();
 227         final BgpDeployerImpl deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.blueprintContainer, this.bundleContext, this.dataBroker, this.mappingService);
 228         final BgpDeployerImpl spyDeployer = spy(deployer);
 229         configureGlobal(IPV4UNICAST.class);
 230         Mockito.doReturn(WRITE).when(this.dObject).getModificationType();
 231
 232         final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> networkInstanceIId = InstanceIdentifier.create(NetworkInstances.class)
 233             .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
 234         final NetworkInstance netII = new NetworkInstanceBuilder().setName(networkInstanceIId.firstKeyOf(NetworkInstance.class).getName())
 235             .setProtocols(new ProtocolsBuilder().build()).build();
 236         verify(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), Mockito.eq(netII));
 237         verify(this.dataBroker).registerDataTreeChangeListener(Mockito.any(DataTreeIdentifier.class), Mockito.any(BgpDeployerImpl.class));
 238
 239         assertEquals(networkInstanceIId, spyDeployer.getInstanceIdentifier());
 240         assertEquals(this.mappingService, spyDeployer.getMappingService());
 241
 242         spyDeployer.onDataTreeChanged(this.collection);
 243         verifyPrivate(spyDeployer).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 244         verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
 245         verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
 246         verify(spyDeployer).onDataTreeChanged(any(Collection.class));
 247         verify(spyDeployer).onGlobalModified(any(InstanceIdentifier.class), any(), any());
 248         verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class), any(BgpDeployer.WriteConfiguration.class));
 249         verifyPrivate(spyDeployer).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer
 250             .WriteConfiguration.class));
 251         verifyPrivate(spyDeployer).invoke("registerRibInstance", any(RibImpl.class), anyString());
 252         verify(spyDeployer).onDataTreeChanged(any(Collection.class));
 253
 254         //change with same rib already existing
 255         spyDeployer.onDataTreeChanged(this.collection);
 256         verifyPrivate(spyDeployer, times(2)).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 257         verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
 258         verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
 259         verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
 260         verify(spyDeployer, times(2)).onGlobalModified(any(InstanceIdentifier.class), any(), any());
 261         verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class), any(BgpDeployer
 262             .WriteConfiguration.class));
 263         verifyPrivate(spyDeployer).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer
 264             .WriteConfiguration.class));
 265         verifyPrivate(spyDeployer).invoke("registerRibInstance", any(RibImpl.class), anyString());
 266         verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
 267
 268         //Update for existing rib
 269         configureGlobal(IPV6UNICAST.class);
 270         spyDeployer.onDataTreeChanged(this.collection);
 271         verifyPrivate(spyDeployer, times(3)).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 272         verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
 273         verify(this.bundleContext, times(2)).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
 274         verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
 275         verify(spyDeployer, times(3)).onGlobalModified(any(InstanceIdentifier.class), any(), any());
 276         verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class),
 277             any(BgpDeployer.WriteConfiguration.class));
 278         verifyPrivate(spyDeployer).invoke("onGlobalUpdated", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer.WriteConfiguration.class));
 279         verify(this.dataTreeRegistration).close();
 280         verify(this.registration).unregister();
 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(3)).onDataTreeChanged(any(Collection.class));
 286
 287         //Delete for existing rib
 288         Mockito.doReturn(DELETE).when(this.dObject).getModificationType();
 289
 290         spyDeployer.onDataTreeChanged(this.collection);
 291         verifyPrivate(spyDeployer, times(4)).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 292         verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
 293         verify(this.bundleContext, times(2)).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
 294         verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
 295         verify(spyDeployer, times(3)).onGlobalModified(any(InstanceIdentifier.class), any(), any());
 296         verify(spyDeployer).onGlobalRemoved(any(InstanceIdentifier.class));
 297         verify(this.dataTreeRegistration, times(2)).close();
 298         verify(this.registration, times(2)).unregister();
 299         verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class),
 300             any(BgpDeployer.WriteConfiguration.class));
 301         verifyPrivate(spyDeployer).invoke("onGlobalUpdated", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer.WriteConfiguration.class));
 302         verifyPrivate(spyDeployer).invoke("closeAllBindedPeers", any(InstanceIdentifier.class));
 303         verifyPrivate(spyDeployer, times(2)).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any
 304             (BgpDeployer.WriteConfiguration.class));
 305         verifyPrivate(spyDeployer, times(2)).invoke("registerRibInstance", any(RibImpl.class), anyString());
 306         verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
 307
 308         deployer.close();
 309     }
 310
 311     private void configureGlobal(final Class<? extends AfiSafiType> afi) {
 312         final Config config = new ConfigBuilder().setAs(AS).setRouterId(BGP_ID).build();
 313         final ArrayList<AfiSafi> afiSafiList = new ArrayList<>();
 314         afiSafiList.add(new AfiSafiBuilder().setAfiSafiName(afi).addAugmentation(AfiSafi2.class, new AfiSafi2Builder().setReceive(true)
 315             .setSendMax(Shorts.checkedCast(0L)).build()).build());
 316         final AfiSafis afiSafi = new AfiSafisBuilder().setAfiSafi(afiSafiList).build();
 317         Mockito.doReturn(new GlobalBuilder().setConfig(config).setAfiSafis(afiSafi).build()).when(this.dObject).getDataAfter();
 318     }
 319
 320     /**
 321      * Test create Rib
 322      */
 323     @Test
 324     public void testDeployerCreateNeighbor() throws Exception {
 325
 326         final BgpDeployerImpl deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.blueprintContainer, this.bundleContext, this.dataBroker, this.mappingService);
 327         final BgpDeployerImpl spyDeployer = spy(deployer);
 328
 329         //First create Rib
 330         Mockito.doReturn(Global.class).when(this.dObject).getDataType();
 331         Mockito.doReturn(WRITE).when(this.dObject).getModificationType();
 332         configureGlobal(IPV4UNICAST.class);
 333
 334         spyDeployer.onDataTreeChanged(this.collection);
 335         verifyPrivate(spyDeployer).invoke("onGlobalChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 336         verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
 337         verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
 338         verify(spyDeployer).onDataTreeChanged(any(Collection.class));
 339         verify(spyDeployer).onGlobalModified(any(InstanceIdentifier.class), any(), any());
 340         verifyPrivate(spyDeployer).invoke("onGlobalCreated", any(InstanceIdentifier.class), any(Global.class), any(BgpDeployer.WriteConfiguration.class));
 341         verifyPrivate(spyDeployer).invoke("initiateRibInstance", any(InstanceIdentifier.class), any(Global.class), any(RibImpl.class), any(BgpDeployer
 342             .WriteConfiguration.class));
 343         verifyPrivate(spyDeployer).invoke("registerRibInstance", any(RibImpl.class), anyString());
 344         verify(spyDeployer).onDataTreeChanged(any(Collection.class));
 345
 346
 347         Mockito.doReturn(Neighbors.class).when(this.dObject).getDataType();
 348         Mockito.doReturn(WRITE).when(this.dObject).getModificationType();
 349         configureNeighbor(IPV4UNICAST.class);
 350
 351
 352         final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> networkInstanceIId = InstanceIdentifier.create(NetworkInstances.class)
 353             .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
 354         final NetworkInstance netII = new NetworkInstanceBuilder().setName(networkInstanceIId.firstKeyOf(NetworkInstance.class).getName())
 355             .setProtocols(new ProtocolsBuilder().build()).build();
 356         verify(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), Mockito.eq(netII));
 357         verify(this.dataBroker).registerDataTreeChangeListener(Mockito.any(DataTreeIdentifier.class), Mockito.any(BgpDeployerImpl.class));
 358
 359         assertEquals(networkInstanceIId, spyDeployer.getInstanceIdentifier());
 360         assertEquals(this.mappingService, spyDeployer.getMappingService());
 361
 362         spyDeployer.onDataTreeChanged(this.collection);
 363         verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
 364         verifyPrivate(spyDeployer).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 365         verify(spyDeployer).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
 366         verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
 367         verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
 368         verifyPrivate(spyDeployer).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
 369             any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
 370         verifyPrivate(spyDeployer).invoke("registerPeerInstance", any(Neighbor.class), anyString());
 371         verify(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
 372
 373         //change with same peer already existing
 374         spyDeployer.onDataTreeChanged(this.collection);
 375         verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
 376         verifyPrivate(spyDeployer, times(2)).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 377         verify(spyDeployer, times(2)).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
 378         verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
 379         verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
 380         verifyPrivate(spyDeployer).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
 381             any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
 382         verifyPrivate(spyDeployer).invoke("registerPeerInstance", any(Neighbor.class), anyString());
 383         verify(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
 384
 385         //Update for existing rib
 386         configureNeighbor(IPV6UNICAST.class);
 387         spyDeployer.onDataTreeChanged(this.collection);
 388         verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
 389         verifyPrivate(spyDeployer, times(3)).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 390         verify(spyDeployer, times(3)).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
 391         verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
 392         verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
 393         verifyPrivate(spyDeployer, times(2)).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
 394             any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
 395         verifyPrivate(spyDeployer, times(2)).invoke("registerPeerInstance", any(Neighbor.class), anyString());
 396         verify(this.bundleContext, times(2)).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
 397
 398        //Delete existing Peer
 399         Mockito.doReturn(DELETE).when(this.dObject).getModificationType();
 400
 401         spyDeployer.onDataTreeChanged(this.collection);
 402         verify(spyDeployer, times(5)).onDataTreeChanged(any(Collection.class));
 403         verify( this.dObject).getDataBefore();
 404         verifyPrivate(spyDeployer, times(4)).invoke("onNeighborsChanged", any(DataObjectModification.class), any(InstanceIdentifier.class));
 405         verify(spyDeployer, times(3)).onNeighborModified(any(InstanceIdentifier.class), any(Neighbor.class), any());
 406         verify(spyDeployer, times(1)).onNeighborRemoved(any(InstanceIdentifier.class), any(Neighbor.class));
 407         verifyPrivate(spyDeployer).invoke("onNeighborCreated", any(InstanceIdentifier.class), any(Neighbor.class), any(BgpDeployer.WriteConfiguration.class));
 408         verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
 409         verifyPrivate(spyDeployer, times(2)).invoke("initiatePeerInstance", any(InstanceIdentifier.class), any(InstanceIdentifier.class), any(Neighbor.class),
 410             any(PeerBean.class), any(BgpDeployer.WriteConfiguration.class));
 411         verifyPrivate(spyDeployer, times(2)).invoke("registerPeerInstance", any(Neighbor.class), anyString());
 412         verify(this.bundleContext, times(2)).registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
 413
 414         deployer.close();
 415     }
 416
 417     private void configureNeighbor(final Class<? extends AfiSafiType> afi) {
 418         Mockito.doReturn(createNeighborExpected(afi)).when(this.dObject).getDataAfter();
 419         Mockito.doReturn(createNeighborExpected(afi)).when(this.dObject).getDataBefore();
 420     }
 421
 422
 423    private Neighbor createNeighborExpected(final Class<? extends AfiSafiType> afi) {
 424         return new NeighborBuilder()
 425             .setAfiSafis(createAfiSafi(afi))
 426             .setConfig(createConfig())
 427             .setNeighborAddress(IPADDRESS)
 428             .setRouteReflector(createRR())
 429             .setTimers(createTimers())
 430             .setTransport(createTransport())
 431             .setAddPaths(createAddPath())
 432             .build();
 433     }
 434
 435     private static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafis createAfiSafi(final Class<? extends AfiSafiType> afi) {
 436         return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder()
 437             .setAfiSafi(Collections.singletonList(new AfiSafiBuilder().setAfiSafiName(afi)
 438             .addAugmentation(AfiSafi1.class, new AfiSafi1Builder().setReceive(true).setSendMax(SHORT).build()).build())).build();
 439     }
 440 }