bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/LocRibWriterTest.java

   1 /*
   2  * Copyright (c) 2015 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;
   9
  10 import static org.mockito.Matchers.any;
  11 import static org.mockito.Matchers.eq;
  12 import static org.mockito.Mockito.doAnswer;
  13 import static org.mockito.Mockito.doReturn;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.times;
  16 import static org.mockito.Mockito.verify;
  17 import static org.opendaylight.protocol.bgp.rib.impl.AbstractRIBTestSetup.PREFIX_QNAME;
  18 import static org.opendaylight.protocol.bgp.rib.spi.PeerRoleUtil.PEER_ROLE_NID;
  19
  20 import com.google.common.base.Optional;
  21 import com.google.common.collect.Lists;
  22 import com.google.common.util.concurrent.CheckedFuture;
  23 import java.util.ArrayList;
  24 import java.util.List;
  25 import org.junit.Before;
  26 import org.junit.Test;
  27 import org.mockito.Mock;
  28 import org.mockito.MockitoAnnotations;
  29 import org.mockito.stubbing.Answer;
  30 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  31 import org.opendaylight.controller.md.sal.dom.api.ClusteredDOMDataTreeChangeListener;
  32 import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService;
  33 import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeIdentifier;
  34 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
  35 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
  36 import org.opendaylight.protocol.bgp.inet.RIBActivator;
  37 import org.opendaylight.protocol.bgp.mode.impl.base.BasePathSelectionModeFactory;
  38 import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContext;
  39 import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContextRegistry;
  40 import org.opendaylight.protocol.bgp.rib.impl.stats.UnsignedInt32Counter;
  41 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext;
  42 import org.opendaylight.protocol.bgp.rib.spi.RibSupportUtils;
  43 import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext;
  44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv4.routes.ipv4.routes.Ipv4Route;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.BgpRib;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.Peer;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.peer.EffectiveRibIn;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.peer.SupportedTables;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  55 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  56 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  57 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
  58 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNode;
  59 import org.opendaylight.yangtools.yang.data.api.schema.tree.ModificationType;
  60
  61 public class LocRibWriterTest {
  62
  63     @Mock
  64     DOMDataWriteTransaction domTransWrite;
  65
  66     private final PolicyDatabase pd = new PolicyDatabase((long) 35, new Ipv4Address("10.25.2.9"), new ClusterIdentifier(new Ipv4Address("10.25.2.9")));
  67
  68     @Mock
  69     private RIBSupportContextRegistry registry;
  70
  71     @Mock
  72     private DOMTransactionChain chain;
  73
  74     @Mock
  75     private DOMDataTreeChangeService service;
  76
  77     @Mock
  78     private RIBSupportContext context;
  79
  80     @Mock
  81     CheckedFuture<?, ?> future;
  82
  83     private LocRibWriter locRibWriter;
  84
  85     private final List<YangInstanceIdentifier> routes = new ArrayList<>();
  86     private final TablesKey tablesKey = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
  87     private final UnsignedInt32Counter routeCounter = new UnsignedInt32Counter("loc-rib");
  88
  89     private RIBActivator ribActivator;
  90
  91     @Before
  92     public void setUp() throws Exception {
  93         MockitoAnnotations.initMocks(this);
  94
  95         this.ribActivator = new RIBActivator();
  96         final RIBExtensionProviderContext ribExtension = new SimpleRIBExtensionProviderContext();
  97         this.ribActivator.startRIBExtensionProvider(ribExtension);
  98
  99         final Answer<Object> answer = invocation -> {
 100             final Object[] args = invocation.getArguments();
 101             final NormalizedNode<?, ?> node = (NormalizedNode<?, ?>) args[2];
 102             if (node.getNodeType().equals(Ipv4Route.QNAME) || node.getNodeType().equals(PREFIX_QNAME)) {
 103                 LocRibWriterTest.this.routes.add((YangInstanceIdentifier) args[1]);
 104             }
 105             return args[1];
 106         };
 107         doAnswer(answer).when(this.domTransWrite).put(eq(LogicalDatastoreType.OPERATIONAL), any(YangInstanceIdentifier.class), any(NormalizedNode.class));
 108         doAnswer(answer).when(this.domTransWrite).merge(eq(LogicalDatastoreType.OPERATIONAL), any(YangInstanceIdentifier.class), any(NormalizedNode.class));
 109         doAnswer(invocation -> {
 110             final Object[] args = invocation.getArguments();
 111             LocRibWriterTest.this.routes.remove(args[1]);
 112             return args[1];
 113         }).when(this.domTransWrite).delete(eq(LogicalDatastoreType.OPERATIONAL), any(YangInstanceIdentifier.class));
 114         doReturn(this.context).when(this.registry).getRIBSupportContext(any(TablesKey.class));
 115         doReturn(ribExtension.getRIBSupport(this.tablesKey)).when(this.context).getRibSupport();
 116         doReturn(this.domTransWrite).when(this.chain).newWriteOnlyTransaction();
 117         doReturn(this.future).when(this.domTransWrite).submit();
 118         doReturn(null).when(this.service).registerDataTreeChangeListener(any(DOMDataTreeIdentifier.class), any(ClusteredDOMDataTreeChangeListener.class));
 119         this.locRibWriter = LocRibWriter.create(this.registry, this.tablesKey, this.chain, YangInstanceIdentifier.of(BgpRib.QNAME),
 120                 new AsNumber((long) 35), this.service, this.pd, new CacheDisconnectedPeersImpl(), BasePathSelectionModeFactory.createBestPathSelectionStrategy(),
 121                 this.routeCounter);
 122     }
 123
 124     private DataTreeCandidate prepareUpdate() {
 125         final DataTreeCandidate candidate = mock(DataTreeCandidate.class);
 126         doReturn("candidate").when(candidate).toString();
 127         final YangInstanceIdentifier rootPath = YangInstanceIdentifier.builder().node(BgpRib.QNAME).node(Peer.QNAME).nodeWithKey(Peer.QNAME, AdjRibInWriter.PEER_ID_QNAME, "12.12.12.12").build();
 128         doReturn(rootPath).when(candidate).getRootPath();
 129         return candidate;
 130     }
 131
 132     @Test
 133     public void testUpdateSupportedTables() {
 134         final DataTreeCandidate candidate = prepareUpdate();
 135         final DataTreeCandidateNode node = mock(DataTreeCandidateNode.class);
 136         doReturn("node").when(node).toString();
 137         final DataTreeCandidateNode tableChange = mock(DataTreeCandidateNode.class);
 138         // add
 139         final DataTreeCandidateNode table = mock(DataTreeCandidateNode.class);
 140         final NormalizedNode<?, ?> dataAfter = mock(NormalizedNode.class);
 141         doReturn(RibSupportUtils.toYangTablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class)).when(dataAfter).getIdentifier();
 142         doReturn(Optional.of(dataAfter)).when(table).getDataAfter();
 143         doReturn(Lists.newArrayList(table)).when(tableChange).getChildNodes();
 144         doReturn("table change").when(tableChange).toString();
 145         doReturn(tableChange).when(node).getModifiedChild(YangInstanceIdentifier.of(SupportedTables.QNAME).getLastPathArgument());
 146         doReturn(null).when(node).getModifiedChild(PEER_ROLE_NID);
 147         doReturn(null).when(node).getModifiedChild(YangInstanceIdentifier.of(EffectiveRibIn.QNAME).getLastPathArgument());
 148         doReturn(node).when(candidate).getRootNode();
 149         doReturn(ModificationType.SUBTREE_MODIFIED).when(node).getModificationType();
 150         this.locRibWriter.onDataTreeChanged(Lists.newArrayList(candidate));
 151         // delete
 152         final DataTreeCandidateNode tableDelete = mock(DataTreeCandidateNode.class);
 153         doReturn(Optional.absent()).when(tableDelete).getDataAfter();
 154         doReturn(Lists.newArrayList(tableDelete)).when(tableChange).getChildNodes();
 155         doReturn("table change").when(tableChange).toString();
 156         doReturn(node).when(candidate).getRootNode();
 157         this.locRibWriter.onDataTreeChanged(Lists.newArrayList(candidate));
 158         verify(node, times(2)).getModifiedChild(YangInstanceIdentifier.of(SupportedTables.QNAME).getLastPathArgument());
 159     }
 160 }