package org.opendaylight.protocol.bgp.rib.impl.config;
-import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
-import static org.opendaylight.controller.md.sal.binding.api.DataObjectModification.ModificationType.DELETE;
-import static org.opendaylight.controller.md.sal.binding.api.DataObjectModification.ModificationType.WRITE;
import static org.opendaylight.protocol.bgp.rib.impl.config.AbstractConfig.TABLES_KEY;
-import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createAddPath;
-import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createConfig;
-import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createRR;
-import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createTimers;
-import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createTransport;
-import static org.powermock.api.mockito.PowerMockito.spy;
-import static org.powermock.api.mockito.PowerMockito.verifyPrivate;
-
-import com.google.common.primitives.Shorts;
-import com.google.common.util.concurrent.CheckedFuture;
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.Collections;
+import static org.opendaylight.protocol.bgp.rib.impl.config.RIBTestsUtil.createGlobalIpv4;
+import static org.opendaylight.protocol.bgp.rib.impl.config.RIBTestsUtil.createGlobalIpv6;
+import static org.opendaylight.protocol.bgp.rib.impl.config.RIBTestsUtil.createNeighbors;
+import static org.opendaylight.protocol.bgp.rib.impl.config.RIBTestsUtil.createNeighborsNoRR;
+import static org.opendaylight.protocol.util.CheckUtil.checkPresentConfiguration;
+
import java.util.Dictionary;
import java.util.Optional;
+import java.util.concurrent.ExecutionException;
import org.junit.Before;
import org.junit.Test;
-import org.junit.runner.RunWith;
import org.mockito.Mock;
-import org.mockito.Mockito;
-import org.mockito.MockitoAnnotations;
-import org.opendaylight.controller.md.sal.binding.api.DataBroker;
-import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
-import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataBrokerExtension;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService;
import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTreeFactory;
import org.opendaylight.mdsal.binding.generator.impl.GeneratedClassLoadingStrategy;
import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration;
import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
+import org.opendaylight.protocol.bgp.rib.impl.DefaultRibPoliciesMockTest;
import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
import org.opendaylight.protocol.bgp.rib.impl.spi.InstanceType;
import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionConsumerContext;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafis;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.Config;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.ConfigBuilder;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV4UNICAST;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV6UNICAST;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
-import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.ProtocolsBuilder;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.BgpTableType;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.AfiSafi1;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.AfiSafi1Builder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.AfiSafi2;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.AfiSafi2Builder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev171207.Protocol1;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
import org.opendaylight.yangtools.concepts.ListenerRegistration;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;
import org.osgi.service.blueprint.container.BlueprintContainer;
-import org.powermock.core.classloader.annotations.PrepareForTest;
-import org.powermock.modules.junit4.PowerMockRunner;
-@RunWith(PowerMockRunner.class)
-@PrepareForTest({BgpDeployerImpl.class})
-public class BgpDeployerImplTest {
- private static final String NETWORK_INSTANCE_NAME = "network-test";
- private static final AsNumber AS = new AsNumber(72L);
- private static final IpAddress IPADDRESS = new IpAddress(new Ipv4Address("127.0.0.1"));
- private static final BgpId BGP_ID = new BgpId(IPADDRESS.getIpv4Address());
+public class BgpDeployerImplTest extends DefaultRibPoliciesMockTest {
private static final BgpTableType TABLE_TYPE = new BgpTableTypeImpl(Ipv4AddressFamily.class,
UnicastSubsequentAddressFamily.class);
+ private static final String NETWORK_INSTANCE_NAME = "network-test";
+ private static final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> NETWORK_II =
+ InstanceIdentifier.create(NetworkInstances.class)
+ .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
+ private static final String KEY = "bgp";
+ private static final InstanceIdentifier<Bgp> BGP_II = NETWORK_II.child(Protocols.class)
+ .child(Protocol.class, new ProtocolKey(BGP.class, KEY))
+ .augmentation(Protocol1.class).child(Bgp.class);
+ private static final InstanceIdentifier<Global> GLOBAL_II = BGP_II.child(Global.class);
+ private static final InstanceIdentifier<Neighbors> NEIGHBORS_II = BGP_II.child(Neighbors.class);
- private static final short SHORT = 0;
-
- @Mock
- private DataObjectModification<?> dObject;
@Mock
private BlueprintContainer blueprintContainer;
@Mock
private BundleContext bundleContext;
@Mock
- private DataBroker dataBroker;
- @Mock
private BGPTableTypeRegistryConsumer tableTypeRegistry;
@Mock
- private WriteTransaction wTx;
- @Mock
private DataTreeModification<Bgp> modification;
@Mock
private ListenerRegistration<?> dataTreeRegistration;
private ServiceRegistration<?> registration;
@Mock
private ClusterSingletonServiceProvider singletonServiceProvider;
-
- private Collection<DataTreeModification<Bgp>> collection = Collections.singleton(this.modification);
-
- private static Neighbor createNeighborExpected(final Class<? extends AfiSafiType> afi) {
- return new NeighborBuilder()
- .setAfiSafis(createAfiSafi(afi))
- .setConfig(createConfig())
- .setNeighborAddress(IPADDRESS)
- .setRouteReflector(createRR())
- .setTimers(createTimers())
- .setTransport(createTransport())
- .setAddPaths(createAddPath())
- .build();
- }
-
- private static org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafis createAfiSafi(final Class<? extends AfiSafiType> afi) {
- return new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder()
- .setAfiSafi(Collections.singletonList(new AfiSafiBuilder().setAfiSafiName(afi)
- .addAugmentation(AfiSafi1.class, new AfiSafi1Builder().setReceive(true).setSendMax(SHORT).build()).build())).build();
- }
+ private BgpDeployerImpl deployer;
@Before
public void setUp() throws Exception {
- MockitoAnnotations.initMocks(this);
- final DOMDataBroker domDataBroker = mock(DOMDataBroker.class);
+ super.setUp();
- doReturn(this.wTx).when(this.dataBroker).newWriteOnlyTransaction();
doReturn("mapping").when(this.tableTypeRegistry).toString();
-
- doReturn(null).when(domDataBroker).createTransactionChain(any());
- doReturn(Collections.singletonMap(DOMDataTreeChangeService.class, mock(DOMDataBrokerExtension.class)))
- .when(domDataBroker).getSupportedExtensions();
-
doReturn(Optional.of(TABLE_TYPE)).when(this.tableTypeRegistry).getTableType(any());
doReturn(Optional.of(TABLES_KEY)).when(this.tableTypeRegistry).getTableKey(any());
- Mockito.doNothing().when(this.registration).unregister();
- doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any()
- , any(Dictionary.class));
- doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()), any()
- , any(Dictionary.class));
-
-
- Mockito.doNothing().when(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class),
- any(NetworkInstance.class));
- final CheckedFuture<?, ?> future = mock(CheckedFuture.class);
- doReturn(future).when(this.wTx).submit();
- Mockito.doNothing().when(future).addListener(any(), any());
- doReturn(this.dataTreeRegistration).when(this.dataBroker).registerDataTreeChangeListener(any(), any());
- Mockito.doNothing().when(this.dataTreeRegistration).close();
-
- final InstanceIdentifier<Bgp> bgpIID = InstanceIdentifier.create(NetworkInstances.class)
- .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME)).child(Protocols.class)
- .child(Protocol.class, new ProtocolKey(BGP.class, "bgp"))
- .augmentation(Protocol1.class).child(Bgp.class);
-
- doReturn(new DataTreeIdentifier<>(LogicalDatastoreType.OPERATIONAL, bgpIID))
- .when(this.modification).getRootPath();
- doReturn(this.dObject).when(this.modification).getRootNode();
- doReturn("bgpPeer").when(this.modification).toString();
-
- doReturn(Collections.singleton(this.dObject)).when(this.dObject).getModifiedChildren();
- doReturn("dObject").when(this.dObject).toString();
+ doNothing().when(this.registration).unregister();
+ doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()),
+ any(), any(Dictionary.class));
+ doReturn(this.registration).when(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()),
+ any(), any(Dictionary.class));
+ doNothing().when(this.dataTreeRegistration).close();
+ doReturn("bgpPeer").when(this.modification).toString();
final RIBExtensionConsumerContext extension = mock(RIBExtensionConsumerContext.class);
- doReturn(mock(GeneratedClassLoadingStrategy.class)).when(extension).getClassLoadingStrategy();
+ doReturn(GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy()).when(extension).getClassLoadingStrategy();
final ClusterSingletonServiceRegistration serviceRegistration = mock(ClusterSingletonServiceRegistration.class);
doReturn(serviceRegistration).when(this.singletonServiceProvider).registerClusterSingletonService(any());
- Mockito.doNothing().when(serviceRegistration).close();
+ doNothing().when(serviceRegistration).close();
final DOMSchemaService schemaService = mock(DOMSchemaService.class);
- Mockito.doNothing().when(this.dataTreeRegistration).close();
+ doNothing().when(this.dataTreeRegistration).close();
doReturn(this.dataTreeRegistration).when(schemaService).registerSchemaContextListener(any());
- final RibImpl ribImpl = new RibImpl(extension, mock(BGPDispatcher.class), mock(BindingCodecTreeFactory.class),
- domDataBroker, schemaService);
+ final RibImpl ribImpl = new RibImpl(extension, mock(BGPDispatcher.class), this.policyProvider,
+ mock(BindingCodecTreeFactory.class), getDomBroker(), schemaService, this.mappingService);
doReturn(ribImpl).when(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
- final BgpPeer bgpPeer = new BgpPeer(mock(RpcProviderRegistry.class));
- doReturn(bgpPeer).when(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
- this.collection = Collections.singleton(this.modification);
+ doReturn(new BgpPeer(mock(RpcProviderRegistry.class)))
+ .when(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
+
+ this.deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.singletonServiceProvider,
+ this.blueprintContainer, this.bundleContext, getDataBroker(), this.tableTypeRegistry);
}
@Test
public void testDeployerRib() throws Exception {
- doReturn(Global.class).when(this.dObject).getDataType();
- final BgpDeployerImpl deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.singletonServiceProvider,
- this.blueprintContainer, this.bundleContext, this.dataBroker, this.tableTypeRegistry);
- final BgpDeployerImpl spyDeployer = spy(deployer);
deployer.init();
- configureGlobal(IPV4UNICAST.class);
- doReturn(WRITE).when(this.dObject).getModificationType();
-
- final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> networkInstanceIId =
- InstanceIdentifier.create(NetworkInstances.class)
- .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
- final NetworkInstance netII = new NetworkInstanceBuilder()
- .setName(networkInstanceIId.firstKeyOf(NetworkInstance.class).getName())
- .setProtocols(new ProtocolsBuilder().build()).build();
- verify(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), eq(netII));
- verify(this.dataBroker).registerDataTreeChangeListener(any(DataTreeIdentifier.class),
- any(BgpDeployerImpl.class));
-
- assertEquals(this.tableTypeRegistry, spyDeployer.getTableTypeRegistry());
-
- spyDeployer.onDataTreeChanged(this.collection);
- verifyPrivate(spyDeployer).invoke("onGlobalChanged", any(DataObjectModification.class),
- any(InstanceIdentifier.class));
+ checkPresentConfiguration(getDataBroker(), NETWORK_II);
+ createRib(createGlobalIpv4());
+
verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
- verify(spyDeployer).onDataTreeChanged(any(Collection.class));
//change with same rib already existing
- spyDeployer.onDataTreeChanged(this.collection);
- verifyPrivate(spyDeployer, times(2)).invoke("onGlobalChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
+ createRib(createGlobalIpv4());
verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
- verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
//Update for existing rib
- configureGlobal(IPV6UNICAST.class);
- spyDeployer.onDataTreeChanged(this.collection);
- verifyPrivate(spyDeployer, times(3)).invoke("onGlobalChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
+ createRib(createGlobalIpv6());
+
verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
verify(this.bundleContext, times(2)).registerService(eq(InstanceType.RIB.getServices()),
any(), any(Dictionary.class));
- verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
verify(this.dataTreeRegistration).close();
verify(this.registration).unregister();
- verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
//Delete for existing rib
- doReturn(DELETE).when(this.dObject).getModificationType();
+ deleteRib();
- spyDeployer.onDataTreeChanged(this.collection);
- verifyPrivate(spyDeployer, times(4)).invoke("onGlobalChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
verify(this.bundleContext, times(2))
.registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
- verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
verify(this.dataTreeRegistration, times(2)).close();
verify(this.registration, times(2)).unregister();
- verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
deployer.close();
}
- private void configureGlobal(final Class<? extends AfiSafiType> afi) {
- final Config config = new ConfigBuilder().setAs(AS).setRouterId(BGP_ID).build();
- final ArrayList<AfiSafi> afiSafiList = new ArrayList<>();
- afiSafiList.add(new AfiSafiBuilder().setAfiSafiName(afi).addAugmentation(AfiSafi2.class, new AfiSafi2Builder().setReceive(true)
- .setSendMax(Shorts.checkedCast(0L)).build()).build());
- final AfiSafis afiSafi = new AfiSafisBuilder().setAfiSafi(afiSafiList).build();
- doReturn(new GlobalBuilder().setConfig(config).setAfiSafis(afiSafi).build()).when(this.dObject).getDataAfter();
- }
-
/**
* Test create Rib
*/
@Test
public void testDeployerCreateNeighbor() throws Exception {
-
- final BgpDeployerImpl deployer = new BgpDeployerImpl(NETWORK_INSTANCE_NAME, this.singletonServiceProvider,
- this.blueprintContainer, this.bundleContext, this.dataBroker, this.tableTypeRegistry);
- final BgpDeployerImpl spyDeployer = spy(deployer);
deployer.init();
- //First create Rib
- doReturn(Global.class).when(this.dObject).getDataType();
- doReturn(WRITE).when(this.dObject).getModificationType();
- configureGlobal(IPV4UNICAST.class);
-
- spyDeployer.onDataTreeChanged(this.collection);
- verifyPrivate(spyDeployer).invoke("onGlobalChanged", any(DataObjectModification.class),
- any(InstanceIdentifier.class));
- verify(this.blueprintContainer).getComponentInstance(eq("ribImpl"));
- verify(this.bundleContext).registerService(eq(InstanceType.RIB.getServices()), any(), any(Dictionary.class));
- verify(spyDeployer).onDataTreeChanged(any(Collection.class));
-
- doReturn(Neighbors.class).when(this.dObject).getDataType();
- doReturn(WRITE).when(this.dObject).getModificationType();
- configureNeighbor(IPV4UNICAST.class);
-
- final KeyedInstanceIdentifier<NetworkInstance, NetworkInstanceKey> networkInstanceIId =
- InstanceIdentifier.create(NetworkInstances.class)
- .child(NetworkInstance.class, new NetworkInstanceKey(NETWORK_INSTANCE_NAME));
- final NetworkInstance netII = new NetworkInstanceBuilder()
- .setName(networkInstanceIId.firstKeyOf(NetworkInstance.class).getName())
- .setProtocols(new ProtocolsBuilder().build()).build();
- verify(this.wTx).merge(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), eq(netII));
- verify(this.dataBroker)
- .registerDataTreeChangeListener(any(DataTreeIdentifier.class), any(BgpDeployerImpl.class));
-
- assertEquals(this.tableTypeRegistry, spyDeployer.getTableTypeRegistry());
-
- spyDeployer.onDataTreeChanged(this.collection);
- verify(spyDeployer, times(2)).onDataTreeChanged(any(Collection.class));
- verifyPrivate(spyDeployer).invoke("onNeighborsChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
+ checkPresentConfiguration(getDataBroker(), NETWORK_II);
+
+ createRib(createGlobalIpv4());
+ createNeighbor(createNeighbors());
verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
verify(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()),
any(BgpPeer.class), any(Dictionary.class));
//change with same peer already existing
- spyDeployer.onDataTreeChanged(this.collection);
- verify(spyDeployer, times(3)).onDataTreeChanged(any(Collection.class));
- verifyPrivate(spyDeployer, times(2)).invoke("onNeighborsChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
+ createNeighbor(createNeighbors());
verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
verify(this.bundleContext).registerService(eq(InstanceType.PEER.getServices()),
any(BgpPeer.class), any(Dictionary.class));
- //Update for existing rib
- configureNeighbor(IPV6UNICAST.class);
- spyDeployer.onDataTreeChanged(this.collection);
- verify(spyDeployer, times(4)).onDataTreeChanged(any(Collection.class));
- verifyPrivate(spyDeployer, times(3)).invoke("onNeighborsChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
+ //Update for peer
+ createNeighbor(createNeighborsNoRR());
+
verify(this.blueprintContainer).getComponentInstance(eq("bgpPeer"));
verify(this.bundleContext, times(2))
.registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
+ verify(this.registration).unregister();
+ deleteNeighbors();
//Delete existing Peer
- doReturn(DELETE).when(this.dObject).getModificationType();
-
- spyDeployer.onDataTreeChanged(this.collection);
- verify(spyDeployer, times(5)).onDataTreeChanged(any(Collection.class));
- verify(this.dObject).getDataBefore();
- verifyPrivate(spyDeployer, times(4)).invoke("onNeighborsChanged",
- any(DataObjectModification.class), any(InstanceIdentifier.class));
verify(this.bundleContext, times(2))
.registerService(eq(InstanceType.PEER.getServices()), any(BgpPeer.class), any(Dictionary.class));
+ verify(this.registration, times(2)).unregister();
deployer.close();
}
- private void configureNeighbor(final Class<? extends AfiSafiType> afi) {
- doReturn(createNeighborExpected(afi)).when(this.dObject).getDataAfter();
- doReturn(createNeighborExpected(afi)).when(this.dObject).getDataBefore();
+ private void createRib(final Global global) throws ExecutionException, InterruptedException {
+ final WriteTransaction wr = getDataBroker().newWriteOnlyTransaction();
+ wr.put(LogicalDatastoreType.CONFIGURATION, GLOBAL_II, global, true);
+ wr.submit().get();
+ }
+
+ private void deleteRib() throws ExecutionException, InterruptedException {
+ final WriteTransaction wr = getDataBroker().newWriteOnlyTransaction();
+ wr.delete(LogicalDatastoreType.CONFIGURATION, BGP_II);
+ wr.submit().get();
+ }
+
+ private void createNeighbor(final Neighbors neighbors) throws ExecutionException, InterruptedException {
+ final WriteTransaction wr = getDataBroker().newWriteOnlyTransaction();
+ wr.put(LogicalDatastoreType.CONFIGURATION, NEIGHBORS_II, neighbors, true);
+ wr.submit().get();
+ }
+
+ private void deleteNeighbors() throws ExecutionException, InterruptedException {
+ final WriteTransaction wr = getDataBroker().newWriteOnlyTransaction();
+ wr.delete(LogicalDatastoreType.CONFIGURATION, NEIGHBORS_II);
+ wr.submit().get();
}
}
\ No newline at end of file
Code Review / bgpcep.git / blobdiff