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