bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/config/AbstractConfig.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.mockito.ArgumentMatchers.any;
  11 import static org.mockito.ArgumentMatchers.anyInt;
  12 import static org.mockito.ArgumentMatchers.eq;
  13 import static org.mockito.Mockito.doNothing;
  14 import static org.mockito.Mockito.doReturn;
  15 import static org.mockito.Mockito.mock;
  16
  17 import io.netty.util.concurrent.Future;
  18 import java.net.InetSocketAddress;
  19 import java.util.Set;
  20 import org.junit.Before;
  21 import org.mockito.Mock;
  22 import org.opendaylight.mdsal.common.api.CommitInfo;
  23 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  24 import org.opendaylight.mdsal.dom.api.DOMDataBroker.DataTreeChangeExtension;
  25 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
  26 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
  27 import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
  28 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  29 import org.opendaylight.protocol.bgp.rib.impl.BGPPeerTrackerImpl;
  30 import org.opendaylight.protocol.bgp.rib.impl.DefaultRibPoliciesMockTest;
  31 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
  32 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
  33 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
  34 import org.opendaylight.protocol.bgp.rib.impl.spi.RIB;
  35 import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContextRegistry;
  36 import org.opendaylight.protocol.bgp.rib.spi.BGPPeerTracker;
  37 import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
  38 import org.opendaylight.protocol.concepts.KeyMapping;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.BgpRib;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.Rib;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.RibId;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.bgp.rib.RibKey;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.BgpId;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
  49 import org.opendaylight.yangtools.concepts.Registration;
  50 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  51 import org.opendaylight.yangtools.yang.common.Uint32;
  52 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  53 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  54 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  55
  56 class AbstractConfig extends DefaultRibPoliciesMockTest {
  57     protected static final AsNumber AS = new AsNumber(Uint32.valueOf(72));
  58     protected static final AsNumber LOCAL_AS = new AsNumber(Uint32.valueOf(73));
  59     protected static final RibId RIB_ID = new RibId("test");
  60     static final TablesKey TABLES_KEY = new TablesKey(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE);
  61     @Mock
  62     protected RIB rib;
  63     @Mock
  64     protected BGPTableTypeRegistryConsumer tableTypeRegistry;
  65     @Mock
  66     protected DOMTransactionChain domTx;
  67     @Mock
  68     protected BGPDispatcher dispatcher;
  69     @Mock
  70     protected BGPPeerRegistry bgpPeerRegistry;
  71     @Mock
  72     protected Registration listener;
  73     @Mock
  74     protected Future<?> future;
  75     @Mock
  76     protected DOMDataTreeWriteTransaction domDW;
  77     @Mock
  78     protected PeerGroupConfigLoader peerGroupLoader;
  79     @Mock
  80     private DataTreeChangeExtension dataTreeChangeService;
  81     private final BGPPeerTracker peerTracker = new BGPPeerTrackerImpl();
  82
  83     @Override
  84     @Before
  85     public void setUp() throws Exception {
  86         super.setUp();
  87         doReturn(InstanceIdentifier.create(BgpRib.class).child(org.opendaylight.yang.gen.v1.urn.opendaylight
  88                 .params.xml.ns.yang.bgp.rib.rev180329.bgp.rib.Rib.class, new RibKey(RIB_ID))).when(rib)
  89                 .getInstanceIdentifier();
  90         doReturn(domTx).when(rib).createPeerDOMChain();
  91         doNothing().when(domTx).addCallback(any());
  92
  93         doReturn(AS).when(rib).getLocalAs();
  94         doReturn(mock(RIBSupportContextRegistry.class)).when(rib).getRibSupportContext();
  95         doReturn(Set.of()).when(rib).getLocalTablesKeys();
  96         doNothing().when(domTx).close();
  97         doReturn(domDW).when(domTx).newWriteOnlyTransaction();
  98         doNothing().when(domDW).put(eq(LogicalDatastoreType.OPERATIONAL),
  99                 any(YangInstanceIdentifier.class), any(MapEntryNode.class));
 100         doNothing().when(domDW).delete(eq(LogicalDatastoreType.OPERATIONAL),
 101                 any(YangInstanceIdentifier.class));
 102         doNothing().when(domDW).merge(eq(LogicalDatastoreType.OPERATIONAL),
 103                 any(YangInstanceIdentifier.class), any(NormalizedNode.class));
 104         doReturn(CommitInfo.emptyFluentFuture()).when(domDW).commit();
 105
 106         doReturn(YangInstanceIdentifier.of(Rib.QNAME)).when(rib).getYangRibId();
 107         doReturn(dataTreeChangeService).when(rib).getService();
 108         doReturn(listener).when(dataTreeChangeService).registerTreeChangeListener(any(), any());
 109         doReturn(new BgpId("127.0.0.1")).when(rib).getBgpIdentifier();
 110         doReturn(true).when(future).cancel(true);
 111         doReturn(future).when(dispatcher).createReconnectingClient(any(InetSocketAddress.class),
 112                 any(), anyInt(), any(KeyMapping.class));
 113         doReturn(dispatcher).when(rib).getDispatcher();
 114
 115         doReturn(new BgpTableTypeImpl(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE))
 116                 .when(tableTypeRegistry).getTableType(any());
 117         doReturn(TABLES_KEY).when(tableTypeRegistry).getTableKey(any());
 118         doReturn(Set.of(new BgpTableTypeImpl(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE)))
 119                 .when(rib).getLocalTables();
 120
 121         doNothing().when(bgpPeerRegistry).addPeer(any(IpAddressNoZone.class),
 122                 any(BGPSessionListener.class), any(BGPSessionPreferences.class));
 123         doNothing().when(bgpPeerRegistry).removePeer(any(IpAddressNoZone.class));
 124         doReturn("registry").when(bgpPeerRegistry).toString();
 125         doNothing().when(listener).close();
 126         doReturn(bgpPeerRegistry).when(dispatcher).getBGPPeerRegistry();
 127         doReturn(peerTracker).when(rib).getPeerTracker();
 128         doReturn(policies).when(rib).getRibPolicies();
 129         doReturn(null).when(peerGroupLoader)
 130                 .getPeerGroup(any(InstanceIdentifier.class), any(String.class));
 131     }
 132 }