bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/config/RibImplTest.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 package org.opendaylight.protocol.bgp.rib.impl.config;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.mockito.ArgumentMatchers.any;
  13 import static org.mockito.Mockito.doReturn;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.verify;
  16
  17 import com.google.common.collect.ImmutableSet;
  18 import java.util.Map;
  19 import java.util.Set;
  20 import java.util.concurrent.ExecutionException;
  21 import org.junit.Before;
  22 import org.junit.Test;
  23 import org.mockito.Mock;
  24 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
  25 import org.opendaylight.mdsal.dom.api.DOMDataBroker.DataTreeChangeExtension;
  26 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  27 import org.opendaylight.protocol.bgp.rib.impl.spi.CodecsRegistry;
  28 import org.opendaylight.protocol.bgp.rib.impl.state.BGPStateCollector;
  29 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionConsumerContext;
  30 import org.opendaylight.protocol.bgp.rib.spi.RIBSupport;
  31 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
  32 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
  33 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiKey;
  34 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
  35 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
  36 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
  37 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV4UNICAST;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.BgpTableType;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.network.instance.protocol.GlobalAddPathsConfigBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.bgp.rib.Rib;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.BgpId;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
  46 import org.opendaylight.yangtools.concepts.Registration;
  47 import org.opendaylight.yangtools.yang.binding.util.BindingMap;
  48 import org.opendaylight.yangtools.yang.common.QName;
  49 import org.opendaylight.yangtools.yang.common.Uint8;
  50 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  51 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  52 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  53
  54 public class RibImplTest extends AbstractConfig {
  55     private static final Map<AfiSafiKey, AfiSafi> AFISAFIS = BindingMap.of(new AfiSafiBuilder()
  56         .setAfiSafiName(IPV4UNICAST.VALUE)
  57         .addAugmentation(new GlobalAddPathsConfigBuilder().setReceive(true).setSendMax(Uint8.ZERO).build())
  58         .build());
  59     private static final BgpTableType TABLE_TYPE = new BgpTableTypeImpl(Ipv4AddressFamily.VALUE,
  60             UnicastSubsequentAddressFamily.VALUE);
  61     private static final BgpId BGP_ID = new BgpId(new Ipv4AddressNoZone("127.0.0.1"));
  62
  63     @Mock
  64     private RIBExtensionConsumerContext extension;
  65     @Mock
  66     private CodecsRegistry codecsRegistry;
  67     @Mock
  68     private DOMDataBroker domDataBroker;
  69     @Mock
  70     private RIBSupport<?, ?> ribSupport;
  71
  72     @Override
  73     @Before
  74     public void setUp() throws Exception {
  75         super.setUp();
  76
  77         doReturn(ribSupport).when(extension).getRIBSupport(any(TablesKey.class));
  78         final NodeIdentifier nii = new NodeIdentifier(QName.create("", "test").intern());
  79         doReturn(nii).when(ribSupport).routeAttributesIdentifier();
  80         doReturn(ImmutableSet.of()).when(ribSupport).cacheableAttributeObjects();
  81         final MapEntryNode emptyTable = mock(MapEntryNode.class);
  82         doReturn(emptyTable).when(ribSupport).emptyTable();
  83         final NodeIdentifierWithPredicates niie = NodeIdentifierWithPredicates.of(Rib.QNAME,
  84                 QName.create("", "test").intern(), "t");
  85         doReturn(niie).when(emptyTable).name();
  86         doReturn(domTx).when(domDataBroker).createMergingTransactionChain();
  87         final DataTreeChangeExtension domDataTreeChangeService = mock(DataTreeChangeExtension.class);
  88         doReturn(domDataTreeChangeService).when(domDataBroker).extension(DataTreeChangeExtension.class);
  89         doReturn(mock(Registration.class)).when(domDataTreeChangeService).registerTreeChangeListener(any(), any());
  90     }
  91
  92     @Test
  93     public void testRibImpl() throws ExecutionException, InterruptedException {
  94         final RibImpl ribImpl = new RibImpl(extension, dispatcher, policyProvider, codecsRegistry,
  95                 new BGPStateCollector(), domDataBroker);
  96         ribImpl.start(createGlobal(), "rib-test", tableTypeRegistry);
  97         verify(domDataBroker).extension(DataTreeChangeExtension.class);
  98         assertEquals("""
  99             RIBImpl{bgpId=Ipv4Address{value=127.0.0.1}, localTables=[BgpTableTypeImpl [\
 100             getAfi()=Ipv4AddressFamily{qname=\
 101             (urn:opendaylight:params:xml:ns:yang:bgp-types?revision=2020-01-20)ipv4-address-family}, \
 102             getSafi()=UnicastSubsequentAddressFamily{qname=\
 103             (urn:opendaylight:params:xml:ns:yang:bgp-types?revision=2020-01-20)unicast-subsequent-address-family}]]}""",
 104             ribImpl.toString());
 105         assertEquals(Set.of(new TablesKey(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE)),
 106             ribImpl.getLocalTablesKeys());
 107         assertNotNull(ribImpl.getService());
 108         assertNotNull(ribImpl.getInstanceIdentifier());
 109         assertEquals(AS, ribImpl.getLocalAs());
 110         assertEquals(BGP_ID, ribImpl.getBgpIdentifier());
 111         assertEquals(Set.of(TABLE_TYPE), ribImpl.getLocalTables());
 112         assertEquals(dispatcher, ribImpl.getDispatcher());
 113         assertEquals(extension, ribImpl.getRibExtensions());
 114         assertNotNull(ribImpl.getRibSupportContext());
 115         assertNotNull(ribImpl.getCodecsRegistry());
 116
 117         ribImpl.stop().get();
 118     }
 119
 120     private static Global createGlobal() {
 121         return new GlobalBuilder()
 122                 .setAfiSafis(new AfiSafisBuilder().setAfiSafi(AFISAFIS).build())
 123                 .setConfig(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base
 124                         .ConfigBuilder().setAs(AS).setRouterId(BGP_ID).build()).build();
 125     }
 126 }