bgp/rib-spi/src/test/java/org/opendaylight/protocol/bgp/rib/spi/RIBSupportTest.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.spi;
   9
  10 import static junit.framework.TestCase.assertFalse;
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNotNull;
  13 import static org.junit.Assert.assertTrue;
  14 import static org.mockito.ArgumentMatchers.any;
  15 import static org.mockito.Mockito.doAnswer;
  16 import static org.mockito.Mockito.doReturn;
  17
  18 import java.util.Collection;
  19 import java.util.HashMap;
  20 import java.util.HashSet;
  21 import java.util.Map;
  22 import java.util.Optional;
  23 import java.util.Set;
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.mockito.Mockito;
  27 import org.mockito.MockitoAnnotations;
  28 import org.opendaylight.mdsal.binding.dom.adapter.AdapterContext;
  29 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractConcurrentDataBrokerTest;
  30 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractDataBrokerTestCustomizer;
  31 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  32 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.UpdateBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.Attributes;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.AttributesBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.AttributesReachBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.AttributesUnreachBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.reach.MpReachNlri;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.reach.MpReachNlriBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.reach.mp.reach.nlri.AdvertizedRoutes;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.reach.mp.reach.nlri.AdvertizedRoutesBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.unreach.MpUnreachNlri;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.unreach.MpUnreachNlriBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.unreach.mp.unreach.nlri.WithdrawnRoutes;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.attributes.unreach.mp.unreach.nlri.WithdrawnRoutesBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.destination.DestinationType;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.BgpRib;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.bgp.rib.rib.LocRib;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.Tables;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.test.rev180515.Ipv4Prefixes;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.test.rev180515.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.test.rev180515.ipv4.routes.Ipv4Routes;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.test.rev180515.ipv4.routes.ipv4.routes.Ipv4Route;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.next.hop.c.next.hop.Ipv4NextHopCase;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
  62 import org.opendaylight.yangtools.yang.common.QName;
  63 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  64 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  65 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  66 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  67 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  68 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  69 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  70 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
  71 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeCandidateNode;
  72
  73 public class RIBSupportTest extends AbstractConcurrentDataBrokerTest {
  74     private static final String ROUTE_KEY = "prefix";
  75     private static final String PREFIX = "1.2.3.4/32";
  76     private static final QName PATH_ID_QNAME = QName.create(Ipv4Route.QNAME, "path-id").intern();
  77     private static final NodeIdentifierWithPredicates PREFIX_NII = NodeIdentifierWithPredicates.of(Ipv4Route.QNAME,
  78         QName.create(Ipv4Route.QNAME, ROUTE_KEY).intern(), PREFIX);
  79     private RIBSupportTestImp ribSupportTestImp;
  80     private static final TablesKey TABLES_KEY = new TablesKey(Ipv4AddressFamily.VALUE,
  81             UnicastSubsequentAddressFamily.VALUE);
  82     private static final YangInstanceIdentifier LOC_RIB_TARGET = YangInstanceIdentifier
  83             .create(YangInstanceIdentifier.of(BgpRib.QNAME)
  84         .node(LocRib.QNAME).node(Tables.QNAME).node(RibSupportUtils.toYangTablesKey(TABLES_KEY)).getPathArguments());
  85     private static final NodeIdentifier ROUTES_IDENTIFIER = new NodeIdentifier(Routes.QNAME);
  86     private static final NodeIdentifier IPV4_ROUTES_IDENTIFIER = new NodeIdentifier(Ipv4Routes.QNAME);
  87     private DataTreeCandidateNode emptyTree;
  88     private DataTreeCandidateNode emptySubTree;
  89     private DataTreeCandidateNode subTree;
  90     private DOMDataTreeWriteTransaction tx;
  91     private ContainerNode nlri;
  92     private final Map<YangInstanceIdentifier, NormalizedNode> routesMap = new HashMap<>();
  93     private ContainerNode attributes;
  94     private MapEntryNode mapEntryNode;
  95     private AdapterContext context;
  96
  97     @Before
  98     public void setUp() throws Exception {
  99         super.setup();
 100         MockitoAnnotations.initMocks(this);
 101         ribSupportTestImp = new RIBSupportTestImp(context.currentSerializer());
 102         emptyTree = Mockito.mock(DataTreeCandidateNode.class);
 103         emptySubTree = Mockito.mock(DataTreeCandidateNode.class);
 104         subTree = Mockito.mock(DataTreeCandidateNode.class);
 105         final DataTreeCandidateNode emptyNode = Mockito.mock(DataTreeCandidateNode.class);
 106         final DataTreeCandidateNode node = Mockito.mock(DataTreeCandidateNode.class);
 107         doReturn(Optional.empty()).when(emptyTree).getModifiedChild(IPV4_ROUTES_IDENTIFIER);
 108
 109         doReturn(Optional.of(emptyNode)).when(emptySubTree).getModifiedChild(IPV4_ROUTES_IDENTIFIER);
 110         doReturn(Optional.empty()).when(emptyNode).getModifiedChild(new NodeIdentifier(Ipv4Route.QNAME));
 111
 112         doReturn(Optional.of(node)).when(subTree).getModifiedChild(IPV4_ROUTES_IDENTIFIER);
 113         doReturn(Optional.of(node)).when(node).getModifiedChild(new NodeIdentifier(Ipv4Route.QNAME));
 114         final Collection<DataTreeCandidateNode> emptyCollection = new HashSet<>();
 115         doReturn(emptyCollection).when(node).getChildNodes();
 116
 117         tx = Mockito.mock(DOMDataTreeWriteTransaction.class);
 118         nlri = Mockito.mock(ContainerNode.class);
 119         attributes = ImmutableContainerNodeBuilder.create()
 120                 .withNodeIdentifier(new NodeIdentifier(QName.create(Ipv4Routes.QNAME, Attributes.QNAME
 121             .getLocalName().intern()))).build();
 122         final ContainerNode destination = Mockito.mock(ContainerNode.class);
 123         final ChoiceNode destinations = Mockito.mock(ChoiceNode.class);
 124         final ContainerNode route = Mockito.mock(ContainerNode.class);
 125
 126         doReturn(destination).when(nlri).childByArg(new NodeIdentifier(WithdrawnRoutes.QNAME));
 127         doReturn(destination).when(nlri).childByArg(new NodeIdentifier(AdvertizedRoutes.QNAME));
 128         doReturn(destinations).when(destination).childByArg(new NodeIdentifier(DestinationType.QNAME));
 129         doReturn(route).when(destinations).childByArg(new NodeIdentifier(Ipv4Prefixes.QNAME));
 130         doReturn(emptyCollection).when(route).body();
 131
 132         doAnswer(invocation -> {
 133             final Object[] args = invocation.getArguments();
 134             routesMap.remove(args[1]);
 135             return args[1];
 136         }).when(tx).delete(Mockito.eq(LogicalDatastoreType.OPERATIONAL), any(YangInstanceIdentifier.class));
 137         doAnswer(invocation -> {
 138             final Object[] args = invocation.getArguments();
 139             final NormalizedNode node1 = (NormalizedNode) args[2];
 140             routesMap.put((YangInstanceIdentifier) args[1], node1);
 141             return args[1];
 142         }).when(tx).put(Mockito.eq(LogicalDatastoreType.OPERATIONAL), any(YangInstanceIdentifier.class),
 143                 any(NormalizedNode.class));
 144
 145         mapEntryNode = Mockito.mock(MapEntryNode.class);
 146     }
 147
 148     @Override
 149     protected final AbstractDataBrokerTestCustomizer createDataBrokerTestCustomizer() {
 150         final AbstractDataBrokerTestCustomizer customizer = super.createDataBrokerTestCustomizer();
 151         context = customizer.getAdapterContext();
 152         return customizer;
 153     }
 154
 155     @Test
 156     public void routesCaseClass() {
 157         assertEquals(Ipv4RoutesCase.class, ribSupportTestImp.routesCaseClass());
 158     }
 159
 160     @Test
 161     public void routesContainerClass() {
 162         assertEquals(Ipv4Routes.class, ribSupportTestImp.routesContainerClass());
 163     }
 164
 165     @Test
 166     public void routesListClass() {
 167         assertEquals(Ipv4Route.class, ribSupportTestImp.routesListClass());
 168     }
 169
 170     @Test
 171     public void routeQName() {
 172         assertEquals(Ipv4Route.QNAME, ribSupportTestImp.routeQName());
 173     }
 174
 175     @Test
 176     public void routeNid() {
 177         assertEquals(new NodeIdentifier(Ipv4Route.QNAME),ribSupportTestImp.routeNid());
 178     }
 179
 180     @Test
 181     public void getAfi() {
 182         assertEquals(Ipv4AddressFamily.VALUE, ribSupportTestImp.getAfi());
 183     }
 184
 185     @Test
 186     public void getSafi() {
 187         assertEquals(UnicastSubsequentAddressFamily.VALUE, ribSupportTestImp.getSafi());
 188     }
 189
 190     @Test
 191     public void routesContainerIdentifier() {
 192         assertEquals(IPV4_ROUTES_IDENTIFIER,ribSupportTestImp.routesContainerIdentifier());
 193
 194     }
 195
 196     @Test
 197     public void routeAttributesIdentifier() {
 198         assertEquals(new NodeIdentifier(QName.create(Ipv4Routes.QNAME,
 199                 Attributes.QNAME.getLocalName().intern())),ribSupportTestImp.routeAttributesIdentifier());
 200     }
 201
 202     @Test
 203     public void routePath() {
 204         assertEquals(LOC_RIB_TARGET.node(ROUTES_IDENTIFIER)
 205                         .node(Ipv4Routes.QNAME).node(Ipv4Route.QNAME).node(PREFIX_NII),
 206                 ribSupportTestImp.routePath(LOC_RIB_TARGET, PREFIX_NII));
 207     }
 208
 209     @Test
 210     public void changedRoutes() {
 211         assertTrue(ribSupportTestImp.changedRoutes(emptyTree).isEmpty());
 212         assertTrue(ribSupportTestImp.changedRoutes(emptySubTree).isEmpty());
 213         assertNotNull(ribSupportTestImp.changedRoutes(subTree));
 214     }
 215
 216     @Test
 217     public void putRoutes() {
 218         ribSupportTestImp.putRoutes(tx, LOC_RIB_TARGET, nlri, attributes);
 219         assertFalse(routesMap.isEmpty());
 220     }
 221
 222     @Test
 223     public void deleteRoutes() {
 224         ribSupportTestImp.deleteRoutes(tx, LOC_RIB_TARGET, nlri);
 225         assertTrue(routesMap.isEmpty());
 226     }
 227
 228
 229     @Test
 230     public void buildUpdate() {
 231         final Ipv4NextHopCase nextHop = new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder()
 232                 .setGlobal(new Ipv4AddressNoZone("10.0.0.2")).build()).build();
 233         final Attributes attr = new AttributesBuilder().setCNextHop(nextHop).build();
 234         final Collection<MapEntryNode> routes = new HashSet<>();
 235
 236         assertEquals(new UpdateBuilder().setAttributes(new AttributesBuilder().build()).build(),
 237                ribSupportTestImp.buildUpdate(routes, routes, attr));
 238
 239         routes.add(mapEntryNode);
 240         final MpReachNlri mpReach = new MpReachNlriBuilder().setAfi(Ipv4AddressFamily.VALUE)
 241                 .setSafi(UnicastSubsequentAddressFamily.VALUE)
 242                 .setCNextHop(nextHop).setAdvertizedRoutes(new AdvertizedRoutesBuilder().build()).build();
 243
 244         final Attributes attMpR = new AttributesBuilder().addAugmentation(
 245             new AttributesReachBuilder().setMpReachNlri(mpReach).build()).build();
 246         assertEquals(new UpdateBuilder().setAttributes(attMpR).build(),
 247                ribSupportTestImp.buildUpdate(routes, Set.of(), attr));
 248
 249         final MpUnreachNlri mpUnreach = new MpUnreachNlriBuilder()
 250                 .setAfi(Ipv4AddressFamily.VALUE)
 251                 .setSafi(UnicastSubsequentAddressFamily.VALUE)
 252                 .setWithdrawnRoutes(new WithdrawnRoutesBuilder().build())
 253                 .build();
 254
 255         final Attributes attMpU = new AttributesBuilder().addAugmentation(
 256                 new AttributesUnreachBuilder().setMpUnreachNlri(mpUnreach).build()).build();
 257         assertEquals(new UpdateBuilder().setAttributes(attMpU).build(),
 258                ribSupportTestImp.buildUpdate(Set.of(), routes, attr));
 259     }
 260 }